forked from lug/matterbridge
Compare commits
485 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
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 | ||
|
|
7b0bc51183 | ||
|
|
53aa076555 | ||
|
|
f57370f33a | ||
|
|
c557d51b6f | ||
|
|
df3fdc26a0 | ||
|
|
af00c34aac | ||
|
|
120bf39f55 | ||
|
|
26a7e35f27 | ||
|
|
d44d2a5f00 | ||
|
|
7f1d86b338 | ||
|
|
d8816280f0 | ||
|
|
b09a73040f | ||
|
|
740b5f2602 | ||
|
|
96841c70c7 | ||
|
|
f92735d35d | ||
|
|
516fd3c92d | ||
|
|
a775b57134 | ||
|
|
bf21604d42 | ||
|
|
1bb39eba87 | ||
|
|
3190703dc8 | ||
|
|
5095db8a43 | ||
|
|
1f1634ea59 | ||
|
|
a7dd033c3b | ||
|
|
95e78ffa05 | ||
|
|
42276ea7d0 | ||
|
|
dffd67eb31 | ||
|
|
55e79063d6 | ||
|
|
46f4bbb3b5 | ||
|
|
240559581a | ||
|
|
48ba829465 | ||
|
|
eef654de98 | ||
|
|
d76a04bd0a | ||
|
|
a8fe54a78d | ||
|
|
0bcb0b882f | ||
|
|
4525fa31aa | ||
|
|
aeaea0574f | ||
|
|
99d71c2177 | ||
|
|
3e60cfafd3 | ||
|
|
3123695869 | ||
|
|
777af73e2b | ||
|
|
716751cf76 | ||
|
|
6ebd5cbbd8 | ||
|
|
077b818d82 | ||
|
|
5af1d80055 | ||
|
|
f236d12166 | ||
|
|
127eb908f3 | ||
|
|
40d76b2296 | ||
|
|
8147815037 | ||
|
|
57f156be83 | ||
|
|
2cfd880cdb | ||
|
|
430b38e770 | ||
|
|
e7f463a082 |
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
|
name: Bug report
|
||||||
about: Create a report to help us improve. (Check the FAQ on the wiki first)
|
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
|
name: Feature request
|
||||||
about: Suggest an idea for this project
|
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@v2
|
||||||
|
with:
|
||||||
|
fetch-depth: 20
|
||||||
|
- name: Run golangci-lint
|
||||||
|
uses: golangci/golangci-lint-action@v2
|
||||||
|
with:
|
||||||
|
version: v1.29
|
||||||
|
args: "-v --new-from-rev HEAD~5"
|
||||||
|
test-build-upload:
|
||||||
|
strategy:
|
||||||
|
matrix:
|
||||||
|
go-version: [1.15.x, 1.16.x]
|
||||||
|
platform: [ubuntu-latest]
|
||||||
|
runs-on: ${{ matrix.platform }}
|
||||||
|
steps:
|
||||||
|
- name: Install Go
|
||||||
|
uses: actions/setup-go@v2
|
||||||
|
with:
|
||||||
|
go-version: ${{ matrix.go-version }}
|
||||||
|
stable: false
|
||||||
|
- name: Checkout code
|
||||||
|
uses: actions/checkout@v2
|
||||||
|
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)
|
||||||
|
GOOS=linux GOARCH=amd64 go build -ldflags "-s -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o output/lin/matterbridge-$VERSION-linux-amd64
|
||||||
|
GOOS=windows GOARCH=amd64 go build -ldflags "-s -X main.githash=$(git log --pretty=format:'%h' -n 1)" -o output/win/matterbridge-$VERSION-windows-amd64.exe
|
||||||
|
GOOS=darwin GOARCH=amd64 go build -ldflags "-s -X main.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.16')
|
||||||
|
uses: actions/upload-artifact@v2
|
||||||
|
with:
|
||||||
|
name: matterbridge-linux-64bit
|
||||||
|
path: output/lin
|
||||||
|
- name: Upload windows 64-bit
|
||||||
|
if: startsWith(matrix.go-version,'1.16')
|
||||||
|
uses: actions/upload-artifact@v2
|
||||||
|
with:
|
||||||
|
name: matterbridge-windows-64bit
|
||||||
|
path: output/win
|
||||||
|
- name: Upload darwin 64-bit
|
||||||
|
if: startsWith(matrix.go-version,'1.16')
|
||||||
|
uses: actions/upload-artifact@v2
|
||||||
|
with:
|
||||||
|
name: matterbridge-darwin-64bit
|
||||||
|
path: output/mac
|
||||||
6
.gitignore
vendored
Normal file
6
.gitignore
vendored
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
# Exclude matterbridge binary
|
||||||
|
/matterbridge
|
||||||
|
/matterbridge.exe
|
||||||
|
|
||||||
|
# Exclude configuration file
|
||||||
|
matterbridge.toml
|
||||||
@@ -7,7 +7,7 @@ run:
|
|||||||
# concurrency: 4
|
# concurrency: 4
|
||||||
|
|
||||||
# timeout for analysis, e.g. 30s, 5m, default is 1m
|
# timeout for analysis, e.g. 30s, 5m, default is 1m
|
||||||
deadline: 1m
|
deadline: 2m
|
||||||
|
|
||||||
# exit code when at least one issue was found, default is 1
|
# exit code when at least one issue was found, default is 1
|
||||||
issues-exit-code: 1
|
issues-exit-code: 1
|
||||||
@@ -23,7 +23,7 @@ run:
|
|||||||
# default value is empty list, but next dirs are always skipped independently
|
# default value is empty list, but next dirs are always skipped independently
|
||||||
# from this option's value:
|
# from this option's value:
|
||||||
# vendor$, third_party$, testdata$, examples$, Godeps$, builtin$
|
# 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
|
# which files to skip: they will be analyzed, but issues from them
|
||||||
# won't be reported. Default value is empty list, but there is
|
# 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.
|
# Correct spellings using locale preferences for US or UK.
|
||||||
# Default is to use a neutral variety of English.
|
# Default is to use a neutral variety of English.
|
||||||
# Setting locale to US will correct the British spelling of 'colour' to 'color'.
|
# Setting locale to US will correct the British spelling of 'colour' to 'color'.
|
||||||
locale: US
|
|
||||||
lll:
|
lll:
|
||||||
# max line length, lines longer will be reported. Default is 120.
|
# 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
|
# '\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.
|
# with golangci-lint call it on a directory with the changed file.
|
||||||
check-exported: false
|
check-exported: false
|
||||||
unparam:
|
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.
|
# 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:
|
# 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
|
# if it's called for subdir of a project it can't find external interfaces. All text editor integrations
|
||||||
@@ -132,6 +127,7 @@ linters-settings:
|
|||||||
# ifElseChain regexpMust singleCaseSwitch sloppyLen switchTrue typeSwitchVar underef
|
# ifElseChain regexpMust singleCaseSwitch sloppyLen switchTrue typeSwitchVar underef
|
||||||
# unlambda unslice rangeValCopy defaultCaseOrder];
|
# unlambda unslice rangeValCopy defaultCaseOrder];
|
||||||
# all checks list: https://github.com/go-critic/checkers
|
# all checks list: https://github.com/go-critic/checkers
|
||||||
|
# disabled for now - hugeParam
|
||||||
enabled-checks:
|
enabled-checks:
|
||||||
- appendAssign
|
- appendAssign
|
||||||
- assignOp
|
- assignOp
|
||||||
@@ -147,7 +143,6 @@ linters-settings:
|
|||||||
- dupSubExpr
|
- dupSubExpr
|
||||||
- elseif
|
- elseif
|
||||||
- emptyFallthrough
|
- emptyFallthrough
|
||||||
- hugeParam
|
|
||||||
- ifElseChain
|
- ifElseChain
|
||||||
- importShadow
|
- importShadow
|
||||||
- indexAlloc
|
- indexAlloc
|
||||||
@@ -158,7 +153,6 @@ linters-settings:
|
|||||||
- regexpMust
|
- regexpMust
|
||||||
- singleCaseSwitch
|
- singleCaseSwitch
|
||||||
- sloppyLen
|
- sloppyLen
|
||||||
- sloppyReassign
|
|
||||||
- switchTrue
|
- switchTrue
|
||||||
- typeSwitchVar
|
- typeSwitchVar
|
||||||
- typeUnparen
|
- typeUnparen
|
||||||
@@ -179,7 +173,15 @@ linters:
|
|||||||
- lll
|
- lll
|
||||||
- maligned
|
- maligned
|
||||||
- prealloc
|
- prealloc
|
||||||
|
- wsl
|
||||||
|
- gomnd
|
||||||
|
- godox
|
||||||
|
- goerr113
|
||||||
|
- testpackage
|
||||||
|
- godot
|
||||||
|
- interfacer
|
||||||
|
- goheader
|
||||||
|
- noctx
|
||||||
|
|
||||||
# rules to deal with reported isues
|
# rules to deal with reported isues
|
||||||
issues:
|
issues:
|
||||||
|
|||||||
38
.goreleaser.yml
Normal file
38
.goreleaser.yml
Normal file
@@ -0,0 +1,38 @@
|
|||||||
|
release:
|
||||||
|
prerelease: auto
|
||||||
|
name_template: "{{.ProjectName}} v{{.Version}}"
|
||||||
|
|
||||||
|
builds:
|
||||||
|
- env:
|
||||||
|
- CGO_ENABLED=0
|
||||||
|
goos:
|
||||||
|
- freebsd
|
||||||
|
- windows
|
||||||
|
- darwin
|
||||||
|
- linux
|
||||||
|
- dragonfly
|
||||||
|
- netbsd
|
||||||
|
- openbsd
|
||||||
|
goarch:
|
||||||
|
- amd64
|
||||||
|
- arm
|
||||||
|
- arm64
|
||||||
|
- 386
|
||||||
|
ldflags:
|
||||||
|
- -s -w -X main.githash={{.ShortCommit}}
|
||||||
|
|
||||||
|
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'
|
||||||
57
.travis.yml
57
.travis.yml
@@ -1,57 +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.12.3"
|
|
||||||
|
|
||||||
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:
|
|
||||||
provider: bintray
|
|
||||||
edge:
|
|
||||||
branch: v1.8.47
|
|
||||||
file: ci/deploy.json
|
|
||||||
user: 42wim
|
|
||||||
key:
|
|
||||||
secure: "CeXXe6JOmt7HYR81MdWLua0ltQHhDdkIeRGBFbgd7hkb1wi8eF9DgpAcQrTso8NIlHNZmSAP46uhFgsRvkuezzX0ygalZ7DCJyAyn3sAMEh+UQSHV1WGThRehTtidqRGjetzsIGSwdrJOWil+XTfbO1Z8DGzfakhSuAZka8CM4BAoe3YeP9rYK8h+84x0GHfczvsLtXZ3mWLvQuwe4pK6+ItBCUg0ae7O7ZUpWHy0xQQkkWztY/6RAzXfaG7DuGjIw+20fhx3WOXRNpHCtZ6Bc3qERCpk0s1HhlQWlrN9wDaFTBWYwlvSnNgvxxMbNXJ6RrRJ0l0bA7FUswYwyroxhzrGLdzWDg8dHaQkypocngdalfhpsnoO9j3ApJhomUFJ3UoEq5nOGRUrKn8MPi+dP0zE4kNQ3e4VNa1ufNrvfpWolMg3xh8OXuhQdD5wIM5zFAbRJLqWSCVAjPq4DDPecmvXBOlIial7oa312lN5qnBnUjvAcxszZ+FUyDHT1Grxzna4tMwxY9obPzZUzm7359AOCCwIQFVB8GLqD2nwIstcXS0zGRz+fhviPipHuBa02q5bGUZwmkvrSNab0s8Jo7pCrel2Rz3nWPKaiCfq2WjbW1CLheSMkOQrjsdUd1hhbqNWFPUjJPInTc77NAKCfm5runv5uyowRLh4NNd0sI="
|
|
||||||
23
Dockerfile
23
Dockerfile
@@ -1,11 +1,14 @@
|
|||||||
FROM alpine:edge
|
FROM alpine AS builder
|
||||||
ENTRYPOINT ["/bin/matterbridge"]
|
|
||||||
|
|
||||||
COPY . /go/src/github.com/42wim/matterbridge
|
COPY . /go/src/matterbridge
|
||||||
RUN apk update && apk add go git gcc musl-dev ca-certificates \
|
RUN apk --no-cache add go git \
|
||||||
&& cd /go/src/github.com/42wim/matterbridge \
|
&& cd /go/src/matterbridge \
|
||||||
&& export GOPATH=/go \
|
&& go build -mod vendor -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge
|
||||||
&& go get \
|
|
||||||
&& go build -x -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge \
|
FROM alpine
|
||||||
&& rm -rf /go \
|
RUN apk --no-cache add ca-certificates mailcap
|
||||||
&& apk del --purge git go gcc musl-dev
|
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"]
|
||||||
|
|||||||
375
README.md
375
README.md
@@ -3,132 +3,217 @@
|
|||||||
# matterbridge
|
# matterbridge
|
||||||
|
|
||||||
<br />
|
<br />
|
||||||
**A simple chat bridge**<br />
|
**A simple chat bridge**<br />
|
||||||
Letting people be where they want to be.<br />
|
Letting people be where they want to be.<br />
|
||||||
<sub>Bridges between a growing number of protocols. Click below to demo.</sub>
|
<sub>Bridges between a growing number of protocols. Click below to demo or join the development chat.</sub>
|
||||||
|
|
||||||
<sup>
|
<sup>
|
||||||
|
|
||||||
[Gitter][mb-gitter] |
|
[Discord][mb-discord] |
|
||||||
[IRC][mb-irc] |
|
[Gitter][mb-gitter] |
|
||||||
[Discord][mb-discord] |
|
[IRC][mb-irc] |
|
||||||
[Matrix][mb-matrix] |
|
[Keybase][mb-keybase] |
|
||||||
[Slack][mb-slack] |
|
[Matrix][mb-matrix] |
|
||||||
[Mattermost][mb-mattermost] |
|
[Mattermost][mb-mattermost] |
|
||||||
[XMPP][mb-xmpp] |
|
[MSTeams][mb-msteams] |
|
||||||
[Twitch][mb-twitch] |
|
[Rocket.Chat][mb-rocketchat] |
|
||||||
[Zulip][mb-zulip] |
|
[Slack][mb-slack] |
|
||||||
And more...
|
[Telegram][mb-telegram] |
|
||||||
</sup>
|
[Twitch][mb-twitch] |
|
||||||
|
[WhatsApp][mb-whatsapp] |
|
||||||
|
[XMPP][mb-xmpp] |
|
||||||
|
[Zulip][mb-zulip] |
|
||||||
|
And more...
|
||||||
|
</sup>
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
----
|
|
||||||
[](https://github.com/42wim/matterbridge/releases/latest)
|
[](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/maintainability)
|
[](https://codeclimate.com/github/42wim/matterbridge/test_coverage)<br />
|
||||||
[](https://codeclimate.com/github/42wim/matterbridge/test_coverage)<br />
|
|
||||||
<hr />
|
<hr />
|
||||||
</div>
|
</div>
|
||||||
<div align="right"><sup>
|
<div align="right"><sup>
|
||||||
|
|
||||||
**Note:** Matter<em>most</em> isn't required to run matter<em>bridge</em>.</sup></div>
|
**Note:** Matter<em>most</em> isn't required to run matter<em>bridge</em>.</sup></div>
|
||||||
|
|
||||||
### Table of Contents
|
<p>
|
||||||
* [Features](https://github.com/42wim/matterbridge/wiki/Features)
|
<a href="https://www.digitalocean.com/">
|
||||||
* [API](#API)
|
<img src="https://opensource.nyc3.cdn.digitaloceanspaces.com/attribution/assets/PoweredByDO/DO_Powered_by_Badge_blue.svg" width="201px">
|
||||||
* [Requirements](#requirements)
|
</a>
|
||||||
* [Screenshots](https://github.com/42wim/matterbridge/wiki/)
|
</p>
|
||||||
* [Installing](#installing)
|
|
||||||
* [Binaries](#binaries)
|
# Table of Contents
|
||||||
* [Building](#building)
|
|
||||||
* [Configuration](#configuration)
|
- [matterbridge](#matterbridge)
|
||||||
* [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config)
|
- [Table of Contents](#table-of-contents)
|
||||||
* [Examples](#examples)
|
- [Features](#features)
|
||||||
* [Running](#running)
|
- [Natively supported](#natively-supported)
|
||||||
* [Docker](#docker)
|
- [3rd party via matterbridge api](#3rd-party-via-matterbridge-api)
|
||||||
* [Changelog](#changelog)
|
- [API](#api)
|
||||||
* [FAQ](#faq)
|
- [Chat with us](#chat-with-us)
|
||||||
* [Related projects](#related-projects)
|
- [Screenshots](#screenshots)
|
||||||
* [Articles](#articles)
|
- [Installing / upgrading](#installing--upgrading)
|
||||||
* [Thanks](#thanks)
|
- [Binaries](#binaries)
|
||||||
|
- [Packages](#packages)
|
||||||
|
- [Building](#building)
|
||||||
|
- [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)
|
||||||
|
- [Changelog](#changelog)
|
||||||
|
- [FAQ](#faq)
|
||||||
|
- [Related projects](#related-projects)
|
||||||
|
- [Articles / Tutorials](#articles--tutorials)
|
||||||
|
- [Thanks](#thanks)
|
||||||
|
|
||||||
## Features
|
## 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)
|
- [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols)
|
||||||
* [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes)
|
- [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols)
|
||||||
* Preserves threading when possible
|
- [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes)
|
||||||
* [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling)
|
- Preserves threading when possible
|
||||||
* [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing)
|
- [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling)
|
||||||
* [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups)
|
- [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing)
|
||||||
* [API](https://github.com/42wim/matterbridge/wiki/Features#api)
|
- [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups)
|
||||||
|
- [API](https://github.com/42wim/matterbridge/wiki/Features#api)
|
||||||
|
|
||||||
|
### Natively supported
|
||||||
|
|
||||||
|
- [Discord](https://discordapp.com)
|
||||||
|
- [Gitter](https://gitter.im)
|
||||||
|
- [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/)
|
||||||
|
- [XMPP](https://xmpp.org)
|
||||||
|
- [Zulip](https://zulipchat.com)
|
||||||
|
|
||||||
|
### 3rd party via matterbridge api
|
||||||
|
|
||||||
|
- [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)
|
||||||
|
- [Minecraft](https://github.com/raws/mattercraft)
|
||||||
|
- [Minecraft](https://gitlab.com/Programie/MatterBukkit)
|
||||||
|
- [Reddit](https://github.com/bonehurtingjuice/mattereddit)
|
||||||
|
- [Counter-Strike, half-life and more](https://forums.alliedmods.net/showthread.php?t=319430)
|
||||||
|
- [MatterAMXX](https://github.com/GabeIggy/MatterAMXX)
|
||||||
|
- [Vintage Story](https://github.com/NikkyAI/vs-matterbridge)
|
||||||
|
|
||||||
### API
|
### 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).
|
More info and examples on the [wiki](https://github.com/42wim/matterbridge/wiki/Api).
|
||||||
|
|
||||||
Used by at least 3 projects. Feel free to make a PR to add your project to this list.
|
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)
|
- [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Forge server chat, archived)
|
||||||
* [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot)
|
- [MatterCraft](https://github.com/raws/mattercraft) (Matterbridge link for Minecraft Forge server chat)
|
||||||
* [Mattereddit](https://github.com/bonehurtingjuice/mattereddit) (Reddit chat support)
|
- [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)
|
||||||
|
|
||||||
## Requirements
|
## Chat with us
|
||||||
Accounts to one of the supported bridges
|
|
||||||
* [Mattermost](https://github.com/mattermost/mattermost-server/) 4.x, 5.x
|
Questions or want to test on your favorite platform? Join below:
|
||||||
* [IRC](http://www.mirc.com/servers.html)
|
|
||||||
* [XMPP](https://xmpp.org)
|
- [Discord][mb-discord]
|
||||||
* [Gitter](https://gitter.im)
|
- [Gitter][mb-gitter]
|
||||||
* [Slack](https://slack.com)
|
- [IRC][mb-irc]
|
||||||
* [Discord](https://discordapp.com)
|
- [Keybase][mb-keybase]
|
||||||
* [Telegram](https://telegram.org)
|
- [Matrix][mb-matrix]
|
||||||
* [Hipchat](https://www.hipchat.com)
|
- [Mattermost][mb-mattermost]
|
||||||
* [Rocket.chat](https://rocket.chat)
|
- [Rocket.Chat][mb-rocketchat]
|
||||||
* [Matrix](https://matrix.org)
|
- [Slack][mb-slack]
|
||||||
* [Steam](https://store.steampowered.com/)
|
- [Telegram][mb-telegram]
|
||||||
* [Twitch](https://twitch.tv)
|
- [Twitch][mb-twitch]
|
||||||
* [Ssh-chat](https://github.com/shazow/ssh-chat)
|
- [XMPP][mb-xmpp] (matterbridge@conference.jabber.de)
|
||||||
* [Zulip](https://zulipchat.com)
|
- [Zulip][mb-zulip]
|
||||||
|
|
||||||
## Screenshots
|
## Screenshots
|
||||||
See https://github.com/42wim/matterbridge/wiki
|
|
||||||
|
|
||||||
## Installing
|
See <https://github.com/42wim/matterbridge/wiki>
|
||||||
|
|
||||||
|
## Installing / upgrading
|
||||||
|
|
||||||
### Binaries
|
### Binaries
|
||||||
* Latest stable release [v1.12.2](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.22.3](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.20.0-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
|
### Packages
|
||||||
* [Overview](https://repology.org/metapackage/matterbridge/versions)
|
|
||||||
|
|
||||||
### Building
|
- [Overview](https://repology.org/metapackage/matterbridge/versions)
|
||||||
Go 1.8+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH](https://golang.org/doc/code.html#GOPATH).
|
- [snap](https://snapcraft.io/matterbridge)
|
||||||
|
- [scoop](https://github.com/42wim/scoop-bucket)
|
||||||
|
|
||||||
After Go is setup, download matterbridge to your $GOPATH directory.
|
## Building
|
||||||
|
|
||||||
```
|
Most people just want to use binaries, you can find those [here](https://github.com/42wim/matterbridge/releases/latest)
|
||||||
cd $GOPATH
|
|
||||||
|
If you really want to build from source, follow these instructions:
|
||||||
|
Go 1.13+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed.
|
||||||
|
|
||||||
|
```bash
|
||||||
go get github.com/42wim/matterbridge
|
go get github.com/42wim/matterbridge
|
||||||
```
|
```
|
||||||
|
|
||||||
You should now have matterbridge binary in the bin directory:
|
You should now have matterbridge binary in the ~/go/bin directory:
|
||||||
|
|
||||||
```
|
```bash
|
||||||
$ ls bin/
|
$ ls ~/go/bin/
|
||||||
matterbridge
|
matterbridge
|
||||||
```
|
```
|
||||||
|
|
||||||
## Configuration
|
## Configuration
|
||||||
|
|
||||||
### Basic configuration
|
### Basic configuration
|
||||||
|
|
||||||
See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration.
|
See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration.
|
||||||
|
|
||||||
|
### Settings
|
||||||
|
|
||||||
|
All possible [settings](https://github.com/42wim/matterbridge/wiki/Settings) for each bridge.
|
||||||
|
|
||||||
### Advanced configuration
|
### Advanced configuration
|
||||||
* [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example.
|
|
||||||
|
- [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example.
|
||||||
|
|
||||||
### Examples
|
### Examples
|
||||||
|
|
||||||
#### Bridge mattermost (off-topic) - irc (#testing)
|
#### Bridge mattermost (off-topic) - irc (#testing)
|
||||||
|
|
||||||
```toml
|
```toml
|
||||||
[irc]
|
[irc]
|
||||||
[irc.freenode]
|
[irc.libera]
|
||||||
Server="irc.freenode.net:6667"
|
Server="irc.libera.chat:6667"
|
||||||
Nick="yourbotname"
|
Nick="yourbotname"
|
||||||
|
|
||||||
[mattermost]
|
[mattermost]
|
||||||
@@ -144,7 +229,7 @@ See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config
|
|||||||
name="mygateway"
|
name="mygateway"
|
||||||
enable=true
|
enable=true
|
||||||
[[gateway.inout]]
|
[[gateway.inout]]
|
||||||
account="irc.freenode"
|
account="irc.libera"
|
||||||
channel="#testing"
|
channel="#testing"
|
||||||
|
|
||||||
[[gateway.inout]]
|
[[gateway.inout]]
|
||||||
@@ -153,6 +238,7 @@ enable=true
|
|||||||
```
|
```
|
||||||
|
|
||||||
#### Bridge slack (#general) - discord (general)
|
#### Bridge slack (#general) - discord (general)
|
||||||
|
|
||||||
```toml
|
```toml
|
||||||
[slack]
|
[slack]
|
||||||
[slack.test]
|
[slack.test]
|
||||||
@@ -184,7 +270,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.
|
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:
|
Usage of ./matterbridge:
|
||||||
-conf string
|
-conf string
|
||||||
config file (default "matterbridge.toml")
|
config file (default "matterbridge.toml")
|
||||||
@@ -197,68 +283,95 @@ Usage of ./matterbridge:
|
|||||||
```
|
```
|
||||||
|
|
||||||
### Docker
|
### Docker
|
||||||
Create your matterbridge.toml file locally eg in `/tmp/matterbridge.toml`
|
|
||||||
```
|
Please take a look at the [Docker Wiki page](https://github.com/42wim/matterbridge/wiki/Deploy:-Docker) for more information.
|
||||||
docker run -ti -v /tmp/matterbridge.toml:/matterbridge.toml 42wim/matterbridge
|
|
||||||
```
|
|
||||||
|
|
||||||
## Changelog
|
## Changelog
|
||||||
|
|
||||||
See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md)
|
See [changelog.md](https://github.com/42wim/matterbridge/blob/master/changelog.md)
|
||||||
|
|
||||||
## FAQ
|
## FAQ
|
||||||
|
|
||||||
See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ)
|
See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ)
|
||||||
|
|
||||||
Want to tip ?
|
|
||||||
* eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f
|
|
||||||
* btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs
|
|
||||||
|
|
||||||
## Related projects
|
## 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)
|
|
||||||
|
|
||||||
## Articles
|
- [jwflory/ansible-role-matterbridge](https://galaxy.ansible.com/jwflory/matterbridge) (Ansible role to simplify deploying Matterbridge)
|
||||||
* [matterbridge on kubernetes](https://medium.freecodecamp.org/using-kubernetes-to-deploy-a-chat-gateway-or-when-technology-works-like-its-supposed-to-a169a8cd69a3)
|
- [matterbridge autoconfig](https://github.com/patcon/matterbridge-autoconfig)
|
||||||
* https://mattermost.com/blog/connect-irc-to-mattermost/
|
- [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer)
|
||||||
* https://blog.valvin.fr/2016/09/17/mattermost-et-un-channel-irc-cest-possible/
|
- [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku)
|
||||||
* https://blog.brightscout.com/top-10-mattermost-integrations/
|
- [mattereddit](https://github.com/bonehurtingjuice/mattereddit)
|
||||||
* http://bencey.co.nz/2018/09/17/bridge/
|
- [matterlink](https://github.com/elytra/MatterLink)
|
||||||
* https://www.algoo.fr/blog/2018/01/19/recouvrez-votre-liberte-en-quittant-slack-pour-un-mattermost-auto-heberge/
|
- [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost
|
||||||
* https://kopano.com/blog/matterbridge-bridging-mattermost-chat/
|
- [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot)
|
||||||
* https://www.stitcher.com/s/?eid=52382713
|
- [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)
|
||||||
|
|
||||||
|
## 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/>
|
||||||
|
- Youtube: [whatsapp - telegram bridging](https://www.youtube.com/watch?v=W-VXISoKtNc)
|
||||||
|
|
||||||
## Thanks
|
## 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:
|
Matterbridge wouldn't exist without these libraries:
|
||||||
* discord - https://github.com/bwmarrin/discordgo
|
|
||||||
* echo - https://github.com/labstack/echo
|
- discord - <https://github.com/bwmarrin/discordgo>
|
||||||
* gitter - https://github.com/sromku/go-gitter
|
- echo - <https://github.com/labstack/echo>
|
||||||
* gops - https://github.com/google/gops
|
- gitter - <https://github.com/sromku/go-gitter>
|
||||||
* gozulipbot - https://github.com/ifo/gozulipbot
|
- gops - <https://github.com/google/gops>
|
||||||
* irc - https://github.com/lrstanley/girc
|
- gozulipbot - <https://github.com/ifo/gozulipbot>
|
||||||
* mattermost - https://github.com/mattermost/mattermost-server
|
- gumble - <https://github.com/layeh/gumble>
|
||||||
* matrix - https://github.com/matrix-org/gomatrix
|
- irc - <https://github.com/lrstanley/girc>
|
||||||
* slack - https://github.com/nlopes/slack
|
- keybase - <https://github.com/keybase/go-keybase-chat-bot>
|
||||||
* steam - https://github.com/Philipp15b/go-steam
|
- matrix - <https://github.com/matrix-org/gomatrix>
|
||||||
* telegram - https://github.com/go-telegram-bot-api/telegram-bot-api
|
- mattermost - <https://github.com/mattermost/mattermost-server>
|
||||||
* xmpp - https://github.com/mattn/go-xmpp
|
- msgraph.go - <https://github.com/yaegashi/msgraph.go>
|
||||||
* zulip - https://github.com/ifo/gozulipbot
|
- mumble - <https://github.com/layeh/gumble>
|
||||||
|
- nctalk - <https://github.com/gary-kim/go-nc-talk>
|
||||||
|
- 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>
|
||||||
|
- xmpp - <https://github.com/mattn/go-xmpp>
|
||||||
|
- zulip - <https://github.com/ifo/gozulipbot>
|
||||||
|
|
||||||
<!-- Links -->
|
<!-- Links -->
|
||||||
|
|
||||||
[mb-gitter]: https://gitter.im/42wim/matterbridge
|
[mb-discord]: https://discord.gg/AkKPtrQ
|
||||||
[mb-irc]: https://webchat.freenode.net/?channels=matterbridgechat
|
[mb-gitter]: https://gitter.im/42wim/matterbridge
|
||||||
[mb-discord]: https://discord.gg/AkKPtrQ
|
[mb-irc]: https://web.libera.chat/#matterbridge
|
||||||
[mb-matrix]: https://riot.im/app/#/room/#matterbridge:matrix.org
|
[mb-keybase]: https://keybase.io/team/matterbridge
|
||||||
[mb-slack]: https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA
|
[mb-matrix]: https://riot.im/app/#/room/#matterbridge:matrix.org
|
||||||
[mb-mattermost]: https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e
|
[mb-mattermost]: https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e
|
||||||
[mb-xmpp]: https://inverse.chat/
|
[mb-msteams]: https://teams.microsoft.com/join/hj92x75gd3y7
|
||||||
[mb-twitch]: https://www.twitch.tv/matterbridge
|
[mb-rocketchat]: https://open.rocket.chat/channel/matterbridge
|
||||||
[mb-zulip]: https://matterbridge.zulipchat.com/register/
|
[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/
|
||||||
|
|||||||
@@ -6,17 +6,20 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"gopkg.in/olahol/melody.v1"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
"github.com/labstack/echo/v4/middleware"
|
"github.com/labstack/echo/v4/middleware"
|
||||||
"github.com/zfjagann/golang-ring"
|
ring "github.com/zfjagann/golang-ring"
|
||||||
)
|
)
|
||||||
|
|
||||||
type API struct {
|
type API struct {
|
||||||
Messages ring.Ring
|
Messages ring.Ring
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
*bridge.Config
|
*bridge.Config
|
||||||
|
mrouter *melody.Melody
|
||||||
}
|
}
|
||||||
|
|
||||||
type Message struct {
|
type Message struct {
|
||||||
@@ -32,6 +35,32 @@ func New(cfg *bridge.Config) bridge.Bridger {
|
|||||||
e := echo.New()
|
e := echo.New()
|
||||||
e.HideBanner = true
|
e.HideBanner = true
|
||||||
e.HidePort = 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{}
|
b.Messages = ring.Ring{}
|
||||||
if b.GetInt("Buffer") != 0 {
|
if b.GetInt("Buffer") != 0 {
|
||||||
b.Messages.SetCapacity(b.GetInt("Buffer"))
|
b.Messages.SetCapacity(b.GetInt("Buffer"))
|
||||||
@@ -41,9 +70,17 @@ func New(cfg *bridge.Config) bridge.Bridger {
|
|||||||
return key == b.GetString("Token"), nil
|
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/health", b.handleHealthcheck)
|
||||||
e.GET("/api/messages", b.handleMessages)
|
e.GET("/api/messages", b.handleMessages)
|
||||||
e.GET("/api/stream", b.handleStream)
|
e.GET("/api/stream", b.handleStream)
|
||||||
|
e.GET("/api/websocket", b.handleWebsocket)
|
||||||
e.POST("/api/message", b.handlePostMessage)
|
e.POST("/api/message", b.handlePostMessage)
|
||||||
go func() {
|
go func() {
|
||||||
if b.GetString("BindAddress") == "" {
|
if b.GetString("BindAddress") == "" {
|
||||||
@@ -58,13 +95,13 @@ func New(cfg *bridge.Config) bridge.Bridger {
|
|||||||
func (b *API) Connect() error {
|
func (b *API) Connect() error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *API) Disconnect() error {
|
func (b *API) Disconnect() error {
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *API) JoinChannel(channel config.ChannelInfo) error {
|
func (b *API) JoinChannel(channel config.ChannelInfo) error {
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *API) Send(msg config.Message) (string, error) {
|
func (b *API) Send(msg config.Message) (string, error) {
|
||||||
@@ -74,7 +111,14 @@ func (b *API) Send(msg config.Message) (string, error) {
|
|||||||
if msg.Event == config.EventMsgDelete {
|
if msg.Event == config.EventMsgDelete {
|
||||||
return "", nil
|
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
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -106,18 +150,23 @@ func (b *API) handleMessages(c echo.Context) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *API) handleStream(c echo.Context) error {
|
func (b *API) getGreeting() config.Message {
|
||||||
c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
return config.Message{
|
||||||
c.Response().WriteHeader(http.StatusOK)
|
|
||||||
greet := config.Message{
|
|
||||||
Event: config.EventAPIConnected,
|
Event: config.EventAPIConnected,
|
||||||
Timestamp: time.Now(),
|
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 {
|
if err := json.NewEncoder(c.Response()).Encode(greet); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
c.Response().Flush()
|
c.Response().Flush()
|
||||||
for {
|
for {
|
||||||
|
// TODO: this causes issues, messages should be broadcasted to all connected clients
|
||||||
msg := b.Messages.Dequeue()
|
msg := b.Messages.Dequeue()
|
||||||
if msg != nil {
|
if msg != nil {
|
||||||
if err := json.NewEncoder(c.Response()).Encode(msg); err != nil {
|
if err := json.NewEncoder(c.Response()).Encode(msg); err != nil {
|
||||||
@@ -128,3 +177,31 @@ func (b *API) handleStream(c echo.Context) error {
|
|||||||
time.Sleep(200 * time.Millisecond)
|
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,11 +1,13 @@
|
|||||||
package bridge
|
package bridge
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"log"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"sync"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Bridger interface {
|
type Bridger interface {
|
||||||
@@ -17,6 +19,8 @@ type Bridger interface {
|
|||||||
|
|
||||||
type Bridge struct {
|
type Bridge struct {
|
||||||
Bridger
|
Bridger
|
||||||
|
*sync.RWMutex
|
||||||
|
|
||||||
Name string
|
Name string
|
||||||
Account string
|
Account string
|
||||||
Protocol string
|
Protocol string
|
||||||
@@ -26,37 +30,38 @@ type Bridge struct {
|
|||||||
Log *logrus.Entry
|
Log *logrus.Entry
|
||||||
Config config.Config
|
Config config.Config
|
||||||
General *config.Protocol
|
General *config.Protocol
|
||||||
*sync.RWMutex
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type Config struct {
|
type Config struct {
|
||||||
// General *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
Log *logrus.Entry
|
|
||||||
*Bridge
|
*Bridge
|
||||||
|
|
||||||
|
Remote chan config.Message
|
||||||
}
|
}
|
||||||
|
|
||||||
// Factory is the factory function to create a bridge
|
// Factory is the factory function to create a bridge
|
||||||
type Factory func(*Config) Bridger
|
type Factory func(*Config) Bridger
|
||||||
|
|
||||||
func New(bridge *config.Bridge) *Bridge {
|
func New(bridge *config.Bridge) *Bridge {
|
||||||
b := &Bridge{
|
|
||||||
Channels: make(map[string]config.ChannelInfo),
|
|
||||||
RWMutex: new(sync.RWMutex),
|
|
||||||
Joined: make(map[string]bool),
|
|
||||||
}
|
|
||||||
accInfo := strings.Split(bridge.Account, ".")
|
accInfo := strings.Split(bridge.Account, ".")
|
||||||
|
if len(accInfo) != 2 {
|
||||||
|
log.Fatalf("config failure, account incorrect: %s", bridge.Account)
|
||||||
|
}
|
||||||
|
|
||||||
protocol := accInfo[0]
|
protocol := accInfo[0]
|
||||||
name := accInfo[1]
|
name := accInfo[1]
|
||||||
b.Name = name
|
|
||||||
b.Protocol = protocol
|
return &Bridge{
|
||||||
b.Account = bridge.Account
|
RWMutex: new(sync.RWMutex),
|
||||||
return b
|
Channels: make(map[string]config.ChannelInfo),
|
||||||
|
Name: name,
|
||||||
|
Protocol: protocol,
|
||||||
|
Account: bridge.Account,
|
||||||
|
Joined: make(map[string]bool),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bridge) JoinChannels() error {
|
func (b *Bridge) JoinChannels() error {
|
||||||
err := b.joinChannels(b.Channels, b.Joined)
|
return b.joinChannels(b.Channels, b.Joined)
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetChannelMembers sets the newMembers to the bridge ChannelMembers
|
// SetChannelMembers sets the newMembers to the bridge ChannelMembers
|
||||||
@@ -70,6 +75,7 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map
|
|||||||
for ID, channel := range channels {
|
for ID, channel := range channels {
|
||||||
if !exists[ID] {
|
if !exists[ID] {
|
||||||
b.Log.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, 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)
|
err := b.JoinChannel(channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -80,8 +86,16 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map
|
|||||||
return nil
|
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 {
|
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 {
|
if !ok {
|
||||||
val, _ = b.Config.GetBool("general." + key)
|
val, _ = b.Config.GetBool("general." + key)
|
||||||
}
|
}
|
||||||
@@ -89,7 +103,7 @@ func (b *Bridge) GetBool(key string) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bridge) GetInt(key string) int {
|
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 {
|
if !ok {
|
||||||
val, _ = b.Config.GetInt("general." + key)
|
val, _ = b.Config.GetInt("general." + key)
|
||||||
}
|
}
|
||||||
@@ -97,7 +111,7 @@ func (b *Bridge) GetInt(key string) int {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bridge) GetString(key string) string {
|
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 {
|
if !ok {
|
||||||
val, _ = b.Config.GetString("general." + key)
|
val, _ = b.Config.GetString("general." + key)
|
||||||
}
|
}
|
||||||
@@ -105,7 +119,7 @@ func (b *Bridge) GetString(key string) string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bridge) GetStringSlice(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 {
|
if !ok {
|
||||||
val, _ = b.Config.GetStringSlice("general." + key)
|
val, _ = b.Config.GetStringSlice("general." + key)
|
||||||
}
|
}
|
||||||
@@ -113,7 +127,7 @@ func (b *Bridge) GetStringSlice(key string) []string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bridge) GetStringSlice2D(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 {
|
if !ok {
|
||||||
val, _ = b.Config.GetStringSlice2D("general." + key)
|
val, _ = b.Config.GetStringSlice2D("general." + key)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -3,12 +3,13 @@ package config
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/fsnotify/fsnotify"
|
"github.com/fsnotify/fsnotify"
|
||||||
prefixed "github.com/matterbridge/logrus-prefixed-formatter"
|
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
)
|
)
|
||||||
@@ -25,8 +26,11 @@ const (
|
|||||||
EventAPIConnected = "api_connected"
|
EventAPIConnected = "api_connected"
|
||||||
EventUserTyping = "user_typing"
|
EventUserTyping = "user_typing"
|
||||||
EventGetChannelMembers = "get_channel_members"
|
EventGetChannelMembers = "get_channel_members"
|
||||||
|
EventNoticeIRC = "notice_irc"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const ParentIDNotFound = "msg-parent-not-found"
|
||||||
|
|
||||||
type Message struct {
|
type Message struct {
|
||||||
Text string `json:"text"`
|
Text string `json:"text"`
|
||||||
Channel string `json:"channel"`
|
Channel string `json:"channel"`
|
||||||
@@ -43,6 +47,14 @@ type Message struct {
|
|||||||
Extra map[string][]interface{}
|
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 {
|
type FileInfo struct {
|
||||||
Name string
|
Name string
|
||||||
Data *[]byte
|
Data *[]byte
|
||||||
@@ -73,33 +85,42 @@ type ChannelMember struct {
|
|||||||
type ChannelMembers []ChannelMember
|
type ChannelMembers []ChannelMember
|
||||||
|
|
||||||
type Protocol struct {
|
type Protocol struct {
|
||||||
AuthCode string // steam
|
AllowMention []string // discord
|
||||||
BindAddress string // mattermost, slack // DEPRECATED
|
AuthCode string // steam
|
||||||
Buffer int // api
|
BindAddress string // mattermost, slack // DEPRECATED
|
||||||
Charset string // irc
|
Buffer int // api
|
||||||
ColorNicks bool // only irc for now
|
Charset string // irc
|
||||||
Debug bool // general
|
ClientID string // msteams
|
||||||
DebugLevel int // only for irc now
|
ColorNicks bool // only irc for now
|
||||||
EditSuffix string // mattermost, slack, discord, telegram, gitter
|
Debug bool // general
|
||||||
EditDisable bool // mattermost, slack, discord, telegram, gitter
|
DebugLevel int // only for irc now
|
||||||
IconURL string // mattermost, slack
|
DisableWebPagePreview bool // telegram
|
||||||
IgnoreFailureOnStart bool // general
|
EditSuffix string // mattermost, slack, discord, telegram, gitter
|
||||||
IgnoreNicks string // all protocols
|
EditDisable bool // mattermost, slack, discord, telegram, gitter
|
||||||
IgnoreMessages string // all protocols
|
HTMLDisable bool // matrix
|
||||||
Jid string // xmpp
|
IconURL string // mattermost, slack
|
||||||
Label string // all protocols
|
IgnoreFailureOnStart bool // general
|
||||||
Login string // mattermost, matrix
|
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
|
MediaDownloadBlackList []string
|
||||||
MediaDownloadPath string // Basically MediaServerUpload, but instead of uploading it, just write it to a file on the same server.
|
MediaDownloadPath string // Basically MediaServerUpload, but instead of uploading it, just write it to a file on the same server.
|
||||||
MediaDownloadSize int // all protocols
|
MediaDownloadSize int // all protocols
|
||||||
MediaServerDownload string
|
MediaServerDownload string
|
||||||
MediaServerUpload string
|
MediaServerUpload string
|
||||||
|
MediaConvertTgs string // telegram
|
||||||
|
MediaConvertWebPToPNG bool // telegram
|
||||||
MessageDelay int // IRC, time in millisecond to wait between messages
|
MessageDelay int // IRC, time in millisecond to wait between messages
|
||||||
MessageFormat string // telegram
|
MessageFormat string // telegram
|
||||||
MessageLength int // IRC, max length of a message allowed
|
MessageLength int // IRC, max length of a message allowed
|
||||||
MessageQueue int // IRC, size of message queue for flood control
|
MessageQueue int // IRC, size of message queue for flood control
|
||||||
MessageSplit bool // IRC, split long messages with newlines on MessageLength instead of clipping
|
MessageSplit bool // IRC, split long messages with newlines on MessageLength instead of clipping
|
||||||
Muc string // xmpp
|
Muc string // xmpp
|
||||||
|
MxID string // matrix
|
||||||
Name string // all protocols
|
Name string // all protocols
|
||||||
Nick string // all protocols
|
Nick string // all protocols
|
||||||
NickFormatter string // mattermost, slack
|
NickFormatter string // mattermost, slack
|
||||||
@@ -109,36 +130,46 @@ type Protocol struct {
|
|||||||
NicksPerRow int // mattermost, slack
|
NicksPerRow int // mattermost, slack
|
||||||
NoHomeServerSuffix bool // matrix
|
NoHomeServerSuffix bool // matrix
|
||||||
NoSendJoinPart bool // all protocols
|
NoSendJoinPart bool // all protocols
|
||||||
NoTLS bool // mattermost
|
NoTLS bool // mattermost, xmpp
|
||||||
Password string // IRC,mattermost,XMPP,matrix
|
Password string // IRC,mattermost,XMPP,matrix
|
||||||
PrefixMessagesWithNick bool // mattemost, slack
|
PrefixMessagesWithNick bool // mattemost, slack
|
||||||
PreserveThreading bool // slack
|
PreserveThreading bool // slack
|
||||||
Protocol string // all protocols
|
Protocol string // all protocols
|
||||||
QuoteDisable bool // telegram
|
QuoteDisable bool // telegram
|
||||||
QuoteFormat string // telegram
|
QuoteFormat string // telegram
|
||||||
|
QuoteLengthLimit int // telegram
|
||||||
RejoinDelay int // IRC
|
RejoinDelay int // IRC
|
||||||
ReplaceMessages [][]string // all protocols
|
ReplaceMessages [][]string // all protocols
|
||||||
ReplaceNicks [][]string // all protocols
|
ReplaceNicks [][]string // all protocols
|
||||||
RemoteNickFormat string // all protocols
|
RemoteNickFormat string // all protocols
|
||||||
RunCommands []string // irc
|
RunCommands []string // IRC
|
||||||
Server string // IRC,mattermost,XMPP,discord
|
Server string // IRC,mattermost,XMPP,discord,matrix
|
||||||
|
SessionFile string // msteams,whatsapp
|
||||||
ShowJoinPart bool // all protocols
|
ShowJoinPart bool // all protocols
|
||||||
ShowTopicChange bool // slack
|
ShowTopicChange bool // slack
|
||||||
ShowUserTyping bool // slack
|
ShowUserTyping bool // slack
|
||||||
ShowEmbeds bool // discord
|
ShowEmbeds bool // discord
|
||||||
SkipTLSVerify bool // IRC, mattermost
|
SkipTLSVerify bool // IRC, mattermost
|
||||||
|
SkipVersionCheck bool // mattermost
|
||||||
StripNick bool // all protocols
|
StripNick bool // all protocols
|
||||||
|
StripMarkdown bool // irc
|
||||||
SyncTopic bool // slack
|
SyncTopic bool // slack
|
||||||
Team string // mattermost
|
TengoModifyMessage string // general
|
||||||
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
|
Topic string // zulip
|
||||||
URL string // mattermost, slack // DEPRECATED
|
URL string // mattermost, slack // DEPRECATED
|
||||||
UseAPI bool // mattermost, slack
|
UseAPI bool // mattermost, slack
|
||||||
|
UseLocalAvatar []string // discord
|
||||||
UseSASL bool // IRC
|
UseSASL bool // IRC
|
||||||
UseTLS bool // IRC
|
UseTLS bool // IRC
|
||||||
|
UseDiscriminator bool // discord
|
||||||
UseFirstName bool // telegram
|
UseFirstName bool // telegram
|
||||||
UseUserName bool // discord
|
UseUserName bool // discord, matrix
|
||||||
UseInsecureURL bool // telegram
|
UseInsecureURL bool // telegram
|
||||||
|
VerboseJoinPart bool // IRC
|
||||||
WebhookBindAddress string // mattermost, slack
|
WebhookBindAddress string // mattermost, slack
|
||||||
WebhookURL string // mattermost, slack
|
WebhookURL string // mattermost, slack
|
||||||
}
|
}
|
||||||
@@ -146,6 +177,7 @@ type Protocol struct {
|
|||||||
type ChannelOptions struct {
|
type ChannelOptions struct {
|
||||||
Key string // irc, xmpp
|
Key string // irc, xmpp
|
||||||
WebhookURL string // discord
|
WebhookURL string // discord
|
||||||
|
Topic string // zulip
|
||||||
}
|
}
|
||||||
|
|
||||||
type Bridge struct {
|
type Bridge struct {
|
||||||
@@ -163,6 +195,13 @@ type Gateway struct {
|
|||||||
InOut []Bridge
|
InOut []Bridge
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Tengo struct {
|
||||||
|
InMessage string
|
||||||
|
Message string
|
||||||
|
RemoteNickFormat string
|
||||||
|
OutMessage string
|
||||||
|
}
|
||||||
|
|
||||||
type SameChannelGateway struct {
|
type SameChannelGateway struct {
|
||||||
Name string
|
Name string
|
||||||
Enable bool
|
Enable bool
|
||||||
@@ -184,14 +223,20 @@ type BridgeValues struct {
|
|||||||
Telegram map[string]Protocol
|
Telegram map[string]Protocol
|
||||||
Rocketchat map[string]Protocol
|
Rocketchat map[string]Protocol
|
||||||
SSHChat map[string]Protocol
|
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
|
Zulip map[string]Protocol
|
||||||
|
Keybase map[string]Protocol
|
||||||
|
Mumble map[string]Protocol
|
||||||
General Protocol
|
General Protocol
|
||||||
|
Tengo Tengo
|
||||||
Gateway []Gateway
|
Gateway []Gateway
|
||||||
SameChannelGateway []SameChannelGateway
|
SameChannelGateway []SameChannelGateway
|
||||||
}
|
}
|
||||||
|
|
||||||
type Config interface {
|
type Config interface {
|
||||||
|
Viper() *viper.Viper
|
||||||
BridgeValues() *BridgeValues
|
BridgeValues() *BridgeValues
|
||||||
|
IsKeySet(key string) bool
|
||||||
GetBool(key string) (bool, bool)
|
GetBool(key string) (bool, bool)
|
||||||
GetInt(key string) (int, bool)
|
GetInt(key string) (int, bool)
|
||||||
GetString(key string) (string, bool)
|
GetString(key string) (string, bool)
|
||||||
@@ -200,63 +245,80 @@ type Config interface {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type config struct {
|
type config struct {
|
||||||
v *viper.Viper
|
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
|
||||||
cv *BridgeValues
|
logger *logrus.Entry
|
||||||
|
v *viper.Viper
|
||||||
|
cv *BridgeValues
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewConfig(cfgfile string) Config {
|
// NewConfig instantiates a new configuration based on the specified configuration file path.
|
||||||
logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false})
|
func NewConfig(rootLogger *logrus.Logger, cfgfile string) Config {
|
||||||
flog := logrus.WithFields(logrus.Fields{"prefix": "config"})
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "config"})
|
||||||
|
|
||||||
viper.SetConfigFile(cfgfile)
|
viper.SetConfigFile(cfgfile)
|
||||||
input, err := getFileContents(cfgfile)
|
input, err := ioutil.ReadFile(cfgfile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
logrus.Fatal(err)
|
logger.Fatalf("Failed to read configuration file: %#v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
mycfg := newConfigFromString(input)
|
|
||||||
if mycfg.cv.General.MediaDownloadSize == 0 {
|
if mycfg.cv.General.MediaDownloadSize == 0 {
|
||||||
mycfg.cv.General.MediaDownloadSize = 1000000
|
mycfg.cv.General.MediaDownloadSize = 1000000
|
||||||
}
|
}
|
||||||
viper.WatchConfig()
|
viper.WatchConfig()
|
||||||
viper.OnConfigChange(func(e fsnotify.Event) {
|
viper.OnConfigChange(func(e fsnotify.Event) {
|
||||||
flog.Println("Config file changed:", e.Name)
|
logger.Println("Config file changed:", e.Name)
|
||||||
})
|
})
|
||||||
return mycfg
|
return mycfg
|
||||||
}
|
}
|
||||||
|
|
||||||
func getFileContents(filename string) ([]byte, error) {
|
// detectConfigType detects JSON and YAML formats, defaults to TOML.
|
||||||
input, err := ioutil.ReadFile(filename)
|
func detectConfigType(cfgfile string) string {
|
||||||
if err != nil {
|
fileExt := filepath.Ext(cfgfile)
|
||||||
logrus.Fatal(err)
|
switch fileExt {
|
||||||
return []byte(nil), err
|
case ".json":
|
||||||
|
return "json"
|
||||||
|
case ".yaml", ".yml":
|
||||||
|
return "yaml"
|
||||||
}
|
}
|
||||||
return input, nil
|
return "toml"
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewConfigFromString(input []byte) Config {
|
// NewConfigFromString instantiates a new configuration based on the specified string.
|
||||||
return newConfigFromString(input)
|
func NewConfigFromString(rootLogger *logrus.Logger, input []byte) Config {
|
||||||
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "config"})
|
||||||
|
return newConfigFromString(logger, input, "toml")
|
||||||
}
|
}
|
||||||
|
|
||||||
func newConfigFromString(input []byte) *config {
|
func newConfigFromString(logger *logrus.Entry, input []byte, cfgtype string) *config {
|
||||||
viper.SetConfigType("toml")
|
viper.SetConfigType(cfgtype)
|
||||||
viper.SetEnvPrefix("matterbridge")
|
viper.SetEnvPrefix("matterbridge")
|
||||||
viper.AddConfigPath(".")
|
|
||||||
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
|
viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
|
||||||
viper.AutomaticEnv()
|
viper.AutomaticEnv()
|
||||||
err := viper.ReadConfig(bytes.NewBuffer(input))
|
|
||||||
if err != nil {
|
if err := viper.ReadConfig(bytes.NewBuffer(input)); err != nil {
|
||||||
logrus.Fatal(err)
|
logger.Fatalf("Failed to parse the configuration: %s", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
cfg := &BridgeValues{}
|
cfg := &BridgeValues{}
|
||||||
err = viper.Unmarshal(cfg)
|
if err := viper.Unmarshal(cfg); err != nil {
|
||||||
if err != nil {
|
logger.Fatalf("Failed to load the configuration: %s", err)
|
||||||
logrus.Fatal(err)
|
|
||||||
}
|
}
|
||||||
return &config{
|
return &config{
|
||||||
v: viper.GetViper(),
|
logger: logger,
|
||||||
cv: cfg,
|
v: viper.GetViper(),
|
||||||
|
cv: cfg,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -264,49 +326,57 @@ func (c *config) BridgeValues() *BridgeValues {
|
|||||||
return c.cv
|
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) {
|
func (c *config) GetBool(key string) (bool, bool) {
|
||||||
c.RLock()
|
c.RLock()
|
||||||
defer c.RUnlock()
|
defer c.RUnlock()
|
||||||
// log.Debugf("getting bool %s = %#v", key, c.v.GetBool(key))
|
|
||||||
return c.v.GetBool(key), c.v.IsSet(key)
|
return c.v.GetBool(key), c.v.IsSet(key)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *config) GetInt(key string) (int, bool) {
|
func (c *config) GetInt(key string) (int, bool) {
|
||||||
c.RLock()
|
c.RLock()
|
||||||
defer c.RUnlock()
|
defer c.RUnlock()
|
||||||
// log.Debugf("getting int %s = %d", key, c.v.GetInt(key))
|
|
||||||
return c.v.GetInt(key), c.v.IsSet(key)
|
return c.v.GetInt(key), c.v.IsSet(key)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *config) GetString(key string) (string, bool) {
|
func (c *config) GetString(key string) (string, bool) {
|
||||||
c.RLock()
|
c.RLock()
|
||||||
defer c.RUnlock()
|
defer c.RUnlock()
|
||||||
// log.Debugf("getting String %s = %s", key, c.v.GetString(key))
|
|
||||||
return c.v.GetString(key), c.v.IsSet(key)
|
return c.v.GetString(key), c.v.IsSet(key)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *config) GetStringSlice(key string) ([]string, bool) {
|
func (c *config) GetStringSlice(key string) ([]string, bool) {
|
||||||
c.RLock()
|
c.RLock()
|
||||||
defer c.RUnlock()
|
defer c.RUnlock()
|
||||||
// log.Debugf("getting StringSlice %s = %#v", key, c.v.GetStringSlice(key))
|
|
||||||
return c.v.GetStringSlice(key), c.v.IsSet(key)
|
return c.v.GetStringSlice(key), c.v.IsSet(key)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *config) GetStringSlice2D(key string) ([][]string, bool) {
|
func (c *config) GetStringSlice2D(key string) ([][]string, bool) {
|
||||||
c.RLock()
|
c.RLock()
|
||||||
defer c.RUnlock()
|
defer c.RUnlock()
|
||||||
result := [][]string{}
|
|
||||||
if res, ok := c.v.Get(key).([]interface{}); ok {
|
res, ok := c.v.Get(key).([]interface{})
|
||||||
for _, entry := range res {
|
if !ok {
|
||||||
result2 := []string{}
|
return nil, false
|
||||||
for _, entry2 := range entry.([]interface{}) {
|
|
||||||
result2 = append(result2, entry2.(string))
|
|
||||||
}
|
|
||||||
result = append(result, result2)
|
|
||||||
}
|
|
||||||
return result, true
|
|
||||||
}
|
}
|
||||||
return result, false
|
var result [][]string
|
||||||
|
for _, entry := range res {
|
||||||
|
result2 := []string{}
|
||||||
|
for _, entry2 := range entry.([]interface{}) {
|
||||||
|
result2 = append(result2, entry2.(string))
|
||||||
|
}
|
||||||
|
result = append(result, result2)
|
||||||
|
}
|
||||||
|
return result, true
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetIconURL(msg *Message, iconURL string) string {
|
func GetIconURL(msg *Message, iconURL string) string {
|
||||||
@@ -325,6 +395,11 @@ type TestConfig struct {
|
|||||||
Overrides map[string]interface{}
|
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) {
|
func (c *TestConfig) GetBool(key string) (bool, bool) {
|
||||||
val, ok := c.Overrides[key]
|
val, ok := c.Overrides[key]
|
||||||
if ok {
|
if ok {
|
||||||
|
|||||||
@@ -2,15 +2,15 @@ package bdiscord
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/discord/transmitter"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/matterbridge/discordgo"
|
||||||
)
|
)
|
||||||
|
|
||||||
const MessageLength = 1950
|
const MessageLength = 1950
|
||||||
@@ -20,12 +20,9 @@ type Bdiscord struct {
|
|||||||
|
|
||||||
c *discordgo.Session
|
c *discordgo.Session
|
||||||
|
|
||||||
nick string
|
nick string
|
||||||
useChannelID bool
|
userID string
|
||||||
guildID string
|
guildID string
|
||||||
webhookID string
|
|
||||||
webhookToken string
|
|
||||||
canEditWebhooks bool
|
|
||||||
|
|
||||||
channelsMutex sync.RWMutex
|
channelsMutex sync.RWMutex
|
||||||
channels []*discordgo.Channel
|
channels []*discordgo.Channel
|
||||||
@@ -34,6 +31,10 @@ type Bdiscord struct {
|
|||||||
membersMutex sync.RWMutex
|
membersMutex sync.RWMutex
|
||||||
userMemberMap map[string]*discordgo.Member
|
userMemberMap map[string]*discordgo.Member
|
||||||
nickMemberMap map[string]*discordgo.Member
|
nickMemberMap map[string]*discordgo.Member
|
||||||
|
|
||||||
|
// Webhook specific logic
|
||||||
|
useAutoWebhooks bool
|
||||||
|
transmitter *transmitter.Transmitter
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
@@ -41,23 +42,18 @@ func New(cfg *bridge.Config) bridge.Bridger {
|
|||||||
b.userMemberMap = make(map[string]*discordgo.Member)
|
b.userMemberMap = make(map[string]*discordgo.Member)
|
||||||
b.nickMemberMap = make(map[string]*discordgo.Member)
|
b.nickMemberMap = make(map[string]*discordgo.Member)
|
||||||
b.channelInfoMap = make(map[string]*config.ChannelInfo)
|
b.channelInfoMap = make(map[string]*config.ChannelInfo)
|
||||||
if b.GetString("WebhookURL") != "" {
|
|
||||||
b.Log.Debug("Configuring Discord Incoming Webhook")
|
b.useAutoWebhooks = b.GetBool("AutoWebhooks")
|
||||||
b.webhookID, b.webhookToken = b.splitURL(b.GetString("WebhookURL"))
|
if b.useAutoWebhooks {
|
||||||
|
b.Log.Debug("Using automatic webhooks")
|
||||||
}
|
}
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bdiscord) Connect() error {
|
func (b *Bdiscord) Connect() error {
|
||||||
var err error
|
var err error
|
||||||
var guildFound bool
|
|
||||||
token := b.GetString("Token")
|
token := b.GetString("Token")
|
||||||
b.Log.Info("Connecting")
|
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 ") {
|
if !strings.HasPrefix(b.GetString("Token"), "Bot ") {
|
||||||
token = "Bot " + b.GetString("Token")
|
token = "Bot " + b.GetString("Token")
|
||||||
}
|
}
|
||||||
@@ -72,9 +68,18 @@ func (b *Bdiscord) Connect() error {
|
|||||||
}
|
}
|
||||||
b.Log.Info("Connection succeeded")
|
b.Log.Info("Connection succeeded")
|
||||||
b.c.AddHandler(b.messageCreate)
|
b.c.AddHandler(b.messageCreate)
|
||||||
|
b.c.AddHandler(b.messageTyping)
|
||||||
b.c.AddHandler(b.memberUpdate)
|
b.c.AddHandler(b.memberUpdate)
|
||||||
b.c.AddHandler(b.messageUpdate)
|
b.c.AddHandler(b.messageUpdate)
|
||||||
b.c.AddHandler(b.messageDelete)
|
b.c.AddHandler(b.messageDelete)
|
||||||
|
b.c.AddHandler(b.messageDeleteBulk)
|
||||||
|
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()
|
err = b.c.Open()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -89,55 +94,108 @@ func (b *Bdiscord) Connect() error {
|
|||||||
}
|
}
|
||||||
serverName := strings.Replace(b.GetString("Server"), "ID:", "", -1)
|
serverName := strings.Replace(b.GetString("Server"), "ID:", "", -1)
|
||||||
b.nick = userinfo.Username
|
b.nick = userinfo.Username
|
||||||
|
b.userID = userinfo.ID
|
||||||
|
|
||||||
|
// Try and find this account's guild, and populate channels
|
||||||
b.channelsMutex.Lock()
|
b.channelsMutex.Lock()
|
||||||
for _, guild := range guilds {
|
for _, guild := range guilds {
|
||||||
if guild.Name == serverName || guild.ID == serverName {
|
// Skip, if the server name does not match the visible name or the ID
|
||||||
b.channels, err = b.c.GuildChannels(guild.ID)
|
if guild.Name != serverName && guild.ID != serverName {
|
||||||
b.guildID = guild.ID
|
continue
|
||||||
guildFound = true
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 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()
|
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
|
return err
|
||||||
}
|
}
|
||||||
b.channelsMutex.RLock()
|
|
||||||
if b.GetString("WebhookURL") == "" {
|
// Legacy note: WebhookURL used to have an actual webhook URL that we would edit,
|
||||||
for _, channel := range b.channels {
|
// but we stopped doing that due to Discord making rate limits more aggressive.
|
||||||
b.Log.Debugf("found channel %#v", channel)
|
//
|
||||||
}
|
// Even older: the same WebhookURL used to be used by every channel, which is usually unexpected.
|
||||||
} else {
|
// This is no longer possible.
|
||||||
b.canEditWebhooks = true
|
if b.GetString("WebhookURL") != "" {
|
||||||
for _, channel := range b.channels {
|
message := "The global WebhookURL setting has been removed. "
|
||||||
b.Log.Debugf("found channel %#v; verifying PermissionManageWebhooks", channel)
|
message += "You can get similar \"webhook editing\" behaviour by replacing this line with `AutoWebhooks=true`. "
|
||||||
perms, permsErr := b.c.State.UserChannelPermissions(userinfo.ID, channel.ID)
|
message += "If you rely on the old-OLD (non-editing) behaviour, can move the WebhookURL to specific channel sections."
|
||||||
manageWebhooks := discordgo.PermissionManageWebhooks
|
b.Log.Errorln(message)
|
||||||
if permsErr != nil || perms&manageWebhooks != manageWebhooks {
|
return fmt.Errorf("use of removed WebhookURL setting")
|
||||||
b.Log.Warnf("Can't manage webhooks in channel \"%s\"", channel.Name)
|
}
|
||||||
b.canEditWebhooks = false
|
|
||||||
|
if b.GetInt("debuglevel") > 0 {
|
||||||
|
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")
|
whID, whToken, ok := b.splitURL(channel.Options.WebhookURL)
|
||||||
} else {
|
if !ok {
|
||||||
b.Log.Warn("Can't manage webhooks; won't edit channel for global webhook on send")
|
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.
|
// Obtaining guild members and initializing nickname mapping.
|
||||||
b.membersMutex.Lock()
|
b.membersMutex.Lock()
|
||||||
@@ -170,10 +228,6 @@ func (b *Bdiscord) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
defer b.channelsMutex.Unlock()
|
defer b.channelsMutex.Unlock()
|
||||||
|
|
||||||
b.channelInfoMap[channel.ID] = &channel
|
b.channelInfoMap[channel.ID] = &channel
|
||||||
idcheck := strings.Split(channel.Name, "ID:")
|
|
||||||
if len(idcheck) > 1 {
|
|
||||||
b.useChannelID = true
|
|
||||||
}
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -185,81 +239,36 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) {
|
|||||||
return "", fmt.Errorf("Could not find channelID for %v", msg.Channel)
|
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
|
// Make a action /me of the message
|
||||||
if msg.Event == config.EventUserAction {
|
if msg.Event == config.EventUserAction {
|
||||||
msg.Text = "_" + msg.Text + "_"
|
msg.Text = "_" + msg.Text + "_"
|
||||||
}
|
}
|
||||||
|
|
||||||
// use initial webhook configured for the entire Discord account
|
// Handle prefix hint for unthreaded messages.
|
||||||
isGlobalWebhook := true
|
if msg.ParentNotFound() {
|
||||||
wID := b.webhookID
|
msg.ParentID = ""
|
||||||
wToken := b.webhookToken
|
msg.Text = fmt.Sprintf("[thread]: %s", msg.Text)
|
||||||
|
|
||||||
// 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
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
b.channelsMutex.RUnlock()
|
|
||||||
|
|
||||||
// Use webhook to send the message
|
// Use webhook to send the message
|
||||||
if wID != "" {
|
useWebhooks := b.shouldMessageUseWebhooks(&msg)
|
||||||
// skip events
|
if useWebhooks && msg.Event != config.EventMsgDelete && msg.ParentID == "" {
|
||||||
if msg.Event != "" && msg.Event != config.EventJoinLeave && msg.Event != config.EventTopicChange {
|
return b.handleEventWebhook(&msg, channelID)
|
||||||
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: %v", err)
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
err := b.c.WebhookExecute(
|
|
||||||
wID,
|
|
||||||
wToken,
|
|
||||||
true,
|
|
||||||
&discordgo.WebhookParams{
|
|
||||||
Content: msg.Text,
|
|
||||||
Username: msg.Username,
|
|
||||||
AvatarURL: msg.Avatar,
|
|
||||||
})
|
|
||||||
return "", err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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)")
|
b.Log.Debugf("Broadcasting using token (API)")
|
||||||
|
|
||||||
// Delete message
|
// Delete message
|
||||||
@@ -273,19 +282,19 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) {
|
|||||||
|
|
||||||
// Upload a file if it exists
|
// Upload a file if it exists
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
for _, rmsg := range helper.HandleExtra(msg, b.General) {
|
||||||
rmsg.Text = helper.ClipMessage(rmsg.Text, MessageLength)
|
rmsg.Text = helper.ClipMessage(rmsg.Text, MessageLength, b.GetString("MessageClipped"))
|
||||||
if _, err := b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text); err != nil {
|
if _, err := b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text); err != nil {
|
||||||
b.Log.Errorf("Could not send message %#v: %v", rmsg, err)
|
b.Log.Errorf("Could not send message %#v: %s", rmsg, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// check if we have files to upload (from slack, telegram or mattermost)
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
if len(msg.Extra["file"]) > 0 {
|
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 = helper.ClipMessage(msg.Text, MessageLength, b.GetString("MessageClipped"))
|
||||||
msg.Text = b.replaceUserMentions(msg.Text)
|
msg.Text = b.replaceUserMentions(msg.Text)
|
||||||
|
|
||||||
// Edit message
|
// Edit message
|
||||||
@@ -294,52 +303,26 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) {
|
|||||||
return msg.ID, err
|
return msg.ID, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
m := discordgo.MessageSend{
|
||||||
|
Content: msg.Username + msg.Text,
|
||||||
|
AllowedMentions: b.getAllowedMentions(),
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.ParentValid() {
|
||||||
|
m.Reference = &discordgo.MessageReference{
|
||||||
|
MessageID: msg.ParentID,
|
||||||
|
ChannelID: channelID,
|
||||||
|
GuildID: b.guildID,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Post normal message
|
// Post normal message
|
||||||
res, err := b.c.ChannelMessageSend(channelID, msg.Username+msg.Text)
|
res, err := b.c.ChannelMessageSendComplex(channelID, &m)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
return res.ID, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// useWebhook returns true if we have a webhook defined somewhere
|
return res.ID, nil
|
||||||
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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// handleUploadFile handles native upload of files
|
// handleUploadFile handles native upload of files
|
||||||
@@ -353,12 +336,13 @@ func (b *Bdiscord) handleUploadFile(msg *config.Message, channelID string) (stri
|
|||||||
Reader: bytes.NewReader(*fi.Data),
|
Reader: bytes.NewReader(*fi.Data),
|
||||||
}
|
}
|
||||||
m := discordgo.MessageSend{
|
m := discordgo.MessageSend{
|
||||||
Content: msg.Username + fi.Comment,
|
Content: msg.Username + fi.Comment,
|
||||||
Files: []*discordgo.File{&file},
|
Files: []*discordgo.File{&file},
|
||||||
|
AllowedMentions: b.getAllowedMentions(),
|
||||||
}
|
}
|
||||||
_, err = b.c.ChannelMessageSendComplex(channelID, &m)
|
_, err = b.c.ChannelMessageSendComplex(channelID, &m)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("file upload failed: %#v", err)
|
return "", fmt.Errorf("file upload failed: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return "", nil
|
return "", nil
|
||||||
|
|||||||
@@ -2,20 +2,50 @@ package bdiscord
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/matterbridge/discordgo"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (b *Bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { //nolint:unparam
|
func (b *Bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { //nolint:unparam
|
||||||
rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EventMsgDelete, Text: config.EventMsgDelete}
|
rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EventMsgDelete, Text: config.EventMsgDelete}
|
||||||
rmsg.Channel = b.getChannelName(m.ChannelID)
|
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("<= Sending message from %s to gateway", b.Account)
|
||||||
b.Log.Debugf("<= Message is %#v", rmsg)
|
b.Log.Debugf("<= Message is %#v", rmsg)
|
||||||
b.Remote <- 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
|
||||||
|
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) messageTyping(s *discordgo.Session, m *discordgo.TypingStart) {
|
||||||
|
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
|
func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { //nolint:unparam
|
||||||
if b.GetBool("EditDisable") {
|
if b.GetBool("EditDisable") {
|
||||||
return
|
return
|
||||||
@@ -24,7 +54,10 @@ func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdat
|
|||||||
if m.Message.EditedTimestamp != "" {
|
if m.Message.EditedTimestamp != "" {
|
||||||
b.Log.Debugf("Sending edit message")
|
b.Log.Debugf("Sending edit message")
|
||||||
m.Content += b.GetString("EditSuffix")
|
m.Content += b.GetString("EditSuffix")
|
||||||
b.messageCreate(s, (*discordgo.MessageCreate)(m))
|
msg := &discordgo.MessageCreate{
|
||||||
|
Message: m.Message,
|
||||||
|
}
|
||||||
|
b.messageCreate(s, msg)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -36,7 +69,7 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
// if using webhooks, do not relay if it's ours
|
// 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
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -51,7 +84,6 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
|
|||||||
|
|
||||||
if m.Content != "" {
|
if m.Content != "" {
|
||||||
b.Log.Debugf("== Receiving event %#v", m.Message)
|
b.Log.Debugf("== Receiving event %#v", m.Message)
|
||||||
m.Message.Content = b.stripCustomoji(m.Message.Content)
|
|
||||||
m.Message.Content = b.replaceChannelMentions(m.Message.Content)
|
m.Message.Content = b.replaceChannelMentions(m.Message.Content)
|
||||||
rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c)
|
rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -62,21 +94,21 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
|
|||||||
|
|
||||||
// set channel name
|
// set channel name
|
||||||
rmsg.Channel = b.getChannelName(m.ChannelID)
|
rmsg.Channel = b.getChannelName(m.ChannelID)
|
||||||
if b.useChannelID {
|
|
||||||
rmsg.Channel = "ID:" + m.ChannelID
|
|
||||||
}
|
|
||||||
|
|
||||||
// set username
|
fromWebhook := m.WebhookID != ""
|
||||||
if !b.GetBool("UseUserName") {
|
if !fromWebhook && !b.GetBool("UseUserName") {
|
||||||
rmsg.Username = b.getNick(m.Author)
|
rmsg.Username = b.getNick(m.Author, m.GuildID)
|
||||||
} else {
|
} else {
|
||||||
rmsg.Username = m.Author.Username
|
rmsg.Username = m.Author.Username
|
||||||
|
if !fromWebhook && b.GetBool("UseDiscriminator") {
|
||||||
|
rmsg.Username += "#" + m.Author.Discriminator
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// if we have embedded content add it to text
|
// if we have embedded content add it to text
|
||||||
if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil {
|
if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil {
|
||||||
for _, embed := range m.Message.Embeds {
|
for _, embed := range m.Message.Embeds {
|
||||||
rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n"
|
rmsg.Text += handleEmbed(embed)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -92,6 +124,14 @@ func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
|
|||||||
rmsg.Event = config.EventUserAction
|
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("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account)
|
||||||
b.Log.Debugf("<= Message is %#v", rmsg)
|
b.Log.Debugf("<= Message is %#v", rmsg)
|
||||||
b.Remote <- rmsg
|
b.Remote <- rmsg
|
||||||
@@ -123,3 +163,75 @@ func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUp
|
|||||||
b.nickMemberMap[m.Member.Nick] = m.Member
|
b.nickMemberMap[m.Member.Nick] = m.Member
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Bdiscord) memberAdd(s *discordgo.Session, m *discordgo.GuildMemberAdd) {
|
||||||
|
if m.Member == nil {
|
||||||
|
b.Log.Warnf("Received member update with no member information: %#v", m)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
username := m.Member.User.Username
|
||||||
|
if m.Member.Nick != "" {
|
||||||
|
username = m.Member.Nick
|
||||||
|
}
|
||||||
|
|
||||||
|
rmsg := config.Message{
|
||||||
|
Account: b.Account,
|
||||||
|
Event: config.EventJoinLeave,
|
||||||
|
Username: "system",
|
||||||
|
Text: username + " joins",
|
||||||
|
}
|
||||||
|
b.Log.Debugf("<= Sending message from %s to gateway", b.Account)
|
||||||
|
b.Log.Debugf("<= Message is %#v", rmsg)
|
||||||
|
b.Remote <- rmsg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bdiscord) memberRemove(s *discordgo.Session, m *discordgo.GuildMemberRemove) {
|
||||||
|
if m.Member == nil {
|
||||||
|
b.Log.Warnf("Received member update with no member information: %#v", m)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
username := m.Member.User.Username
|
||||||
|
if m.Member.Nick != "" {
|
||||||
|
username = m.Member.Nick
|
||||||
|
}
|
||||||
|
|
||||||
|
rmsg := config.Message{
|
||||||
|
Account: b.Account,
|
||||||
|
Event: config.EventJoinLeave,
|
||||||
|
Username: "system",
|
||||||
|
Text: username + " leaves",
|
||||||
|
}
|
||||||
|
b.Log.Debugf("<= Sending message from %s to gateway", b.Account)
|
||||||
|
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/matterbridge/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)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -6,10 +6,34 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
"unicode"
|
"unicode"
|
||||||
|
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/matterbridge/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()
|
b.membersMutex.RLock()
|
||||||
defer b.membersMutex.RUnlock()
|
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.
|
// 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 {
|
if err != nil {
|
||||||
b.Log.Warnf("Failed to fetch information for member %#v: %#v", user, err)
|
b.Log.Warnf("Failed to fetch information for member %#v on guild %#v: %s", user, guildID, err)
|
||||||
return user.Username
|
return user.Username
|
||||||
} else if member == nil {
|
} else if member == nil {
|
||||||
b.Log.Warnf("Got no information for member %#v", user)
|
b.Log.Warnf("Got no information for member %#v", user)
|
||||||
@@ -51,6 +75,9 @@ func (b *Bdiscord) getGuildMemberByNick(nick string) (*discordgo.Member, error)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bdiscord) getChannelID(name string) string {
|
func (b *Bdiscord) getChannelID(name string) string {
|
||||||
|
if strings.Contains(name, "/") {
|
||||||
|
return b.getCategoryChannelID(name)
|
||||||
|
}
|
||||||
b.channelsMutex.RLock()
|
b.channelsMutex.RLock()
|
||||||
defer b.channelsMutex.RUnlock()
|
defer b.channelsMutex.RUnlock()
|
||||||
|
|
||||||
@@ -59,40 +86,92 @@ func (b *Bdiscord) getChannelID(name string) string {
|
|||||||
return idcheck[1]
|
return idcheck[1]
|
||||||
}
|
}
|
||||||
for _, channel := range b.channels {
|
for _, channel := range b.channels {
|
||||||
if channel.Name == name {
|
if channel.Name == name && channel.Type == discordgo.ChannelTypeGuildText {
|
||||||
return channel.ID
|
return channel.ID
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ""
|
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 {
|
func (b *Bdiscord) getChannelName(id string) string {
|
||||||
b.channelsMutex.RLock()
|
b.channelsMutex.RLock()
|
||||||
defer b.channelsMutex.RUnlock()
|
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 {
|
for _, channel := range b.channels {
|
||||||
if channel.ID == id {
|
if channel.ID == id {
|
||||||
return channel.Name
|
return b.getCategoryChannelName(channel.Name, channel.ParentID)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ""
|
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 (
|
var (
|
||||||
// See https://discordapp.com/developers/docs/reference#message-formatting.
|
// See https://discordapp.com/developers/docs/reference#message-formatting.
|
||||||
channelMentionRE = regexp.MustCompile("<#[0-9]+>")
|
channelMentionRE = regexp.MustCompile("<#[0-9]+>")
|
||||||
emojiRE = regexp.MustCompile("<(:.*?:)[0-9]+>")
|
|
||||||
userMentionRE = regexp.MustCompile("@[^@\n]{1,32}")
|
userMentionRE = regexp.MustCompile("@[^@\n]{1,32}")
|
||||||
|
emoteRE = regexp.MustCompile(`<a?(:\w+:)\d+>`)
|
||||||
)
|
)
|
||||||
|
|
||||||
func (b *Bdiscord) replaceChannelMentions(text string) string {
|
func (b *Bdiscord) replaceChannelMentions(text string) string {
|
||||||
replaceChannelMentionFunc := func(match string) string {
|
replaceChannelMentionFunc := func(match string) string {
|
||||||
var err error
|
|
||||||
channelID := match[2 : len(match)-1]
|
channelID := match[2 : len(match)-1]
|
||||||
|
|
||||||
channelName := b.getChannelName(channelID)
|
channelName := b.getChannelName(channelID)
|
||||||
|
|
||||||
// If we don't have the channel refresh our list.
|
// If we don't have the channel refresh our list.
|
||||||
if channelName == "" {
|
if channelName == "" {
|
||||||
|
var err error
|
||||||
b.channels, err = b.c.GuildChannels(b.guildID)
|
b.channels, err = b.c.GuildChannels(b.guildID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "#unknownchannel"
|
return "#unknownchannel"
|
||||||
@@ -128,19 +207,20 @@ func (b *Bdiscord) replaceUserMentions(text string) string {
|
|||||||
return userMentionRE.ReplaceAllStringFunc(text, replaceUserMentionFunc)
|
return userMentionRE.ReplaceAllStringFunc(text, replaceUserMentionFunc)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bdiscord) stripCustomoji(text string) string {
|
func replaceEmotes(text string) string {
|
||||||
return emojiRE.ReplaceAllString(text, `$1`)
|
return emoteRE.ReplaceAllString(text, "$1")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bdiscord) replaceAction(text string) (string, bool) {
|
func (b *Bdiscord) replaceAction(text string) (string, bool) {
|
||||||
if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") {
|
length := len(text)
|
||||||
return text[1:], true
|
if length > 1 && text[0] == '_' && text[length-1] == '_' {
|
||||||
|
return text[1 : length-1], true
|
||||||
}
|
}
|
||||||
return text, false
|
return text, false
|
||||||
}
|
}
|
||||||
|
|
||||||
// splitURL splits a webhookURL and returns the ID and token.
|
// 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 (
|
const (
|
||||||
expectedWebhookSplitCount = 7
|
expectedWebhookSplitCount = 7
|
||||||
webhookIdxID = 5
|
webhookIdxID = 5
|
||||||
@@ -148,9 +228,9 @@ func (b *Bdiscord) splitURL(url string) (string, string) {
|
|||||||
)
|
)
|
||||||
webhookURLSplit := strings.Split(url, "/")
|
webhookURLSplit := strings.Split(url, "/")
|
||||||
if len(webhookURLSplit) != expectedWebhookSplitCount {
|
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 {
|
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/matterbridge/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/matterbridge/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
|
||||||
|
}
|
||||||
150
bridge/discord/webhook.go
Normal file
150
bridge/discord/webhook.go
Normal file
@@ -0,0 +1,150 @@
|
|||||||
|
package bdiscord
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
"github.com/matterbridge/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 ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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) (*discordgo.Message, error) {
|
||||||
|
var (
|
||||||
|
res *discordgo.Message
|
||||||
|
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.transmitter.Send(
|
||||||
|
channelID,
|
||||||
|
&discordgo.WebhookParams{
|
||||||
|
Content: msg.Text,
|
||||||
|
Username: msg.Username,
|
||||||
|
AvatarURL: msg.Avatar,
|
||||||
|
AllowedMentions: b.getAllowedMentions(),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Errorf("Could not send text (%s) for message %#v: %s", msg.Text, msg, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
file := discordgo.File{
|
||||||
|
Name: fi.Name,
|
||||||
|
ContentType: "",
|
||||||
|
Reader: bytes.NewReader(*fi.Data),
|
||||||
|
}
|
||||||
|
content := ""
|
||||||
|
if msg.Text == "" {
|
||||||
|
content = fi.Comment
|
||||||
|
}
|
||||||
|
_, e2 := b.transmitter.Send(
|
||||||
|
channelID,
|
||||||
|
&discordgo.WebhookParams{
|
||||||
|
Username: msg.Username,
|
||||||
|
AvatarURL: msg.Avatar,
|
||||||
|
File: &file,
|
||||||
|
Content: content,
|
||||||
|
AllowedMentions: b.getAllowedMentions(),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
if e2 != nil {
|
||||||
|
b.Log.Errorf("Could not send file %#v for message %#v: %s", file, msg, e2)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
msg.Text = helper.ClipMessage(msg.Text, MessageLength, b.GetString("MessageClipped"))
|
||||||
|
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 msg.ID != "" {
|
||||||
|
b.Log.Debugf("Editing webhook message")
|
||||||
|
err := b.transmitter.Edit(channelID, msg.ID, &discordgo.WebhookParams{
|
||||||
|
Content: msg.Text,
|
||||||
|
Username: msg.Username,
|
||||||
|
AllowedMentions: b.getAllowedMentions(),
|
||||||
|
})
|
||||||
|
if err == nil {
|
||||||
|
return msg.ID, nil
|
||||||
|
}
|
||||||
|
b.Log.Errorf("Could not edit webhook message: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Log.Debugf("Processing webhook sending for message %#v", msg)
|
||||||
|
discordMsg, err := b.webhookSend(msg, channelID)
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Errorf("Could not broadcast via webhook for message %#v: %s", msg, err)
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if discordMsg == nil {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return discordMsg.ID, nil
|
||||||
|
}
|
||||||
@@ -3,22 +3,32 @@ package helper
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"image/png"
|
||||||
"io"
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"os/exec"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
"unicode/utf8"
|
"unicode/utf8"
|
||||||
|
|
||||||
|
"golang.org/x/image/webp"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/gomarkdown/markdown"
|
||||||
|
"github.com/gomarkdown/markdown/html"
|
||||||
|
"github.com/gomarkdown/markdown/parser"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
"gitlab.com/golang-commonmark/markdown"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// DownloadFile downloads the given non-authenticated URL.
|
||||||
func DownloadFile(url string) (*[]byte, error) {
|
func DownloadFile(url string) (*[]byte, error) {
|
||||||
return DownloadFileAuth(url, "")
|
return DownloadFileAuth(url, "")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// DownloadFileAuth downloads the given URL using the specified authentication token.
|
||||||
func DownloadFileAuth(url string, auth string) (*[]byte, error) {
|
func DownloadFileAuth(url string, auth string) (*[]byte, error) {
|
||||||
var buf bytes.Buffer
|
var buf bytes.Buffer
|
||||||
client := &http.Client{
|
client := &http.Client{
|
||||||
@@ -41,15 +51,41 @@ func DownloadFileAuth(url string, auth string) (*[]byte, error) {
|
|||||||
return &data, nil
|
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
|
// GetSubLines splits messages in newline-delimited lines. If maxLineLength is
|
||||||
// specified as non-zero GetSubLines will and also clip long lines to the
|
// specified as non-zero GetSubLines will also clip long lines to the maximum
|
||||||
// maximum length and insert a warning marker that the line was clipped.
|
// length and insert a warning marker that the line was clipped.
|
||||||
//
|
//
|
||||||
// TODO: The current implementation has the inconvenient that it disregards
|
// TODO: The current implementation has the inconvenient that it disregards
|
||||||
// word boundaries when splitting but this is hard to solve without potentially
|
// word boundaries when splitting but this is hard to solve without potentially
|
||||||
// breaking formatting and other stylistic effects.
|
// breaking formatting and other stylistic effects.
|
||||||
func GetSubLines(message string, maxLineLength int) []string {
|
func GetSubLines(message string, maxLineLength int, clippingMessage string) []string {
|
||||||
const clippingMessage = " <clipped message>"
|
if clippingMessage == "" {
|
||||||
|
clippingMessage = " <clipped message>"
|
||||||
|
}
|
||||||
|
|
||||||
var lines []string
|
var lines []string
|
||||||
for _, line := range strings.Split(strings.TrimSpace(message), "\n") {
|
for _, line := range strings.Split(strings.TrimSpace(message), "\n") {
|
||||||
@@ -79,18 +115,24 @@ func GetSubLines(message string, maxLineLength int) []string {
|
|||||||
return lines
|
return lines
|
||||||
}
|
}
|
||||||
|
|
||||||
// handle all the stuff we put into extra
|
// HandleExtra manages the supplementary details stored inside a message's 'Extra' field map.
|
||||||
func HandleExtra(msg *config.Message, general *config.Protocol) []config.Message {
|
func HandleExtra(msg *config.Message, general *config.Protocol) []config.Message {
|
||||||
extra := msg.Extra
|
extra := msg.Extra
|
||||||
rmsg := []config.Message{}
|
rmsg := []config.Message{}
|
||||||
for _, f := range extra[config.EventFileFailureSize] {
|
for _, f := range extra[config.EventFileFailureSize] {
|
||||||
fi := f.(config.FileInfo)
|
fi := f.(config.FileInfo)
|
||||||
text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize)
|
text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize)
|
||||||
rmsg = append(rmsg, config.Message{Text: text, Username: "<system> ", Channel: msg.Channel, Account: msg.Account})
|
rmsg = append(rmsg, config.Message{
|
||||||
|
Text: text,
|
||||||
|
Username: "<system> ",
|
||||||
|
Channel: msg.Channel,
|
||||||
|
Account: msg.Account,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
return rmsg
|
return rmsg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 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 {
|
func GetAvatar(av map[string]string, userid string, general *config.Protocol) string {
|
||||||
if sha, ok := av[userid]; ok {
|
if sha, ok := av[userid]; ok {
|
||||||
return general.MediaServerDownload + "/" + sha + "/" + userid + ".png"
|
return general.MediaServerDownload + "/" + sha + "/" + userid + ".png"
|
||||||
@@ -98,13 +140,15 @@ func GetAvatar(av map[string]string, userid string, general *config.Protocol) st
|
|||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func HandleDownloadSize(flog *logrus.Entry, msg *config.Message, name string, size int64, general *config.Protocol) error {
|
// HandleDownloadSize checks a specified filename against the configured download blacklist
|
||||||
|
// and checks a specified file-size against the configure limit.
|
||||||
|
func HandleDownloadSize(logger *logrus.Entry, msg *config.Message, name string, size int64, general *config.Protocol) error {
|
||||||
// check blacklist here
|
// check blacklist here
|
||||||
for _, entry := range general.MediaDownloadBlackList {
|
for _, entry := range general.MediaDownloadBlackList {
|
||||||
if entry != "" {
|
if entry != "" {
|
||||||
re, err := regexp.Compile(entry)
|
re, err := regexp.Compile(entry)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("incorrect regexp %s for %s", entry, msg.Account)
|
logger.Errorf("incorrect regexp %s for %s", entry, msg.Account)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if re.MatchString(name) {
|
if re.MatchString(name) {
|
||||||
@@ -112,48 +156,149 @@ func HandleDownloadSize(flog *logrus.Entry, msg *config.Message, name string, si
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
flog.Debugf("Trying to download %#v with size %#v", name, size)
|
logger.Debugf("Trying to download %#v with size %#v", name, size)
|
||||||
if int(size) > general.MediaDownloadSize {
|
if int(size) > general.MediaDownloadSize {
|
||||||
msg.Event = config.EventFileFailureSize
|
msg.Event = config.EventFileFailureSize
|
||||||
msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: name, Comment: msg.Text, Size: size})
|
msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{
|
||||||
|
Name: name,
|
||||||
|
Comment: msg.Text,
|
||||||
|
Size: size,
|
||||||
|
})
|
||||||
return fmt.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, general.MediaDownloadSize)
|
return fmt.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, general.MediaDownloadSize)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func HandleDownloadData(flog *logrus.Entry, msg *config.Message, name, comment, url string, data *[]byte, general *config.Protocol) {
|
// 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) {
|
||||||
var avatar bool
|
var avatar bool
|
||||||
flog.Debugf("Download OK %#v %#v", name, len(*data))
|
logger.Debugf("Download OK %#v %#v", name, len(*data))
|
||||||
if msg.Event == config.EventAvatarDownload {
|
if msg.Event == config.EventAvatarDownload {
|
||||||
avatar = true
|
avatar = true
|
||||||
}
|
}
|
||||||
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, URL: url, Comment: comment, Avatar: avatar})
|
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{
|
||||||
|
Name: name,
|
||||||
|
Data: data,
|
||||||
|
URL: url,
|
||||||
|
Comment: comment,
|
||||||
|
Avatar: avatar,
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var emptyLineMatcher = regexp.MustCompile("\n+")
|
||||||
|
|
||||||
|
// RemoveEmptyNewLines collapses consecutive newline characters into a single one and
|
||||||
|
// trims any preceding or trailing newline characters as well.
|
||||||
func RemoveEmptyNewLines(msg string) string {
|
func RemoveEmptyNewLines(msg string) string {
|
||||||
lines := ""
|
return emptyLineMatcher.ReplaceAllString(strings.Trim(msg, "\n"), "\n")
|
||||||
for _, line := range strings.Split(msg, "\n") {
|
|
||||||
if line != "" {
|
|
||||||
lines += line + "\n"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
lines = strings.TrimRight(lines, "\n")
|
|
||||||
return lines
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func ClipMessage(text string, length int) string {
|
// ClipMessage trims a message to the specified length if it exceeds it and adds a warning
|
||||||
// clip too long messages
|
// to the message in case it does so.
|
||||||
|
func ClipMessage(text string, length int, clippingMessage string) string {
|
||||||
|
if clippingMessage == "" {
|
||||||
|
clippingMessage = " <clipped message>"
|
||||||
|
}
|
||||||
|
|
||||||
if len(text) > length {
|
if len(text) > length {
|
||||||
text = text[:length-len(" *message clipped*")]
|
text = text[:length-len(clippingMessage)]
|
||||||
if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError {
|
if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError {
|
||||||
text = text[:len(text)-size]
|
text = text[:len(text)-size]
|
||||||
}
|
}
|
||||||
text += " *message clipped*"
|
text += clippingMessage
|
||||||
}
|
}
|
||||||
return text
|
return text
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ParseMarkdown takes in an input string as markdown and parses it to html
|
||||||
func ParseMarkdown(input string) string {
|
func ParseMarkdown(input string) string {
|
||||||
md := markdown.New(markdown.XHTMLOutput(true), markdown.Breaks(true))
|
extensions := parser.HardLineBreak | parser.NoIntraEmphasis | parser.FencedCode
|
||||||
return (md.RenderToString([]byte(input)))
|
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)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
var output []byte
|
||||||
|
w := bytes.NewBuffer(output)
|
||||||
|
if err := png.Encode(w, m); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
*data = w.Bytes()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,8 @@
|
|||||||
package helper
|
package helper
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
@@ -8,98 +10,118 @@ import (
|
|||||||
|
|
||||||
const testLineLength = 64
|
const testLineLength = 64
|
||||||
|
|
||||||
var (
|
var lineSplittingTestCases = map[string]struct {
|
||||||
lineSplittingTestCases = map[string]struct {
|
input string
|
||||||
input string
|
splitOutput []string
|
||||||
splitOutput []string
|
nonSplitOutput []string
|
||||||
nonSplitOutput []string
|
}{
|
||||||
}{
|
"Short single-line message": {
|
||||||
"Short single-line message": {
|
input: "short",
|
||||||
input: "short",
|
splitOutput: []string{"short"},
|
||||||
splitOutput: []string{"short"},
|
nonSplitOutput: []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": {
|
nonSplitOutput: []string{"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."},
|
||||||
input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
|
},
|
||||||
splitOutput: []string{
|
"Short multi-line message": {
|
||||||
"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>",
|
input: "I\ncan't\nget\nno\nsatisfaction!",
|
||||||
"cing elit, sed do eiusmod tempor incididunt ut <clipped message>",
|
splitOutput: []string{
|
||||||
" labore et dolore magna aliqua.",
|
"I",
|
||||||
},
|
"can't",
|
||||||
nonSplitOutput: []string{"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."},
|
"get",
|
||||||
|
"no",
|
||||||
|
"satisfaction!",
|
||||||
},
|
},
|
||||||
"Short multi-line message": {
|
nonSplitOutput: []string{
|
||||||
input: "I\ncan't\nget\nno\nsatisfaction!",
|
"I",
|
||||||
splitOutput: []string{
|
"can't",
|
||||||
"I",
|
"get",
|
||||||
"can't",
|
"no",
|
||||||
"get",
|
"satisfaction!",
|
||||||
"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" +
|
"Long multi-line message": {
|
||||||
"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.\n" +
|
input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.\n" +
|
||||||
"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.\n" +
|
"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.\n" +
|
||||||
"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.",
|
"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.\n" +
|
||||||
splitOutput: []string{
|
"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.",
|
||||||
"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>",
|
splitOutput: []string{
|
||||||
"cing elit, sed do eiusmod tempor incididunt ut <clipped message>",
|
"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>",
|
||||||
" labore et dolore magna aliqua.",
|
"cing elit, sed do eiusmod tempor incididunt ut <clipped message>",
|
||||||
"Ut enim ad minim veniam, quis nostrud exercita <clipped message>",
|
" labore et dolore magna aliqua.",
|
||||||
"tion ullamco laboris nisi ut aliquip ex ea com <clipped message>",
|
"Ut enim ad minim veniam, quis nostrud exercita <clipped message>",
|
||||||
"modo consequat.",
|
"tion ullamco laboris nisi ut aliquip ex ea com <clipped message>",
|
||||||
"Duis aute irure dolor in reprehenderit in volu <clipped message>",
|
"modo consequat.",
|
||||||
"ptate velit esse cillum dolore eu fugiat nulla <clipped message>",
|
"Duis aute irure dolor in reprehenderit in volu <clipped message>",
|
||||||
" pariatur.",
|
"ptate velit esse cillum dolore eu fugiat nulla <clipped message>",
|
||||||
"Excepteur sint occaecat cupidatat non proident <clipped message>",
|
" pariatur.",
|
||||||
", sunt in culpa qui officia deserunt mollit an <clipped message>",
|
"Excepteur sint occaecat cupidatat non proident <clipped message>",
|
||||||
"im id est laborum.",
|
", sunt in culpa qui officia deserunt mollit an <clipped message>",
|
||||||
},
|
"im id est laborum.",
|
||||||
nonSplitOutput: []string{
|
|
||||||
"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
|
|
||||||
"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.",
|
|
||||||
"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.",
|
|
||||||
"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.",
|
|
||||||
},
|
|
||||||
},
|
},
|
||||||
"Message ending with new-line.": {
|
nonSplitOutput: []string{
|
||||||
input: "Newline ending\n",
|
"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.",
|
||||||
splitOutput: []string{"Newline ending"},
|
"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.",
|
||||||
nonSplitOutput: []string{"Newline ending"},
|
"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: "不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說",
|
"Message ending with new-line.": {
|
||||||
splitOutput: []string{
|
input: "Newline ending\n",
|
||||||
"不布人個我此而及單石業喜資富下 <clipped message>",
|
splitOutput: []string{"Newline ending"},
|
||||||
"我河下日沒一我臺空達的常景便物 <clipped message>",
|
nonSplitOutput: []string{"Newline ending"},
|
||||||
"沒為……子大我別名解成?生賣的 <clipped message>",
|
},
|
||||||
"全直黑,我自我結毛分洲了世當, <clipped message>",
|
"Long message containing UTF-8 multi-byte runes": {
|
||||||
"是政福那是東;斯說",
|
input: "不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說",
|
||||||
},
|
splitOutput: []string{
|
||||||
nonSplitOutput: []string{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"},
|
"不布人個我此而及單石業喜資富下 <clipped message>",
|
||||||
|
"我河下日沒一我臺空達的常景便物 <clipped message>",
|
||||||
|
"沒為……子大我別名解成?生賣的 <clipped message>",
|
||||||
|
"全直黑,我自我結毛分洲了世當, <clipped message>",
|
||||||
|
"是政福那是東;斯說",
|
||||||
},
|
},
|
||||||
}
|
nonSplitOutput: []string{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"},
|
||||||
)
|
},
|
||||||
|
}
|
||||||
|
|
||||||
func TestGetSubLines(t *testing.T) {
|
func TestGetSubLines(t *testing.T) {
|
||||||
for testname, testcase := range lineSplittingTestCases {
|
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)
|
assert.Equalf(t, testcase.splitOutput, splitLines, "'%s' testcase should give expected lines with splitting.", testname)
|
||||||
for _, splitLine := range splitLines {
|
for _, splitLine := range splitLines {
|
||||||
byteLength := len([]byte(splitLine))
|
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)
|
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)
|
assert.Equalf(t, testcase.nonSplitOutput, nonSplitLines, "'%s' testcase should give expected lines without splitting.", testname)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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, 0o644) // nolint:gosec
|
||||||
|
if err != nil {
|
||||||
|
t.Fail()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -4,15 +4,14 @@ import (
|
|||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"regexp"
|
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/dfordsoft/golib/ic"
|
|
||||||
"github.com/lrstanley/girc"
|
"github.com/lrstanley/girc"
|
||||||
|
"github.com/missdeer/golib/ic"
|
||||||
"github.com/paulrosania/go-charset/charset"
|
"github.com/paulrosania/go-charset/charset"
|
||||||
"github.com/saintfish/chardet"
|
"github.com/saintfish/chardet"
|
||||||
|
|
||||||
@@ -55,12 +54,12 @@ func (b *Birc) handleFiles(msg *config.Message) bool {
|
|||||||
for _, f := range msg.Extra["file"] {
|
for _, f := range msg.Extra["file"] {
|
||||||
fi := f.(config.FileInfo)
|
fi := f.(config.FileInfo)
|
||||||
if fi.Comment != "" {
|
if fi.Comment != "" {
|
||||||
msg.Text += fi.Comment + ": "
|
msg.Text += fi.Comment + " : "
|
||||||
}
|
}
|
||||||
if fi.URL != "" {
|
if fi.URL != "" {
|
||||||
msg.Text = fi.URL
|
msg.Text = fi.URL
|
||||||
if fi.Comment != "" {
|
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}
|
b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event}
|
||||||
@@ -68,6 +67,20 @@ func (b *Birc) handleFiles(msg *config.Message) bool {
|
|||||||
return true
|
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) {
|
func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) {
|
||||||
if len(event.Params) == 0 {
|
if len(event.Params) == 0 {
|
||||||
b.Log.Debugf("handleJoinPart: empty Params? %#v", event)
|
b.Log.Debugf("handleJoinPart: empty Params? %#v", event)
|
||||||
@@ -81,7 +94,7 @@ func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
if event.Command == "QUIT" {
|
if event.Command == "QUIT" {
|
||||||
if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") {
|
if event.Source.Name == b.Nick && strings.Contains(event.Last(), "Ping timeout") {
|
||||||
b.Log.Infof("%s reconnecting ..", b.Account)
|
b.Log.Infof("%s reconnecting ..", b.Account)
|
||||||
b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EventFailure}
|
b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EventFailure}
|
||||||
return
|
return
|
||||||
@@ -91,8 +104,13 @@ func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) {
|
|||||||
if b.GetBool("nosendjoinpart") {
|
if b.GetBool("nosendjoinpart") {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account)
|
|
||||||
msg := config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EventJoinLeave}
|
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.Log.Debugf("<= Message is %#v", msg)
|
||||||
b.Remote <- msg
|
b.Remote <- msg
|
||||||
return
|
return
|
||||||
@@ -105,14 +123,15 @@ func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) {
|
|||||||
i := b.i
|
i := b.i
|
||||||
b.Nick = event.Params[0]
|
b.Nick = event.Params[0]
|
||||||
|
|
||||||
i.Handlers.Add("PRIVMSG", b.handlePrivMsg)
|
i.Handlers.AddBg("PRIVMSG", b.handlePrivMsg)
|
||||||
i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg)
|
i.Handlers.AddBg("CTCP_ACTION", b.handlePrivMsg)
|
||||||
i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
|
i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime)
|
||||||
i.Handlers.Add(girc.NOTICE, b.handleNotice)
|
i.Handlers.AddBg(girc.NOTICE, b.handleNotice)
|
||||||
i.Handlers.Add("JOIN", b.handleJoinPart)
|
i.Handlers.AddBg("JOIN", b.handleJoinPart)
|
||||||
i.Handlers.Add("PART", b.handleJoinPart)
|
i.Handlers.AddBg("PART", b.handleJoinPart)
|
||||||
i.Handlers.Add("QUIT", b.handleJoinPart)
|
i.Handlers.AddBg("QUIT", b.handleJoinPart)
|
||||||
i.Handlers.Add("KICK", b.handleJoinPart)
|
i.Handlers.AddBg("KICK", b.handleJoinPart)
|
||||||
|
i.Handlers.Add("INVITE", b.handleInvite)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Birc) handleNickServ() {
|
func (b *Birc) handleNickServ() {
|
||||||
@@ -156,28 +175,39 @@ func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) {
|
|||||||
b.handleNickServ()
|
b.handleNickServ()
|
||||||
b.handleRunCommands()
|
b.handleRunCommands()
|
||||||
// we are now fully connected
|
// 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) {
|
func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) {
|
||||||
if b.skipPrivMsg(event) {
|
if b.skipPrivMsg(event) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host}
|
|
||||||
b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Trailing, event)
|
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
|
// set action event
|
||||||
if event.IsAction() {
|
if event.IsAction() {
|
||||||
rmsg.Event = config.EventUserAction
|
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
|
// strip action, we made an event if it was an action
|
||||||
rmsg.Text += event.StripAction()
|
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
|
// start detecting the charset
|
||||||
mycharset := b.GetString("Charset")
|
mycharset := b.GetString("Charset")
|
||||||
if mycharset == "" {
|
if mycharset == "" {
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"fmt"
|
"fmt"
|
||||||
"hash/crc32"
|
"hash/crc32"
|
||||||
|
"io/ioutil"
|
||||||
"net"
|
"net"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
"strconv"
|
||||||
@@ -14,6 +15,7 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/lrstanley/girc"
|
"github.com/lrstanley/girc"
|
||||||
|
stripmd "github.com/writeas/go-strip-markdown"
|
||||||
|
|
||||||
// We need to import the 'data' package as an implicit dependency.
|
// We need to import the 'data' package as an implicit dependency.
|
||||||
// See: https://godoc.org/github.com/paulrosania/go-charset/charset
|
// See: https://godoc.org/github.com/paulrosania/go-charset/charset
|
||||||
@@ -28,6 +30,7 @@ type Birc struct {
|
|||||||
Local chan config.Message // local queue for flood control
|
Local chan config.Message // local queue for flood control
|
||||||
FirstConnection, authDone bool
|
FirstConnection, authDone bool
|
||||||
MessageDelay, MessageQueue, MessageLength int
|
MessageDelay, MessageQueue, MessageLength int
|
||||||
|
channels map[string]bool
|
||||||
|
|
||||||
*bridge.Config
|
*bridge.Config
|
||||||
}
|
}
|
||||||
@@ -38,6 +41,8 @@ func New(cfg *bridge.Config) bridge.Bridger {
|
|||||||
b.Nick = b.GetString("Nick")
|
b.Nick = b.GetString("Nick")
|
||||||
b.names = make(map[string][]string)
|
b.names = make(map[string][]string)
|
||||||
b.connected = make(chan error)
|
b.connected = make(chan error)
|
||||||
|
b.channels = make(map[string]bool)
|
||||||
|
|
||||||
if b.GetInt("MessageDelay") == 0 {
|
if b.GetInt("MessageDelay") == 0 {
|
||||||
b.MessageDelay = 1300
|
b.MessageDelay = 1300
|
||||||
} else {
|
} else {
|
||||||
@@ -110,6 +115,7 @@ func (b *Birc) Disconnect() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Birc) JoinChannel(channel config.ChannelInfo) 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
|
// need to check if we have nickserv auth done before joining channels
|
||||||
for {
|
for {
|
||||||
if b.authDone {
|
if b.authDone {
|
||||||
@@ -137,6 +143,7 @@ func (b *Birc) Send(msg config.Message) (string, error) {
|
|||||||
// we can be in between reconnects #385
|
// we can be in between reconnects #385
|
||||||
if !b.i.IsConnected() {
|
if !b.i.IsConnected() {
|
||||||
b.Log.Error("Not connected to server, dropping message")
|
b.Log.Error("Not connected to server, dropping message")
|
||||||
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Execute a command
|
// Execute a command
|
||||||
@@ -155,10 +162,14 @@ func (b *Birc) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var msgLines []string
|
var msgLines []string
|
||||||
|
if b.GetBool("StripMarkdown") {
|
||||||
|
msg.Text = stripmd.Strip(msg.Text)
|
||||||
|
}
|
||||||
|
|
||||||
if b.GetBool("MessageSplit") {
|
if b.GetBool("MessageSplit") {
|
||||||
msgLines = helper.GetSubLines(msg.Text, b.MessageLength)
|
msgLines = helper.GetSubLines(msg.Text, b.MessageLength, b.GetString("MessageClipped"))
|
||||||
} else {
|
} else {
|
||||||
msgLines = helper.GetSubLines(msg.Text, 0)
|
msgLines = helper.GetSubLines(msg.Text, 0, b.GetString("MessageClipped"))
|
||||||
}
|
}
|
||||||
for i := range msgLines {
|
for i := range msgLines {
|
||||||
if len(b.Local) >= b.MessageQueue {
|
if len(b.Local) >= b.MessageQueue {
|
||||||
@@ -166,12 +177,8 @@ func (b *Birc) Send(msg config.Message) (string, error) {
|
|||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
b.Local <- config.Message{
|
msg.Text = msgLines[i]
|
||||||
Text: msgLines[i],
|
b.Local <- msg
|
||||||
Username: msg.Username,
|
|
||||||
Channel: msg.Channel,
|
|
||||||
Event: msg.Event,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
@@ -198,22 +205,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() {
|
func (b *Birc) doSend() {
|
||||||
rate := time.Millisecond * time.Duration(b.MessageDelay)
|
rate := time.Millisecond * time.Duration(b.MessageDelay)
|
||||||
throttle := time.NewTicker(rate)
|
throttle := time.NewTicker(rate)
|
||||||
for msg := range b.Local {
|
for msg := range b.Local {
|
||||||
<-throttle.C
|
<-throttle.C
|
||||||
username := msg.Username
|
username := msg.Username
|
||||||
if b.GetBool("Colornicks") {
|
// Optional support for the proposed RELAYMSG extension, described at
|
||||||
checksum := crc32.ChecksumIEEE([]byte(msg.Username))
|
// https://github.com/jlu5/ircv3-specifications/blob/master/extensions/relaymsg.md
|
||||||
colorCode := checksum%14 + 2 // quick fix - prevent white or black color codes
|
// nolint:nestif
|
||||||
username = fmt.Sprintf("\x03%02d%s\x0F", colorCode, msg.Username)
|
if (b.i.HasCapability("overdrivenetworks.com/relaymsg") || b.i.HasCapability("draft/relaymsg")) &&
|
||||||
}
|
b.GetBool("UseRelayMsg") {
|
||||||
if msg.Event == config.EventUserAction {
|
username = sanitizeNick(username)
|
||||||
b.i.Cmd.Action(msg.Channel, username+msg.Text)
|
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 {
|
} else {
|
||||||
b.Log.Debugf("Sending to channel %s", msg.Channel)
|
if b.GetBool("Colornicks") {
|
||||||
b.i.Cmd.Message(msg.Channel, username+msg.Text)
|
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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -231,13 +274,25 @@ func (b *Birc) getClient() (*girc.Client, error) {
|
|||||||
// fix strict user handling of girc
|
// fix strict user handling of girc
|
||||||
user := b.GetString("Nick")
|
user := b.GetString("Nick")
|
||||||
for !girc.IsValidUser(user) {
|
for !girc.IsValidUser(user) {
|
||||||
if len(user) == 1 {
|
if len(user) == 1 || len(user) == 0 {
|
||||||
user = "matterbridge"
|
user = "matterbridge"
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
user = user[1:]
|
user = user[1:]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
|
||||||
i := girc.New(girc.Config{
|
i := girc.New(girc.Config{
|
||||||
Server: server,
|
Server: server,
|
||||||
ServerPass: b.GetString("Password"),
|
ServerPass: b.GetString("Password"),
|
||||||
@@ -247,7 +302,11 @@ func (b *Birc) getClient() (*girc.Client, error) {
|
|||||||
Name: b.GetString("Nick"),
|
Name: b.GetString("Nick"),
|
||||||
SSL: b.GetBool("UseTLS"),
|
SSL: b.GetBool("UseTLS"),
|
||||||
TLSConfig: &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, //nolint:gosec
|
TLSConfig: &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, //nolint:gosec
|
||||||
PingDelay: time.Minute,
|
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
|
return i, nil
|
||||||
}
|
}
|
||||||
@@ -257,12 +316,16 @@ func (b *Birc) endNames(client *girc.Client, event girc.Event) {
|
|||||||
sort.Strings(b.names[channel])
|
sort.Strings(b.names[channel])
|
||||||
maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
|
maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow()
|
||||||
for len(b.names[channel]) > maxNamesPerPost {
|
for len(b.names[channel]) > maxNamesPerPost {
|
||||||
b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]),
|
b.Remote <- config.Message{
|
||||||
Channel: channel, Account: b.Account}
|
Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]),
|
||||||
|
Channel: channel, Account: b.Account,
|
||||||
|
}
|
||||||
b.names[channel] = b.names[channel][maxNamesPerPost:]
|
b.names[channel] = b.names[channel][maxNamesPerPost:]
|
||||||
}
|
}
|
||||||
b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel]),
|
b.Remote <- config.Message{
|
||||||
Channel: channel, Account: b.Account}
|
Username: b.Nick, Text: b.formatnicks(b.names[channel]),
|
||||||
|
Channel: channel, Account: b.Account,
|
||||||
|
}
|
||||||
b.names[channel] = nil
|
b.names[channel] = nil
|
||||||
b.i.Handlers.Clear(girc.RPL_NAMREPLY)
|
b.i.Handlers.Clear(girc.RPL_NAMREPLY)
|
||||||
b.i.Handlers.Clear(girc.RPL_ENDOFNAMES)
|
b.i.Handlers.Clear(girc.RPL_ENDOFNAMES)
|
||||||
@@ -273,7 +336,7 @@ func (b *Birc) skipPrivMsg(event girc.Event) bool {
|
|||||||
b.Nick = b.i.GetNick()
|
b.Nick = b.i.GetNick()
|
||||||
|
|
||||||
// freenode doesn't send 001 as first reply
|
// freenode doesn't send 001 as first reply
|
||||||
if event.Command == "NOTICE" {
|
if event.Command == "NOTICE" && len(event.Params) != 2 {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
// don't forward queries to the bot
|
// don't forward queries to the bot
|
||||||
@@ -284,6 +347,15 @@ func (b *Birc) skipPrivMsg(event girc.Event) bool {
|
|||||||
if event.Source.Name == b.Nick {
|
if event.Source.Name == b.Nick {
|
||||||
return true
|
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
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -295,7 +367,7 @@ func (b *Birc) storeNames(client *girc.Client, event girc.Event) {
|
|||||||
channel := event.Params[2]
|
channel := event.Params[2]
|
||||||
b.names[channel] = append(
|
b.names[channel] = append(
|
||||||
b.names[channel],
|
b.names[channel],
|
||||||
strings.Split(strings.TrimSpace(event.Trailing), " ")...)
|
strings.Split(strings.TrimSpace(event.Last()), " ")...)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Birc) formatnicks(nicks []string) string {
|
func (b *Birc) formatnicks(nicks []string) string {
|
||||||
|
|||||||
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/matrix-org/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 homserver 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,50 +3,105 @@ package bmatrix
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"html"
|
|
||||||
"mime"
|
"mime"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
matrix "github.com/matterbridge/gomatrix"
|
matrix "github.com/matrix-org/gomatrix"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
htmlTag = regexp.MustCompile("</.*?>")
|
||||||
|
htmlReplacementTag = regexp.MustCompile("<[^>]*>")
|
||||||
|
)
|
||||||
|
|
||||||
|
type NicknameCacheEntry struct {
|
||||||
|
displayName string
|
||||||
|
lastUpdated time.Time
|
||||||
|
}
|
||||||
|
|
||||||
type Bmatrix struct {
|
type Bmatrix struct {
|
||||||
mc *matrix.Client
|
mc *matrix.Client
|
||||||
UserID string
|
UserID string
|
||||||
RoomMap map[string]string
|
NicknameMap map[string]NicknameCacheEntry
|
||||||
|
RoomMap map[string]string
|
||||||
|
rateMutex sync.RWMutex
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
*bridge.Config
|
*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"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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
|
||||||
|
}
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
b := &Bmatrix{Config: cfg}
|
b := &Bmatrix{Config: cfg}
|
||||||
b.RoomMap = make(map[string]string)
|
b.RoomMap = make(map[string]string)
|
||||||
|
b.NicknameMap = make(map[string]NicknameCacheEntry)
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) Connect() error {
|
func (b *Bmatrix) Connect() error {
|
||||||
var err error
|
var err error
|
||||||
b.Log.Infof("Connecting %s", b.GetString("Server"))
|
b.Log.Infof("Connecting %s", b.GetString("Server"))
|
||||||
b.mc, err = matrix.NewClient(b.GetString("Server"), "", "")
|
if b.GetString("MxID") != "" && b.GetString("Token") != "" {
|
||||||
if err != nil {
|
b.mc, err = matrix.NewClient(
|
||||||
return err
|
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()
|
go b.handlematrix()
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -56,14 +111,18 @@ func (b *Bmatrix) Disconnect() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error {
|
func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error {
|
||||||
resp, err := b.mc.JoinRoom(channel.Name, "", nil)
|
return b.retry(func() error {
|
||||||
if err != nil {
|
resp, err := b.mc.JoinRoom(channel.Name, "", nil)
|
||||||
return err
|
if err != nil {
|
||||||
}
|
return err
|
||||||
b.Lock()
|
}
|
||||||
b.RoomMap[resp.RoomID] = channel.Name
|
|
||||||
b.Unlock()
|
b.Lock()
|
||||||
return err
|
b.RoomMap[resp.RoomID] = channel.Name
|
||||||
|
b.Unlock()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
||||||
@@ -72,17 +131,30 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
|||||||
channel := b.getRoomID(msg.Channel)
|
channel := b.getRoomID(msg.Channel)
|
||||||
b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel)
|
b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel)
|
||||||
|
|
||||||
|
username := newMatrixUsername(msg.Username)
|
||||||
|
|
||||||
// Make a action /me of the message
|
// Make a action /me of the message
|
||||||
if msg.Event == config.EventUserAction {
|
if msg.Event == config.EventUserAction {
|
||||||
m := matrix.TextMessage{
|
m := matrix.TextMessage{
|
||||||
MsgType: "m.emote",
|
MsgType: "m.emote",
|
||||||
Body: msg.Username + msg.Text,
|
Body: username.plain + msg.Text,
|
||||||
|
FormattedBody: username.formatted + msg.Text,
|
||||||
}
|
}
|
||||||
resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m)
|
|
||||||
if err != nil {
|
msgID := ""
|
||||||
return "", err
|
|
||||||
}
|
err := b.retry(func() error {
|
||||||
return resp.EventID, err
|
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
|
// Delete message
|
||||||
@@ -90,17 +162,34 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
|||||||
if msg.ID == "" {
|
if msg.ID == "" {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{})
|
|
||||||
if err != nil {
|
msgID := ""
|
||||||
return "", err
|
|
||||||
}
|
err := b.retry(func() error {
|
||||||
return resp.EventID, err
|
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
|
// Upload a file if it exists
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
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)
|
b.Log.Errorf("sendText failed: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -111,31 +200,105 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Edit message if we have an ID
|
// Edit message if we have an ID
|
||||||
// matrix has no editing support
|
if msg.ID != "" {
|
||||||
|
rmsg := EditedMessage{TextMessage: matrix.TextMessage{
|
||||||
|
Body: username.plain + msg.Text,
|
||||||
|
MsgType: "m.text",
|
||||||
|
}}
|
||||||
|
if b.GetBool("HTMLDisable") {
|
||||||
|
rmsg.TextMessage.FormattedBody = username.formatted + "* " + msg.Text
|
||||||
|
} else {
|
||||||
|
rmsg.Format = "org.matrix.custom.html"
|
||||||
|
rmsg.TextMessage.FormattedBody = username.formatted + "* " + helper.ParseMarkdown(msg.Text)
|
||||||
|
}
|
||||||
|
rmsg.NewContent = SubTextMessage{
|
||||||
|
Body: rmsg.TextMessage.Body,
|
||||||
|
MsgType: "m.text",
|
||||||
|
}
|
||||||
|
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: username.plain + msg.Text,
|
||||||
|
FormattedBody: username.formatted + msg.Text,
|
||||||
|
}
|
||||||
|
|
||||||
|
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, username.plain+msg.Text)
|
||||||
|
|
||||||
|
return err
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return resp.EventID, err
|
||||||
|
}
|
||||||
|
|
||||||
// Post normal message with HTML support (eg riot.im)
|
// Post normal message with HTML support (eg riot.im)
|
||||||
resp, err := b.mc.SendHTML(channel, msg.Username+msg.Text, html.EscapeString(msg.Username)+helper.ParseMarkdown(msg.Text))
|
var (
|
||||||
|
resp *matrix.RespSendEvent
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
|
||||||
|
err = b.retry(func() error {
|
||||||
|
resp, err = b.mc.SendFormattedText(channel, username.plain+msg.Text,
|
||||||
|
username.formatted+helper.ParseMarkdown(msg.Text))
|
||||||
|
|
||||||
|
return err
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
return resp.EventID, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *Bmatrix) getRoomID(channel string) string {
|
return resp.EventID, err
|
||||||
b.RLock()
|
|
||||||
defer b.RUnlock()
|
|
||||||
for ID, name := range b.RoomMap {
|
|
||||||
if name == channel {
|
|
||||||
return ID
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) handlematrix() {
|
func (b *Bmatrix) handlematrix() {
|
||||||
syncer := b.mc.Syncer.(*matrix.DefaultSyncer)
|
syncer := b.mc.Syncer.(*matrix.DefaultSyncer)
|
||||||
syncer.OnEventType("m.room.redaction", b.handleEvent)
|
syncer.OnEventType("m.room.redaction", b.handleEvent)
|
||||||
syncer.OnEventType("m.room.message", b.handleEvent)
|
syncer.OnEventType("m.room.message", b.handleEvent)
|
||||||
|
syncer.OnEventType("m.room.member", b.handleMemberChange)
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
if err := b.mc.Sync(); err != nil {
|
if err := b.mc.Sync(); err != nil {
|
||||||
@@ -145,6 +308,45 @@ 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) 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) {
|
func (b *Bmatrix) handleEvent(ev *matrix.Event) {
|
||||||
b.Log.Debugf("== Receiving event: %#v", ev)
|
b.Log.Debugf("== Receiving event: %#v", ev)
|
||||||
if ev.Sender != b.UserID {
|
if ev.Sender != b.UserID {
|
||||||
@@ -156,16 +358,14 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO download avatar
|
|
||||||
|
|
||||||
// Create our message
|
// Create our message
|
||||||
rmsg := config.Message{Username: ev.Sender[1:], Channel: channel, Account: b.Account, UserID: ev.Sender, ID: ev.ID}
|
rmsg := config.Message{
|
||||||
|
Username: b.getDisplayName(ev.Sender),
|
||||||
// Text must be a string
|
Channel: channel,
|
||||||
if rmsg.Text, ok = ev.Content["body"].(string); !ok {
|
Account: b.Account,
|
||||||
b.Log.Errorf("Content[body] is not a string: %T\n%#v",
|
UserID: ev.Sender,
|
||||||
ev.Content["body"], ev.Content)
|
ID: ev.ID,
|
||||||
return
|
Avatar: b.getAvatarURL(ev.Sender),
|
||||||
}
|
}
|
||||||
|
|
||||||
// Remove homeserver suffix if configured
|
// Remove homeserver suffix if configured
|
||||||
@@ -183,11 +383,23 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) {
|
|||||||
return
|
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
|
// Do we have a /me action
|
||||||
if ev.Content["msgtype"].(string) == "m.emote" {
|
if ev.Content["msgtype"].(string) == "m.emote" {
|
||||||
rmsg.Event = config.EventUserAction
|
rmsg.Event = config.EventUserAction
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Is it an edit?
|
||||||
|
if b.handleEdit(ev, rmsg) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// Do we have attachments
|
// Do we have attachments
|
||||||
if b.containsAttachment(ev.Content) {
|
if b.containsAttachment(ev.Content) {
|
||||||
err := b.handleDownloadFile(&rmsg, ev.Content)
|
err := b.handleDownloadFile(&rmsg, ev.Content)
|
||||||
@@ -198,6 +410,11 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) {
|
|||||||
|
|
||||||
b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Sender, b.Account)
|
b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Sender, b.Account)
|
||||||
b.Remote <- rmsg
|
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())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -272,20 +489,30 @@ func (b *Bmatrix) handleUploadFiles(msg *config.Message, channel string) (string
|
|||||||
|
|
||||||
// handleUploadFile handles native upload of a file.
|
// handleUploadFile handles native upload of a file.
|
||||||
func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) {
|
func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) {
|
||||||
|
username := newMatrixUsername(msg.Username)
|
||||||
content := bytes.NewReader(*fi.Data)
|
content := bytes.NewReader(*fi.Data)
|
||||||
sp := strings.Split(fi.Name, ".")
|
sp := strings.Split(fi.Name, ".")
|
||||||
mtype := mime.TypeByExtension("." + sp[len(sp)-1])
|
mtype := mime.TypeByExtension("." + sp[len(sp)-1])
|
||||||
if !strings.Contains(mtype, "image") && !strings.Contains(mtype, "video") {
|
// image and video uploads send no username, we have to do this ourself here #715
|
||||||
return
|
err := b.retry(func() error {
|
||||||
}
|
_, err := b.mc.SendFormattedText(channel, username.plain+fi.Comment, username.formatted+fi.Comment)
|
||||||
if fi.Comment != "" {
|
|
||||||
_, err := b.mc.SendText(channel, msg.Username+fi.Comment)
|
return err
|
||||||
if err != nil {
|
})
|
||||||
b.Log.Errorf("file comment failed: %#v", err)
|
if err != nil {
|
||||||
}
|
b.Log.Errorf("file comment failed: %#v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
b.Log.Debugf("uploading file: %s %s", fi.Name, mtype)
|
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 {
|
if err != nil {
|
||||||
b.Log.Errorf("file upload failed: %#v", err)
|
b.Log.Errorf("file upload failed: %#v", err)
|
||||||
return
|
return
|
||||||
@@ -294,32 +521,60 @@ func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *conf
|
|||||||
switch {
|
switch {
|
||||||
case strings.Contains(mtype, "video"):
|
case strings.Contains(mtype, "video"):
|
||||||
b.Log.Debugf("sendVideo %s", res.ContentURI)
|
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 {
|
if err != nil {
|
||||||
b.Log.Errorf("sendVideo failed: %#v", err)
|
b.Log.Errorf("sendVideo failed: %#v", err)
|
||||||
}
|
}
|
||||||
case strings.Contains(mtype, "image"):
|
case strings.Contains(mtype, "image"):
|
||||||
b.Log.Debugf("sendImage %s", res.ContentURI)
|
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 {
|
if err != nil {
|
||||||
b.Log.Errorf("sendImage failed: %#v", err)
|
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)
|
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)
|
||||||
|
}
|
||||||
@@ -4,7 +4,7 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/matterclient"
|
"github.com/42wim/matterbridge/matterclient"
|
||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/v5/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
// handleDownloadAvatar downloads the avatar of userid from channel
|
// handleDownloadAvatar downloads the avatar of userid from channel
|
||||||
@@ -66,6 +66,10 @@ func (b *Bmattermost) handleMatter() {
|
|||||||
} else {
|
} else {
|
||||||
b.Log.Debugf("Choosing login/password based receiving")
|
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)
|
go b.handleMatterClient(messages)
|
||||||
}
|
}
|
||||||
var ok bool
|
var ok bool
|
||||||
@@ -104,7 +108,7 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {
|
|||||||
Channel: message.Channel,
|
Channel: message.Channel,
|
||||||
Text: message.Text,
|
Text: message.Text,
|
||||||
ID: message.Post.Id,
|
ID: message.Post.Id,
|
||||||
ParentID: message.Post.ParentId,
|
ParentID: message.Post.RootId, // ParentID is obsolete with mattermost
|
||||||
Extra: make(map[string][]interface{}),
|
Extra: make(map[string][]interface{}),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/matterclient"
|
"github.com/42wim/matterbridge/matterclient"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/v5/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (b *Bmattermost) doConnectWebhookBind() error {
|
func (b *Bmattermost) doConnectWebhookBind() error {
|
||||||
@@ -70,6 +70,7 @@ func (b *Bmattermost) apiLogin() error {
|
|||||||
b.mc.SetLogLevel("debug")
|
b.mc.SetLogLevel("debug")
|
||||||
}
|
}
|
||||||
b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify")
|
b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify")
|
||||||
|
b.mc.SkipVersionCheck = b.GetBool("SkipVersionCheck")
|
||||||
b.mc.NoTLS = b.GetBool("NoTLS")
|
b.mc.NoTLS = b.GetBool("NoTLS")
|
||||||
b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))
|
b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))
|
||||||
err := b.mc.Login()
|
err := b.mc.Login()
|
||||||
@@ -186,6 +187,12 @@ func (b *Bmattermost) skipMessage(message *matterclient.Message) bool {
|
|||||||
return true
|
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
|
// Ignore messages sent from matterbridge
|
||||||
if message.Post.Props != nil {
|
if message.Post.Props != nil {
|
||||||
if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok {
|
if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok {
|
||||||
|
|||||||
@@ -121,6 +121,21 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
|
|||||||
return msg.ID, b.mc.DeleteMessage(msg.ID)
|
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, res := b.mc.Client.GetPost(msg.ParentID, "")
|
||||||
|
if res.Error != nil {
|
||||||
|
b.Log.Errorf("getting post %s failed: %s", msg.ParentID, res.Error.DetailedError)
|
||||||
|
}
|
||||||
|
msg.ParentID = post.RootId
|
||||||
|
}
|
||||||
|
|
||||||
// Upload a file if it exists
|
// Upload a file if it exists
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
|||||||
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"
|
||||||
|
}
|
||||||
227
bridge/msteams/msteams.go
Normal file
227
bridge/msteams/msteams.go
Normal file
@@ -0,0 +1,227 @@
|
|||||||
|
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"}
|
||||||
|
var 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, 0600)
|
||||||
|
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.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()
|
||||||
|
}
|
||||||
96
bridge/mumble/handlers.go
Normal file
96
bridge/mumble/handlers.go
Normal file
@@ -0,0 +1,96 @@
|
|||||||
|
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 {
|
||||||
|
fname := b.Account + "_" + strconv.FormatInt(now.UnixNano(), 10) + "_" + strconv.Itoa(i) + part.FileExtension
|
||||||
|
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)
|
||||||
|
event.Client.Self.SetSelfMuted(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) handleUserChange(event *gumble.UserChangeEvent) {
|
||||||
|
// Only care about changes to self
|
||||||
|
if event.User != event.Client.Self {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Someone attempted to move the user out of the configured channel; attempt to join back
|
||||||
|
if b.Channel != nil {
|
||||||
|
if err := b.doJoin(event.Client, *b.Channel); err != nil {
|
||||||
|
b.Log.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
259
bridge/mumble/mumble.go
Normal file
259
bridge/mumble/mumble.go
Normal file
@@ -0,0 +1,259 @@
|
|||||||
|
package bmumble
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/tls"
|
||||||
|
"crypto/x509"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
|
"net"
|
||||||
|
"strconv"
|
||||||
|
"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 {
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
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 {
|
||||||
|
msgLines = helper.GetSubLines(msg.Text, *maxLength-len(msg.Username), b.GetString("MessageClipped"))
|
||||||
|
} else {
|
||||||
|
msgLines = helper.GetSubLines(msg.Text, 0, b.GetString("MessageClipped"))
|
||||||
|
}
|
||||||
|
// Send the individual lindes
|
||||||
|
for i := range msgLines {
|
||||||
|
b.client.Self.Channel.Send(msg.Username+msgLines[i], false)
|
||||||
|
}
|
||||||
|
}
|
||||||
284
bridge/nctalk/nctalk.go
Normal file
284
bridge/nctalk/nctalk.go
Normal file
@@ -0,0 +1,284 @@
|
|||||||
|
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 {
|
||||||
|
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 := r.room.SendMessage(msg.Username + 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) 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 := msg.Username
|
||||||
|
if fi.Comment != "" {
|
||||||
|
message += fi.Comment + " "
|
||||||
|
}
|
||||||
|
message += fi.URL
|
||||||
|
_, err := r.room.SendMessage(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
|
||||||
|
}
|
||||||
136
bridge/rocketchat/handlers.go
Normal file
136
bridge/rocketchat/handlers.go
Normal file
@@ -0,0 +1,136 @@
|
|||||||
|
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() {
|
||||||
|
messages := make(chan *config.Message)
|
||||||
|
if b.GetString("WebhookBindAddress") != "" {
|
||||||
|
b.Log.Debugf("Choosing webhooks based receiving")
|
||||||
|
go b.handleRocketHook(messages)
|
||||||
|
} else {
|
||||||
|
b.Log.Debugf("Choosing login/password based receiving")
|
||||||
|
go b.handleRocketClient(messages)
|
||||||
|
}
|
||||||
|
for message := range messages {
|
||||||
|
message.Account = b.Account
|
||||||
|
b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
|
||||||
|
b.Log.Debugf("<= Message is %#v", message)
|
||||||
|
b.Remote <- *message
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) handleRocketHook(messages chan *config.Message) {
|
||||||
|
for {
|
||||||
|
message := b.rh.Receive()
|
||||||
|
b.Log.Debugf("Receiving from rockethook %#v", message)
|
||||||
|
// do not loop
|
||||||
|
if message.UserName == b.GetString("Nick") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
messages <- &config.Message{
|
||||||
|
UserID: message.UserID,
|
||||||
|
Username: message.UserName,
|
||||||
|
Text: message.Text,
|
||||||
|
Channel: message.ChannelName,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
b.cache.Add(message.ID, true)
|
||||||
|
b.Log.Debugf("message %#v", message)
|
||||||
|
m := message
|
||||||
|
if b.skipMessage(&m) {
|
||||||
|
b.Log.Debugf("Skipped message: %#v", message)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
rmsg := &config.Message{Text: message.Msg,
|
||||||
|
Username: message.User.UserName,
|
||||||
|
Channel: b.getChannelName(message.RoomID),
|
||||||
|
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
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
if err := b.uploadFile(&fi, b.getChannelID(msg.Channel)); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
201
bridge/rocketchat/helpers.go
Normal file
201
bridge/rocketchat/helpers.go
Normal file
@@ -0,0 +1,201 @@
|
|||||||
|
package brocketchat
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"io/ioutil"
|
||||||
|
"mime"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
"github.com/42wim/matterbridge/hook/rockethook"
|
||||||
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/models"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/realtime"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/rest"
|
||||||
|
"github.com/nelsonken/gomf"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (b *Brocketchat) doConnectWebhookBind() error {
|
||||||
|
switch {
|
||||||
|
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"),
|
||||||
|
DisableServer: true})
|
||||||
|
b.rh = rockethook.New(b.GetString("WebhookURL"), rockethook.Config{BindAddress: b.GetString("WebhookBindAddress")})
|
||||||
|
case b.GetString("Login") != "":
|
||||||
|
b.Log.Info("Connecting using login/password (sending)")
|
||||||
|
err := b.apiLogin()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
b.Log.Info("Connecting using webhookbindaddress (receiving)")
|
||||||
|
b.rh = rockethook.New(b.GetString("WebhookURL"), rockethook.Config{BindAddress: b.GetString("WebhookBindAddress")})
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) doConnectWebhookURL() error {
|
||||||
|
b.Log.Info("Connecting using webhookurl (sending)")
|
||||||
|
b.mh = matterhook.New(b.GetString("WebhookURL"),
|
||||||
|
matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
|
||||||
|
DisableServer: true})
|
||||||
|
if b.GetString("Login") != "" {
|
||||||
|
b.Log.Info("Connecting using login/password (receiving)")
|
||||||
|
err := b.apiLogin()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
client, err := realtime.NewClient(myURL, b.GetBool("debug"))
|
||||||
|
b.c = client
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
restclient := rest.NewClient(myURL, b.GetBool("debug"))
|
||||||
|
user, err := b.c.Login(credentials)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b.user = user
|
||||||
|
b.r = restclient
|
||||||
|
err = b.r.Login(credentials)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
b.Log.Info("Connection succeeded")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) getChannelName(id string) string {
|
||||||
|
b.RLock()
|
||||||
|
defer b.RUnlock()
|
||||||
|
if name, ok := b.channelMap[id]; ok {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) getChannelID(name string) string {
|
||||||
|
b.RLock()
|
||||||
|
defer b.RUnlock()
|
||||||
|
for k, v := range b.channelMap {
|
||||||
|
if v == name || v == "#"+name {
|
||||||
|
return k
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) skipMessage(message *models.Message) bool {
|
||||||
|
return message.User.ID == b.user.ID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Brocketchat) uploadFile(fi *config.FileInfo, channel string) error {
|
||||||
|
fb := gomf.New()
|
||||||
|
if err := fb.WriteField("description", fi.Comment); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
sp := strings.Split(fi.Name, ".")
|
||||||
|
mtype := mime.TypeByExtension("." + sp[len(sp)-1])
|
||||||
|
if !strings.Contains(mtype, "image") && !strings.Contains(mtype, "video") {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if err := fb.WriteFile("file", fi.Name, mtype, *fi.Data); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
req, err := fb.GetHTTPRequest(context.TODO(), b.GetString("server")+"/api/v1/rooms.upload/"+channel)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
req.Header.Add("X-Auth-Token", b.user.Token)
|
||||||
|
req.Header.Add("X-User-Id", b.user.ID)
|
||||||
|
client := &http.Client{
|
||||||
|
Timeout: time.Second * 5,
|
||||||
|
}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
body, err := ioutil.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.StatusCode != 200 {
|
||||||
|
b.Log.Errorf("failed: %#v", string(body))
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// sendWebhook uses the configured WebhookURL to send the message
|
||||||
|
func (b *Brocketchat) sendWebhook(msg *config.Message) error {
|
||||||
|
// skip events
|
||||||
|
if msg.Event != "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if b.GetBool("PrefixMessagesWithNick") {
|
||||||
|
msg.Text = msg.Username + msg.Text
|
||||||
|
}
|
||||||
|
if msg.Extra != nil {
|
||||||
|
// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE
|
||||||
|
for _, rmsg := range helper.HandleExtra(msg, b.General) {
|
||||||
|
rmsg := rmsg // scopelint
|
||||||
|
iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
|
||||||
|
matterMessage := matterhook.OMessage{
|
||||||
|
IconURL: iconURL,
|
||||||
|
Channel: rmsg.Channel,
|
||||||
|
UserName: rmsg.Username,
|
||||||
|
Text: rmsg.Text,
|
||||||
|
Props: make(map[string]interface{}),
|
||||||
|
}
|
||||||
|
if err := b.mh.Send(matterMessage); err != nil {
|
||||||
|
b.Log.Errorf("sendWebhook failed: %s ", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// webhook doesn't support file uploads, so we add the url manually
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.URL != "" {
|
||||||
|
msg.Text += fi.URL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
iconURL := config.GetIconURL(msg, b.GetString("iconurl"))
|
||||||
|
matterMessage := matterhook.OMessage{
|
||||||
|
IconURL: iconURL,
|
||||||
|
Channel: msg.Channel,
|
||||||
|
UserName: msg.Username,
|
||||||
|
Text: msg.Text,
|
||||||
|
}
|
||||||
|
if msg.Avatar != "" {
|
||||||
|
matterMessage.IconURL = msg.Avatar
|
||||||
|
}
|
||||||
|
err := b.mh.Send(matterMessage)
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Info(err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -1,21 +1,53 @@
|
|||||||
package brocketchat
|
package brocketchat
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/hook/rockethook"
|
"github.com/42wim/matterbridge/hook/rockethook"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
|
lru "github.com/hashicorp/golang-lru"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/models"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/realtime"
|
||||||
|
"github.com/matterbridge/Rocket.Chat.Go.SDK/rest"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Brocketchat struct {
|
type Brocketchat struct {
|
||||||
mh *matterhook.Client
|
mh *matterhook.Client
|
||||||
rh *rockethook.Client
|
rh *rockethook.Client
|
||||||
|
c *realtime.Client
|
||||||
|
r *rest.Client
|
||||||
|
cache *lru.Cache
|
||||||
*bridge.Config
|
*bridge.Config
|
||||||
|
messageChan chan models.Message
|
||||||
|
channelMap map[string]string
|
||||||
|
user *models.User
|
||||||
|
sync.RWMutex
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
sUserJoined = "uj"
|
||||||
|
sUserLeft = "ul"
|
||||||
|
sRoomChangedTopic = "room_changed_topic"
|
||||||
|
)
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
return &Brocketchat{Config: cfg}
|
newCache, err := lru.New(100)
|
||||||
|
if err != nil {
|
||||||
|
cfg.Log.Fatalf("Could not create LRU cache for rocketchat bridge: %v", err)
|
||||||
|
}
|
||||||
|
b := &Brocketchat{
|
||||||
|
Config: cfg,
|
||||||
|
messageChan: make(chan models.Message),
|
||||||
|
channelMap: make(map[string]string),
|
||||||
|
cache: newCache,
|
||||||
|
}
|
||||||
|
b.Log.Debugf("enabling rocketchat")
|
||||||
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) Command(cmd string) string {
|
func (b *Brocketchat) Command(cmd string) string {
|
||||||
@@ -23,70 +55,127 @@ func (b *Brocketchat) Command(cmd string) string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) Connect() error {
|
func (b *Brocketchat) Connect() error {
|
||||||
b.Log.Info("Connecting webhooks")
|
if b.GetString("WebhookBindAddress") != "" {
|
||||||
b.mh = matterhook.New(b.GetString("WebhookURL"),
|
if err := b.doConnectWebhookBind(); err != nil {
|
||||||
matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
|
return err
|
||||||
DisableServer: true})
|
}
|
||||||
b.rh = rockethook.New(b.GetString("WebhookURL"), rockethook.Config{BindAddress: b.GetString("WebhookBindAddress")})
|
go b.handleRocket()
|
||||||
go b.handleRocketHook()
|
return nil
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case b.GetString("WebhookURL") != "":
|
||||||
|
if err := b.doConnectWebhookURL(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
go b.handleRocket()
|
||||||
|
return nil
|
||||||
|
case b.GetString("Login") != "":
|
||||||
|
b.Log.Info("Connecting using login/password (sending and receiving)")
|
||||||
|
err := b.apiLogin()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
go b.handleRocket()
|
||||||
|
}
|
||||||
|
if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" &&
|
||||||
|
b.GetString("Login") == "" {
|
||||||
|
return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Login/Password/Server configured")
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) Disconnect() error {
|
func (b *Brocketchat) Disconnect() error {
|
||||||
return nil
|
return nil
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error {
|
func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error {
|
||||||
|
if b.c == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
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: strings.TrimPrefix(channel.Name, "#")}
|
||||||
|
if err := b.c.JoinChannel(id); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := b.c.SubscribeToMessageStream(mychannel, b.messageChan); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) Send(msg config.Message) (string, error) {
|
func (b *Brocketchat) Send(msg config.Message) (string, error) {
|
||||||
// ignore delete messages
|
// strip the # if people has set this
|
||||||
if msg.Event == config.EventMsgDelete {
|
msg.Channel = strings.TrimPrefix(msg.Channel, "#")
|
||||||
return "", nil
|
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 + "_"
|
||||||
}
|
}
|
||||||
b.Log.Debugf("=> Receiving %#v", msg)
|
|
||||||
|
// Delete message
|
||||||
|
if msg.Event == config.EventMsgDelete {
|
||||||
|
if msg.ID == "" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
return msg.ID, b.c.DeleteMessage(&models.Message{ID: msg.ID})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Use webhook to send the message
|
||||||
|
if b.GetString("WebhookURL") != "" {
|
||||||
|
return "", b.sendWebhook(&msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Prepend nick if configured
|
||||||
|
if b.GetBool("PrefixMessagesWithNick") {
|
||||||
|
msg.Text = msg.Username + msg.Text
|
||||||
|
}
|
||||||
|
|
||||||
|
// Edit message if we have an ID
|
||||||
|
if msg.ID != "" {
|
||||||
|
return msg.ID, b.c.EditMessage(&models.Message{ID: msg.ID, Msg: msg.Text, RoomID: b.getChannelID(msg.Channel)})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Upload a file if it exists
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
rmsg := rmsg // scopelint
|
// strip the # if people has set this
|
||||||
iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
|
rmsg.Channel = strings.TrimPrefix(rmsg.Channel, "#")
|
||||||
matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text}
|
smsg := &models.Message{
|
||||||
b.mh.Send(matterMessage)
|
RoomID: b.getChannelID(rmsg.Channel),
|
||||||
}
|
Msg: rmsg.Username + rmsg.Text,
|
||||||
if len(msg.Extra["file"]) > 0 {
|
PostMessage: models.PostMessage{
|
||||||
for _, f := range msg.Extra["file"] {
|
Avatar: rmsg.Avatar,
|
||||||
fi := f.(config.FileInfo)
|
Alias: rmsg.Username,
|
||||||
if fi.URL != "" {
|
},
|
||||||
msg.Text += fi.URL
|
}
|
||||||
}
|
if _, err := b.c.SendMessage(smsg); err != nil {
|
||||||
|
b.Log.Errorf("SendMessage failed: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
return "", b.handleUploadFile(&msg)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
|
smsg := &models.Message{
|
||||||
matterMessage := matterhook.OMessage{IconURL: iconURL}
|
RoomID: channel.ID,
|
||||||
matterMessage.Channel = msg.Channel
|
Msg: msg.Text,
|
||||||
matterMessage.UserName = msg.Username
|
PostMessage: models.PostMessage{
|
||||||
matterMessage.Type = ""
|
Avatar: msg.Avatar,
|
||||||
matterMessage.Text = msg.Text
|
Alias: msg.Username,
|
||||||
err := b.mh.Send(matterMessage)
|
},
|
||||||
if err != nil {
|
}
|
||||||
b.Log.Info(err)
|
|
||||||
|
rmsg, err := b.c.SendMessage(smsg)
|
||||||
|
if rmsg == nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
return "", nil
|
return rmsg.ID, err
|
||||||
}
|
|
||||||
|
|
||||||
func (b *Brocketchat) handleRocketHook() {
|
|
||||||
for {
|
|
||||||
message := b.rh.Receive()
|
|
||||||
b.Log.Debugf("Receiving from rockethook %#v", message)
|
|
||||||
// do not loop
|
|
||||||
if message.UserName == b.GetString("Nick") {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
b.Log.Debugf("<= Sending message from %s on %s to gateway", message.UserName, b.Account)
|
|
||||||
b.Remote <- config.Message{Text: message.Text, Username: message.UserName, Channel: message.ChannelName, Account: b.Account, UserID: message.UserID}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,18 +1,22 @@
|
|||||||
package bslack
|
package bslack
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"html"
|
"html"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"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() {
|
func (b *Bslack) handleSlack() {
|
||||||
messages := make(chan *config.Message)
|
messages := make(chan *config.Message)
|
||||||
if b.GetString(incomingWebhookConfig) != "" {
|
if b.GetString(incomingWebhookConfig) != "" && b.GetString(tokenConfig) == "" {
|
||||||
b.Log.Debugf("Choosing webhooks based receiving")
|
b.Log.Debugf("Choosing webhooks based receiving")
|
||||||
go b.handleMatterHook(messages)
|
go b.handleMatterHook(messages)
|
||||||
} else {
|
} else {
|
||||||
@@ -22,20 +26,21 @@ func (b *Bslack) handleSlack() {
|
|||||||
time.Sleep(time.Second)
|
time.Sleep(time.Second)
|
||||||
b.Log.Debug("Start listening for Slack messages")
|
b.Log.Debug("Start listening for Slack messages")
|
||||||
for message := range 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 {
|
||||||
b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
|
b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
|
||||||
|
// 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.Log.Debugf("<= Message is %#v", message)
|
||||||
b.Remote <- *message
|
b.Remote <- *message
|
||||||
}
|
}
|
||||||
@@ -43,7 +48,7 @@ func (b *Bslack) handleSlack() {
|
|||||||
|
|
||||||
func (b *Bslack) handleSlackClient(messages chan *config.Message) {
|
func (b *Bslack) handleSlackClient(messages chan *config.Message) {
|
||||||
for msg := range b.rtm.IncomingEvents {
|
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)
|
b.Log.Debugf("== Receiving event %#v", msg.Data)
|
||||||
}
|
}
|
||||||
switch ev := msg.Data.(type) {
|
switch ev := msg.Data.(type) {
|
||||||
@@ -52,7 +57,9 @@ func (b *Bslack) handleSlackClient(messages chan *config.Message) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
rmsg, err := b.handleTypingEvent(ev)
|
rmsg, err := b.handleTypingEvent(ev)
|
||||||
if err != nil {
|
if err == ErrEventIgnored {
|
||||||
|
continue
|
||||||
|
} else if err != nil {
|
||||||
b.Log.Errorf("%#v", err)
|
b.Log.Errorf("%#v", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -75,21 +82,18 @@ func (b *Bslack) handleSlackClient(messages chan *config.Message) {
|
|||||||
// When we join a channel we update the full list of users as
|
// 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
|
// well as the information for the channel that we joined as this
|
||||||
// should now tell that we are a member of it.
|
// should now tell that we are a member of it.
|
||||||
b.channelsMutex.Lock()
|
b.channels.registerChannel(ev.Channel)
|
||||||
b.channelsByID[ev.Channel.ID] = &ev.Channel
|
|
||||||
b.channelsByName[ev.Channel.Name] = &ev.Channel
|
|
||||||
b.channelsMutex.Unlock()
|
|
||||||
case *slack.ConnectedEvent:
|
case *slack.ConnectedEvent:
|
||||||
b.si = ev.Info
|
b.si = ev.Info
|
||||||
b.populateChannels(true)
|
b.channels.populateChannels(true)
|
||||||
b.populateUsers(true)
|
b.users.populateUsers(true)
|
||||||
case *slack.InvalidAuthEvent:
|
case *slack.InvalidAuthEvent:
|
||||||
b.Log.Fatalf("Invalid Token %#v", ev)
|
b.Log.Fatalf("Invalid Token %#v", ev)
|
||||||
case *slack.ConnectionErrorEvent:
|
case *slack.ConnectionErrorEvent:
|
||||||
b.Log.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj)
|
b.Log.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj)
|
||||||
case *slack.MemberJoinedChannelEvent:
|
case *slack.MemberJoinedChannelEvent:
|
||||||
b.populateUser(ev.User)
|
b.users.populateUser(ev.User)
|
||||||
case *slack.LatencyReport:
|
case *slack.HelloEvent, *slack.LatencyReport, *slack.ConnectingEvent:
|
||||||
continue
|
continue
|
||||||
default:
|
default:
|
||||||
b.Log.Debugf("Unhandled incoming event: %T", ev)
|
b.Log.Debugf("Unhandled incoming event: %T", ev)
|
||||||
@@ -126,18 +130,34 @@ func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Skip any messages that we made ourselves or from 'slackbot' (see #527).
|
// Check for our callback ID
|
||||||
if ev.Username == sSlackBotUser ||
|
hasOurCallbackID := false
|
||||||
(b.rtm != nil && ev.Username == b.si.User.Name) ||
|
if len(ev.Blocks.BlockSet) == 1 {
|
||||||
(len(ev.Attachments) > 0 && ev.Attachments[0].CallbackID == "matterbridge_"+b.uuid) {
|
block, ok := ev.Blocks.BlockSet[0].(*slack.SectionBlock)
|
||||||
return true
|
hasOurCallbackID = ok && block.BlockID == "matterbridge_"+b.uuid
|
||||||
}
|
}
|
||||||
|
|
||||||
// It seems ev.SubMessage.Edited == nil when slack unfurls.
|
if ev.SubMessage != nil {
|
||||||
// Do not forward these messages. See Github issue #266.
|
// It seems ev.SubMessage.Edited == nil when slack unfurls.
|
||||||
if ev.SubMessage != nil &&
|
// Do not forward these messages. See Github issue #266.
|
||||||
ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp &&
|
if ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp &&
|
||||||
ev.SubMessage.Edited == nil {
|
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
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -192,6 +212,9 @@ func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, er
|
|||||||
// This is probably a webhook we couldn't resolve.
|
// This is probably a webhook we couldn't resolve.
|
||||||
return nil, fmt.Errorf("message handling resulted in an empty bot message (probably an incoming webhook we couldn't resolve): %#v", ev)
|
return nil, fmt.Errorf("message handling resulted in an empty bot message (probably an incoming webhook we couldn't resolve): %#v", ev)
|
||||||
}
|
}
|
||||||
|
if ev.SubMessage != nil {
|
||||||
|
return nil, fmt.Errorf("message handling resulted in an empty message: %#v with submessage %#v", ev, ev.SubMessage)
|
||||||
|
}
|
||||||
return nil, fmt.Errorf("message handling resulted in an empty message: %#v", ev)
|
return nil, fmt.Errorf("message handling resulted in an empty message: %#v", ev)
|
||||||
}
|
}
|
||||||
return rmsg, nil
|
return rmsg, nil
|
||||||
@@ -207,7 +230,7 @@ func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message)
|
|||||||
rmsg.Username = sSystemUser
|
rmsg.Username = sSystemUser
|
||||||
rmsg.Event = config.EventJoinLeave
|
rmsg.Event = config.EventJoinLeave
|
||||||
case sChannelTopic, sChannelPurpose:
|
case sChannelTopic, sChannelPurpose:
|
||||||
b.populateChannels(false)
|
b.channels.populateChannels(false)
|
||||||
rmsg.Event = config.EventTopicChange
|
rmsg.Event = config.EventTopicChange
|
||||||
case sMessageChanged:
|
case sMessageChanged:
|
||||||
rmsg.Text = ev.SubMessage.Text
|
rmsg.Text = ev.SubMessage.Text
|
||||||
@@ -263,7 +286,10 @@ func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message, error) {
|
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 {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -313,36 +339,7 @@ func (b *Bslack) handleGetChannelMembers(rmsg *config.Message) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
cMembers := config.ChannelMembers{}
|
cMembers := b.channels.getChannelMembers(b.users)
|
||||||
|
|
||||||
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()
|
|
||||||
|
|
||||||
extra := make(map[string][]interface{})
|
extra := make(map[string][]interface{})
|
||||||
extra[config.EventGetChannelMembers] = append(extra[config.EventGetChannelMembers], cMembers)
|
extra[config.EventGetChannelMembers] = append(extra[config.EventGetChannelMembers], cMembers)
|
||||||
|
|||||||
@@ -1,233 +1,21 @@
|
|||||||
package bslack
|
package bslack
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"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
|
// populateReceivedMessage shapes the initial Matterbridge message that we will forward to the
|
||||||
// router before we apply message-dependent modifications.
|
// router before we apply message-dependent modifications.
|
||||||
func (b *Bslack) populateReceivedMessage(ev *slack.MessageEvent) (*config.Message, error) {
|
func (b *Bslack) populateReceivedMessage(ev *slack.MessageEvent) (*config.Message, error) {
|
||||||
// Use our own func because rtm.GetChannelInfo doesn't work for private channels.
|
// Use our own func because rtm.GetChannelInfo doesn't work for private channels.
|
||||||
channel, err := b.getChannelByID(ev.Channel)
|
channel, err := b.channels.getChannelByID(ev.Channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -254,6 +42,13 @@ func (b *Bslack) populateReceivedMessage(ev *slack.MessageEvent) (*config.Messag
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// For edits, only submessage has thread ts.
|
||||||
|
// Ensures edits to threaded messages maintain their prefix hint on the
|
||||||
|
// unthreaded end.
|
||||||
|
if ev.SubMessage != nil {
|
||||||
|
rmsg.ParentID = ev.SubMessage.ThreadTimestamp
|
||||||
|
}
|
||||||
|
|
||||||
if err = b.populateMessageWithUserInfo(ev, rmsg); err != nil {
|
if err = b.populateMessageWithUserInfo(ev, rmsg); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -282,7 +77,7 @@ func (b *Bslack) populateMessageWithUserInfo(ev *slack.MessageEvent, rmsg *confi
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
user := b.getUser(userID)
|
user := b.users.getUser(userID)
|
||||||
if user == nil {
|
if user == nil {
|
||||||
return fmt.Errorf("could not find information for user with id %s", ev.User)
|
return fmt.Errorf("could not find information for user with id %s", ev.User)
|
||||||
}
|
}
|
||||||
@@ -308,7 +103,7 @@ func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config
|
|||||||
break
|
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)
|
b.Log.Errorf("Could not retrieve bot information: %#v", err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -353,7 +148,7 @@ func (b *Bslack) extractTopicOrPurpose(text string) (string, string) {
|
|||||||
func (b *Bslack) replaceMention(text string) string {
|
func (b *Bslack) replaceMention(text string) string {
|
||||||
replaceFunc := func(match string) string {
|
replaceFunc := func(match string) string {
|
||||||
userID := strings.Trim(match, "@<>")
|
userID := strings.Trim(match, "@<>")
|
||||||
if username := b.getUsername(userID); userID != "" {
|
if username := b.users.getUsername(userID); userID != "" {
|
||||||
return "@" + username
|
return "@" + username
|
||||||
}
|
}
|
||||||
return match
|
return match
|
||||||
@@ -393,18 +188,38 @@ func (b *Bslack) replaceURL(text string) string {
|
|||||||
return text
|
return text
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bslack) replaceCodeFence(text string) string {
|
func (b *Bslack) replaceb0rkedMarkDown(text string) string {
|
||||||
return codeFenceRE.ReplaceAllString(text, "```")
|
// 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
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bslack) handleRateLimit(err error) error {
|
func (b *Bslack) replaceCodeFence(text string) string {
|
||||||
rateLimit, ok := err.(*slack.RateLimitedError)
|
return codeFenceRE.ReplaceAllString(text, "```")
|
||||||
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
|
// getUsersInConversation returns an array of userIDs that are members of channelID
|
||||||
@@ -417,7 +232,7 @@ func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) {
|
|||||||
|
|
||||||
members, nextCursor, err := b.sc.GetUsersInConversation(queryParams)
|
members, nextCursor, err := b.sc.GetUsersInConversation(queryParams)
|
||||||
if err != nil {
|
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)
|
return channelMembers, fmt.Errorf("Could not retrieve users in channels: %#v", err)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
@@ -432,3 +247,13 @@ func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) {
|
|||||||
}
|
}
|
||||||
return channelMembers, nil
|
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
|
||||||
|
}
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ func TestExtractTopicOrPurpose(t *testing.T) {
|
|||||||
|
|
||||||
logger := logrus.New()
|
logger := logrus.New()
|
||||||
logger.SetOutput(ioutil.Discard)
|
logger.SetOutput(ioutil.Discard)
|
||||||
cfg := &bridge.Config{Log: logger.WithFields(nil)}
|
cfg := &bridge.Config{Bridge: &bridge.Bridge{Log: logrus.NewEntry(logger)}}
|
||||||
b := newBridge(cfg)
|
b := newBridge(cfg)
|
||||||
for name, tc := range testcases {
|
for name, tc := range testcases {
|
||||||
gotChangeType, gotOutput := b.extractTopicOrPurpose(tc.input)
|
gotChangeType, gotOutput := b.extractTopicOrPurpose(tc.input)
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import (
|
|||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
"github.com/nlopes/slack"
|
"github.com/slack-go/slack"
|
||||||
)
|
)
|
||||||
|
|
||||||
type BLegacy struct {
|
type BLegacy struct {
|
||||||
@@ -13,7 +13,9 @@ type BLegacy struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func NewLegacy(cfg *bridge.Config) bridge.Bridger {
|
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 {
|
func (b *BLegacy) Connect() error {
|
||||||
@@ -55,14 +57,18 @@ func (b *BLegacy) Connect() error {
|
|||||||
})
|
})
|
||||||
if b.GetString(tokenConfig) != "" {
|
if b.GetString(tokenConfig) != "" {
|
||||||
b.Log.Info("Connecting using token (receiving)")
|
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()
|
b.rtm = b.sc.NewRTM()
|
||||||
go b.rtm.ManageConnection()
|
go b.rtm.ManageConnection()
|
||||||
go b.handleSlack()
|
go b.handleSlack()
|
||||||
}
|
}
|
||||||
} else if b.GetString(tokenConfig) != "" {
|
} else if b.GetString(tokenConfig) != "" {
|
||||||
b.Log.Info("Connecting using token (sending and receiving)")
|
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()
|
b.rtm = b.sc.NewRTM()
|
||||||
go b.rtm.ManageConnection()
|
go b.rtm.ManageConnection()
|
||||||
go b.handleSlack()
|
go b.handleSlack()
|
||||||
|
|||||||
@@ -12,9 +12,9 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
"github.com/hashicorp/golang-lru"
|
lru "github.com/hashicorp/golang-lru"
|
||||||
"github.com/nlopes/slack"
|
|
||||||
"github.com/rs/xid"
|
"github.com/rs/xid"
|
||||||
|
"github.com/slack-go/slack"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Bslack struct {
|
type Bslack struct {
|
||||||
@@ -30,23 +30,13 @@ type Bslack struct {
|
|||||||
uuid string
|
uuid string
|
||||||
useChannelID bool
|
useChannelID bool
|
||||||
|
|
||||||
users map[string]*slack.User
|
channels *channels
|
||||||
usersMutex sync.RWMutex
|
users *users
|
||||||
|
legacy bool
|
||||||
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
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
sHello = "hello"
|
||||||
sChannelJoin = "channel_join"
|
sChannelJoin = "channel_join"
|
||||||
sChannelLeave = "channel_leave"
|
sChannelLeave = "channel_leave"
|
||||||
sChannelJoined = "channel_joined"
|
sChannelJoined = "channel_joined"
|
||||||
@@ -74,6 +64,7 @@ const (
|
|||||||
editSuffixConfig = "EditSuffix"
|
editSuffixConfig = "EditSuffix"
|
||||||
iconURLConfig = "iconurl"
|
iconURLConfig = "iconurl"
|
||||||
noSendJoinConfig = "nosendjoinpart"
|
noSendJoinConfig = "nosendjoinpart"
|
||||||
|
messageLength = 3000
|
||||||
)
|
)
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
@@ -94,14 +85,9 @@ func newBridge(cfg *bridge.Config) *Bslack {
|
|||||||
cfg.Log.Fatalf("Could not create LRU cache for Slack bridge: %v", err)
|
cfg.Log.Fatalf("Could not create LRU cache for Slack bridge: %v", err)
|
||||||
}
|
}
|
||||||
b := &Bslack{
|
b := &Bslack{
|
||||||
Config: cfg,
|
Config: cfg,
|
||||||
uuid: xid.New().String(),
|
uuid: xid.New().String(),
|
||||||
cache: newCache,
|
cache: newCache,
|
||||||
users: map[string]*slack.User{},
|
|
||||||
channelsByID: map[string]*slack.Channel{},
|
|
||||||
channelsByName: map[string]*slack.Channel{},
|
|
||||||
earliestChannelRefresh: time.Now(),
|
|
||||||
earliestUserRefresh: time.Now(),
|
|
||||||
}
|
}
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
@@ -121,7 +107,12 @@ func (b *Bslack) Connect() error {
|
|||||||
// If we have a token we use the Slack websocket-based RTM for both sending and receiving.
|
// If we have a token we use the Slack websocket-based RTM for both sending and receiving.
|
||||||
if token := b.GetString(tokenConfig); token != "" {
|
if token := b.GetString(tokenConfig); token != "" {
|
||||||
b.Log.Info("Connecting using token")
|
b.Log.Info("Connecting using token")
|
||||||
b.sc = slack.New(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()
|
b.rtm = b.sc.NewRTM()
|
||||||
go b.rtm.ManageConnection()
|
go b.rtm.ManageConnection()
|
||||||
go b.handleSlack()
|
go b.handleSlack()
|
||||||
@@ -163,9 +154,21 @@ func (b *Bslack) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
return nil
|
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 {
|
if err != nil {
|
||||||
return fmt.Errorf("could not join channel: %#v", err)
|
return fmt.Errorf("could not join channel: %#v", err)
|
||||||
}
|
}
|
||||||
@@ -175,7 +178,8 @@ func (b *Bslack) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
channel.Name = channelInfo.Name
|
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 fmt.Errorf("slack integration that matterbridge is using is not member of channel '%s', please add it manually", channelInfo.Name)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
@@ -191,6 +195,7 @@ func (b *Bslack) Send(msg config.Message) (string, error) {
|
|||||||
b.Log.Debugf("=> Receiving %#v", msg)
|
b.Log.Debugf("=> Receiving %#v", msg)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
msg.Text = helper.ClipMessage(msg.Text, messageLength, b.GetString("MessageClipped"))
|
||||||
msg.Text = b.replaceCodeFence(msg.Text)
|
msg.Text = b.replaceCodeFence(msg.Text)
|
||||||
|
|
||||||
// Make a action /me of the message
|
// Make a action /me of the message
|
||||||
@@ -199,7 +204,7 @@ func (b *Bslack) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Use webhook to send the message
|
// Use webhook to send the message
|
||||||
if b.GetString(outgoingWebhookConfig) != "" {
|
if b.GetString(outgoingWebhookConfig) != "" && b.GetString(tokenConfig) == "" {
|
||||||
return "", b.sendWebhook(msg)
|
return "", b.sendWebhook(msg)
|
||||||
}
|
}
|
||||||
return b.sendRTM(msg)
|
return b.sendRTM(msg)
|
||||||
@@ -275,7 +280,7 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) {
|
|||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
channelInfo, err := b.getChannel(msg.Channel)
|
channelInfo, err := b.channels.getChannel(msg.Channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("could not send message: %v", err)
|
return "", fmt.Errorf("could not send message: %v", err)
|
||||||
}
|
}
|
||||||
@@ -293,6 +298,12 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) {
|
|||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Handle prefix hint for unthreaded messages.
|
||||||
|
if msg.ParentNotFound() {
|
||||||
|
msg.ParentID = ""
|
||||||
|
msg.Text = fmt.Sprintf("[thread]: %s", msg.Text)
|
||||||
|
}
|
||||||
|
|
||||||
// Handle message deletions.
|
// Handle message deletions.
|
||||||
if handled, err = b.deleteMessage(&msg, channelInfo); handled {
|
if handled, err = b.deleteMessage(&msg, channelInfo); handled {
|
||||||
return msg.ID, err
|
return msg.ID, err
|
||||||
@@ -345,7 +356,7 @@ func (b *Bslack) updateTopicOrPurpose(msg *config.Message, channelInfo *slack.Ch
|
|||||||
if err == nil {
|
if err == nil {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if err = b.handleRateLimit(err); err != nil {
|
if err = handleRateLimit(b.Log, err); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -386,7 +397,7 @@ func (b *Bslack) deleteMessage(msg *config.Message, channelInfo *slack.Channel)
|
|||||||
return true, nil
|
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)
|
b.Log.Errorf("Failed to delete user message from Slack: %#v", err)
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
@@ -399,13 +410,12 @@ func (b *Bslack) editMessage(msg *config.Message, channelInfo *slack.Channel) (b
|
|||||||
}
|
}
|
||||||
messageOptions := b.prepareMessageOptions(msg)
|
messageOptions := b.prepareMessageOptions(msg)
|
||||||
for {
|
for {
|
||||||
messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false))
|
|
||||||
_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, messageOptions...)
|
_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, messageOptions...)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return true, 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)
|
b.Log.Errorf("Failed to edit user message on Slack: %#v", err)
|
||||||
return true, err
|
return true, err
|
||||||
}
|
}
|
||||||
@@ -418,14 +428,13 @@ func (b *Bslack) postMessage(msg *config.Message, channelInfo *slack.Channel) (s
|
|||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
messageOptions := b.prepareMessageOptions(msg)
|
messageOptions := b.prepareMessageOptions(msg)
|
||||||
messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false))
|
|
||||||
for {
|
for {
|
||||||
_, id, err := b.rtm.PostMessage(channelInfo.ID, messageOptions...)
|
_, id, err := b.rtm.PostMessage(channelInfo.ID, messageOptions...)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return id, 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)
|
b.Log.Errorf("Failed to sent user message to Slack: %#v", err)
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
@@ -484,8 +493,6 @@ func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var attachments []slack.Attachment
|
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
|
// add file attachments
|
||||||
attachments = append(attachments, b.createAttach(msg.Extra)...)
|
attachments = append(attachments, b.createAttach(msg.Extra)...)
|
||||||
// add slack attachments (from another slack bridge)
|
// add slack attachments (from another slack bridge)
|
||||||
@@ -496,6 +503,19 @@ func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption {
|
|||||||
}
|
}
|
||||||
|
|
||||||
var opts []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.MsgOptionAttachments(attachments...))
|
||||||
opts = append(opts, slack.MsgOptionPostMessageParameters(params))
|
opts = append(opts, slack.MsgOptionPostMessageParameters(params))
|
||||||
return opts
|
return opts
|
||||||
|
|||||||
336
bridge/slack/users_channels.go
Normal file
336
bridge/slack/users_channels.go
Normal file
@@ -0,0 +1,336 @@
|
|||||||
|
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) 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"},
|
||||||
|
}
|
||||||
|
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
|
||||||
|
}
|
||||||
@@ -9,7 +9,6 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/shazow/ssh-chat/sshd"
|
"github.com/shazow/ssh-chat/sshd"
|
||||||
"github.com/sirupsen/logrus"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type Bsshchat struct {
|
type Bsshchat struct {
|
||||||
@@ -131,10 +130,14 @@ func (b *Bsshchat) handleSSHChat() error {
|
|||||||
if strings.Contains(b.r.Text(), "Rate limiting is in effect") {
|
if strings.Contains(b.r.Text(), "Rate limiting is in effect") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
// skip our own messages
|
||||||
|
if !strings.HasPrefix(b.r.Text(), "["+b.GetString("Nick")+"] \x1b") {
|
||||||
|
continue
|
||||||
|
}
|
||||||
res := strings.Split(stripPrompt(b.r.Text()), ":")
|
res := strings.Split(stripPrompt(b.r.Text()), ":")
|
||||||
if res[0] == "-> Set theme" {
|
if res[0] == "-> Set theme" {
|
||||||
wait = false
|
wait = false
|
||||||
logrus.Debugf("mono found, allowing")
|
b.Log.Debugf("mono found, allowing")
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if !wait {
|
if !wait {
|
||||||
|
|||||||
@@ -85,7 +85,7 @@ func (b *Bsteam) handleEvents() {
|
|||||||
|
|
||||||
func (b *Bsteam) handleLogOnFailed(e *steam.LogOnFailedEvent, myLoginInfo *steam.LogOnDetails) error {
|
func (b *Bsteam) handleLogOnFailed(e *steam.LogOnFailedEvent, myLoginInfo *steam.LogOnDetails) error {
|
||||||
switch e.Result {
|
switch e.Result {
|
||||||
case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode:
|
case steamlang.EResult_AccountLoginDeniedNeedTwoFactor:
|
||||||
b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:")
|
b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:")
|
||||||
var code string
|
var code string
|
||||||
fmt.Scanf("%s", &code)
|
fmt.Scanf("%s", &code)
|
||||||
|
|||||||
@@ -2,13 +2,14 @@ package btelegram
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"html"
|
"html"
|
||||||
"regexp"
|
"path/filepath"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"unicode/utf16"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/go-telegram-bot-api/telegram-bot-api"
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (b *Btelegram) handleUpdate(rmsg *config.Message, message, posted, edited *tgbotapi.Message) *tgbotapi.Message {
|
func (b *Btelegram) handleUpdate(rmsg *config.Message, message, posted, edited *tgbotapi.Message) *tgbotapi.Message {
|
||||||
@@ -38,22 +39,32 @@ func (b *Btelegram) handleGroups(rmsg *config.Message, message *tgbotapi.Message
|
|||||||
|
|
||||||
// handleForwarded handles forwarded messages
|
// handleForwarded handles forwarded messages
|
||||||
func (b *Btelegram) handleForwarded(rmsg *config.Message, message *tgbotapi.Message) {
|
func (b *Btelegram) handleForwarded(rmsg *config.Message, message *tgbotapi.Message) {
|
||||||
if message.ForwardFrom != nil {
|
if message.ForwardDate == 0 {
|
||||||
usernameForward := ""
|
return
|
||||||
if b.GetBool("UseFirstName") {
|
}
|
||||||
|
|
||||||
|
if message.ForwardFrom == nil {
|
||||||
|
rmsg.Text = "Forwarded from " + unknownUser + ": " + rmsg.Text
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
usernameForward := ""
|
||||||
|
if b.GetBool("UseFirstName") {
|
||||||
|
usernameForward = message.ForwardFrom.FirstName
|
||||||
|
}
|
||||||
|
|
||||||
|
if usernameForward == "" {
|
||||||
|
usernameForward = message.ForwardFrom.UserName
|
||||||
|
if usernameForward == "" {
|
||||||
usernameForward = message.ForwardFrom.FirstName
|
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
|
// handleQuoting handles quoting of previous messages
|
||||||
@@ -94,7 +105,7 @@ func (b *Btelegram) handleUsername(rmsg *config.Message, message *tgbotapi.Messa
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
// only download avatars if we have a place to upload them (configured mediaserver)
|
// 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)
|
b.handleDownloadAvatar(message.From.ID, rmsg.Channel)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -125,6 +136,11 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
|||||||
// handle groups
|
// handle groups
|
||||||
message = b.handleGroups(&rmsg, message, update)
|
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
|
// set the ID's from the channel or group message
|
||||||
rmsg.ID = strconv.Itoa(message.MessageID)
|
rmsg.ID = strconv.Itoa(message.MessageID)
|
||||||
rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10)
|
rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10)
|
||||||
@@ -144,6 +160,9 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
|||||||
// quote the previous message
|
// quote the previous message
|
||||||
b.handleQuoting(&rmsg, message)
|
b.handleQuoting(&rmsg, message)
|
||||||
|
|
||||||
|
// handle entities (adding URLs)
|
||||||
|
b.handleEntities(&rmsg, message)
|
||||||
|
|
||||||
if rmsg.Text != "" || len(rmsg.Extra) > 0 {
|
if rmsg.Text != "" || len(rmsg.Extra) > 0 {
|
||||||
rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text)
|
rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text)
|
||||||
// channels don't have (always?) user information. see #410
|
// channels don't have (always?) user information. see #410
|
||||||
@@ -162,13 +181,15 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
|||||||
// sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful.
|
// sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful.
|
||||||
// logs an error message if it fails
|
// logs an error message if it fails
|
||||||
func (b *Btelegram) handleDownloadAvatar(userid int, channel string) {
|
func (b *Btelegram) handleDownloadAvatar(userid int, channel string) {
|
||||||
rmsg := config.Message{Username: "system",
|
rmsg := config.Message{
|
||||||
Text: "avatar",
|
Username: "system",
|
||||||
Channel: channel,
|
Text: "avatar",
|
||||||
Account: b.Account,
|
Channel: channel,
|
||||||
UserID: strconv.Itoa(userid),
|
Account: b.Account,
|
||||||
Event: config.EventAvatarDownload,
|
UserID: strconv.Itoa(userid),
|
||||||
Extra: make(map[string][]interface{})}
|
Event: config.EventAvatarDownload,
|
||||||
|
Extra: make(map[string][]interface{}),
|
||||||
|
}
|
||||||
|
|
||||||
if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok {
|
if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok {
|
||||||
photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1})
|
photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1})
|
||||||
@@ -198,6 +219,46 @@ func (b *Btelegram) handleDownloadAvatar(userid int, channel string) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Btelegram) maybeConvertTgs(name *string, data *[]byte) {
|
||||||
|
var format string
|
||||||
|
switch b.GetString("MediaConvertTgs") {
|
||||||
|
case FormatWebp:
|
||||||
|
b.Log.Debugf("Tgs to WebP conversion enabled, converting %v", name)
|
||||||
|
format = FormatWebp
|
||||||
|
case FormatPng:
|
||||||
|
// The WebP to PNG converter can't handle animated webp files yet,
|
||||||
|
// and I'm not going to write a path for x/image/webp.
|
||||||
|
// The error message would be:
|
||||||
|
// conversion failed: webp: non-Alpha VP8X is not implemented
|
||||||
|
// So instead, we tell lottie to directly go to PNG.
|
||||||
|
b.Log.Debugf("Tgs to PNG conversion enabled, converting %v", name)
|
||||||
|
format = FormatPng
|
||||||
|
default:
|
||||||
|
// 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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
// handleDownloadFile handles file download
|
||||||
func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Message) error {
|
func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Message) error {
|
||||||
size := 0
|
size := 0
|
||||||
@@ -245,6 +306,18 @@ func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Messa
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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)
|
helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@@ -294,6 +367,9 @@ func (b *Btelegram) handleEdit(msg *config.Message, chatid int64) (string, error
|
|||||||
case "Markdown":
|
case "Markdown":
|
||||||
b.Log.Debug("Using mode markdown")
|
b.Log.Debug("Using mode markdown")
|
||||||
m.ParseMode = tgbotapi.ModeMarkdown
|
m.ParseMode = tgbotapi.ModeMarkdown
|
||||||
|
case MarkdownV2:
|
||||||
|
b.Log.Debug("Using mode MarkdownV2")
|
||||||
|
m.ParseMode = MarkdownV2
|
||||||
}
|
}
|
||||||
if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick {
|
if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick {
|
||||||
b.Log.Debug("Using mode HTML - nick only")
|
b.Log.Debug("Using mode HTML - nick only")
|
||||||
@@ -315,21 +391,32 @@ func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) string {
|
|||||||
Name: fi.Name,
|
Name: fi.Name,
|
||||||
Bytes: *fi.Data,
|
Bytes: *fi.Data,
|
||||||
}
|
}
|
||||||
re := regexp.MustCompile(".(jpg|png)$")
|
switch filepath.Ext(fi.Name) {
|
||||||
if re.MatchString(fi.Name) {
|
case ".jpg", ".jpe", ".png":
|
||||||
c = tgbotapi.NewPhotoUpload(chatid, file)
|
pc := tgbotapi.NewPhotoUpload(chatid, file)
|
||||||
} else {
|
pc.Caption, pc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment)
|
||||||
c = tgbotapi.NewDocumentUpload(chatid, file)
|
c = pc
|
||||||
|
case ".mp4", ".m4v":
|
||||||
|
vc := tgbotapi.NewVideoUpload(chatid, file)
|
||||||
|
vc.Caption, vc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment)
|
||||||
|
c = vc
|
||||||
|
case ".mp3", ".oga":
|
||||||
|
ac := tgbotapi.NewAudioUpload(chatid, file)
|
||||||
|
ac.Caption, ac.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment)
|
||||||
|
c = ac
|
||||||
|
case ".ogg":
|
||||||
|
voc := tgbotapi.NewVoiceUpload(chatid, file)
|
||||||
|
voc.Caption, voc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment)
|
||||||
|
c = voc
|
||||||
|
default:
|
||||||
|
dc := tgbotapi.NewDocumentUpload(chatid, file)
|
||||||
|
dc.Caption, dc.ParseMode = TGGetParseMode(b, msg.Username, fi.Comment)
|
||||||
|
c = dc
|
||||||
}
|
}
|
||||||
_, err := b.c.Send(c)
|
_, err := b.c.Send(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
b.Log.Errorf("file upload failed: %#v", err)
|
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
@@ -339,8 +426,40 @@ func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string
|
|||||||
if format == "" {
|
if format == "" {
|
||||||
format = "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})"
|
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, "{MESSAGE}", message, -1)
|
||||||
format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1)
|
format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1)
|
||||||
format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1)
|
format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1)
|
||||||
return format
|
return format
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// handleEntities handles messageEntities
|
||||||
|
func (b *Btelegram) handleEntities(rmsg *config.Message, message *tgbotapi.Message) {
|
||||||
|
if message.Entities == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// for now only do URL replacements
|
||||||
|
for _, e := range *message.Entities {
|
||||||
|
if e.Type == "text_link" {
|
||||||
|
url, err := e.ParseURL()
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Errorf("entity text_link url parse failed: %s", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
utfEncodedString := utf16.Encode([]rune(rmsg.Text))
|
||||||
|
if e.Offset+e.Length > len(utfEncodedString) {
|
||||||
|
b.Log.Errorf("entity length is too long %d > %d", e.Offset+e.Length, len(utfEncodedString))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
link := utf16.Decode(utfEncodedString[e.Offset : e.Offset+e.Length])
|
||||||
|
rmsg.Text = strings.Replace(rmsg.Text, string(link), url.String(), 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ package btelegram
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"html"
|
"html"
|
||||||
"io"
|
|
||||||
|
|
||||||
"github.com/russross/blackfriday"
|
"github.com/russross/blackfriday"
|
||||||
)
|
)
|
||||||
@@ -33,7 +32,7 @@ func (options *customHTML) Header(out *bytes.Buffer, text func() bool, level int
|
|||||||
options.Paragraph(out, text)
|
options.Paragraph(out, text)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (options *customHTML) HRule(out io.ByteWriter) {
|
func (options *customHTML) HRule(out *bytes.Buffer) {
|
||||||
out.WriteByte('\n') //nolint:errcheck
|
out.WriteByte('\n') //nolint:errcheck
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -54,16 +53,13 @@ func (options *customHTML) ListItem(out *bytes.Buffer, text []byte, flags int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func makeHTML(input string) string {
|
func makeHTML(input string) string {
|
||||||
extensions := blackfriday.NoIntraEmphasis |
|
return string(blackfriday.Markdown([]byte(input),
|
||||||
blackfriday.FencedCode |
|
&customHTML{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")},
|
||||||
blackfriday.Autolink |
|
blackfriday.EXTENSION_NO_INTRA_EMPHASIS|
|
||||||
blackfriday.SpaceHeadings |
|
blackfriday.EXTENSION_FENCED_CODE|
|
||||||
blackfriday.HeadingIDs |
|
blackfriday.EXTENSION_AUTOLINK|
|
||||||
blackfriday.BackslashLineBreak |
|
blackfriday.EXTENSION_SPACE_HEADERS|
|
||||||
blackfriday.DefinitionLists
|
blackfriday.EXTENSION_HEADER_IDS|
|
||||||
|
blackfriday.EXTENSION_BACKSLASH_LINE_BREAK|
|
||||||
renderer := &customHTML{blackfriday.NewHTMLRenderer(blackfriday.HTMLRendererParameters{
|
blackfriday.EXTENSION_DEFINITION_LISTS))
|
||||||
Flags: blackfriday.UseXHTML | blackfriday.SkipImages,
|
|
||||||
})}
|
|
||||||
return string(blackfriday.Run([]byte(input), blackfriday.WithExtensions(extensions), blackfriday.WithRenderer(renderer)))
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,19 +2,23 @@ package btelegram
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"html"
|
"html"
|
||||||
|
"log"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/go-telegram-bot-api/telegram-bot-api"
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
unknownUser = "unknown"
|
unknownUser = "unknown"
|
||||||
HTMLFormat = "HTML"
|
HTMLFormat = "HTML"
|
||||||
HTMLNick = "htmlnick"
|
HTMLNick = "htmlnick"
|
||||||
|
MarkdownV2 = "MarkdownV2"
|
||||||
|
FormatPng = "png"
|
||||||
|
FormatWebp = "webp"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Btelegram struct {
|
type Btelegram struct {
|
||||||
@@ -24,6 +28,16 @@ type Btelegram struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
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 lottie does not appear to work:\n%#v", tgsConvertFormat, err)
|
||||||
|
}
|
||||||
|
if tgsConvertFormat != FormatPng && tgsConvertFormat != FormatWebp {
|
||||||
|
log.Fatalf("Telegram bridge configured to convert .tgs files to '%s', but only '%s' and '%s' are supported.", FormatPng, FormatWebp, tgsConvertFormat)
|
||||||
|
}
|
||||||
|
}
|
||||||
return &Btelegram{Config: cfg, avatarMap: make(map[string]string)}
|
return &Btelegram{Config: cfg, avatarMap: make(map[string]string)}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -55,6 +69,28 @@ func (b *Btelegram) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
return nil
|
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) Send(msg config.Message) (string, error) {
|
func (b *Btelegram) Send(msg config.Message) (string, error) {
|
||||||
b.Log.Debugf("=> Receiving %#v", msg)
|
b.Log.Debugf("=> Receiving %#v", msg)
|
||||||
|
|
||||||
@@ -81,8 +117,8 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
// Upload a file if it exists
|
// Upload a file if it exists
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
if _, err := b.sendMessage(chatid, rmsg.Username, rmsg.Text); err != nil {
|
if _, msgErr := b.sendMessage(chatid, rmsg.Username, rmsg.Text); msgErr != nil {
|
||||||
b.Log.Errorf("sendMessage failed: %s", err)
|
b.Log.Errorf("sendMessage failed: %s", msgErr)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// check if we have files to upload (from slack, telegram or mattermost)
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
@@ -97,7 +133,14 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Post normal message
|
// 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, msg.Username, msg.Text)
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Btelegram) getFileDirectURL(id string) string {
|
func (b *Btelegram) getFileDirectURL(id string) string {
|
||||||
@@ -110,20 +153,10 @@ func (b *Btelegram) getFileDirectURL(id string) string {
|
|||||||
|
|
||||||
func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) {
|
func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) {
|
||||||
m := tgbotapi.NewMessage(chatid, "")
|
m := tgbotapi.NewMessage(chatid, "")
|
||||||
m.Text = username + text
|
m.Text, m.ParseMode = TGGetParseMode(b, username, text)
|
||||||
if b.GetString("MessageFormat") == HTMLFormat {
|
|
||||||
b.Log.Debug("Using mode HTML")
|
m.DisableWebPagePreview = b.GetBool("DisableWebPagePreview")
|
||||||
m.ParseMode = tgbotapi.ModeHTML
|
|
||||||
}
|
|
||||||
if b.GetString("MessageFormat") == "Markdown" {
|
|
||||||
b.Log.Debug("Using mode markdown")
|
|
||||||
m.ParseMode = tgbotapi.ModeMarkdown
|
|
||||||
}
|
|
||||||
if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick {
|
|
||||||
b.Log.Debug("Using mode HTML - nick only")
|
|
||||||
m.Text = username + html.EscapeString(text)
|
|
||||||
m.ParseMode = tgbotapi.ModeHTML
|
|
||||||
}
|
|
||||||
res, err := b.c.Send(m)
|
res, err := b.c.Send(m)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
|
|||||||
327
bridge/vk/vk.go
Normal file
327
bridge/vk/vk.go
Normal file
@@ -0,0 +1,327 @@
|
|||||||
|
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
|
||||||
|
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"))
|
||||||
|
lp, err := longpoll.NewLongPoll(b.c, b.GetInt("GroupID"))
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Debugf("%#v", err)
|
||||||
|
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
lp.MessageNew(func(ctx context.Context, obj events.MessageNewObject) {
|
||||||
|
b.handleMessage(obj.Message, false)
|
||||||
|
})
|
||||||
|
|
||||||
|
b.Log.Info("Connection succeeded")
|
||||||
|
|
||||||
|
go func() {
|
||||||
|
err := lp.Run()
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Fatal("Enable longpoll in group management")
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bvk) Disconnect() error {
|
||||||
|
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.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) {
|
||||||
|
p, err := b.c.UploadMessagesPhoto(peerID, 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)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
380
bridge/whatsapp/handlers.go
Normal file
380
bridge/whatsapp/handlers.go
Normal file
@@ -0,0 +1,380 @@
|
|||||||
|
package bwhatsapp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"mime"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
"github.com/Rhymen/go-whatsapp"
|
||||||
|
"github.com/jpillora/backoff"
|
||||||
|
)
|
||||||
|
|
||||||
|
/*
|
||||||
|
Implement handling messages coming from WhatsApp
|
||||||
|
Check:
|
||||||
|
- https://github.com/Rhymen/go-whatsapp#add-message-handlers
|
||||||
|
- https://github.com/Rhymen/go-whatsapp/blob/master/handler.go
|
||||||
|
- https://github.com/tulir/mautrix-whatsapp/tree/master/whatsapp-ext for more advanced command handling
|
||||||
|
*/
|
||||||
|
|
||||||
|
// HandleError received from WhatsApp
|
||||||
|
func (b *Bwhatsapp) HandleError(err error) {
|
||||||
|
// 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 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// whatsapp sends last messages to show context , cut them
|
||||||
|
if message.Info.Timestamp < b.startedAt {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
groupJID := message.Info.RemoteJid
|
||||||
|
senderJID := message.Info.SenderJid
|
||||||
|
|
||||||
|
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)
|
||||||
|
if senderName == "" {
|
||||||
|
senderName = "Someone" // don't expose telephone number
|
||||||
|
}
|
||||||
|
|
||||||
|
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)
|
||||||
|
|
||||||
|
// 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] + "@s.whatsapp.net")
|
||||||
|
if mention == "" {
|
||||||
|
mention = "someone"
|
||||||
|
}
|
||||||
|
|
||||||
|
message.Text = strings.Replace(message.Text, "@"+numberAndSuffix[0], "@"+mention, 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
rmsg := config.Message{
|
||||||
|
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 {
|
||||||
|
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(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
|
||||||
|
}
|
||||||
163
bridge/whatsapp/helpers.go
Normal file
163
bridge/whatsapp/helpers.go
Normal file
@@ -0,0 +1,163 @@
|
|||||||
|
package bwhatsapp
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/gob"
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
qrcodeTerminal "github.com/Baozisoftware/qrcode-terminal-go"
|
||||||
|
"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)
|
||||||
|
}
|
||||||
|
|
||||||
|
terminal.Get(<-qr).Print()
|
||||||
|
}()
|
||||||
|
|
||||||
|
return qr
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bwhatsapp) readSession() (whatsapp.Session, error) {
|
||||||
|
session := whatsapp.Session{}
|
||||||
|
sessionFile := b.Config.GetString(sessionFile)
|
||||||
|
|
||||||
|
if sessionFile == "" {
|
||||||
|
return session, errors.New("if you won't set SessionFile then you will need to scan QR code on every restart")
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Open(sessionFile)
|
||||||
|
if err != nil {
|
||||||
|
return session, err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
decoder := gob.NewDecoder(file)
|
||||||
|
|
||||||
|
return session, decoder.Decode(&session)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bwhatsapp) writeSession(session whatsapp.Session) error {
|
||||||
|
sessionFile := b.Config.GetString(sessionFile)
|
||||||
|
|
||||||
|
if sessionFile == "" {
|
||||||
|
// we already sent a warning while starting the bridge, so let's be quiet here
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Create(sessionFile)
|
||||||
|
if err != nil {
|
||||||
|
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 {
|
||||||
|
if sender, exists := b.users[senderJid]; exists {
|
||||||
|
if sender.Name != "" {
|
||||||
|
return sender.Name
|
||||||
|
}
|
||||||
|
// 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
|
||||||
|
if sender.Notify != "" {
|
||||||
|
return sender.Notify
|
||||||
|
}
|
||||||
|
|
||||||
|
if sender.Short != "" {
|
||||||
|
return sender.Short
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// try to reload this contact
|
||||||
|
_, err := b.conn.Contacts()
|
||||||
|
if 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 ""
|
||||||
|
}
|
||||||
|
|
||||||
|
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")
|
||||||
|
}
|
||||||
332
bridge/whatsapp/whatsapp.go
Normal file
332
bridge/whatsapp/whatsapp.go
Normal file
@@ -0,0 +1,332 @@
|
|||||||
|
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/Rhymen/go-whatsapp"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Account config parameters
|
||||||
|
cfgNumber = "Number"
|
||||||
|
qrOnWhiteTerminal = "QrOnWhiteTerminal"
|
||||||
|
sessionFile = "SessionFile"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Bwhatsapp Bridge structure keeping all the information needed for relying
|
||||||
|
type Bwhatsapp struct {
|
||||||
|
*bridge.Config
|
||||||
|
|
||||||
|
session *whatsapp.Session
|
||||||
|
conn *whatsapp.Conn
|
||||||
|
startedAt uint64
|
||||||
|
|
||||||
|
users map[string]whatsapp.Contact
|
||||||
|
userAvatars map[string]string
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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]whatsapp.Contact),
|
||||||
|
userAvatars: make(map[string]string),
|
||||||
|
}
|
||||||
|
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connect to WhatsApp. Required implementation of the Bridger interface
|
||||||
|
func (b *Bwhatsapp) Connect() error {
|
||||||
|
number := b.GetString(cfgNumber)
|
||||||
|
if number == "" {
|
||||||
|
return errors.New("whatsapp's telephone number need to be configured")
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Log.Debugln("Connecting to WhatsApp..")
|
||||||
|
conn, err := whatsapp.NewConn(20 * time.Second)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("failed to connect to WhatsApp: " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
b.conn = conn
|
||||||
|
|
||||||
|
b.conn.AddHandler(b)
|
||||||
|
b.Log.Debugln("WhatsApp connection successful")
|
||||||
|
|
||||||
|
// load existing session in order to keep it between restarts
|
||||||
|
b.session, err = b.restoreSession()
|
||||||
|
if err != nil {
|
||||||
|
b.Log.Warn(err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// login to a new session
|
||||||
|
if b.session == nil {
|
||||||
|
if err = b.Login(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
b.startedAt = uint64(time.Now().Unix())
|
||||||
|
|
||||||
|
_, err = b.conn.Contacts()
|
||||||
|
if err != nil {
|
||||||
|
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" {
|
||||||
|
// it is user
|
||||||
|
b.users[id] = contact
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// get user avatar asynchronously
|
||||||
|
go func() {
|
||||||
|
b.Log.Debug("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()
|
||||||
|
b.userAvatars[jid] = info.URL
|
||||||
|
b.Unlock()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Log.Debug("Finished getting avatars..")
|
||||||
|
}()
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Login to WhatsApp creating a new session. This will require to scan a QR code on your mobile device
|
||||||
|
func (b *Bwhatsapp) Login() error {
|
||||||
|
b.Log.Debugln("Logging in..")
|
||||||
|
|
||||||
|
invert := b.GetBool(qrOnWhiteTerminal) // false is the default
|
||||||
|
qrChan := qrFromTerminal(invert)
|
||||||
|
|
||||||
|
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)
|
||||||
|
b.Log.Infof("Connection: %#v", b.conn)
|
||||||
|
|
||||||
|
err = b.writeSession(session)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Fprintf(os.Stderr, "error saving session: %v\n", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Disconnect is called while reconnecting to the bridge
|
||||||
|
// Required implementation of the Bridger interface
|
||||||
|
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
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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
|
||||||
|
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
|
||||||
|
// Required implementation of the Bridger interface
|
||||||
|
// https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16
|
||||||
|
func (b *Bwhatsapp) Send(msg config.Message) (string, error) {
|
||||||
|
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.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)"
|
||||||
|
}
|
||||||
|
|
||||||
|
// 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
|
||||||
|
message := whatsapp.TextMessage{
|
||||||
|
Info: whatsapp.MessageInfo{
|
||||||
|
RemoteJid: msg.Channel, // which equals to group id
|
||||||
|
},
|
||||||
|
Text: msg.Username + msg.Text,
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Log.Debugf("=> Sending %#v", msg)
|
||||||
|
|
||||||
|
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
|
||||||
|
//func (b *Bwhatsapp) Command(cmd string) string {
|
||||||
|
// return ""
|
||||||
|
//}
|
||||||
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,8 +1,14 @@
|
|||||||
package bxmpp
|
package bxmpp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
@@ -14,49 +20,36 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Bxmpp struct {
|
type Bxmpp struct {
|
||||||
xc *xmpp.Client
|
|
||||||
xmppMap map[string]string
|
|
||||||
*bridge.Config
|
*bridge.Config
|
||||||
|
|
||||||
|
startTime time.Time
|
||||||
|
xc *xmpp.Client
|
||||||
|
xmppMap map[string]string
|
||||||
|
connected bool
|
||||||
|
sync.RWMutex
|
||||||
|
|
||||||
|
avatarAvailability map[string]bool
|
||||||
|
avatarMap map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
b := &Bxmpp{Config: cfg}
|
return &Bxmpp{
|
||||||
b.xmppMap = make(map[string]string)
|
Config: cfg,
|
||||||
return b
|
xmppMap: make(map[string]string),
|
||||||
|
avatarAvailability: make(map[string]bool),
|
||||||
|
avatarMap: make(map[string]string),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bxmpp) Connect() error {
|
func (b *Bxmpp) Connect() error {
|
||||||
var err error
|
|
||||||
b.Log.Infof("Connecting %s", b.GetString("Server"))
|
b.Log.Infof("Connecting %s", b.GetString("Server"))
|
||||||
b.xc, err = b.createXMPP()
|
if err := b.createXMPP(); err != nil {
|
||||||
if err != nil {
|
|
||||||
b.Log.Debugf("%#v", err)
|
b.Log.Debugf("%#v", err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
b.Log.Info("Connection succeeded")
|
b.Log.Info("Connection succeeded")
|
||||||
go func() {
|
go b.manageConnection()
|
||||||
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()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -75,58 +68,181 @@ func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bxmpp) Send(msg config.Message) (string, 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
|
// ignore delete messages
|
||||||
if msg.Event == config.EventMsgDelete {
|
if msg.Event == config.EventMsgDelete {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
b.Log.Debugf("=> Receiving %#v", msg)
|
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 {
|
if msg.Extra != nil {
|
||||||
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
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 {
|
if len(msg.Extra["file"]) > 0 {
|
||||||
return b.handleUploadFile(&msg)
|
return "", b.handleUploadFile(&msg)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
var msgreplaceid string
|
if b.GetString("WebhookURL") != "" {
|
||||||
msgid := xid.New().String()
|
b.Log.Debugf("Sending message using Webhook")
|
||||||
if msg.ID != "" {
|
err := b.postSlackCompatibleWebhook(msg)
|
||||||
msgid = msg.ID
|
if err != nil {
|
||||||
msgreplaceid = msg.ID
|
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})
|
// Post normal message.
|
||||||
if err != nil {
|
var msgReplaceID string
|
||||||
|
msgID := xid.New().String()
|
||||||
|
if msg.ID != "" {
|
||||||
|
msgID = msg.ID
|
||||||
|
msgReplaceID = msg.ID
|
||||||
|
}
|
||||||
|
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,
|
||||||
|
}); err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
return msgid, nil
|
return msgID, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bxmpp) createXMPP() (*xmpp.Client, error) {
|
func (b *Bxmpp) postSlackCompatibleWebhook(msg config.Message) error {
|
||||||
tc := new(tls.Config)
|
type XMPPWebhook struct {
|
||||||
tc.InsecureSkipVerify = b.GetBool("SkipTLSVerify")
|
Username string `json:"username"`
|
||||||
tc.ServerName = strings.Split(b.GetString("Server"), ":")[0]
|
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 {
|
||||||
|
if !strings.Contains(b.GetString("Jid"), "@") {
|
||||||
|
return fmt.Errorf("the Jid %s doesn't contain an @", b.GetString("Jid"))
|
||||||
|
}
|
||||||
|
tc := &tls.Config{
|
||||||
|
ServerName: strings.Split(b.GetString("Jid"), "@")[1],
|
||||||
|
InsecureSkipVerify: b.GetBool("SkipTLSVerify"), // nolint: gosec
|
||||||
|
}
|
||||||
|
|
||||||
|
xmpp.DebugWriter = b.Log.Writer()
|
||||||
|
|
||||||
options := xmpp.Options{
|
options := xmpp.Options{
|
||||||
Host: b.GetString("Server"),
|
Host: b.GetString("Server"),
|
||||||
User: b.GetString("Jid"),
|
User: b.GetString("Jid"),
|
||||||
Password: b.GetString("Password"),
|
Password: b.GetString("Password"),
|
||||||
NoTLS: true,
|
NoTLS: true,
|
||||||
StartTLS: true,
|
StartTLS: !b.GetBool("NoTLS"),
|
||||||
TLSConfig: tc,
|
TLSConfig: tc,
|
||||||
Debug: b.GetBool("debug"),
|
Debug: b.GetBool("debug"),
|
||||||
Logger: b.Log.Writer(),
|
|
||||||
Session: true,
|
Session: true,
|
||||||
Status: "",
|
Status: "",
|
||||||
StatusMessage: "",
|
StatusMessage: "",
|
||||||
Resource: "",
|
Resource: "",
|
||||||
InsecureAllowUnencryptedAuth: false,
|
InsecureAllowUnencryptedAuth: b.GetBool("NoTLS"),
|
||||||
}
|
}
|
||||||
var err error
|
var err error
|
||||||
b.xc, err = options.NewClient()
|
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 {
|
func (b *Bxmpp) xmppKeepAlive() chan bool {
|
||||||
@@ -138,8 +254,7 @@ func (b *Bxmpp) xmppKeepAlive() chan bool {
|
|||||||
select {
|
select {
|
||||||
case <-ticker.C:
|
case <-ticker.C:
|
||||||
b.Log.Debugf("PING")
|
b.Log.Debugf("PING")
|
||||||
err := b.xc.PingC2S("", "")
|
if err := b.xc.PingC2S("", ""); err != nil {
|
||||||
if err != nil {
|
|
||||||
b.Log.Debugf("PING failed %#v", err)
|
b.Log.Debugf("PING failed %#v", err)
|
||||||
}
|
}
|
||||||
case <-done:
|
case <-done:
|
||||||
@@ -151,40 +266,74 @@ func (b *Bxmpp) xmppKeepAlive() chan bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bxmpp) handleXMPP() error {
|
func (b *Bxmpp) handleXMPP() error {
|
||||||
var ok bool
|
b.startTime = time.Now()
|
||||||
var msgid string
|
|
||||||
done := b.xmppKeepAlive()
|
done := b.xmppKeepAlive()
|
||||||
defer close(done)
|
defer close(done)
|
||||||
|
|
||||||
for {
|
for {
|
||||||
m, err := b.xc.Recv()
|
m, err := b.xc.Recv()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
switch v := m.(type) {
|
switch v := m.(type) {
|
||||||
case xmpp.Chat:
|
case xmpp.Chat:
|
||||||
if v.Type == "groupchat" {
|
if v.Type == "groupchat" {
|
||||||
b.Log.Debugf("== Receiving %#v", v)
|
b.Log.Debugf("== Receiving %#v", v)
|
||||||
// skip invalid messages
|
|
||||||
|
// Skip invalid messages.
|
||||||
if b.skipMessage(v) {
|
if b.skipMessage(v) {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
msgid = v.ID
|
|
||||||
if v.ReplaceID != "" {
|
|
||||||
msgid = v.ReplaceID
|
|
||||||
}
|
|
||||||
rmsg := config.Message{Username: b.parseNick(v.Remote), Text: v.Text, Channel: b.parseChannel(v.Remote), Account: b.Account, UserID: v.Remote, ID: msgid}
|
|
||||||
|
|
||||||
// check if we have an action event
|
var event string
|
||||||
|
if strings.Contains(v.Text, "has set the subject to:") {
|
||||||
|
event = config.EventTopicChange
|
||||||
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
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,
|
||||||
|
Event: event,
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if we have an action event.
|
||||||
|
var ok bool
|
||||||
rmsg.Text, ok = b.replaceAction(rmsg.Text)
|
rmsg.Text, ok = b.replaceAction(rmsg.Text)
|
||||||
if ok {
|
if ok {
|
||||||
rmsg.Event = config.EventUserAction
|
rmsg.Event = config.EventUserAction
|
||||||
}
|
}
|
||||||
|
|
||||||
b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
|
b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
|
||||||
b.Log.Debugf("<= Message is %#v", rmsg)
|
b.Log.Debugf("<= Message is %#v", rmsg)
|
||||||
b.Remote <- 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:
|
case xmpp.Presence:
|
||||||
// do nothing
|
// Do nothing.
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -197,30 +346,41 @@ func (b *Bxmpp) replaceAction(text string) (string, bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// handleUploadFile handles native upload of files
|
// handleUploadFile handles native upload of files
|
||||||
func (b *Bxmpp) handleUploadFile(msg *config.Message) (string, error) {
|
func (b *Bxmpp) handleUploadFile(msg *config.Message) error {
|
||||||
var urldesc = ""
|
var urlDesc string
|
||||||
|
|
||||||
for _, f := range msg.Extra["file"] {
|
for _, file := range msg.Extra["file"] {
|
||||||
fi := f.(config.FileInfo)
|
fileInfo := file.(config.FileInfo)
|
||||||
if fi.Comment != "" {
|
if fileInfo.Comment != "" {
|
||||||
msg.Text += fi.Comment + ": "
|
msg.Text += fileInfo.Comment + ": "
|
||||||
}
|
}
|
||||||
if fi.URL != "" {
|
if fileInfo.URL != "" {
|
||||||
msg.Text = fi.URL
|
msg.Text = fileInfo.URL
|
||||||
if fi.Comment != "" {
|
if fileInfo.Comment != "" {
|
||||||
msg.Text = fi.Comment + ": " + fi.URL
|
msg.Text = fileInfo.Comment + ": " + fileInfo.URL
|
||||||
urldesc = fi.Comment
|
urlDesc = fileInfo.Comment
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, err := b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.GetString("Muc"), Text: msg.Username + msg.Text})
|
if _, err := b.xc.Send(xmpp.Chat{
|
||||||
if err != nil {
|
Type: "groupchat",
|
||||||
return "", err
|
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 {
|
func (b *Bxmpp) parseNick(remote string) string {
|
||||||
@@ -259,7 +419,28 @@ func (b *Bxmpp) skipMessage(message xmpp.Chat) bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// do not show subjects on connect #732
|
||||||
|
if strings.Contains(message.Text, "has set the subject to:") && time.Since(b.startTime) < time.Second*5 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ignore messages posted by our webhook
|
||||||
|
if b.GetString("WebhookURL") != "" && strings.Contains(message.ID, "webhookbot") {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
// skip delayed messages
|
// skip delayed messages
|
||||||
t := time.Time{}
|
return !message.Stamp.IsZero() && time.Since(message.Stamp).Minutes() > 5
|
||||||
return message.Stamp != t
|
}
|
||||||
|
|
||||||
|
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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -4,6 +4,8 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
@@ -17,6 +19,7 @@ type Bzulip struct {
|
|||||||
bot *gzb.Bot
|
bot *gzb.Bot
|
||||||
streams map[int]string
|
streams map[int]string
|
||||||
*bridge.Config
|
*bridge.Config
|
||||||
|
sync.RWMutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg *bridge.Config) bridge.Bridger {
|
func New(cfg *bridge.Config) bridge.Bridger {
|
||||||
@@ -100,27 +103,67 @@ func (b *Bzulip) getChannel(id int) string {
|
|||||||
|
|
||||||
func (b *Bzulip) handleQueue() error {
|
func (b *Bzulip) handleQueue() error {
|
||||||
for {
|
for {
|
||||||
messages, _ := b.q.GetEvents()
|
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 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
for _, m := range messages {
|
for _, m := range messages {
|
||||||
b.Log.Debugf("== Receiving %#v", m)
|
b.Log.Debugf("== Receiving %#v", m)
|
||||||
// ignore our own messages
|
// ignore our own messages
|
||||||
if m.SenderEmail == b.GetString("login") {
|
if m.SenderEmail == b.GetString("login") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
rmsg := config.Message{Username: m.SenderFullName, Text: m.Content, Channel: b.getChannel(m.StreamID), Account: b.Account, UserID: strconv.Itoa(m.SenderID), Avatar: m.AvatarURL}
|
|
||||||
|
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: avatarURL,
|
||||||
|
}
|
||||||
b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
|
b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account)
|
||||||
b.Log.Debugf("<= Message is %#v", rmsg)
|
b.Log.Debugf("<= Message is %#v", rmsg)
|
||||||
b.Remote <- rmsg
|
b.Remote <- rmsg
|
||||||
b.q.LastEventID = m.ID
|
|
||||||
}
|
}
|
||||||
|
|
||||||
time.Sleep(time.Second * 3)
|
time.Sleep(time.Second * 3)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bzulip) sendMessage(msg config.Message) (string, error) {
|
func (b *Bzulip) sendMessage(msg config.Message) (string, error) {
|
||||||
topic := "matterbridge"
|
topic := ""
|
||||||
if b.GetString("topic") != "" {
|
if strings.Contains(msg.Channel, "/topic:") {
|
||||||
topic = b.GetString("topic")
|
res := strings.Split(msg.Channel, "/topic:")
|
||||||
|
topic = res[1]
|
||||||
|
msg.Channel = res[0]
|
||||||
}
|
}
|
||||||
m := gzb.Message{
|
m := gzb.Message{
|
||||||
Stream: msg.Channel,
|
Stream: msg.Channel,
|
||||||
|
|||||||
1628
changelog.md
1628
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
|
|
||||||
|
|
||||||
2
contrib/example.tengo
Normal file
2
contrib/example.tengo
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
text := import("text")
|
||||||
|
msgText=text.re_replace("matterbridge",msgText,"matterbridge (https://github.com/42wim/matterbridge)")
|
||||||
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)
|
||||||
|
}
|
||||||
5
gateway/bench.tengo
Normal file
5
gateway/bench.tengo
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
text := import("text")
|
||||||
|
if text.re_match("blah",msgText) {
|
||||||
|
msgText="replaced by this"
|
||||||
|
msgUsername="fakeuser"
|
||||||
|
}
|
||||||
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{}{}
|
||||||
|
}
|
||||||
11
gateway/bridgemap/bgitter.go
Normal file
11
gateway/bridgemap/bgitter.go
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
// +build !nogitter
|
||||||
|
|
||||||
|
package bridgemap
|
||||||
|
|
||||||
|
import (
|
||||||
|
bgitter "github.com/42wim/matterbridge/bridge/gitter"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
FullMap["gitter"] = bgitter.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,34 +2,9 @@ package bridgemap
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"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/xmpp"
|
|
||||||
"github.com/42wim/matterbridge/bridge/zulip"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
var FullMap = map[string]bridge.Factory{
|
var (
|
||||||
"api": api.New,
|
FullMap = map[string]bridge.Factory{}
|
||||||
"discord": bdiscord.New,
|
UserTypingSupport = map[string]struct{}{}
|
||||||
"gitter": bgitter.New,
|
)
|
||||||
"irc": birc.New,
|
|
||||||
"mattermost": bmattermost.New,
|
|
||||||
"matrix": bmatrix.New,
|
|
||||||
"rocketchat": brocketchat.New,
|
|
||||||
"slack-legacy": bslack.NewLegacy,
|
|
||||||
"slack": bslack.New,
|
|
||||||
"sshchat": bsshchat.New,
|
|
||||||
"steam": bsteam.New,
|
|
||||||
"telegram": btelegram.New,
|
|
||||||
"xmpp": bxmpp.New,
|
|
||||||
"zulip": bzulip.New,
|
|
||||||
}
|
|
||||||
|
|||||||
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
|
||||||
|
}
|
||||||
11
gateway/bridgemap/bwhatsapp.go
Normal file
11
gateway/bridgemap/bwhatsapp.go
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
// +build !nowhatsapp
|
||||||
|
|
||||||
|
package bridgemap
|
||||||
|
|
||||||
|
import (
|
||||||
|
bwhatsapp "github.com/42wim/matterbridge/bridge/whatsapp"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
FullMap["whatsapp"] = bwhatsapp.New
|
||||||
|
}
|
||||||
11
gateway/bridgemap/bxmpp.go
Normal file
11
gateway/bridgemap/bxmpp.go
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
// +build !noxmpp
|
||||||
|
|
||||||
|
package bridgemap
|
||||||
|
|
||||||
|
import (
|
||||||
|
bxmpp "github.com/42wim/matterbridge/bridge/xmpp"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
FullMap["xmpp"] = bxmpp.New
|
||||||
|
}
|
||||||
11
gateway/bridgemap/bzulip.go
Normal file
11
gateway/bridgemap/bzulip.go
Normal file
@@ -0,0 +1,11 @@
|
|||||||
|
// +build !nozulip
|
||||||
|
|
||||||
|
package bridgemap
|
||||||
|
|
||||||
|
import (
|
||||||
|
bzulip "github.com/42wim/matterbridge/bridge/zulip"
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
FullMap["zulip"] = bzulip.New
|
||||||
|
}
|
||||||
@@ -1,6 +1,8 @@
|
|||||||
package gateway
|
package gateway
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
@@ -8,8 +10,11 @@ import (
|
|||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/hashicorp/golang-lru"
|
"github.com/42wim/matterbridge/internal"
|
||||||
"github.com/peterhellberg/emojilib"
|
"github.com/d5/tengo/v2"
|
||||||
|
"github.com/d5/tengo/v2/stdlib"
|
||||||
|
lru "github.com/hashicorp/golang-lru"
|
||||||
|
"github.com/kyokomi/emoji/v2"
|
||||||
"github.com/sirupsen/logrus"
|
"github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -24,6 +29,8 @@ type Gateway struct {
|
|||||||
Message chan config.Message
|
Message chan config.Message
|
||||||
Name string
|
Name string
|
||||||
Messages *lru.Cache
|
Messages *lru.Cache
|
||||||
|
|
||||||
|
logger *logrus.Entry
|
||||||
}
|
}
|
||||||
|
|
||||||
type BrMsgID struct {
|
type BrMsgID struct {
|
||||||
@@ -32,25 +39,30 @@ type BrMsgID struct {
|
|||||||
ChannelID string
|
ChannelID string
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *logrus.Entry
|
const apiProtocol = "api"
|
||||||
|
|
||||||
const (
|
// New creates a new Gateway object associated with the specified router and
|
||||||
apiProtocol = "api"
|
// following the given configuration.
|
||||||
)
|
func New(rootLogger *logrus.Logger, cfg *config.Gateway, r *Router) *Gateway {
|
||||||
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "gateway"})
|
||||||
|
|
||||||
func New(cfg config.Gateway, r *Router) *Gateway {
|
|
||||||
flog = logrus.WithFields(logrus.Fields{"prefix": "gateway"})
|
|
||||||
gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message,
|
|
||||||
Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config}
|
|
||||||
cache, _ := lru.New(5000)
|
cache, _ := lru.New(5000)
|
||||||
gw.Messages = cache
|
gw := &Gateway{
|
||||||
if err := gw.AddConfig(&cfg); err != nil {
|
Channels: make(map[string]*config.ChannelInfo),
|
||||||
flog.Errorf("AddConfig failed: %s", err)
|
Message: r.Message,
|
||||||
|
Router: r,
|
||||||
|
Bridges: make(map[string]*bridge.Bridge),
|
||||||
|
Config: r.Config,
|
||||||
|
Messages: cache,
|
||||||
|
logger: logger,
|
||||||
|
}
|
||||||
|
if err := gw.AddConfig(cfg); err != nil {
|
||||||
|
logger.Errorf("Failed to add configuration to gateway: %#v", err)
|
||||||
}
|
}
|
||||||
return gw
|
return gw
|
||||||
}
|
}
|
||||||
|
|
||||||
// Find the canonical ID that the message is keyed under in cache
|
// FindCanonicalMsgID returns the ID under which a message was stored in the cache.
|
||||||
func (gw *Gateway) FindCanonicalMsgID(protocol string, mID string) string {
|
func (gw *Gateway) FindCanonicalMsgID(protocol string, mID string) string {
|
||||||
ID := protocol + " " + mID
|
ID := protocol + " " + mID
|
||||||
if gw.Messages.Contains(ID) {
|
if gw.Messages.Contains(ID) {
|
||||||
@@ -70,16 +82,23 @@ func (gw *Gateway) FindCanonicalMsgID(protocol string, mID string) string {
|
|||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// AddBridge sets up a new bridge in the gateway object with the specified configuration.
|
||||||
func (gw *Gateway) AddBridge(cfg *config.Bridge) error {
|
func (gw *Gateway) AddBridge(cfg *config.Bridge) error {
|
||||||
br := gw.Router.getBridge(cfg.Account)
|
br := gw.Router.getBridge(cfg.Account)
|
||||||
if br == nil {
|
if br == nil {
|
||||||
|
gw.checkConfig(cfg)
|
||||||
br = bridge.New(cfg)
|
br = bridge.New(cfg)
|
||||||
br.Config = gw.Router.Config
|
br.Config = gw.Router.Config
|
||||||
br.General = &gw.BridgeValues().General
|
br.General = &gw.BridgeValues().General
|
||||||
// set logging
|
br.Log = gw.logger.WithFields(logrus.Fields{"prefix": br.Protocol})
|
||||||
br.Log = logrus.WithFields(logrus.Fields{"prefix": "bridge"})
|
brconfig := &bridge.Config{
|
||||||
brconfig := &bridge.Config{Remote: gw.Message, Log: logrus.WithFields(logrus.Fields{"prefix": br.Protocol}), Bridge: br}
|
Remote: gw.Message,
|
||||||
|
Bridge: br,
|
||||||
|
}
|
||||||
// add the actual bridger for this protocol to this bridge using the bridgeMap
|
// add the actual bridger for this protocol to this bridge using the bridgeMap
|
||||||
|
if _, ok := gw.Router.BridgeMap[br.Protocol]; !ok {
|
||||||
|
gw.logger.Fatalf("Incorrect protocol %s specified in gateway configuration %s, exiting.", br.Protocol, cfg.Account)
|
||||||
|
}
|
||||||
br.Bridger = gw.Router.BridgeMap[br.Protocol](brconfig)
|
br.Bridger = gw.Router.BridgeMap[br.Protocol](brconfig)
|
||||||
}
|
}
|
||||||
gw.mapChannelsToBridge(br)
|
gw.mapChannelsToBridge(br)
|
||||||
@@ -87,14 +106,28 @@ func (gw *Gateway) AddBridge(cfg *config.Bridge) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (gw *Gateway) checkConfig(cfg *config.Bridge) {
|
||||||
|
match := false
|
||||||
|
for _, key := range gw.Router.Config.Viper().AllKeys() {
|
||||||
|
if strings.HasPrefix(key, strings.ToLower(cfg.Account)) {
|
||||||
|
match = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !match {
|
||||||
|
gw.logger.Fatalf("Account %s defined in gateway %s but no configuration found, exiting.", cfg.Account, gw.Name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddConfig associates a new configuration with the gateway object.
|
||||||
func (gw *Gateway) AddConfig(cfg *config.Gateway) error {
|
func (gw *Gateway) AddConfig(cfg *config.Gateway) error {
|
||||||
gw.Name = cfg.Name
|
gw.Name = cfg.Name
|
||||||
gw.MyConfig = cfg
|
gw.MyConfig = cfg
|
||||||
if err := gw.mapChannels(); err != nil {
|
if err := gw.mapChannels(); err != nil {
|
||||||
flog.Errorf("mapChannels() failed: %s", err)
|
gw.logger.Errorf("mapChannels() failed: %s", err)
|
||||||
}
|
}
|
||||||
for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) {
|
for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) {
|
||||||
br := br //scopelint
|
br := br // scopelint
|
||||||
err := gw.AddBridge(&br)
|
err := gw.AddBridge(&br)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@@ -113,20 +146,20 @@ func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) {
|
|||||||
|
|
||||||
func (gw *Gateway) reconnectBridge(br *bridge.Bridge) {
|
func (gw *Gateway) reconnectBridge(br *bridge.Bridge) {
|
||||||
if err := br.Disconnect(); err != nil {
|
if err := br.Disconnect(); err != nil {
|
||||||
flog.Errorf("Disconnect() %s failed: %s", br.Account, err)
|
gw.logger.Errorf("Disconnect() %s failed: %s", br.Account, err)
|
||||||
}
|
}
|
||||||
time.Sleep(time.Second * 5)
|
time.Sleep(time.Second * 5)
|
||||||
RECONNECT:
|
RECONNECT:
|
||||||
flog.Infof("Reconnecting %s", br.Account)
|
gw.logger.Infof("Reconnecting %s", br.Account)
|
||||||
err := br.Connect()
|
err := br.Connect()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err)
|
gw.logger.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err)
|
||||||
time.Sleep(time.Second * 60)
|
time.Sleep(time.Second * 60)
|
||||||
goto RECONNECT
|
goto RECONNECT
|
||||||
}
|
}
|
||||||
br.Joined = make(map[string]bool)
|
br.Joined = make(map[string]bool)
|
||||||
if err := br.JoinChannels(); err != nil {
|
if err := br.JoinChannels(); err != nil {
|
||||||
flog.Errorf("JoinChannels() %s failed: %s", br.Account, err)
|
gw.logger.Errorf("JoinChannels() %s failed: %s", br.Account, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -140,13 +173,23 @@ func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) {
|
|||||||
br.Channel = strings.ToLower(br.Channel)
|
br.Channel = strings.ToLower(br.Channel)
|
||||||
}
|
}
|
||||||
if strings.HasPrefix(br.Account, "mattermost.") && strings.HasPrefix(br.Channel, "#") {
|
if strings.HasPrefix(br.Account, "mattermost.") && strings.HasPrefix(br.Channel, "#") {
|
||||||
flog.Errorf("Mattermost channels do not start with a #: remove the # in %s", br.Channel)
|
gw.logger.Errorf("Mattermost channels do not start with a #: remove the # in %s", br.Channel)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(br.Account, "zulip.") && !strings.Contains(br.Channel, "/topic:") {
|
||||||
|
gw.logger.Errorf("Breaking change, since matterbridge 1.14.0 zulip channels need to specify the topic with channel/topic:mytopic in %s of %s", br.Channel, br.Account)
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
ID := br.Channel + br.Account
|
ID := br.Channel + br.Account
|
||||||
if _, ok := gw.Channels[ID]; !ok {
|
if _, ok := gw.Channels[ID]; !ok {
|
||||||
channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account,
|
channel := &config.ChannelInfo{
|
||||||
SameChannel: make(map[string]bool)}
|
Name: br.Channel,
|
||||||
|
Direction: direction,
|
||||||
|
ID: ID,
|
||||||
|
Options: br.Options,
|
||||||
|
Account: br.Account,
|
||||||
|
SameChannel: make(map[string]bool),
|
||||||
|
}
|
||||||
channel.SameChannel[gw.Name] = br.SameChannel
|
channel.SameChannel[gw.Name] = br.SameChannel
|
||||||
gw.Channels[channel.ID] = channel
|
gw.Channels[channel.ID] = channel
|
||||||
} else {
|
} else {
|
||||||
@@ -174,10 +217,21 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con
|
|||||||
return channels
|
return channels
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// discord join/leave is for the whole bridge, isn't a per channel join/leave
|
||||||
|
if msg.Event == config.EventJoinLeave && getProtocol(msg) == "discord" && msg.Channel == "" {
|
||||||
|
for _, channel := range gw.Channels {
|
||||||
|
if channel.Account == dest.Account && strings.Contains(channel.Direction, "out") &&
|
||||||
|
gw.validGatewayDest(msg) {
|
||||||
|
channels = append(channels, *channel)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return channels
|
||||||
|
}
|
||||||
|
|
||||||
// if source channel is in only, do nothing
|
// if source channel is in only, do nothing
|
||||||
for _, channel := range gw.Channels {
|
for _, channel := range gw.Channels {
|
||||||
// lookup the channel from the message
|
// lookup the channel from the message
|
||||||
if channel.ID == getChannelID(*msg) {
|
if channel.ID == getChannelID(msg) {
|
||||||
// we only have destinations if the original message is from an "in" (sending) channel
|
// we only have destinations if the original message is from an "in" (sending) channel
|
||||||
if !strings.Contains(channel.Direction, "in") {
|
if !strings.Contains(channel.Direction, "in") {
|
||||||
return channels
|
return channels
|
||||||
@@ -186,11 +240,11 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
for _, channel := range gw.Channels {
|
for _, channel := range gw.Channels {
|
||||||
if _, ok := gw.Channels[getChannelID(*msg)]; !ok {
|
if _, ok := gw.Channels[getChannelID(msg)]; !ok {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// do samechannelgateway flogic
|
// do samechannelgateway logic
|
||||||
if channel.SameChannel[msg.Gateway] {
|
if channel.SameChannel[msg.Gateway] {
|
||||||
if msg.Channel == channel.Name && msg.Account != dest.Account {
|
if msg.Channel == channel.Name && msg.Account != dest.Account {
|
||||||
channels = append(channels, *channel)
|
channels = append(channels, *channel)
|
||||||
@@ -204,7 +258,7 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con
|
|||||||
return channels
|
return channels
|
||||||
}
|
}
|
||||||
|
|
||||||
func (gw *Gateway) getDestMsgID(msgID string, dest *bridge.Bridge, channel config.ChannelInfo) string {
|
func (gw *Gateway) getDestMsgID(msgID string, dest *bridge.Bridge, channel *config.ChannelInfo) string {
|
||||||
if res, ok := gw.Messages.Get(msgID); ok {
|
if res, ok := gw.Messages.Get(msgID); ok {
|
||||||
IDs := res.([]*BrMsgID)
|
IDs := res.([]*BrMsgID)
|
||||||
for _, id := range IDs {
|
for _, id := range IDs {
|
||||||
@@ -233,42 +287,10 @@ func (gw *Gateway) ignoreTextEmpty(msg *config.Message) bool {
|
|||||||
len(msg.Extra[config.EventFileFailureSize]) > 0) {
|
len(msg.Extra[config.EventFileFailureSize]) > 0) {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account)
|
gw.logger.Debugf("ignoring empty message %#v from %s", msg, msg.Account)
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
// ignoreTexts returns true if msg.Text matches any of the input regexes.
|
|
||||||
func (gw *Gateway) ignoreTexts(msg *config.Message, input []string) bool {
|
|
||||||
for _, entry := range input {
|
|
||||||
if entry == "" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
// TODO do not compile regexps everytime
|
|
||||||
re, err := regexp.Compile(entry)
|
|
||||||
if err != nil {
|
|
||||||
flog.Errorf("incorrect regexp %s for %s", entry, msg.Account)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if re.MatchString(msg.Text) {
|
|
||||||
flog.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account)
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// ignoreNicks returns true if msg.Username matches any of the input regexes.
|
|
||||||
func (gw *Gateway) ignoreNicks(msg *config.Message, input []string) bool {
|
|
||||||
// is the username in IgnoreNicks field
|
|
||||||
for _, entry := range input {
|
|
||||||
if msg.Username == entry {
|
|
||||||
flog.Debugf("ignoring %s from %s", msg.Username, msg.Account)
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
|
func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
|
||||||
// if we don't have the bridge, ignore it
|
// if we don't have the bridge, ignore it
|
||||||
if _, ok := gw.Bridges[msg.Account]; !ok {
|
if _, ok := gw.Bridges[msg.Account]; !ok {
|
||||||
@@ -277,16 +299,14 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
|
|||||||
|
|
||||||
igNicks := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks"))
|
igNicks := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks"))
|
||||||
igMessages := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages"))
|
igMessages := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages"))
|
||||||
if gw.ignoreTextEmpty(msg) || gw.ignoreNicks(msg, igNicks) || gw.ignoreTexts(msg, igMessages) {
|
if gw.ignoreTextEmpty(msg) || gw.ignoreText(msg.Username, igNicks) || gw.ignoreText(msg.Text, igMessages) {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) string {
|
func (gw *Gateway) modifyUsername(msg *config.Message, dest *bridge.Bridge) string {
|
||||||
br := gw.Bridges[msg.Account]
|
|
||||||
msg.Protocol = br.Protocol
|
|
||||||
if dest.GetBool("StripNick") {
|
if dest.GetBool("StripNick") {
|
||||||
re := regexp.MustCompile("[^a-zA-Z0-9]+")
|
re := regexp.MustCompile("[^a-zA-Z0-9]+")
|
||||||
msg.Username = re.ReplaceAllString(msg.Username, "")
|
msg.Username = re.ReplaceAllString(msg.Username, "")
|
||||||
@@ -294,13 +314,14 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin
|
|||||||
nick := dest.GetString("RemoteNickFormat")
|
nick := dest.GetString("RemoteNickFormat")
|
||||||
|
|
||||||
// loop to replace nicks
|
// loop to replace nicks
|
||||||
|
br := gw.Bridges[msg.Account]
|
||||||
for _, outer := range br.GetStringSlice2D("ReplaceNicks") {
|
for _, outer := range br.GetStringSlice2D("ReplaceNicks") {
|
||||||
search := outer[0]
|
search := outer[0]
|
||||||
replace := outer[1]
|
replace := outer[1]
|
||||||
// TODO move compile to bridge init somewhere
|
// TODO move compile to bridge init somewhere
|
||||||
re, err := regexp.Compile(search)
|
re, err := regexp.Compile(search)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("regexp in %s failed: %s", msg.Account, err)
|
gw.logger.Errorf("regexp in %s failed: %s", msg.Account, err)
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
msg.Username = re.ReplaceAllString(msg.Username, replace)
|
msg.Username = re.ReplaceAllString(msg.Username, replace)
|
||||||
@@ -316,19 +337,25 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin
|
|||||||
}
|
}
|
||||||
i++
|
i++
|
||||||
}
|
}
|
||||||
nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1)
|
nick = strings.ReplaceAll(nick, "{NOPINGNICK}", msg.Username[:i]+"\u200b"+msg.Username[i:])
|
||||||
}
|
}
|
||||||
|
|
||||||
nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
|
nick = strings.ReplaceAll(nick, "{BRIDGE}", br.Name)
|
||||||
nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
|
nick = strings.ReplaceAll(nick, "{PROTOCOL}", br.Protocol)
|
||||||
nick = strings.Replace(nick, "{GATEWAY}", gw.Name, -1)
|
nick = strings.ReplaceAll(nick, "{GATEWAY}", gw.Name)
|
||||||
nick = strings.Replace(nick, "{LABEL}", br.GetString("Label"), -1)
|
nick = strings.ReplaceAll(nick, "{LABEL}", br.GetString("Label"))
|
||||||
nick = strings.Replace(nick, "{NICK}", msg.Username, -1)
|
nick = strings.ReplaceAll(nick, "{NICK}", msg.Username)
|
||||||
nick = strings.Replace(nick, "{CHANNEL}", msg.Channel, -1)
|
nick = strings.ReplaceAll(nick, "{USERID}", msg.UserID)
|
||||||
|
nick = strings.ReplaceAll(nick, "{CHANNEL}", msg.Channel)
|
||||||
|
tengoNick, err := gw.modifyUsernameTengo(msg, br)
|
||||||
|
if err != nil {
|
||||||
|
gw.logger.Errorf("modifyUsernameTengo error: %s", err)
|
||||||
|
}
|
||||||
|
nick = strings.ReplaceAll(nick, "{TENGO}", tengoNick)
|
||||||
return nick
|
return nick
|
||||||
}
|
}
|
||||||
|
|
||||||
func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string {
|
func (gw *Gateway) modifyAvatar(msg *config.Message, dest *bridge.Bridge) string {
|
||||||
iconurl := dest.GetString("IconURL")
|
iconurl := dest.GetString("IconURL")
|
||||||
iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1)
|
iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1)
|
||||||
if msg.Avatar == "" {
|
if msg.Avatar == "" {
|
||||||
@@ -338,8 +365,29 @@ func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (gw *Gateway) modifyMessage(msg *config.Message) {
|
func (gw *Gateway) modifyMessage(msg *config.Message) {
|
||||||
|
if gw.BridgeValues().General.TengoModifyMessage != "" {
|
||||||
|
gw.logger.Warnf("General TengoModifyMessage=%s is deprecated and will be removed in v1.20.0, please move to Tengo InMessage=%s", gw.BridgeValues().General.TengoModifyMessage, gw.BridgeValues().General.TengoModifyMessage)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := modifyInMessageTengo(gw.BridgeValues().General.TengoModifyMessage, msg); err != nil {
|
||||||
|
gw.logger.Errorf("TengoModifyMessage failed: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
inMessage := gw.BridgeValues().Tengo.InMessage
|
||||||
|
if inMessage == "" {
|
||||||
|
inMessage = gw.BridgeValues().Tengo.Message
|
||||||
|
if inMessage != "" {
|
||||||
|
gw.logger.Warnf("Tengo Message=%s is deprecated and will be removed in v1.20.0, please move to Tengo InMessage=%s", inMessage, inMessage)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := modifyInMessageTengo(inMessage, msg); err != nil {
|
||||||
|
gw.logger.Errorf("Tengo.Message failed: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
// replace :emoji: to unicode
|
// replace :emoji: to unicode
|
||||||
msg.Text = emojilib.Replace(msg.Text)
|
emoji.ReplacePadding = ""
|
||||||
|
msg.Text = emoji.Sprint(msg.Text)
|
||||||
|
|
||||||
br := gw.Bridges[msg.Account]
|
br := gw.Bridges[msg.Account]
|
||||||
// loop to replace messages
|
// loop to replace messages
|
||||||
@@ -349,61 +397,97 @@ func (gw *Gateway) modifyMessage(msg *config.Message) {
|
|||||||
// TODO move compile to bridge init somewhere
|
// TODO move compile to bridge init somewhere
|
||||||
re, err := regexp.Compile(search)
|
re, err := regexp.Compile(search)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("regexp in %s failed: %s", msg.Account, err)
|
gw.logger.Errorf("regexp in %s failed: %s", msg.Account, err)
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
msg.Text = re.ReplaceAllString(msg.Text, replace)
|
msg.Text = re.ReplaceAllString(msg.Text, replace)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
gw.handleExtractNicks(msg)
|
||||||
|
|
||||||
// messages from api have Gateway specified, don't overwrite
|
// messages from api have Gateway specified, don't overwrite
|
||||||
if msg.Protocol != apiProtocol {
|
if msg.Protocol != apiProtocol {
|
||||||
msg.Gateway = gw.Name
|
msg.Gateway = gw.Name
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendMessage sends a message (with specified parentID) to the channel on the selected destination bridge.
|
// SendMessage sends a message (with specified parentID) to the channel on the selected
|
||||||
// returns a message id and error.
|
// destination bridge and returns a message ID or an error.
|
||||||
func (gw *Gateway) SendMessage(origmsg config.Message, dest *bridge.Bridge, channel config.ChannelInfo, canonicalParentMsgID string) (string, error) {
|
func (gw *Gateway) SendMessage(
|
||||||
msg := origmsg
|
rmsg *config.Message,
|
||||||
|
dest *bridge.Bridge,
|
||||||
|
channel *config.ChannelInfo,
|
||||||
|
canonicalParentMsgID string,
|
||||||
|
) (string, error) {
|
||||||
|
msg := *rmsg
|
||||||
// Only send the avatar download event to ourselves.
|
// Only send the avatar download event to ourselves.
|
||||||
if msg.Event == config.EventAvatarDownload {
|
if msg.Event == config.EventAvatarDownload {
|
||||||
if channel.ID != getChannelID(origmsg) {
|
if channel.ID != getChannelID(rmsg) {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// do not send to ourself for any other event
|
// do not send to ourself for any other event
|
||||||
if channel.ID == getChannelID(origmsg) {
|
if channel.ID == getChannelID(rmsg) {
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Only send irc notices to irc
|
||||||
|
if msg.Event == config.EventNoticeIRC && dest.Protocol != "irc" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
// Too noisy to log like other events
|
// Too noisy to log like other events
|
||||||
|
debugSendMessage := ""
|
||||||
if msg.Event != config.EventUserTyping {
|
if msg.Event != config.EventUserTyping {
|
||||||
flog.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, origmsg.Channel, dest.Account, channel.Name)
|
debugSendMessage = fmt.Sprintf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, rmsg.Channel, dest.Account, channel.Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
msg.Channel = channel.Name
|
msg.Channel = channel.Name
|
||||||
msg.Avatar = gw.modifyAvatar(origmsg, dest)
|
msg.Avatar = gw.modifyAvatar(rmsg, dest)
|
||||||
msg.Username = gw.modifyUsername(origmsg, dest)
|
msg.Username = gw.modifyUsername(rmsg, dest)
|
||||||
|
|
||||||
msg.ID = gw.getDestMsgID(origmsg.Protocol+" "+origmsg.ID, dest, channel)
|
msg.ID = gw.getDestMsgID(rmsg.Protocol+" "+rmsg.ID, dest, channel)
|
||||||
|
|
||||||
// for api we need originchannel as channel
|
// for api we need originchannel as channel
|
||||||
if dest.Protocol == apiProtocol {
|
if dest.Protocol == apiProtocol {
|
||||||
msg.Channel = origmsg.Channel
|
msg.Channel = rmsg.Channel
|
||||||
}
|
}
|
||||||
|
|
||||||
msg.ParentID = gw.getDestMsgID(origmsg.Protocol+" "+canonicalParentMsgID, dest, channel)
|
msg.ParentID = gw.getDestMsgID(rmsg.Protocol+" "+canonicalParentMsgID, dest, channel)
|
||||||
if msg.ParentID == "" {
|
if msg.ParentID == "" {
|
||||||
msg.ParentID = canonicalParentMsgID
|
msg.ParentID = canonicalParentMsgID
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// if the parentID is still empty and we have a parentID set in the original message
|
||||||
|
// this means that we didn't find it in the cache so set it to a "msg-parent-not-found" constant
|
||||||
|
if msg.ParentID == "" && rmsg.ParentID != "" {
|
||||||
|
msg.ParentID = config.ParentIDNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
drop, err := gw.modifyOutMessageTengo(rmsg, &msg, dest)
|
||||||
|
if err != nil {
|
||||||
|
gw.logger.Errorf("modifySendMessageTengo: %s", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if drop {
|
||||||
|
gw.logger.Debugf("=> Tengo dropping %#v from %s (%s) to %s (%s)", msg, msg.Account, rmsg.Channel, dest.Account, channel.Name)
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if debugSendMessage != "" {
|
||||||
|
gw.logger.Debug(debugSendMessage)
|
||||||
|
}
|
||||||
// if we are using mattermost plugin account, send messages to MattermostPlugin channel
|
// if we are using mattermost plugin account, send messages to MattermostPlugin channel
|
||||||
// that can be picked up by the mattermost matterbridge plugin
|
// that can be picked up by the mattermost matterbridge plugin
|
||||||
if dest.Account == "mattermost.plugin" {
|
if dest.Account == "mattermost.plugin" {
|
||||||
gw.Router.MattermostPlugin <- msg
|
gw.Router.MattermostPlugin <- msg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
defer func(t time.Time) {
|
||||||
|
gw.logger.Debugf("=> Send from %s (%s) to %s (%s) took %s", msg.Account, rmsg.Channel, dest.Account, channel.Name, time.Since(t))
|
||||||
|
}(time.Now())
|
||||||
|
|
||||||
mID, err := dest.Send(msg)
|
mID, err := dest.Send(msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return mID, err
|
return mID, err
|
||||||
@@ -411,9 +495,9 @@ func (gw *Gateway) SendMessage(origmsg config.Message, dest *bridge.Bridge, chan
|
|||||||
|
|
||||||
// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice
|
// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice
|
||||||
if mID != "" {
|
if mID != "" {
|
||||||
flog.Debugf("mID %s: %s", dest.Account, mID)
|
gw.logger.Debugf("mID %s: %s", dest.Account, mID)
|
||||||
return mID, nil
|
return mID, nil
|
||||||
//brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + mID, channel.ID})
|
// brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + mID, channel.ID})
|
||||||
}
|
}
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
@@ -422,10 +506,149 @@ func (gw *Gateway) validGatewayDest(msg *config.Message) bool {
|
|||||||
return msg.Gateway == gw.Name
|
return msg.Gateway == gw.Name
|
||||||
}
|
}
|
||||||
|
|
||||||
func getChannelID(msg config.Message) string {
|
func getChannelID(msg *config.Message) string {
|
||||||
return msg.Channel + msg.Account
|
return msg.Channel + msg.Account
|
||||||
}
|
}
|
||||||
|
|
||||||
func isAPI(account string) bool {
|
func isAPI(account string) bool {
|
||||||
return strings.HasPrefix(account, "api.")
|
return strings.HasPrefix(account, "api.")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ignoreText returns true if text matches any of the input regexes.
|
||||||
|
func (gw *Gateway) ignoreText(text string, input []string) bool {
|
||||||
|
for _, entry := range input {
|
||||||
|
if entry == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
// TODO do not compile regexps everytime
|
||||||
|
re, err := regexp.Compile(entry)
|
||||||
|
if err != nil {
|
||||||
|
gw.logger.Errorf("incorrect regexp %s", entry)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if re.MatchString(text) {
|
||||||
|
gw.logger.Debugf("matching %s. ignoring %s", entry, text)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func getProtocol(msg *config.Message) string {
|
||||||
|
p := strings.Split(msg.Account, ".")
|
||||||
|
return p[0]
|
||||||
|
}
|
||||||
|
|
||||||
|
func modifyInMessageTengo(filename string, msg *config.Message) error {
|
||||||
|
if filename == "" {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
res, err := ioutil.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s := tengo.NewScript(res)
|
||||||
|
s.SetImports(stdlib.GetModuleMap(stdlib.AllModuleNames()...))
|
||||||
|
_ = s.Add("msgText", msg.Text)
|
||||||
|
_ = s.Add("msgUsername", msg.Username)
|
||||||
|
_ = s.Add("msgUserID", msg.UserID)
|
||||||
|
_ = s.Add("msgAccount", msg.Account)
|
||||||
|
_ = s.Add("msgChannel", msg.Channel)
|
||||||
|
c, err := s.Compile()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := c.Run(); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
msg.Text = c.Get("msgText").String()
|
||||||
|
msg.Username = c.Get("msgUsername").String()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gw *Gateway) modifyUsernameTengo(msg *config.Message, br *bridge.Bridge) (string, error) {
|
||||||
|
filename := gw.BridgeValues().Tengo.RemoteNickFormat
|
||||||
|
if filename == "" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
res, err := ioutil.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
s := tengo.NewScript(res)
|
||||||
|
s.SetImports(stdlib.GetModuleMap(stdlib.AllModuleNames()...))
|
||||||
|
_ = s.Add("result", "")
|
||||||
|
_ = s.Add("msgText", msg.Text)
|
||||||
|
_ = s.Add("msgUsername", msg.Username)
|
||||||
|
_ = s.Add("msgUserID", msg.UserID)
|
||||||
|
_ = s.Add("nick", msg.Username)
|
||||||
|
_ = s.Add("msgAccount", msg.Account)
|
||||||
|
_ = s.Add("msgChannel", msg.Channel)
|
||||||
|
_ = s.Add("channel", msg.Channel)
|
||||||
|
_ = s.Add("msgProtocol", msg.Protocol)
|
||||||
|
_ = s.Add("remoteAccount", br.Account)
|
||||||
|
_ = s.Add("protocol", br.Protocol)
|
||||||
|
_ = s.Add("bridge", br.Name)
|
||||||
|
_ = s.Add("gateway", gw.Name)
|
||||||
|
c, err := s.Compile()
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if err := c.Run(); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return c.Get("result").String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gw *Gateway) modifyOutMessageTengo(origmsg *config.Message, msg *config.Message, br *bridge.Bridge) (bool, error) {
|
||||||
|
filename := gw.BridgeValues().Tengo.OutMessage
|
||||||
|
var (
|
||||||
|
res []byte
|
||||||
|
err error
|
||||||
|
drop bool
|
||||||
|
)
|
||||||
|
|
||||||
|
if filename == "" {
|
||||||
|
res, err = internal.Asset("tengo/outmessage.tengo")
|
||||||
|
if err != nil {
|
||||||
|
return drop, err
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
res, err = ioutil.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
return drop, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
s := tengo.NewScript(res)
|
||||||
|
|
||||||
|
s.SetImports(stdlib.GetModuleMap(stdlib.AllModuleNames()...))
|
||||||
|
_ = s.Add("inAccount", origmsg.Account)
|
||||||
|
_ = s.Add("inProtocol", origmsg.Protocol)
|
||||||
|
_ = s.Add("inChannel", origmsg.Channel)
|
||||||
|
_ = s.Add("inGateway", origmsg.Gateway)
|
||||||
|
_ = s.Add("inEvent", origmsg.Event)
|
||||||
|
_ = s.Add("outAccount", br.Account)
|
||||||
|
_ = s.Add("outProtocol", br.Protocol)
|
||||||
|
_ = s.Add("outChannel", msg.Channel)
|
||||||
|
_ = s.Add("outGateway", gw.Name)
|
||||||
|
_ = s.Add("outEvent", msg.Event)
|
||||||
|
_ = s.Add("msgText", msg.Text)
|
||||||
|
_ = s.Add("msgUsername", msg.Username)
|
||||||
|
_ = s.Add("msgUserID", msg.UserID)
|
||||||
|
_ = s.Add("msgDrop", drop)
|
||||||
|
c, err := s.Compile()
|
||||||
|
if err != nil {
|
||||||
|
return drop, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := c.Run(); err != nil {
|
||||||
|
return drop, err
|
||||||
|
}
|
||||||
|
|
||||||
|
drop = c.Get("msgDrop").Bool()
|
||||||
|
msg.Text = c.Get("msgText").String()
|
||||||
|
msg.Username = c.Get("msgUsername").String()
|
||||||
|
|
||||||
|
return drop, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,20 +2,28 @@ package gateway
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
"strconv"
|
"strconv"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/gateway/bridgemap"
|
"github.com/42wim/matterbridge/gateway/bridgemap"
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/suite"
|
||||||
)
|
)
|
||||||
|
|
||||||
var testconfig = []byte(`
|
var testconfig = []byte(`
|
||||||
[irc.freenode]
|
[irc.freenode]
|
||||||
|
server=""
|
||||||
[mattermost.test]
|
[mattermost.test]
|
||||||
|
server=""
|
||||||
[gitter.42wim]
|
[gitter.42wim]
|
||||||
|
server=""
|
||||||
[discord.test]
|
[discord.test]
|
||||||
|
server=""
|
||||||
[slack.test]
|
[slack.test]
|
||||||
|
server=""
|
||||||
|
|
||||||
[[gateway]]
|
[[gateway]]
|
||||||
name = "bridge1"
|
name = "bridge1"
|
||||||
@@ -41,10 +49,15 @@ var testconfig = []byte(`
|
|||||||
|
|
||||||
var testconfig2 = []byte(`
|
var testconfig2 = []byte(`
|
||||||
[irc.freenode]
|
[irc.freenode]
|
||||||
|
server=""
|
||||||
[mattermost.test]
|
[mattermost.test]
|
||||||
|
server=""
|
||||||
[gitter.42wim]
|
[gitter.42wim]
|
||||||
|
server=""
|
||||||
[discord.test]
|
[discord.test]
|
||||||
|
server=""
|
||||||
[slack.test]
|
[slack.test]
|
||||||
|
server=""
|
||||||
|
|
||||||
[[gateway]]
|
[[gateway]]
|
||||||
name = "bridge1"
|
name = "bridge1"
|
||||||
@@ -84,8 +97,11 @@ var testconfig2 = []byte(`
|
|||||||
|
|
||||||
var testconfig3 = []byte(`
|
var testconfig3 = []byte(`
|
||||||
[irc.zzz]
|
[irc.zzz]
|
||||||
|
server=""
|
||||||
[telegram.zzz]
|
[telegram.zzz]
|
||||||
|
server=""
|
||||||
[slack.zzz]
|
[slack.zzz]
|
||||||
|
server=""
|
||||||
[[gateway]]
|
[[gateway]]
|
||||||
name="bridge"
|
name="bridge"
|
||||||
enable=true
|
enable=true
|
||||||
@@ -159,8 +175,10 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func maketestRouter(input []byte) *Router {
|
func maketestRouter(input []byte) *Router {
|
||||||
cfg := config.NewConfigFromString(input)
|
logger := logrus.New()
|
||||||
r, err := NewRouter(cfg, bridgemap.FullMap)
|
logger.SetOutput(ioutil.Discard)
|
||||||
|
cfg := config.NewConfigFromString(logger, input)
|
||||||
|
r, err := NewRouter(logger, cfg, bridgemap.FullMap)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
}
|
}
|
||||||
@@ -171,7 +189,6 @@ func TestNewRouter(t *testing.T) {
|
|||||||
assert.Equal(t, 1, len(r.Gateways))
|
assert.Equal(t, 1, len(r.Gateways))
|
||||||
assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges))
|
assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges))
|
||||||
assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels))
|
assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels))
|
||||||
|
|
||||||
r = maketestRouter(testconfig2)
|
r = maketestRouter(testconfig2)
|
||||||
assert.Equal(t, 2, len(r.Gateways))
|
assert.Equal(t, 2, len(r.Gateways))
|
||||||
assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges))
|
assert.Equal(t, 4, len(r.Gateways["bridge1"].Bridges))
|
||||||
@@ -387,7 +404,23 @@ func TestGetDestChannelAdvanced(t *testing.T) {
|
|||||||
assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits)
|
assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestIgnoreTextEmpty(t *testing.T) {
|
type ignoreTestSuite struct {
|
||||||
|
suite.Suite
|
||||||
|
|
||||||
|
gw *Gateway
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIgnoreSuite(t *testing.T) {
|
||||||
|
s := &ignoreTestSuite{}
|
||||||
|
suite.Run(t, s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *ignoreTestSuite) SetupSuite() {
|
||||||
|
logger := logrus.New()
|
||||||
|
logger.SetOutput(ioutil.Discard)
|
||||||
|
s.gw = &Gateway{logger: logrus.NewEntry(logger)}
|
||||||
|
}
|
||||||
|
func (s *ignoreTestSuite) TestIgnoreTextEmpty() {
|
||||||
extraFile := make(map[string][]interface{})
|
extraFile := make(map[string][]interface{})
|
||||||
extraAttach := make(map[string][]interface{})
|
extraAttach := make(map[string][]interface{})
|
||||||
extraFailure := make(map[string][]interface{})
|
extraFailure := make(map[string][]interface{})
|
||||||
@@ -424,78 +457,85 @@ func TestIgnoreTextEmpty(t *testing.T) {
|
|||||||
output: true,
|
output: true,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
gw := &Gateway{}
|
|
||||||
for testname, testcase := range msgTests {
|
for testname, testcase := range msgTests {
|
||||||
output := gw.ignoreTextEmpty(testcase.input)
|
output := s.gw.ignoreTextEmpty(testcase.input)
|
||||||
assert.Equalf(t, testcase.output, output, "case '%s' failed", testname)
|
s.Assert().Equalf(testcase.output, output, "case '%s' failed", testname)
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestIgnoreTexts(t *testing.T) {
|
func (s *ignoreTestSuite) TestIgnoreTexts() {
|
||||||
msgTests := map[string]struct {
|
msgTests := map[string]struct {
|
||||||
input *config.Message
|
input string
|
||||||
re []string
|
re []string
|
||||||
output bool
|
output bool
|
||||||
}{
|
}{
|
||||||
"no regex": {
|
"no regex": {
|
||||||
input: &config.Message{Text: "a text message"},
|
input: "a text message",
|
||||||
re: []string{},
|
re: []string{},
|
||||||
output: false,
|
output: false,
|
||||||
},
|
},
|
||||||
"simple regex": {
|
"simple regex": {
|
||||||
input: &config.Message{Text: "a text message"},
|
input: "a text message",
|
||||||
re: []string{"text"},
|
re: []string{"text"},
|
||||||
output: true,
|
output: true,
|
||||||
},
|
},
|
||||||
"multiple regex fail": {
|
"multiple regex fail": {
|
||||||
input: &config.Message{Text: "a text message"},
|
input: "a text message",
|
||||||
re: []string{"abc", "123$"},
|
re: []string{"abc", "123$"},
|
||||||
output: false,
|
output: false,
|
||||||
},
|
},
|
||||||
"multiple regex pass": {
|
"multiple regex pass": {
|
||||||
input: &config.Message{Text: "a text message"},
|
input: "a text message",
|
||||||
re: []string{"lala", "sage$"},
|
re: []string{"lala", "sage$"},
|
||||||
output: true,
|
output: true,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
gw := &Gateway{}
|
|
||||||
for testname, testcase := range msgTests {
|
for testname, testcase := range msgTests {
|
||||||
output := gw.ignoreTexts(testcase.input, testcase.re)
|
output := s.gw.ignoreText(testcase.input, testcase.re)
|
||||||
assert.Equalf(t, testcase.output, output, "case '%s' failed", testname)
|
s.Assert().Equalf(testcase.output, output, "case '%s' failed", testname)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestIgnoreNicks(t *testing.T) {
|
func (s *ignoreTestSuite) TestIgnoreNicks() {
|
||||||
msgTests := map[string]struct {
|
msgTests := map[string]struct {
|
||||||
input *config.Message
|
input string
|
||||||
re []string
|
re []string
|
||||||
output bool
|
output bool
|
||||||
}{
|
}{
|
||||||
"no entry": {
|
"no entry": {
|
||||||
input: &config.Message{Username: "user", Text: "a text message"},
|
input: "user",
|
||||||
re: []string{},
|
re: []string{},
|
||||||
output: false,
|
output: false,
|
||||||
},
|
},
|
||||||
"one entry": {
|
"one entry": {
|
||||||
input: &config.Message{Username: "user", Text: "a text message"},
|
input: "user",
|
||||||
re: []string{"user"},
|
re: []string{"user"},
|
||||||
output: true,
|
output: true,
|
||||||
},
|
},
|
||||||
"multiple entries": {
|
"multiple entries": {
|
||||||
input: &config.Message{Username: "user", Text: "a text message"},
|
input: "user",
|
||||||
re: []string{"abc", "user"},
|
re: []string{"abc", "user"},
|
||||||
output: true,
|
output: true,
|
||||||
},
|
},
|
||||||
"multiple entries fail": {
|
"multiple entries fail": {
|
||||||
input: &config.Message{Username: "user", Text: "a text message"},
|
input: "user",
|
||||||
re: []string{"abc", "def"},
|
re: []string{"abc", "def"},
|
||||||
output: false,
|
output: false,
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
gw := &Gateway{}
|
|
||||||
for testname, testcase := range msgTests {
|
for testname, testcase := range msgTests {
|
||||||
output := gw.ignoreNicks(testcase.input, testcase.re)
|
output := s.gw.ignoreText(testcase.input, testcase.re)
|
||||||
assert.Equalf(t, testcase.output, output, "case '%s' failed", testname)
|
s.Assert().Equalf(testcase.output, output, "case '%s' failed", testname)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func BenchmarkTengo(b *testing.B) {
|
||||||
|
msg := &config.Message{Username: "user", Text: "blah testing", Account: "protocol.account", Channel: "mychannel"}
|
||||||
|
for n := 0; n < b.N; n++ {
|
||||||
|
err := modifyInMessageTengo("bench.tengo", msg)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,10 +9,12 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/gateway/bridgemap"
|
||||||
)
|
)
|
||||||
|
|
||||||
// handleEventFailure handles failures and reconnects bridges.
|
// handleEventFailure handles failures and reconnects bridges.
|
||||||
@@ -39,7 +41,7 @@ func (r *Router) handleEventGetChannelMembers(msg *config.Message) {
|
|||||||
for _, br := range gw.Bridges {
|
for _, br := range gw.Bridges {
|
||||||
if msg.Account == br.Account {
|
if msg.Account == br.Account {
|
||||||
cMembers := msg.Extra[config.EventGetChannelMembers][0].(config.ChannelMembers)
|
cMembers := msg.Extra[config.EventGetChannelMembers][0].(config.ChannelMembers)
|
||||||
flog.Debugf("Syncing channelmembers from %s", msg.Account)
|
r.logger.Debugf("Syncing channelmembers from %s", msg.Account)
|
||||||
br.SetChannelMembers(&cMembers)
|
br.SetChannelMembers(&cMembers)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -57,7 +59,7 @@ func (r *Router) handleEventRejoinChannels(msg *config.Message) {
|
|||||||
if msg.Account == br.Account {
|
if msg.Account == br.Account {
|
||||||
br.Joined = make(map[string]bool)
|
br.Joined = make(map[string]bool)
|
||||||
if err := br.JoinChannels(); err != nil {
|
if err := br.JoinChannels(); err != nil {
|
||||||
flog.Errorf("channel join failed for %s: %s", msg.Account, err)
|
r.logger.Errorf("channel join failed for %s: %s", msg.Account, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -93,13 +95,13 @@ func (gw *Gateway) handleFiles(msg *config.Message) {
|
|||||||
if gw.BridgeValues().General.MediaServerUpload != "" {
|
if gw.BridgeValues().General.MediaServerUpload != "" {
|
||||||
// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth.
|
// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth.
|
||||||
if err := gw.handleFilesUpload(&fi); err != nil {
|
if err := gw.handleFilesUpload(&fi); err != nil {
|
||||||
flog.Error(err)
|
gw.logger.Error(err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// Use MediaServerPath. Place the file on the current filesystem.
|
// Use MediaServerPath. Place the file on the current filesystem.
|
||||||
if err := gw.handleFilesLocal(&fi); err != nil {
|
if err := gw.handleFilesLocal(&fi); err != nil {
|
||||||
flog.Error(err)
|
gw.logger.Error(err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -107,7 +109,7 @@ func (gw *Gateway) handleFiles(msg *config.Message) {
|
|||||||
// Download URL.
|
// Download URL.
|
||||||
durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name
|
durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name
|
||||||
|
|
||||||
flog.Debugf("mediaserver download URL = %s", durl)
|
gw.logger.Debugf("mediaserver download URL = %s", durl)
|
||||||
|
|
||||||
// We uploaded/placed the file successfully. Add the SHA and URL.
|
// We uploaded/placed the file successfully. Add the SHA and URL.
|
||||||
extra := msg.Extra["file"][i].(config.FileInfo)
|
extra := msg.Extra["file"][i].(config.FileInfo)
|
||||||
@@ -132,7 +134,7 @@ func (gw *Gateway) handleFilesUpload(fi *config.FileInfo) error {
|
|||||||
return fmt.Errorf("mediaserver upload failed, could not create request: %#v", err)
|
return fmt.Errorf("mediaserver upload failed, could not create request: %#v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
flog.Debugf("mediaserver upload url: %s", url)
|
gw.logger.Debugf("mediaserver upload url: %s", url)
|
||||||
|
|
||||||
req.Header.Set("Content-Type", "binary/octet-stream")
|
req.Header.Set("Content-Type", "binary/octet-stream")
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
@@ -153,7 +155,7 @@ func (gw *Gateway) handleFilesLocal(fi *config.FileInfo) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
path := dir + "/" + fi.Name
|
path := dir + "/" + fi.Name
|
||||||
flog.Debugf("mediaserver path placing file: %s", path)
|
gw.logger.Debugf("mediaserver path placing file: %s", path)
|
||||||
|
|
||||||
err = ioutil.WriteFile(path, *fi.Data, os.ModePerm)
|
err = ioutil.WriteFile(path, *fi.Data, os.ModePerm)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -167,7 +169,7 @@ func (gw *Gateway) ignoreEvent(event string, dest *bridge.Bridge) bool {
|
|||||||
switch event {
|
switch event {
|
||||||
case config.EventAvatarDownload:
|
case config.EventAvatarDownload:
|
||||||
// Avatar downloads are only relevant for telegram and mattermost for now
|
// Avatar downloads are only relevant for telegram and mattermost for now
|
||||||
if dest.Protocol != "mattermost" && dest.Protocol != "telegram" {
|
if dest.Protocol != "mattermost" && dest.Protocol != "telegram" && dest.Protocol != "xmpp" {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
case config.EventJoinLeave:
|
case config.EventJoinLeave:
|
||||||
@@ -177,7 +179,7 @@ func (gw *Gateway) ignoreEvent(event string, dest *bridge.Bridge) bool {
|
|||||||
}
|
}
|
||||||
case config.EventTopicChange:
|
case config.EventTopicChange:
|
||||||
// only relay topic change when used in some way on other side
|
// only relay topic change when used in some way on other side
|
||||||
if dest.GetBool("ShowTopicChange") && dest.GetBool("SyncTopic") {
|
if !dest.GetBool("ShowTopicChange") && !dest.GetBool("SyncTopic") {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -186,36 +188,44 @@ func (gw *Gateway) ignoreEvent(event string, dest *bridge.Bridge) bool {
|
|||||||
|
|
||||||
// handleMessage makes sure the message get sent to the correct bridge/channels.
|
// handleMessage makes sure the message get sent to the correct bridge/channels.
|
||||||
// Returns an array of msg ID's
|
// Returns an array of msg ID's
|
||||||
func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID {
|
func (gw *Gateway) handleMessage(rmsg *config.Message, dest *bridge.Bridge) []*BrMsgID {
|
||||||
var brMsgIDs []*BrMsgID
|
var brMsgIDs []*BrMsgID
|
||||||
|
|
||||||
|
// Not all bridges support "user is typing" indications so skip the message
|
||||||
|
// if the targeted bridge does not support it.
|
||||||
|
if rmsg.Event == config.EventUserTyping {
|
||||||
|
if _, ok := bridgemap.UserTypingSupport[dest.Protocol]; !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// if we have an attached file, or other info
|
// if we have an attached file, or other info
|
||||||
if msg.Extra != nil && len(msg.Extra[config.EventFileFailureSize]) != 0 && msg.Text == "" {
|
if rmsg.Extra != nil && len(rmsg.Extra[config.EventFileFailureSize]) != 0 && rmsg.Text == "" {
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
|
|
||||||
if gw.ignoreEvent(msg.Event, dest) {
|
if gw.ignoreEvent(rmsg.Event, dest) {
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
|
|
||||||
// broadcast to every out channel (irc QUIT)
|
// broadcast to every out channel (irc QUIT)
|
||||||
if msg.Channel == "" && msg.Event != config.EventJoinLeave {
|
if rmsg.Channel == "" && rmsg.Event != config.EventJoinLeave {
|
||||||
flog.Debug("empty channel")
|
gw.logger.Debug("empty channel")
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get the ID of the parent message in thread
|
// Get the ID of the parent message in thread
|
||||||
var canonicalParentMsgID string
|
var canonicalParentMsgID string
|
||||||
if msg.ParentID != "" && dest.GetBool("PreserveThreading") {
|
if rmsg.ParentID != "" && dest.GetBool("PreserveThreading") {
|
||||||
canonicalParentMsgID = gw.FindCanonicalMsgID(msg.Protocol, msg.ParentID)
|
canonicalParentMsgID = gw.FindCanonicalMsgID(rmsg.Protocol, rmsg.ParentID)
|
||||||
}
|
}
|
||||||
|
|
||||||
origmsg := msg
|
channels := gw.getDestChannel(rmsg, *dest)
|
||||||
channels := gw.getDestChannel(&msg, *dest)
|
for idx := range channels {
|
||||||
for _, channel := range channels {
|
channel := &channels[idx]
|
||||||
msgID, err := gw.SendMessage(origmsg, dest, channel, canonicalParentMsgID)
|
msgID, err := gw.SendMessage(rmsg, dest, channel, canonicalParentMsgID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("SendMessage failed: %s", err)
|
gw.logger.Errorf("SendMessage failed: %s", err)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if msgID == "" {
|
if msgID == "" {
|
||||||
@@ -225,3 +235,41 @@ func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrM
|
|||||||
}
|
}
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (gw *Gateway) handleExtractNicks(msg *config.Message) {
|
||||||
|
var err error
|
||||||
|
br := gw.Bridges[msg.Account]
|
||||||
|
for _, outer := range br.GetStringSlice2D("ExtractNicks") {
|
||||||
|
search := outer[0]
|
||||||
|
replace := outer[1]
|
||||||
|
msg.Username, msg.Text, err = extractNick(search, replace, msg.Username, msg.Text)
|
||||||
|
if err != nil {
|
||||||
|
gw.logger.Errorf("regexp in %s failed: %s", msg.Account, err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// extractNick searches for a username (based on "search" a regular expression).
|
||||||
|
// if this matches it extracts a nick (based on "extract" another regular expression) from text
|
||||||
|
// and replaces username with this result.
|
||||||
|
// returns error if the regexp doesn't compile.
|
||||||
|
func extractNick(search, extract, username, text string) (string, string, error) {
|
||||||
|
re, err := regexp.Compile(search)
|
||||||
|
if err != nil {
|
||||||
|
return username, text, err
|
||||||
|
}
|
||||||
|
if re.MatchString(username) {
|
||||||
|
re, err = regexp.Compile(extract)
|
||||||
|
if err != nil {
|
||||||
|
return username, text, err
|
||||||
|
}
|
||||||
|
res := re.FindAllStringSubmatch(text, 1)
|
||||||
|
// only replace if we have exactly 1 match
|
||||||
|
if len(res) > 0 && len(res[0]) == 2 {
|
||||||
|
username = res[0][1]
|
||||||
|
text = strings.Replace(text, res[0][0], "", 1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return username, text, nil
|
||||||
|
}
|
||||||
|
|||||||
75
gateway/handlers_test.go
Normal file
75
gateway/handlers_test.go
Normal file
@@ -0,0 +1,75 @@
|
|||||||
|
package gateway
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/42wim/matterbridge/bridge"
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestIgnoreEvent(t *testing.T) {
|
||||||
|
eventTests := map[string]struct {
|
||||||
|
input string
|
||||||
|
dest *bridge.Bridge
|
||||||
|
output bool
|
||||||
|
}{
|
||||||
|
"avatar mattermost": {
|
||||||
|
input: config.EventAvatarDownload,
|
||||||
|
dest: &bridge.Bridge{Protocol: "mattermost"},
|
||||||
|
output: false,
|
||||||
|
},
|
||||||
|
"avatar slack": {
|
||||||
|
input: config.EventAvatarDownload,
|
||||||
|
dest: &bridge.Bridge{Protocol: "slack"},
|
||||||
|
output: true,
|
||||||
|
},
|
||||||
|
"avatar telegram": {
|
||||||
|
input: config.EventAvatarDownload,
|
||||||
|
dest: &bridge.Bridge{Protocol: "telegram"},
|
||||||
|
output: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
gw := &Gateway{}
|
||||||
|
for testname, testcase := range eventTests {
|
||||||
|
output := gw.ignoreEvent(testcase.input, testcase.dest)
|
||||||
|
assert.Equalf(t, testcase.output, output, "case '%s' failed", testname)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestExtractNick(t *testing.T) {
|
||||||
|
eventTests := map[string]struct {
|
||||||
|
search string
|
||||||
|
extract string
|
||||||
|
username string
|
||||||
|
text string
|
||||||
|
resultUsername string
|
||||||
|
resultText string
|
||||||
|
}{
|
||||||
|
"test1": {
|
||||||
|
search: "fromgitter",
|
||||||
|
extract: "<(.*?)>\\s+",
|
||||||
|
username: "fromgitter",
|
||||||
|
text: "<userx> blahblah",
|
||||||
|
resultUsername: "userx",
|
||||||
|
resultText: "blahblah",
|
||||||
|
},
|
||||||
|
"test2": {
|
||||||
|
search: "<.*?bot>",
|
||||||
|
//extract: `\((.*?)\)\s+`,
|
||||||
|
extract: "\\((.*?)\\)\\s+",
|
||||||
|
username: "<matterbot>",
|
||||||
|
text: "(userx) blahblah (abc) test",
|
||||||
|
resultUsername: "userx",
|
||||||
|
resultText: "blahblah (abc) test",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
// gw := &Gateway{}
|
||||||
|
for testname, testcase := range eventTests {
|
||||||
|
resultUsername, resultText, _ := extractNick(testcase.search, testcase.extract, testcase.username, testcase.text)
|
||||||
|
assert.Equalf(t, testcase.resultUsername, resultUsername, "case '%s' failed", testname)
|
||||||
|
assert.Equalf(t, testcase.resultText, resultText, "case '%s' failed", testname)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
@@ -7,31 +7,40 @@ import (
|
|||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
samechannelgateway "github.com/42wim/matterbridge/gateway/samechannel"
|
"github.com/42wim/matterbridge/gateway/samechannel"
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Router struct {
|
type Router struct {
|
||||||
config.Config
|
config.Config
|
||||||
|
sync.RWMutex
|
||||||
|
|
||||||
BridgeMap map[string]bridge.Factory
|
BridgeMap map[string]bridge.Factory
|
||||||
Gateways map[string]*Gateway
|
Gateways map[string]*Gateway
|
||||||
Message chan config.Message
|
Message chan config.Message
|
||||||
MattermostPlugin chan config.Message
|
MattermostPlugin chan config.Message
|
||||||
sync.RWMutex
|
|
||||||
|
logger *logrus.Entry
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewRouter(cfg config.Config, bridgeMap map[string]bridge.Factory) (*Router, error) {
|
// NewRouter initializes a new Matterbridge router for the specified configuration and
|
||||||
|
// sets up all required gateways.
|
||||||
|
func NewRouter(rootLogger *logrus.Logger, cfg config.Config, bridgeMap map[string]bridge.Factory) (*Router, error) {
|
||||||
|
logger := rootLogger.WithFields(logrus.Fields{"prefix": "router"})
|
||||||
|
|
||||||
r := &Router{
|
r := &Router{
|
||||||
Config: cfg,
|
Config: cfg,
|
||||||
BridgeMap: bridgeMap,
|
BridgeMap: bridgeMap,
|
||||||
Message: make(chan config.Message),
|
Message: make(chan config.Message),
|
||||||
MattermostPlugin: make(chan config.Message),
|
MattermostPlugin: make(chan config.Message),
|
||||||
Gateways: make(map[string]*Gateway),
|
Gateways: make(map[string]*Gateway),
|
||||||
|
logger: logger,
|
||||||
}
|
}
|
||||||
sgw := samechannelgateway.New(cfg)
|
sgw := samechannel.New(cfg)
|
||||||
gwconfigs := sgw.GetConfig()
|
gwconfigs := append(sgw.GetConfig(), cfg.BridgeValues().Gateway...)
|
||||||
|
|
||||||
for _, entry := range append(gwconfigs, cfg.BridgeValues().Gateway...) {
|
for idx := range gwconfigs {
|
||||||
|
entry := &gwconfigs[idx]
|
||||||
if !entry.Enable {
|
if !entry.Enable {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@@ -41,21 +50,29 @@ func NewRouter(cfg config.Config, bridgeMap map[string]bridge.Factory) (*Router,
|
|||||||
if _, ok := r.Gateways[entry.Name]; ok {
|
if _, ok := r.Gateways[entry.Name]; ok {
|
||||||
return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name)
|
return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name)
|
||||||
}
|
}
|
||||||
r.Gateways[entry.Name] = New(entry, r)
|
r.Gateways[entry.Name] = New(rootLogger, entry, r)
|
||||||
}
|
}
|
||||||
return r, nil
|
return r, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Start will connect all gateways belonging to this router and subsequently route messages
|
||||||
|
// between them.
|
||||||
func (r *Router) Start() error {
|
func (r *Router) Start() error {
|
||||||
m := make(map[string]*bridge.Bridge)
|
m := make(map[string]*bridge.Bridge)
|
||||||
|
if len(r.Gateways) == 0 {
|
||||||
|
return fmt.Errorf("no [[gateway]] configured. See https://github.com/42wim/matterbridge/wiki/How-to-create-your-config for more info")
|
||||||
|
}
|
||||||
for _, gw := range r.Gateways {
|
for _, gw := range r.Gateways {
|
||||||
flog.Infof("Parsing gateway %s", gw.Name)
|
r.logger.Infof("Parsing gateway %s", gw.Name)
|
||||||
|
if len(gw.Bridges) == 0 {
|
||||||
|
return fmt.Errorf("no bridges configured for gateway %s. See https://github.com/42wim/matterbridge/wiki/How-to-create-your-config for more info", gw.Name)
|
||||||
|
}
|
||||||
for _, br := range gw.Bridges {
|
for _, br := range gw.Bridges {
|
||||||
m[br.Account] = br
|
m[br.Account] = br
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for _, br := range m {
|
for _, br := range m {
|
||||||
flog.Infof("Starting bridge: %s ", br.Account)
|
r.logger.Infof("Starting bridge: %s ", br.Account)
|
||||||
err := br.Connect()
|
err := br.Connect()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
e := fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
|
e := fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
|
||||||
@@ -77,13 +94,13 @@ func (r *Router) Start() error {
|
|||||||
for _, gw := range r.Gateways {
|
for _, gw := range r.Gateways {
|
||||||
for i, br := range gw.Bridges {
|
for i, br := range gw.Bridges {
|
||||||
if br.Bridger == nil {
|
if br.Bridger == nil {
|
||||||
flog.Errorf("removing failed bridge %s", i)
|
r.logger.Errorf("removing failed bridge %s", i)
|
||||||
delete(gw.Bridges, i)
|
delete(gw.Bridges, i)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
go r.handleReceive()
|
go r.handleReceive()
|
||||||
go r.updateChannelMembers()
|
//go r.updateChannelMembers()
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -91,7 +108,7 @@ func (r *Router) Start() error {
|
|||||||
// otherwise returns false
|
// otherwise returns false
|
||||||
func (r *Router) disableBridge(br *bridge.Bridge, err error) bool {
|
func (r *Router) disableBridge(br *bridge.Bridge, err error) bool {
|
||||||
if r.BridgeValues().General.IgnoreFailureOnStart {
|
if r.BridgeValues().General.IgnoreFailureOnStart {
|
||||||
flog.Error(err)
|
r.logger.Error(err)
|
||||||
// setting this bridge empty
|
// setting this bridge empty
|
||||||
*br = bridge.Bridge{}
|
*br = bridge.Bridge{}
|
||||||
return true
|
return true
|
||||||
@@ -114,6 +131,11 @@ func (r *Router) handleReceive() {
|
|||||||
r.handleEventGetChannelMembers(&msg)
|
r.handleEventGetChannelMembers(&msg)
|
||||||
r.handleEventFailure(&msg)
|
r.handleEventFailure(&msg)
|
||||||
r.handleEventRejoinChannels(&msg)
|
r.handleEventRejoinChannels(&msg)
|
||||||
|
|
||||||
|
// Set message protocol based on the account it came from
|
||||||
|
msg.Protocol = r.getBridge(msg.Account).Protocol
|
||||||
|
|
||||||
|
filesHandled := false
|
||||||
for _, gw := range r.Gateways {
|
for _, gw := range r.Gateways {
|
||||||
// record all the message ID's of the different bridges
|
// record all the message ID's of the different bridges
|
||||||
var msgIDs []*BrMsgID
|
var msgIDs []*BrMsgID
|
||||||
@@ -122,13 +144,25 @@ func (r *Router) handleReceive() {
|
|||||||
}
|
}
|
||||||
msg.Timestamp = time.Now()
|
msg.Timestamp = time.Now()
|
||||||
gw.modifyMessage(&msg)
|
gw.modifyMessage(&msg)
|
||||||
gw.handleFiles(&msg)
|
if !filesHandled {
|
||||||
for _, br := range gw.Bridges {
|
gw.handleFiles(&msg)
|
||||||
msgIDs = append(msgIDs, gw.handleMessage(msg, br)...)
|
filesHandled = true
|
||||||
}
|
}
|
||||||
// only add the message ID if it doesn't already exists
|
for _, br := range gw.Bridges {
|
||||||
if _, ok := gw.Messages.Get(msg.Protocol + " " + msg.ID); !ok && msg.ID != "" {
|
msgIDs = append(msgIDs, gw.handleMessage(&msg, br)...)
|
||||||
gw.Messages.Add(msg.Protocol+" "+msg.ID, msgIDs)
|
}
|
||||||
|
|
||||||
|
if msg.ID != "" {
|
||||||
|
_, exists := gw.Messages.Get(msg.Protocol + " " + msg.ID)
|
||||||
|
|
||||||
|
// Only add the message ID if it doesn't already exist
|
||||||
|
//
|
||||||
|
// For some bridges we always add/update the message ID.
|
||||||
|
// This is necessary as msgIDs will change if a bridge returns
|
||||||
|
// a different ID in response to edits.
|
||||||
|
if !exists {
|
||||||
|
gw.Messages.Add(msg.Protocol+" "+msg.ID, msgIDs)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -146,9 +180,9 @@ func (r *Router) updateChannelMembers() {
|
|||||||
if br.Protocol != "slack" {
|
if br.Protocol != "slack" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
flog.Debugf("sending %s to %s", config.EventGetChannelMembers, br.Account)
|
r.logger.Debugf("sending %s to %s", config.EventGetChannelMembers, br.Account)
|
||||||
if _, err := br.Send(config.Message{Event: config.EventGetChannelMembers}); err != nil {
|
if _, err := br.Send(config.Message{Event: config.EventGetChannelMembers}); err != nil {
|
||||||
flog.Errorf("updateChannelMembers: %s", err)
|
r.logger.Errorf("updateChannelMembers: %s", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
package samechannelgateway
|
package samechannel
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
|||||||
@@ -1,9 +1,11 @@
|
|||||||
package samechannelgateway
|
package samechannel
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"io/ioutil"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/sirupsen/logrus"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -66,7 +68,9 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func TestGetConfig(t *testing.T) {
|
func TestGetConfig(t *testing.T) {
|
||||||
cfg := config.NewConfigFromString([]byte(testConfig))
|
logger := logrus.New()
|
||||||
|
logger.SetOutput(ioutil.Discard)
|
||||||
|
cfg := config.NewConfigFromString(logger, []byte(testConfig))
|
||||||
sgw := New(cfg)
|
sgw := New(cfg)
|
||||||
configs := sgw.GetConfig()
|
configs := sgw.GetConfig()
|
||||||
assert.Equal(t, []config.Gateway{expectedConfig}, configs)
|
assert.Equal(t, []config.Gateway{expectedConfig}, configs)
|
||||||
|
|||||||
114
go.mod
114
go.mod
@@ -2,69 +2,59 @@ module github.com/42wim/matterbridge
|
|||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557
|
github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557
|
||||||
github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 // indirect
|
github.com/Baozisoftware/qrcode-terminal-go v0.0.0-20170407111555-c0650d8dff0f
|
||||||
github.com/Philipp15b/go-steam v1.0.1-0.20180818081528-681bd9573329
|
github.com/Jeffail/gabs v1.4.0 // indirect
|
||||||
github.com/bwmarrin/discordgo v0.19.0
|
github.com/Philipp15b/go-steam v1.0.1-0.20200727090957-6ae9b3c0a560
|
||||||
github.com/dfordsoft/golib v0.0.0-20180902042739-76ee6ab99bec
|
github.com/Rhymen/go-whatsapp v0.1.2-0.20210615184944-2b8a3e9b8aa2
|
||||||
github.com/fsnotify/fsnotify v1.4.7
|
github.com/SevereCloud/vksdk/v2 v2.10.0
|
||||||
github.com/go-telegram-bot-api/telegram-bot-api v4.6.5-0.20181225215658-ec221ba9ea45+incompatible
|
github.com/d5/tengo/v2 v2.7.0
|
||||||
github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc // indirect
|
github.com/davecgh/go-spew v1.1.1
|
||||||
github.com/google/gops v0.3.5
|
github.com/fsnotify/fsnotify v1.4.9
|
||||||
github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f // indirect
|
github.com/go-telegram-bot-api/telegram-bot-api v1.0.1-0.20200524105306-7434b0456e81
|
||||||
github.com/gorilla/schema v1.0.2
|
github.com/gomarkdown/markdown v0.0.0-20210514010506-3b9f47219fe7
|
||||||
github.com/gorilla/websocket v1.4.0
|
github.com/google/gops v0.3.18
|
||||||
github.com/hashicorp/golang-lru v0.5.0
|
github.com/gopackage/ddp v0.0.0-20170117053602-652027933df4 // indirect
|
||||||
github.com/hpcloud/tail v1.0.0 // indirect
|
github.com/gorilla/schema v1.2.0
|
||||||
github.com/jpillora/backoff v0.0.0-20180909062703-3050d21c67d7
|
github.com/gorilla/websocket v1.4.2
|
||||||
github.com/jtolds/gls v4.2.1+incompatible // indirect
|
github.com/hashicorp/golang-lru v0.5.4
|
||||||
github.com/kardianos/osext v0.0.0-20170510131534-ae77be60afb1 // indirect
|
github.com/jpillora/backoff v1.0.0
|
||||||
github.com/kr/pretty v0.1.0 // indirect
|
github.com/keybase/go-keybase-chat-bot v0.0.0-20200505163032-5cacf52379da
|
||||||
github.com/labstack/echo/v4 v4.0.0
|
github.com/kyokomi/emoji/v2 v2.2.8
|
||||||
github.com/lrstanley/girc v0.0.0-20190102153329-c1e59a02f488
|
github.com/labstack/echo/v4 v4.3.0
|
||||||
github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 // indirect
|
github.com/lrstanley/girc v0.0.0-20210611213246-771323f1624b
|
||||||
github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 // indirect
|
github.com/matrix-org/gomatrix v0.0.0-20210324163249-be2af5ef2e16
|
||||||
github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91
|
github.com/matterbridge/Rocket.Chat.Go.SDK v0.0.0-20210403163225-761e8622445d
|
||||||
github.com/matterbridge/gomatrix v0.0.0-20190102230110-6f9631ca6dea
|
github.com/matterbridge/discordgo v0.21.2-0.20210201201054-fb39a175b4f7
|
||||||
github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544
|
github.com/matterbridge/go-xmpp v0.0.0-20200418225040-c8a3a57b4050
|
||||||
github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61
|
github.com/matterbridge/gozulipbot v0.0.0-20200820220548-be5824faa913
|
||||||
github.com/mattermost/mattermost-server v5.5.0+incompatible
|
github.com/matterbridge/logrus-prefixed-formatter v0.5.3-0.20200523233437-d971309a77ba
|
||||||
github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect
|
github.com/mattermost/mattermost-server/v5 v5.30.1
|
||||||
github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 // indirect
|
github.com/mattn/godown v0.0.1
|
||||||
|
github.com/mgutz/ansi v0.0.0-20200706080929-d51e80ef957d // indirect
|
||||||
|
github.com/missdeer/golib v1.0.4
|
||||||
|
github.com/mreiferson/go-httpclient v0.0.0-20201222173833-5e475fde3a4d // indirect
|
||||||
github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff // indirect
|
github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff // indirect
|
||||||
github.com/nicksnyder/go-i18n v1.4.0 // indirect
|
github.com/nelsonken/gomf v0.0.0-20180504123937-a9dd2f9deae9
|
||||||
github.com/nlopes/slack v0.5.0
|
github.com/paulrosania/go-charset v0.0.0-20190326053356-55c9d7a5834c
|
||||||
github.com/onsi/ginkgo v1.6.0 // indirect
|
github.com/rs/xid v1.3.0
|
||||||
github.com/onsi/gomega v1.4.1 // indirect
|
github.com/russross/blackfriday v1.6.0
|
||||||
github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83
|
|
||||||
github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606 // indirect
|
|
||||||
github.com/peterhellberg/emojilib v0.0.0-20190124112554-c18758d55320
|
|
||||||
github.com/pkg/errors v0.8.0 // indirect
|
|
||||||
github.com/rs/xid v1.2.1
|
|
||||||
github.com/russross/blackfriday v2.0.0+incompatible
|
|
||||||
github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca
|
github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca
|
||||||
github.com/shazow/ssh-chat v0.0.0-20190125184227-81d7e1686296
|
github.com/shazow/ssh-chat v1.10.1
|
||||||
github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95 // indirect
|
github.com/sirupsen/logrus v1.8.1
|
||||||
github.com/sirupsen/logrus v1.3.0
|
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e // indirect
|
||||||
github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 // indirect
|
github.com/slack-go/slack v0.9.1
|
||||||
github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a // indirect
|
github.com/spf13/viper v1.8.0
|
||||||
github.com/spf13/viper v1.3.1
|
github.com/stretchr/testify v1.7.0
|
||||||
github.com/stretchr/testify v1.3.0
|
github.com/vincent-petithory/dataurl v0.0.0-20191104211930-d1553a71de50
|
||||||
github.com/technoweenie/multipartstreamer v1.0.1 // indirect
|
github.com/writeas/go-strip-markdown v2.0.1+incompatible
|
||||||
github.com/x-cray/logrus-prefixed-formatter v0.5.2 // indirect
|
github.com/x-cray/logrus-prefixed-formatter v0.5.2 // indirect
|
||||||
github.com/zfjagann/golang-ring v0.0.0-20190106091943-a88bb6aef447
|
github.com/yaegashi/msgraph.go v0.1.4
|
||||||
gitlab.com/golang-commonmark/html v0.0.0-20180917080848-cfaf75183c4a // indirect
|
github.com/zfjagann/golang-ring v0.0.0-20210116075443-7c86fdb43134
|
||||||
gitlab.com/golang-commonmark/linkify v0.0.0-20180917065525-c22b7bdb1179 // indirect
|
golang.org/x/image v0.0.0-20210607152325-775e3b0c77b9
|
||||||
gitlab.com/golang-commonmark/markdown v0.0.0-20181102083822-772775880e1f
|
golang.org/x/oauth2 v0.0.0-20210615190721-d04028783cf1
|
||||||
gitlab.com/golang-commonmark/mdurl v0.0.0-20180912090424-e5bce34c34f2 // indirect
|
gomod.garykim.dev/nc-talk v0.2.2
|
||||||
gitlab.com/golang-commonmark/puny v0.0.0-20180912090636-2cd490539afe // indirect
|
gopkg.in/olahol/melody.v1 v1.0.0-20170518105555-d52139073376
|
||||||
gitlab.com/opennota/wd v0.0.0-20180912061657-c5d65f63c638 // indirect
|
layeh.com/gumble v0.0.0-20200818122324-146f9205029b
|
||||||
go.uber.org/atomic v1.3.2 // indirect
|
|
||||||
go.uber.org/multierr v1.1.0 // indirect
|
|
||||||
go.uber.org/zap v1.9.1 // indirect
|
|
||||||
golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 // indirect
|
|
||||||
golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f // indirect
|
|
||||||
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect
|
|
||||||
gopkg.in/fsnotify.v1 v1.4.7 // indirect
|
|
||||||
gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect
|
|
||||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 // indirect
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
go 1.15
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user