Compare commits
	
		
			200 Commits
		
	
	
		
			v1.11.3
			...
			v1.14.0-rc
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | 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 | ||
|   | 1d39c771e4 | ||
|   | c81c0dd22a | ||
|   | f8a1ab4622 | ||
|   | 5d3309fdcd | ||
|   | 4ae028fe73 | ||
|   | 707db950c8 | ||
|   | 94812d8648 | ||
|   | 8548b69e6e | ||
|   | e3cb665d92 | ||
|   | fb713ed91b | ||
|   | d99eacc2e1 | ||
|   | 62e55214fc | ||
|   | 464d27ad7e | ||
|   | f88c5f6c08 | ||
|   | da6ce791bc | ||
|   | b33b50987b | ||
|   | 5193634a52 | ||
|   | 0d94746f4a | ||
|   | 85680935d4 | ||
|   | 46e2683995 | ||
|   | 492722af8b | ||
|   | 56749dfb20 | ||
|   | 04567c765e | ||
|   | 048158ad6d | ||
|   | 7326b9e10d | ||
|   | 8522d8f29c | ||
|   | bab385c342 | ||
|   | bb27ef7939 | ||
|   | d2044c647b | ||
|   | c585d00f16 | ||
|   | 015c076315 | ||
|   | 426aa33723 | ||
|   | da8e415ae1 | ||
|   | 1b834c6858 | ||
|   | d82726cd1b | ||
|   | 288f0a06bb | ||
|   | 0121d75032 | ||
|   | 53c86702a3 | ||
|   | 192fe89789 | ||
|   | 959ca3cef3 | ||
|   | ccd55d2a28 | ||
|   | bfa9a83d31 | ||
|   | 2f7b4d7f68 | ||
|   | d887855e16 | ||
|   | 1a1e68ec98 | ||
|   | a2754f15fc | ||
|   | b6d81f34ba | ||
|   | f9fb33e696 | ||
|   | f72d5de2d7 | ||
|   | 0365c0786a | ||
|   | af7a00d030 | ||
|   | 8a7efce941 | ||
|   | ce73aa5a74 | ||
|   | 64d63a25cc | ||
|   | 2cef9c4fcf | ||
|   | 4265d43096 | ||
|   | 25857591a2 | ||
|   | 27f5a1a685 | ||
|   | 84da2d6a29 | ||
|   | 859ebad55d | ||
|   | e538a4d304 | ||
|   | f94c2b40a3 | ||
|   | 47d29ecf63 | ||
|   | f2088a687e | ||
|   | faeeee2948 | ||
|   | 7923cfe8f8 | ||
|   | b51d0a9b05 | ||
|   | 09f22a801e | ||
|   | 3a824c5f9d | ||
|   | df02f51c56 | ||
|   | fc5e3a6728 | ||
|   | 57fbd3c723 | ||
|   | 25cd1e2cc1 | ||
|   | f5659d455d | ||
|   | 5ed7abdbeb | ||
|   | 09875fe160 | ||
|   | f716b8fc0f | ||
|   | 9f66f93641 | ||
|   | f00d4d7d3f | ||
|   | 0929535b2e | ||
|   | 8869e253ca | ||
|   | f3a5ea2956 | ||
|   | f4d4dc91b1 | ||
|   | c6fd65d1d7 | ||
|   | 0795906533 | ||
|   | a2b45bc799 | ||
|   | 757657f29c | ||
|   | 219c7659e1 | ||
|   | ae32bae791 | ||
|   | 57eba77561 | ||
|   | d5bc7c4343 | ||
|   | 32f57b7c26 | ||
|   | 692bb8faa7 | ||
|   | 455a0fc239 | ||
|   | b2cbd13251 | ||
|   | ce21ba1545 | ||
|   | c89085bf44 | ||
|   | 4254ed3c63 | ||
|   | 85564a35fd | ||
|   | 09713d40ba | ||
|   | 16d5aeac7c | ||
|   | e19ba5a06a | ||
|   | f7a5077d5d | ||
|   | f8dc24bc09 | ||
|   | e9419f10d3 | ||
|   | cded603c27 | ||
|   | d2ae3ebf9e | ||
|   | 730ccdd456 | ||
|   | 2f042ad915 | ||
|   | ba70691877 | ||
|   | ed11686a99 | ||
|   | 5c50d86908 | ||
|   | fea31753b0 | ||
|   | 0d64cd8bab | ||
|   | 9be0f8f000 | ||
|   | 78401214b0 | ||
|   | b2a07aba3a | ||
|   | 1e0bb3da95 | ||
|   | 59994da176 | ||
|   | 3d281b3316 | ||
|   | ea86849a58 | ||
|   | 399789811e | ||
|   | 8d117cb0a4 | ||
|   | 588b8e0303 | ||
|   | 1794922263 | ||
|   | 0ededb8863 | ||
|   | aa59bb1a41 | ||
|   | f2703979a4 | ||
|   | d2a1dc792f | ||
|   | 06d66a0b2b | ||
|   | 0e2522279e | ||
|   | 141a42a75b | ||
|   | a1bf37e457 | ||
|   | a20b7895a9 | ||
|   | 5666821e7b | ||
|   | 5132d8f097 | ||
|   | b81ff9c008 | ||
|   | 7e62bc4819 | ||
|   | d058be25ad | ||
|   | 1269be1d04 | ||
|   | 3b8837a16b | ||
|   | 32f478e4a0 | ||
|   | e2b50d6194 | ||
|   | 74e33b0a51 | ||
|   | 107969c09a | ||
|   | d379118772 | ||
|   | 291594b99c | ||
|   | f2cdda7278 | ||
|   | 6911458d15 | ||
|   | 6238effdc2 | ||
|   | 498377a230 | ||
|   | 3dd4ec57ff | ||
|   | e15b0e04b8 | ||
|   | 97b1fc813b | ||
|   | 917040b044 | ||
|   | 69646a160d | ||
|   | 54adb0509e | ||
|   | bd3a3b6eaf | ||
|   | 296428d53e | ||
|   | e0ca876de2 | ||
|   | a431a4fa04 | ||
|   | cc2bd03ec9 | ||
|   | 1fe81b7d1e | 
							
								
								
									
										213
									
								
								.golangci.yaml
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,213 @@ | ||||
| # For full documentation of the configuration options please | ||||
| # see: https://github.com/golangci/golangci-lint#config-file. | ||||
|  | ||||
| # options for analysis running | ||||
| run: | ||||
|   # default concurrency is the available CPU number | ||||
|   # concurrency: 4 | ||||
|  | ||||
|   # timeout for analysis, e.g. 30s, 5m, default is 1m | ||||
|   deadline: 1m | ||||
|  | ||||
|   # exit code when at least one issue was found, default is 1 | ||||
|   issues-exit-code: 1 | ||||
|  | ||||
|   # include test files or not, default is true | ||||
|   tests: true | ||||
|  | ||||
|   # list of build tags, all linters use it. Default is empty list. | ||||
|   build-tags: | ||||
|  | ||||
|   # which dirs to skip: they won't be analyzed; | ||||
|   # can use regexp here: generated.*, regexp is applied on full path; | ||||
|   # default value is empty list, but next dirs are always skipped independently | ||||
|   # from this option's value: | ||||
|   #   	vendor$, third_party$, testdata$, examples$, Godeps$, builtin$ | ||||
|   skip-dirs: | ||||
|  | ||||
|   # which files to skip: they will be analyzed, but issues from them | ||||
|   # won't be reported. Default value is empty list, but there is | ||||
|   # no need to include all autogenerated files, we confidently recognize | ||||
|   # autogenerated files. If it's not please let us know. | ||||
|   skip-files: | ||||
|  | ||||
|  | ||||
| # output configuration options | ||||
| output: | ||||
|   # colored-line-number|line-number|json|tab|checkstyle, default is "colored-line-number" | ||||
|   format: colored-line-number | ||||
|  | ||||
|   # print lines of code with issue, default is true | ||||
|   print-issued-lines: true | ||||
|  | ||||
|   # print linter name in the end of issue text, default is true | ||||
|   print-linter-name: true | ||||
|  | ||||
|  | ||||
| # all available settings of specific linters, we can set an option for | ||||
| # a given linter even if we deactivate that same linter at runtime | ||||
| linters-settings: | ||||
|   errcheck: | ||||
|     # report about not checking of errors in type assertions: `a := b.(MyStruct)`; | ||||
|     # default is false: such cases aren't reported by default. | ||||
|     check-type-assertions: false | ||||
|  | ||||
|     # report about assignment of errors to blank identifier: `num, _ := strconv.Atoi(numStr)`; | ||||
|     # default is false: such cases aren't reported by default. | ||||
|     check-blank: false | ||||
|   govet: | ||||
|     # report about shadowed variables | ||||
|     check-shadowing: true | ||||
|   golint: | ||||
|     # minimal confidence for issues, default is 0.8 | ||||
|     min-confidence: 0.8 | ||||
|   gofmt: | ||||
|     # simplify code: gofmt with `-s` option, true by default | ||||
|     simplify: true | ||||
|   goimports: | ||||
|     # put imports beginning with prefix after 3rd-party packages; | ||||
|     # it's a comma-separated list of prefixes | ||||
|     local-prefixes: github.com | ||||
|   gocyclo: | ||||
|     # minimal code complexity to report, 30 by default (but we recommend 10-20) | ||||
|     min-complexity: 15 | ||||
|   maligned: | ||||
|     # print struct with more effective memory layout or not, false by default | ||||
|     suggest-new: true | ||||
|   dupl: | ||||
|     # tokens count to trigger issue, 150 by default | ||||
|     threshold: 150 | ||||
|   goconst: | ||||
|     # minimal length of string constant, 3 by default | ||||
|     min-len: 3 | ||||
|     # minimal occurrences count to trigger, 3 by default | ||||
|     min-occurrences: 3 | ||||
|   depguard: | ||||
|     list-type: blacklist | ||||
|     include-go-root: false | ||||
|     packages: | ||||
|     # List of packages that we would want to blacklist for... reasons. | ||||
|   misspell: | ||||
|     # Correct spellings using locale preferences for US or UK. | ||||
|     # Default is to use a neutral variety of English. | ||||
|     # Setting locale to US will correct the British spelling of 'colour' to 'color'. | ||||
|     locale: US | ||||
|   lll: | ||||
|     # max line length, lines longer will be reported. Default is 120. | ||||
|     # '\t' is counted as 1 character by default, and can be changed with the tab-width option | ||||
|     line-length: 150 | ||||
|     # tab width in spaces. Default to 1. | ||||
|     tab-width: 1 | ||||
|   unused: | ||||
|     # treat code as a program (not a library) and report unused exported identifiers; default is false. | ||||
|     # XXX: if you enable this setting, unused will report a lot of false-positives in text editors: | ||||
|     # if it's called for subdir of a project it can't find funcs usages. All text editor integrations | ||||
|     # with golangci-lint call it on a directory with the changed file. | ||||
|     check-exported: false | ||||
|   unparam: | ||||
|     # call graph construction algorithm (cha, rta). In general, use cha for libraries, | ||||
|     # and rta for programs with main packages. Default is cha. | ||||
|     algo: rta | ||||
|  | ||||
|     # Inspect exported functions, default is false. Set to true if no external program/library imports your code. | ||||
|     # XXX: if you enable this setting, unparam will report a lot of false-positives in text editors: | ||||
|     # if it's called for subdir of a project it can't find external interfaces. All text editor integrations | ||||
|     # with golangci-lint call it on a directory with the changed file. | ||||
|     check-exported: false | ||||
|   nakedret: | ||||
|     # make an issue if func has more lines of code than this setting and it has naked returns; default is 30 | ||||
|     max-func-lines: 0 # Warn on all naked returns. | ||||
|   prealloc: | ||||
|     # XXX: we don't recommend using this linter before doing performance profiling. | ||||
|     # For most programs usage of prealloc will be a premature optimization. | ||||
|  | ||||
|     # Report preallocation suggestions only on simple loops that have no returns/breaks/continues/gotos in them. | ||||
|     # True by default. | ||||
|     simple: true | ||||
|     range-loops: true # Report preallocation suggestions on range loops, true by default | ||||
|     for-loops: false # Report preallocation suggestions on for loops, false by default | ||||
|   gocritic: | ||||
|     # which checks should be enabled; can't be combined with 'disabled-checks'; | ||||
|     # default are: [appendAssign assignOp caseOrder dupArg dupBranchBody dupCase flagDeref | ||||
|     # ifElseChain regexpMust singleCaseSwitch sloppyLen switchTrue typeSwitchVar underef | ||||
|     # unlambda unslice rangeValCopy defaultCaseOrder]; | ||||
|     # all checks list: https://github.com/go-critic/checkers | ||||
|     # disabled for now - hugeParam | ||||
|     enabled-checks: | ||||
|       - appendAssign | ||||
|       - assignOp | ||||
|       - boolExprSimplify | ||||
|       - builtinShadow | ||||
|       - captLocal | ||||
|       - caseOrder | ||||
|       - commentedOutImport | ||||
|       - defaultCaseOrder | ||||
|       - dupArg | ||||
|       - dupBranchBody | ||||
|       - dupCase | ||||
|       - dupSubExpr | ||||
|       - elseif | ||||
|       - emptyFallthrough | ||||
|       - ifElseChain | ||||
|       - importShadow | ||||
|       - indexAlloc | ||||
|       - methodExprCall | ||||
|       - nestingReduce | ||||
|       - offBy1 | ||||
|       - ptrToRefParam | ||||
|       - regexpMust | ||||
|       - singleCaseSwitch | ||||
|       - sloppyLen | ||||
|       - sloppyReassign | ||||
|       - switchTrue | ||||
|       - typeSwitchVar | ||||
|       - typeUnparen | ||||
|       - underef | ||||
|       - unlambda | ||||
|       - unnecessaryBlock | ||||
|       - unslice | ||||
|       - valSwap | ||||
|       - wrapperFunc | ||||
|       - yodaStyleExpr | ||||
|  | ||||
|  | ||||
| # linters that we should / shouldn't run | ||||
| linters: | ||||
|   enable-all: true | ||||
|   disable: | ||||
|     - gochecknoglobals | ||||
|     - lll | ||||
|     - maligned | ||||
|     - prealloc | ||||
|  | ||||
|  | ||||
| # rules to deal with reported isues | ||||
| issues: | ||||
|   # List of regexps of issue texts to exclude, empty list by default. | ||||
|   # But independently from this option we use default exclude patterns, | ||||
|   # it can be disabled by `exclude-use-default: false`. To list all | ||||
|   # excluded by default patterns execute `golangci-lint run --help` | ||||
|   exclude: | ||||
|  | ||||
|   # Independently from option `exclude` we use default exclude patterns, | ||||
|   # it can be disabled by this option. To list all | ||||
|   # excluded by default patterns execute `golangci-lint run --help`. | ||||
|   # Default value for this option is true. | ||||
|   exclude-use-default: true | ||||
|  | ||||
|   # Maximum issues count per one linter. Set to 0 to disable. Default is 50. | ||||
|   max-per-linter: 0 | ||||
|  | ||||
|   # Maximum count of issues with the same text. Set to 0 to disable. Default is 3. | ||||
|   max-same-issues: 0 | ||||
|  | ||||
|   # Show only new issues: if there are unstaged changes or untracked files, | ||||
|   # only those changes are analyzed, else only changes in HEAD~ are analyzed. | ||||
|   # It's a super-useful option for integration of golangci-lint into existing | ||||
|   # large codebase. It's not practical to fix all existing issues at the moment | ||||
|   # of integration: much better don't allow issues in new code. | ||||
|   # Default is false. | ||||
|   new: false | ||||
|  | ||||
|   # Show only new issues created after git revision `REV` | ||||
|   new-from-rev: "HEAD~1" | ||||
							
								
								
									
										42
									
								
								.travis.yml
									
									
									
									
									
								
							
							
						
						| @@ -1,16 +1,18 @@ | ||||
| language: go | ||||
| go: | ||||
|     #- 1.7.x | ||||
|     - 1.10.x | ||||
|       # - tip | ||||
|     - 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 | ||||
|       #    - GOOS=windows GOARCH=amd64 | ||||
|       #- GOOS=linux GOARCH=arm | ||||
|     - GOLANGCI_VERSION="v1.14.0" | ||||
|  | ||||
| matrix: | ||||
|   # It's ok if our code fails on unstable development versions of Go. | ||||
| @@ -24,22 +26,26 @@ notifications: | ||||
|       email: false | ||||
|  | ||||
| before_script: | ||||
|   - MY_VERSION=$(git describe --tags) | ||||
|   - GO_FILES=$(find . -iname '*.go' | grep -v /vendor/)  # All the .go files, excluding vendor/ | ||||
|   - PKGS=$(go list ./... | grep -v /vendor/)             # All the import paths, excluding vendor/ | ||||
| #  - go get github.com/golang/lint/golint                 # Linter | ||||
|   - go get honnef.co/go/tools/cmd/megacheck              # Badass static analyzer/linter | ||||
|   # 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 | ||||
|  | ||||
| # Anything in before_script: that returns a nonzero exit code will | ||||
| # flunk the build and immediately stop. It's sorta like having | ||||
| # set -e enabled in bash.  | ||||
| script: | ||||
|  #- test -z $(gofmt -s -l $GO_FILES)  # Fail if a .go file hasn't been formatted with gofmt | ||||
|   - go test -v -race $PKGS            # Run all the tests with the race detector enabled | ||||
|  #  - go vet $PKGS                      # go vet is the official Go static analyzer | ||||
|   - megacheck $PKGS                   # "go vet on steroids" + linter | ||||
|   # 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 | ||||
|   #- golint -set_exit_status $PKGS     # one last linter | ||||
|  | ||||
| after_script: | ||||
|   # Upload test coverage to CodeClimate. | ||||
|   - ./cc-test-reporter after-build --exit-code ${TRAVIS_TEST_RESULT} | ||||
|  | ||||
| deploy: | ||||
|   provider: bintray | ||||
|   | ||||
							
								
								
									
										191
									
								
								README.md
									
									
									
									
									
								
							
							
						
						| @@ -1,57 +1,76 @@ | ||||
| <div align="center"> | ||||
|  | ||||
| # matterbridge | ||||
| Click on one of the badges below to join the chat    | ||||
|  | ||||
| [](https://gitter.im/42wim/matterbridge) [](https://webchat.freenode.net/?channels=matterbridgechat) [](https://discord.gg/AkKPtrQ) [](https://riot.im/app/#/room/#matterbridge:matrix.org) [](https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA) [](https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e) [](https://inverse.chat) [](https://www.twitch.tv/matterbridge) [](https://matterbridge.zulipchat.com/register/) | ||||
| <br /> | ||||
|    **A simple chat bridge**<br /> | ||||
|    Letting people be where they want to be.<br /> | ||||
|    <sub>Bridges between a growing number of protocols. Click below to demo or join the development chat.</sub> | ||||
|  | ||||
| [](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion) | ||||
|    <sup> | ||||
|  | ||||
|  | ||||
|    [Gitter][mb-gitter] | | ||||
|    [IRC][mb-irc] | | ||||
|       [Discord][mb-discord] | | ||||
|       [Matrix][mb-matrix] | | ||||
|       [Slack][mb-slack] | | ||||
|       [Mattermost][mb-mattermost] | | ||||
|       [Rocket.Chat][mb-rocketchat] | | ||||
|       [XMPP][mb-xmpp] | | ||||
|       [Twitch][mb-twitch] | | ||||
|       [WhatsApp][mb-whatsapp] | | ||||
|       [Zulip][mb-zulip] | | ||||
|       [Telegram][mb-telegram] | | ||||
|       And more... | ||||
|    </sup> | ||||
|  | ||||
| Simple bridge between IRC, XMPP, Gitter, Mattermost, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix, Steam, ssh-chat and Zulip | ||||
| Has a REST API.    | ||||
| Minecraft server chat support via [MatterLink](https://github.com/elytra/MatterLink) | ||||
| ---- | ||||
| [](https://github.com/42wim/matterbridge/releases/latest) | ||||
|    [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion) | ||||
|    [](https://codeclimate.com/github/42wim/matterbridge/maintainability) | ||||
|    [](https://codeclimate.com/github/42wim/matterbridge/test_coverage)<br /> | ||||
|   <hr /> | ||||
| </div> | ||||
| <div align="right"><sup> | ||||
|  | ||||
| **Mattermost isn't required to run matterbridge. It bridges between any supported protocol.**    | ||||
| (The name matterbridge is a remnant when it was only bridging mattermost) | ||||
| **Note:** Matter<em>most</em> isn't required to run matter<em>bridge</em>.</sup></div> | ||||
|  | ||||
| # Table of Contents | ||||
| ### Table of Contents | ||||
|  * [Features](https://github.com/42wim/matterbridge/wiki/Features) | ||||
|  * [Requirements](#requirements) | ||||
|    * [Natively supported](#natively-supported) | ||||
|    * [3rd party via matterbridge api](#3rd-party-via-matterbridge-api) | ||||
|    * [API](#API) | ||||
|  * [Chat with us](#chat-with-us) | ||||
|  * [Screenshots](https://github.com/42wim/matterbridge/wiki/) | ||||
|  * [Installing](#installing) | ||||
|    * [Binaries](#binaries) | ||||
|    * [Building](#building) | ||||
|  * [Configuration](#configuration) | ||||
|    * [Howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) | ||||
|    * [Examples](#examples)  | ||||
|    * [Examples](#examples) | ||||
|  * [Running](#running) | ||||
|    * [Docker](#docker) | ||||
|  * [Changelog](#changelog) | ||||
|  * [FAQ](#faq) | ||||
|  * [Related projects](#related-projects) | ||||
|  * [Articles](#articles) | ||||
|  * [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) | ||||
| * [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes) | ||||
| * Preserves threading when possible | ||||
| * [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling) | ||||
| * [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing) | ||||
| * [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups) | ||||
| * [API](https://github.com/42wim/matterbridge/wiki/Features#api) | ||||
|  | ||||
| ## API | ||||
| The API is very basic at the moment and rather undocumented. | ||||
| ### Natively supported | ||||
|  | ||||
| Used by at least 2 projects. Feel free to make a PR to add your project to this list. | ||||
|  | ||||
| * [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Server chat) | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
|  | ||||
| # Requirements | ||||
| Accounts to one of the supported bridges | ||||
| * [Mattermost](https://github.com/mattermost/platform/) 3.8.x - 3.10.x, 4.x, 5.x | ||||
| * [Mattermost](https://github.com/mattermost/mattermost-server/) 4.x, 5.x | ||||
| * [IRC](http://www.mirc.com/servers.html) | ||||
| * [XMPP](https://jabber.org) | ||||
| * [XMPP](https://xmpp.org) | ||||
| * [Gitter](https://gitter.im) | ||||
| * [Slack](https://slack.com) | ||||
| * [Discord](https://discordapp.com) | ||||
| @@ -62,20 +81,56 @@ Accounts to one of the supported bridges | ||||
| * [Steam](https://store.steampowered.com/) | ||||
| * [Twitch](https://twitch.tv) | ||||
| * [Ssh-chat](https://github.com/shazow/ssh-chat) | ||||
| * [WhatsApp](https://www.whatsapp.com/) | ||||
| * [Zulip](https://zulipchat.com) | ||||
|  | ||||
| # Screenshots | ||||
| ### 3rd party via matterbridge api | ||||
| * [Minecraft](https://github.com/elytra/MatterLink) | ||||
| * [Reddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| * [Facebook messenger](https://github.com/VictorNine/fbridge) | ||||
|  | ||||
| ### API | ||||
| The API is very basic at the moment.    | ||||
| More info and examples on the [wiki](https://github.com/42wim/matterbridge/wiki/Api). | ||||
|  | ||||
| Used by the projects below. Feel free to make a PR to add your project to this list. | ||||
|  | ||||
| * [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Server chat) | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| * [Mattereddit](https://github.com/bonehurtingjuice/mattereddit) (Reddit chat support) | ||||
| * [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
|  | ||||
| ## Chat with us | ||||
|  | ||||
| Questions or want to test on your favorite platform? Join below: | ||||
|  | ||||
| * [Gitter][mb-gitter] | ||||
| * [IRC][mb-irc] | ||||
| * [Discord][mb-discord] | ||||
| * [Matrix][mb-matrix] | ||||
| * [Slack][mb-slack] | ||||
| * [Mattermost][mb-mattermost] | ||||
| * [Rocket.Chat][mb-rocketchat] | ||||
| * [XMPP][mb-xmpp] | ||||
| * [Twitch][mb-twitch] | ||||
| * [Zulip][mb-zulip] | ||||
| * [Telegram][mb-telegram] | ||||
|  | ||||
| ## Screenshots | ||||
| See https://github.com/42wim/matterbridge/wiki | ||||
|  | ||||
| # Installing | ||||
| ## Binaries | ||||
| * Latest stable release [v1.11.3](https://github.com/42wim/matterbridge/releases/latest) | ||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)   | ||||
| ## Installing | ||||
| ### Binaries | ||||
| * Latest stable release [v1.13.1](https://github.com/42wim/matterbridge/releases/latest) | ||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/) | ||||
|  | ||||
| ## Building | ||||
| 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). | ||||
| ### Packages | ||||
| * [Overview](https://repology.org/metapackage/matterbridge/versions) | ||||
|  | ||||
| After Go is setup, download matterbridge to your $GOPATH directory.  | ||||
| ### Building | ||||
| Go 1.9+ is required. Make sure you have [Go](https://golang.org/doc/install) properly installed, including setting up your [GOPATH](https://golang.org/doc/code.html#GOPATH). | ||||
|  | ||||
| After Go is setup, download matterbridge to your $GOPATH directory. | ||||
|  | ||||
| ``` | ||||
| cd $GOPATH | ||||
| @@ -89,16 +144,16 @@ $ ls bin/ | ||||
| matterbridge | ||||
| ``` | ||||
|  | ||||
| # Configuration | ||||
| ## Basic configuration | ||||
| ## Configuration | ||||
| ### Basic configuration | ||||
| See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||
|  | ||||
| ## Advanced configuration | ||||
| ### Advanced configuration | ||||
| * [matterbridge.toml.sample](https://github.com/42wim/matterbridge/blob/master/matterbridge.toml.sample) for documentation and an example. | ||||
|  | ||||
| ## Examples  | ||||
| ### Bridge mattermost (off-topic) - irc (#testing) | ||||
| ``` | ||||
| ### Examples | ||||
| #### Bridge mattermost (off-topic) - irc (#testing) | ||||
| ```toml | ||||
| [irc] | ||||
|     [irc.freenode] | ||||
|     Server="irc.freenode.net:6667" | ||||
| @@ -125,8 +180,8 @@ enable=true | ||||
|     channel="off-topic" | ||||
| ``` | ||||
|  | ||||
| ### Bridge slack (#general) - discord (general) | ||||
| ``` | ||||
| #### Bridge slack (#general) - discord (general) | ||||
| ```toml | ||||
| [slack] | ||||
| [slack.test] | ||||
| Token="yourslacktoken" | ||||
| @@ -153,7 +208,7 @@ RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | ||||
|     channel = "general" | ||||
| ``` | ||||
|  | ||||
| # Running | ||||
| ## Running | ||||
|  | ||||
| See [howto](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) for a step by step walkthrough for creating your configuration. | ||||
|  | ||||
| @@ -169,24 +224,46 @@ Usage of ./matterbridge: | ||||
|         show version | ||||
| ``` | ||||
|  | ||||
| ## Docker | ||||
| Create your matterbridge.toml file locally eg in ```/tmp/matterbridge.toml``` | ||||
| ### Docker | ||||
| Create your matterbridge.toml file locally eg in `/tmp/matterbridge.toml` | ||||
| ``` | ||||
| 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) | ||||
|  | ||||
| # FAQ | ||||
| ## FAQ | ||||
|  | ||||
| See [FAQ](https://github.com/42wim/matterbridge/wiki/FAQ) | ||||
|  | ||||
| Want to tip ?  | ||||
| Want to tip ? | ||||
| * eth: 0xb3f9b5387c66ad6be892bcb7bbc67862f3abc16f | ||||
| * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | ||||
|  | ||||
| # Thanks | ||||
| ## Related projects | ||||
| * [FOSSRIT/infrastructure - roles/matterbridge](https://github.com/FOSSRIT/infrastructure/tree/master/roles/matterbridge) (Ansible role used to automate deployments of Matterbridge) | ||||
| * [matterbridge autoconfig](https://github.com/patcon/matterbridge-autoconfig) | ||||
| * [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||
| * [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||
| * [mattereddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| * [matterlink](https://github.com/elytra/MatterLink) | ||||
| * [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| * [fbridge](https://github.com/VictorNine/fbridge) (Facebook messenger support) | ||||
|  | ||||
| ## Articles | ||||
| * [matterbridge on kubernetes](https://medium.freecodecamp.org/using-kubernetes-to-deploy-a-chat-gateway-or-when-technology-works-like-its-supposed-to-a169a8cd69a3) | ||||
| * https://mattermost.com/blog/connect-irc-to-mattermost/ | ||||
| * https://blog.valvin.fr/2016/09/17/mattermost-et-un-channel-irc-cest-possible/ | ||||
| * https://blog.brightscout.com/top-10-mattermost-integrations/ | ||||
| * http://bencey.co.nz/2018/09/17/bridge/ | ||||
| * https://www.algoo.fr/blog/2018/01/19/recouvrez-votre-liberte-en-quittant-slack-pour-un-mattermost-auto-heberge/ | ||||
| * https://kopano.com/blog/matterbridge-bridging-mattermost-chat/ | ||||
| * https://www.stitcher.com/s/?eid=52382713 | ||||
| * https://daniele.tech/2019/02/how-to-use-matterbridge-to-connect-2-different-slack-workspaces/ | ||||
|  | ||||
| ## Thanks | ||||
| [](https://www.digitalocean.com/) for sponsoring demo/testing droplets. | ||||
|  | ||||
| Matterbridge wouldn't exist without these libraries: | ||||
| @@ -196,10 +273,28 @@ Matterbridge wouldn't exist without these libraries: | ||||
| * gops - https://github.com/google/gops | ||||
| * gozulipbot - https://github.com/ifo/gozulipbot | ||||
| * irc - https://github.com/lrstanley/girc | ||||
| * mattermost - https://github.com/mattermost/platform | ||||
| * mattermost - https://github.com/mattermost/mattermost-server | ||||
| * matrix - https://github.com/matrix-org/gomatrix | ||||
| * sshchat - https://github.com/shazow/ssh-chat | ||||
| * slack - https://github.com/nlopes/slack | ||||
| * steam - https://github.com/Philipp15b/go-steam | ||||
| * telegram - https://github.com/go-telegram-bot-api/telegram-bot-api | ||||
| * xmpp - https://github.com/mattn/go-xmpp | ||||
| * whatsapp - https://github.com/Rhymen/go-whatsapp/ | ||||
| * zulip - https://github.com/ifo/gozulipbot | ||||
| * tengo - https://github.com/d5/tengo | ||||
|  | ||||
| <!-- Links --> | ||||
|  | ||||
|    [mb-gitter]: https://gitter.im/42wim/matterbridge | ||||
|    [mb-irc]: https://webchat.freenode.net/?channels=matterbridgechat | ||||
|    [mb-discord]: https://discord.gg/AkKPtrQ | ||||
|    [mb-matrix]: https://riot.im/app/#/room/#matterbridge:matrix.org | ||||
|    [mb-slack]: https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA | ||||
|    [mb-mattermost]: https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e | ||||
|    [mb-rocketchat]: https://open.rocket.chat/channel/matterbridge | ||||
|    [mb-xmpp]: https://inverse.chat/ | ||||
|    [mb-twitch]: https://www.twitch.tv/matterbridge | ||||
|    [mb-whatsapp]: https://www.whatsapp.com/ | ||||
|    [mb-zulip]: https://matterbridge.zulipchat.com/register/ | ||||
|    [mb-telegram]: https://t.me/Matterbridge | ||||
|   | ||||
| @@ -8,18 +8,18 @@ import ( | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/labstack/echo" | ||||
| 	"github.com/labstack/echo/middleware" | ||||
| 	"github.com/labstack/echo/v4" | ||||
| 	"github.com/labstack/echo/v4/middleware" | ||||
| 	"github.com/zfjagann/golang-ring" | ||||
| ) | ||||
|  | ||||
| type Api struct { | ||||
| type API struct { | ||||
| 	Messages ring.Ring | ||||
| 	sync.RWMutex | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| type ApiMessage struct { | ||||
| type Message struct { | ||||
| 	Text     string `json:"text"` | ||||
| 	Username string `json:"username"` | ||||
| 	UserID   string `json:"userid"` | ||||
| @@ -28,17 +28,20 @@ type ApiMessage struct { | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Api{Config: cfg} | ||||
| 	b := &API{Config: cfg} | ||||
| 	e := echo.New() | ||||
| 	e.HideBanner = true | ||||
| 	e.HidePort = true | ||||
| 	b.Messages = ring.Ring{} | ||||
| 	b.Messages.SetCapacity(b.GetInt("Buffer")) | ||||
| 	if b.GetInt("Buffer") != 0 { | ||||
| 		b.Messages.SetCapacity(b.GetInt("Buffer")) | ||||
| 	} | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) { | ||||
| 			return key == b.GetString("Token"), nil | ||||
| 		})) | ||||
| 	} | ||||
| 	e.GET("/api/health", b.handleHealthcheck) | ||||
| 	e.GET("/api/messages", b.handleMessages) | ||||
| 	e.GET("/api/stream", b.handleStream) | ||||
| 	e.POST("/api/message", b.handlePostMessage) | ||||
| @@ -52,30 +55,34 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Api) Connect() error { | ||||
| func (b *API) Connect() error { | ||||
| 	return nil | ||||
| } | ||||
| func (b *Api) Disconnect() error { | ||||
| func (b *API) Disconnect() error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
| func (b *Api) JoinChannel(channel config.ChannelInfo) error { | ||||
| func (b *API) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *Api) Send(msg config.Message) (string, error) { | ||||
| func (b *API) Send(msg config.Message) (string, error) { | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	b.Messages.Enqueue(&msg) | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Api) handlePostMessage(c echo.Context) error { | ||||
| func (b *API) handleHealthcheck(c echo.Context) error { | ||||
| 	return c.String(http.StatusOK, "OK") | ||||
| } | ||||
|  | ||||
| func (b *API) handlePostMessage(c echo.Context) error { | ||||
| 	message := config.Message{} | ||||
| 	if err := c.Bind(&message); err != nil { | ||||
| 		return err | ||||
| @@ -91,7 +98,7 @@ func (b *Api) handlePostMessage(c echo.Context) error { | ||||
| 	return c.JSON(http.StatusOK, message) | ||||
| } | ||||
|  | ||||
| func (b *Api) handleMessages(c echo.Context) error { | ||||
| func (b *API) handleMessages(c echo.Context) error { | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
| 	c.JSONPretty(http.StatusOK, b.Messages.Values(), " ") | ||||
| @@ -99,23 +106,25 @@ func (b *Api) handleMessages(c echo.Context) error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Api) handleStream(c echo.Context) error { | ||||
| func (b *API) handleStream(c echo.Context) error { | ||||
| 	c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON) | ||||
| 	c.Response().WriteHeader(http.StatusOK) | ||||
| 	closeNotifier := c.Response().CloseNotify() | ||||
| 	greet := config.Message{ | ||||
| 		Event:     config.EventAPIConnected, | ||||
| 		Timestamp: time.Now(), | ||||
| 	} | ||||
| 	if err := json.NewEncoder(c.Response()).Encode(greet); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	c.Response().Flush() | ||||
| 	for { | ||||
| 		select { | ||||
| 		case <-closeNotifier: | ||||
| 			return nil | ||||
| 		default: | ||||
| 			msg := b.Messages.Dequeue() | ||||
| 			if msg != nil { | ||||
| 				if err := json.NewEncoder(c.Response()).Encode(msg); err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 				c.Response().Flush() | ||||
| 		msg := b.Messages.Dequeue() | ||||
| 		if msg != nil { | ||||
| 			if err := json.NewEncoder(c.Response()).Encode(msg); err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			time.Sleep(200 * time.Millisecond) | ||||
| 			c.Response().Flush() | ||||
| 		} | ||||
| 		time.Sleep(200 * time.Millisecond) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,10 +1,11 @@ | ||||
| package bridge | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
|  | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Bridger interface { | ||||
| @@ -16,42 +17,52 @@ type Bridger interface { | ||||
|  | ||||
| type Bridge struct { | ||||
| 	Bridger | ||||
| 	Name     string | ||||
| 	Account  string | ||||
| 	Protocol string | ||||
| 	Channels map[string]config.ChannelInfo | ||||
| 	Joined   map[string]bool | ||||
| 	Log      *log.Entry | ||||
| 	Config   *config.Config | ||||
| 	General  *config.Protocol | ||||
| 	*sync.RWMutex | ||||
|  | ||||
| 	Name           string | ||||
| 	Account        string | ||||
| 	Protocol       string | ||||
| 	Channels       map[string]config.ChannelInfo | ||||
| 	Joined         map[string]bool | ||||
| 	ChannelMembers *config.ChannelMembers | ||||
| 	Log            *logrus.Entry | ||||
| 	Config         config.Config | ||||
| 	General        *config.Protocol | ||||
| } | ||||
|  | ||||
| type Config struct { | ||||
| 	//	General *config.Protocol | ||||
| 	Remote chan config.Message | ||||
| 	Log    *log.Entry | ||||
| 	*Bridge | ||||
|  | ||||
| 	Remote chan config.Message | ||||
| } | ||||
|  | ||||
| // Factory is the factory function to create a bridge | ||||
| type Factory func(*Config) Bridger | ||||
|  | ||||
| func New(bridge *config.Bridge) *Bridge { | ||||
| 	b := new(Bridge) | ||||
| 	b.Channels = make(map[string]config.ChannelInfo) | ||||
| 	accInfo := strings.Split(bridge.Account, ".") | ||||
| 	protocol := accInfo[0] | ||||
| 	name := accInfo[1] | ||||
| 	b.Name = name | ||||
| 	b.Protocol = protocol | ||||
| 	b.Account = bridge.Account | ||||
| 	b.Joined = make(map[string]bool) | ||||
| 	return b | ||||
|  | ||||
| 	return &Bridge{ | ||||
| 		RWMutex:  new(sync.RWMutex), | ||||
| 		Channels: make(map[string]config.ChannelInfo), | ||||
| 		Name:     name, | ||||
| 		Protocol: protocol, | ||||
| 		Account:  bridge.Account, | ||||
| 		Joined:   make(map[string]bool), | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bridge) JoinChannels() error { | ||||
| 	err := b.joinChannels(b.Channels, b.Joined) | ||||
| 	return err | ||||
| 	return b.joinChannels(b.Channels, b.Joined) | ||||
| } | ||||
|  | ||||
| // SetChannelMembers sets the newMembers to the bridge ChannelMembers | ||||
| func (b *Bridge) SetChannelMembers(newMembers *config.ChannelMembers) { | ||||
| 	b.Lock() | ||||
| 	b.ChannelMembers = newMembers | ||||
| 	b.Unlock() | ||||
| } | ||||
|  | ||||
| func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error { | ||||
| @@ -69,36 +80,41 @@ func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetBool(key string) bool { | ||||
| 	if b.Config.GetBool(b.Account + "." + key) { | ||||
| 		return b.Config.GetBool(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetBool(b.Account + "." + key) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetBool("general." + key) | ||||
| 	} | ||||
| 	return b.Config.GetBool("general." + key) | ||||
| 	return val | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetInt(key string) int { | ||||
| 	if b.Config.GetInt(b.Account+"."+key) != 0 { | ||||
| 		return b.Config.GetInt(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetInt(b.Account + "." + key) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetInt("general." + key) | ||||
| 	} | ||||
| 	return b.Config.GetInt("general." + key) | ||||
| 	return val | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetString(key string) string { | ||||
| 	if b.Config.GetString(b.Account+"."+key) != "" { | ||||
| 		return b.Config.GetString(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetString(b.Account + "." + key) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetString("general." + key) | ||||
| 	} | ||||
| 	return b.Config.GetString("general." + key) | ||||
| 	return val | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetStringSlice(key string) []string { | ||||
| 	if len(b.Config.GetStringSlice(b.Account+"."+key)) != 0 { | ||||
| 		return b.Config.GetStringSlice(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetStringSlice(b.Account + "." + key) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetStringSlice("general." + key) | ||||
| 	} | ||||
| 	return b.Config.GetStringSlice("general." + key) | ||||
| 	return val | ||||
| } | ||||
|  | ||||
| func (b *Bridge) GetStringSlice2D(key string) [][]string { | ||||
| 	if len(b.Config.GetStringSlice2D(b.Account+"."+key)) != 0 { | ||||
| 		return b.Config.GetStringSlice2D(b.Account + "." + key) | ||||
| 	val, ok := b.Config.GetStringSlice2D(b.Account + "." + key) | ||||
| 	if !ok { | ||||
| 		val, _ = b.Config.GetStringSlice2D("general." + key) | ||||
| 	} | ||||
| 	return b.Config.GetStringSlice2D("general." + key) | ||||
| 	return val | ||||
| } | ||||
|   | ||||
| @@ -2,26 +2,28 @@ package config | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"os" | ||||
| 	"io/ioutil" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/fsnotify/fsnotify" | ||||
| 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/spf13/viper" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	EVENT_JOIN_LEAVE        = "join_leave" | ||||
| 	EVENT_TOPIC_CHANGE      = "topic_change" | ||||
| 	EVENT_FAILURE           = "failure" | ||||
| 	EVENT_FILE_FAILURE_SIZE = "file_failure_size" | ||||
| 	EVENT_AVATAR_DOWNLOAD   = "avatar_download" | ||||
| 	EVENT_REJOIN_CHANNELS   = "rejoin_channels" | ||||
| 	EVENT_USER_ACTION       = "user_action" | ||||
| 	EVENT_MSG_DELETE        = "msg_delete" | ||||
| 	EventJoinLeave         = "join_leave" | ||||
| 	EventTopicChange       = "topic_change" | ||||
| 	EventFailure           = "failure" | ||||
| 	EventFileFailureSize   = "file_failure_size" | ||||
| 	EventAvatarDownload    = "avatar_download" | ||||
| 	EventRejoinChannels    = "rejoin_channels" | ||||
| 	EventUserAction        = "user_action" | ||||
| 	EventMsgDelete         = "msg_delete" | ||||
| 	EventAPIConnected      = "api_connected" | ||||
| 	EventUserTyping        = "user_typing" | ||||
| 	EventGetChannelMembers = "get_channel_members" | ||||
| ) | ||||
|  | ||||
| type Message struct { | ||||
| @@ -34,6 +36,7 @@ type Message struct { | ||||
| 	Event     string    `json:"event"` | ||||
| 	Protocol  string    `json:"protocol"` | ||||
| 	Gateway   string    `json:"gateway"` | ||||
| 	ParentID  string    `json:"parent_id"` | ||||
| 	Timestamp time.Time `json:"timestamp"` | ||||
| 	ID        string    `json:"id"` | ||||
| 	Extra     map[string][]interface{} | ||||
| @@ -58,6 +61,16 @@ type ChannelInfo struct { | ||||
| 	Options     ChannelOptions | ||||
| } | ||||
|  | ||||
| type ChannelMember struct { | ||||
| 	Username    string | ||||
| 	Nick        string | ||||
| 	UserID      string | ||||
| 	ChannelID   string | ||||
| 	ChannelName string | ||||
| } | ||||
|  | ||||
| type ChannelMembers []ChannelMember | ||||
|  | ||||
| type Protocol struct { | ||||
| 	AuthCode               string // steam | ||||
| 	BindAddress            string // mattermost, slack // DEPRECATED | ||||
| @@ -69,6 +82,7 @@ type Protocol struct { | ||||
| 	EditSuffix             string // mattermost, slack, discord, telegram, gitter | ||||
| 	EditDisable            bool   // mattermost, slack, discord, telegram, gitter | ||||
| 	IconURL                string // mattermost, slack | ||||
| 	IgnoreFailureOnStart   bool   // general | ||||
| 	IgnoreNicks            string // all protocols | ||||
| 	IgnoreMessages         string // all protocols | ||||
| 	Jid                    string // xmpp | ||||
| @@ -97,6 +111,7 @@ type Protocol struct { | ||||
| 	NoTLS                  bool       // mattermost | ||||
| 	Password               string     // IRC,mattermost,XMPP,matrix | ||||
| 	PrefixMessagesWithNick bool       // mattemost, slack | ||||
| 	PreserveThreading      bool       // slack | ||||
| 	Protocol               string     // all protocols | ||||
| 	QuoteDisable           bool       // telegram | ||||
| 	QuoteFormat            string     // telegram | ||||
| @@ -104,12 +119,16 @@ type Protocol struct { | ||||
| 	ReplaceMessages        [][]string // all protocols | ||||
| 	ReplaceNicks           [][]string // all protocols | ||||
| 	RemoteNickFormat       string     // all protocols | ||||
| 	RunCommands            []string   // irc | ||||
| 	Server                 string     // IRC,mattermost,XMPP,discord | ||||
| 	ShowJoinPart           bool       // all protocols | ||||
| 	ShowTopicChange        bool       // slack | ||||
| 	ShowUserTyping         bool       // slack | ||||
| 	ShowEmbeds             bool       // discord | ||||
| 	SkipTLSVerify          bool       // IRC, mattermost | ||||
| 	StripNick              bool       // all protocols | ||||
| 	SyncTopic              bool       // slack | ||||
| 	TengoModifyMessage     string     // general | ||||
| 	Team                   string     // mattermost | ||||
| 	Token                  string     // gitter, slack, discord, api | ||||
| 	Topic                  string     // zulip | ||||
| @@ -117,17 +136,18 @@ type Protocol struct { | ||||
| 	UseAPI                 bool       // mattermost, slack | ||||
| 	UseSASL                bool       // IRC | ||||
| 	UseTLS                 bool       // IRC | ||||
| 	UseDiscriminator       bool       // discord | ||||
| 	UseFirstName           bool       // telegram | ||||
| 	UseUserName            bool       // discord | ||||
| 	UseInsecureURL         bool       // telegram | ||||
| 	WebhookBindAddress     string     // mattermost, slack | ||||
| 	WebhookURL             string     // mattermost, slack | ||||
| 	WebhookUse             string     // mattermost, slack, discord | ||||
| } | ||||
|  | ||||
| type ChannelOptions struct { | ||||
| 	Key        string // irc, xmpp | ||||
| 	WebhookURL string // discord | ||||
| 	Topic      string // zulip | ||||
| } | ||||
|  | ||||
| type Bridge struct { | ||||
| @@ -152,127 +172,137 @@ type SameChannelGateway struct { | ||||
| 	Accounts []string | ||||
| } | ||||
|  | ||||
| type ConfigValues struct { | ||||
| 	Api                map[string]Protocol | ||||
| 	Irc                map[string]Protocol | ||||
| type BridgeValues struct { | ||||
| 	API                map[string]Protocol | ||||
| 	IRC                map[string]Protocol | ||||
| 	Mattermost         map[string]Protocol | ||||
| 	Matrix             map[string]Protocol | ||||
| 	Slack              map[string]Protocol | ||||
| 	SlackLegacy        map[string]Protocol | ||||
| 	Steam              map[string]Protocol | ||||
| 	Gitter             map[string]Protocol | ||||
| 	Xmpp               map[string]Protocol | ||||
| 	XMPP               map[string]Protocol | ||||
| 	Discord            map[string]Protocol | ||||
| 	Telegram           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 | ||||
| 	General            Protocol | ||||
| 	Gateway            []Gateway | ||||
| 	SameChannelGateway []SameChannelGateway | ||||
| } | ||||
|  | ||||
| type Config struct { | ||||
| 	v *viper.Viper | ||||
| 	*ConfigValues | ||||
| 	sync.RWMutex | ||||
| type Config interface { | ||||
| 	BridgeValues() *BridgeValues | ||||
| 	GetBool(key string) (bool, bool) | ||||
| 	GetInt(key string) (int, bool) | ||||
| 	GetString(key string) (string, bool) | ||||
| 	GetStringSlice(key string) ([]string, bool) | ||||
| 	GetStringSlice2D(key string) ([][]string, bool) | ||||
| } | ||||
|  | ||||
| func NewConfig(cfgfile string) *Config { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||
| 	flog := log.WithFields(log.Fields{"prefix": "config"}) | ||||
| 	var cfg ConfigValues | ||||
| 	viper.SetConfigType("toml") | ||||
| type config struct { | ||||
| 	sync.RWMutex | ||||
|  | ||||
| 	logger *logrus.Entry | ||||
| 	v      *viper.Viper | ||||
| 	cv     *BridgeValues | ||||
| } | ||||
|  | ||||
| // NewConfig instantiates a new configuration based on the specified configuration file path. | ||||
| func NewConfig(rootLogger *logrus.Logger, cfgfile string) Config { | ||||
| 	logger := rootLogger.WithFields(logrus.Fields{"prefix": "config"}) | ||||
|  | ||||
| 	viper.SetConfigFile(cfgfile) | ||||
| 	viper.SetEnvPrefix("matterbridge") | ||||
| 	viper.AddConfigPath(".") | ||||
| 	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_")) | ||||
| 	viper.AutomaticEnv() | ||||
| 	f, err := os.Open(cfgfile) | ||||
| 	input, err := ioutil.ReadFile(cfgfile) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 		logger.Fatalf("Failed to read configuration file: %#v", err) | ||||
| 	} | ||||
| 	err = viper.ReadConfig(f) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 	} | ||||
| 	err = viper.Unmarshal(&cfg) | ||||
| 	if err != nil { | ||||
| 		log.Fatal("blah", err) | ||||
| 	} | ||||
| 	mycfg := new(Config) | ||||
| 	mycfg.v = viper.GetViper() | ||||
| 	if cfg.General.MediaDownloadSize == 0 { | ||||
| 		cfg.General.MediaDownloadSize = 1000000 | ||||
|  | ||||
| 	mycfg := newConfigFromString(logger, input) | ||||
| 	if mycfg.cv.General.MediaDownloadSize == 0 { | ||||
| 		mycfg.cv.General.MediaDownloadSize = 1000000 | ||||
| 	} | ||||
| 	viper.WatchConfig() | ||||
| 	viper.OnConfigChange(func(e fsnotify.Event) { | ||||
| 		flog.Println("Config file changed:", e.Name) | ||||
| 		logger.Println("Config file changed:", e.Name) | ||||
| 	}) | ||||
|  | ||||
| 	mycfg.ConfigValues = &cfg | ||||
| 	return mycfg | ||||
| } | ||||
|  | ||||
| func NewConfigFromString(input []byte) *Config { | ||||
| 	var cfg ConfigValues | ||||
| // NewConfigFromString instantiates a new configuration based on the specified string. | ||||
| func NewConfigFromString(rootLogger *logrus.Logger, input []byte) Config { | ||||
| 	logger := rootLogger.WithFields(logrus.Fields{"prefix": "config"}) | ||||
| 	return newConfigFromString(logger, input) | ||||
| } | ||||
|  | ||||
| func newConfigFromString(logger *logrus.Entry, input []byte) *config { | ||||
| 	viper.SetConfigType("toml") | ||||
| 	err := viper.ReadConfig(bytes.NewBuffer(input)) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 	viper.SetEnvPrefix("matterbridge") | ||||
| 	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_")) | ||||
| 	viper.AutomaticEnv() | ||||
|  | ||||
| 	if err := viper.ReadConfig(bytes.NewBuffer(input)); err != nil { | ||||
| 		logger.Fatalf("Failed to parse the configuration: %#v", err) | ||||
| 	} | ||||
| 	err = viper.Unmarshal(&cfg) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
|  | ||||
| 	cfg := &BridgeValues{} | ||||
| 	if err := viper.Unmarshal(cfg); err != nil { | ||||
| 		logger.Fatalf("Failed to load the configuration: %#v", err) | ||||
| 	} | ||||
| 	return &config{ | ||||
| 		logger: logger, | ||||
| 		v:      viper.GetViper(), | ||||
| 		cv:     cfg, | ||||
| 	} | ||||
| 	mycfg := new(Config) | ||||
| 	mycfg.v = viper.GetViper() | ||||
| 	mycfg.ConfigValues = &cfg | ||||
| 	return mycfg | ||||
| } | ||||
|  | ||||
| func (c *Config) GetBool(key string) bool { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	//	log.Debugf("getting bool %s = %#v", key, c.v.GetBool(key)) | ||||
| 	return c.v.GetBool(key) | ||||
| func (c *config) BridgeValues() *BridgeValues { | ||||
| 	return c.cv | ||||
| } | ||||
|  | ||||
| func (c *Config) GetInt(key string) int { | ||||
| func (c *config) GetBool(key string) (bool, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	//	log.Debugf("getting int %s = %d", key, c.v.GetInt(key)) | ||||
| 	return c.v.GetInt(key) | ||||
| 	return c.v.GetBool(key), c.v.IsSet(key) | ||||
| } | ||||
|  | ||||
| func (c *Config) GetString(key string) string { | ||||
| func (c *config) GetInt(key string) (int, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	//	log.Debugf("getting String %s = %s", key, c.v.GetString(key)) | ||||
| 	return c.v.GetString(key) | ||||
| 	return c.v.GetInt(key), c.v.IsSet(key) | ||||
| } | ||||
|  | ||||
| func (c *Config) GetStringSlice(key string) []string { | ||||
| func (c *config) GetString(key string) (string, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	// log.Debugf("getting StringSlice %s = %#v", key, c.v.GetStringSlice(key)) | ||||
| 	return c.v.GetStringSlice(key) | ||||
| 	return c.v.GetString(key), c.v.IsSet(key) | ||||
| } | ||||
|  | ||||
| func (c *Config) GetStringSlice2D(key string) [][]string { | ||||
| func (c *config) GetStringSlice(key string) ([]string, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
| 	result := [][]string{} | ||||
| 	if res, ok := c.v.Get(key).([]interface{}); ok { | ||||
| 		for _, entry := range res { | ||||
| 			result2 := []string{} | ||||
| 			for _, entry2 := range entry.([]interface{}) { | ||||
| 				result2 = append(result2, entry2.(string)) | ||||
| 			} | ||||
| 			result = append(result, result2) | ||||
| 	return c.v.GetStringSlice(key), c.v.IsSet(key) | ||||
| } | ||||
|  | ||||
| func (c *config) GetStringSlice2D(key string) ([][]string, bool) { | ||||
| 	c.RLock() | ||||
| 	defer c.RUnlock() | ||||
|  | ||||
| 	res, ok := c.v.Get(key).([]interface{}) | ||||
| 	if !ok { | ||||
| 		return nil, false | ||||
| 	} | ||||
| 	var result [][]string | ||||
| 	for _, entry := range res { | ||||
| 		result2 := []string{} | ||||
| 		for _, entry2 := range entry.([]interface{}) { | ||||
| 			result2 = append(result2, entry2.(string)) | ||||
| 		} | ||||
| 		return result | ||||
| 		result = append(result, result2) | ||||
| 	} | ||||
| 	return result | ||||
| 	return result, true | ||||
| } | ||||
|  | ||||
| func GetIconURL(msg *Message, iconURL string) string { | ||||
| @@ -284,3 +314,45 @@ func GetIconURL(msg *Message, iconURL string) string { | ||||
| 	iconURL = strings.Replace(iconURL, "{PROTOCOL}", protocol, -1) | ||||
| 	return iconURL | ||||
| } | ||||
|  | ||||
| type TestConfig struct { | ||||
| 	Config | ||||
|  | ||||
| 	Overrides map[string]interface{} | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetBool(key string) (bool, bool) { | ||||
| 	val, ok := c.Overrides[key] | ||||
| 	if ok { | ||||
| 		return val.(bool), true | ||||
| 	} | ||||
| 	return c.Config.GetBool(key) | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetInt(key string) (int, bool) { | ||||
| 	if val, ok := c.Overrides[key]; ok { | ||||
| 		return val.(int), true | ||||
| 	} | ||||
| 	return c.Config.GetInt(key) | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetString(key string) (string, bool) { | ||||
| 	if val, ok := c.Overrides[key]; ok { | ||||
| 		return val.(string), true | ||||
| 	} | ||||
| 	return c.Config.GetString(key) | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetStringSlice(key string) ([]string, bool) { | ||||
| 	if val, ok := c.Overrides[key]; ok { | ||||
| 		return val.([]string), true | ||||
| 	} | ||||
| 	return c.Config.GetStringSlice(key) | ||||
| } | ||||
|  | ||||
| func (c *TestConfig) GetStringSlice2D(key string) ([][]string, bool) { | ||||
| 	if val, ok := c.Overrides[key]; ok { | ||||
| 		return val.([][]string), true | ||||
| 	} | ||||
| 	return c.Config.GetStringSlice2D(key) | ||||
| } | ||||
|   | ||||
| @@ -2,36 +2,44 @@ package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/matterbridge/discordgo" | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| const MessageLength = 1950 | ||||
|  | ||||
| type Bdiscord struct { | ||||
| 	c              *discordgo.Session | ||||
| 	Channels       []*discordgo.Channel | ||||
| 	Nick           string | ||||
| 	UseChannelID   bool | ||||
| 	userMemberMap  map[string]*discordgo.Member | ||||
| 	guildID        string | ||||
| 	webhookID      string | ||||
| 	webhookToken   string | ||||
| 	channelInfoMap map[string]*config.ChannelInfo | ||||
| 	sync.RWMutex | ||||
| 	*bridge.Config | ||||
|  | ||||
| 	c *discordgo.Session | ||||
|  | ||||
| 	nick            string | ||||
| 	useChannelID    bool | ||||
| 	guildID         string | ||||
| 	webhookID       string | ||||
| 	webhookToken    string | ||||
| 	canEditWebhooks bool | ||||
|  | ||||
| 	channelsMutex  sync.RWMutex | ||||
| 	channels       []*discordgo.Channel | ||||
| 	channelInfoMap map[string]*config.ChannelInfo | ||||
|  | ||||
| 	membersMutex  sync.RWMutex | ||||
| 	userMemberMap map[string]*discordgo.Member | ||||
| 	nickMemberMap map[string]*discordgo.Member | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bdiscord{Config: cfg} | ||||
| 	b.userMemberMap = make(map[string]*discordgo.Member) | ||||
| 	b.nickMemberMap = make(map[string]*discordgo.Member) | ||||
| 	b.channelInfoMap = make(map[string]*config.ChannelInfo) | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		b.Log.Debug("Configuring Discord Incoming Webhook") | ||||
| @@ -42,7 +50,8 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
|  | ||||
| func (b *Bdiscord) Connect() error { | ||||
| 	var err error | ||||
| 	var token string | ||||
| 	var guildFound bool | ||||
| 	token := b.GetString("Token") | ||||
| 	b.Log.Info("Connecting") | ||||
| 	if b.GetString("WebhookURL") == "" { | ||||
| 		b.Log.Info("Connecting using token") | ||||
| @@ -52,6 +61,11 @@ func (b *Bdiscord) Connect() error { | ||||
| 	if !strings.HasPrefix(b.GetString("Token"), "Bot ") { | ||||
| 		token = "Bot " + b.GetString("Token") | ||||
| 	} | ||||
| 	// if we have a User token, remove the `Bot` prefix | ||||
| 	if strings.HasPrefix(b.GetString("Token"), "User ") { | ||||
| 		token = strings.Replace(b.GetString("Token"), "User ", "", -1) | ||||
| 	} | ||||
|  | ||||
| 	b.c, err = discordgo.New(token) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| @@ -61,6 +75,8 @@ func (b *Bdiscord) Connect() error { | ||||
| 	b.c.AddHandler(b.memberUpdate) | ||||
| 	b.c.AddHandler(b.messageUpdate) | ||||
| 	b.c.AddHandler(b.messageDelete) | ||||
| 	b.c.AddHandler(b.memberAdd) | ||||
| 	b.c.AddHandler(b.memberRemove) | ||||
| 	err = b.c.Open() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| @@ -73,18 +89,76 @@ func (b *Bdiscord) Connect() error { | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Nick = userinfo.Username | ||||
| 	serverName := strings.Replace(b.GetString("Server"), "ID:", "", -1) | ||||
| 	b.nick = userinfo.Username | ||||
| 	b.channelsMutex.Lock() | ||||
| 	for _, guild := range guilds { | ||||
| 		if guild.Name == b.GetString("Server") { | ||||
| 			b.Channels, err = b.c.GuildChannels(guild.ID) | ||||
| 		if guild.Name == serverName || guild.ID == serverName { | ||||
| 			b.channels, err = b.c.GuildChannels(guild.ID) | ||||
| 			b.guildID = guild.ID | ||||
| 			guildFound = true | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	for _, channel := range b.Channels { | ||||
| 		b.Log.Debugf("found channel %#v", channel) | ||||
| 	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 { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.channelsMutex.RLock() | ||||
| 	if b.GetString("WebhookURL") == "" { | ||||
| 		for _, channel := range b.channels { | ||||
| 			b.Log.Debugf("found channel %#v", channel) | ||||
| 		} | ||||
| 	} else { | ||||
| 		b.canEditWebhooks = true | ||||
| 		for _, channel := range b.channels { | ||||
| 			b.Log.Debugf("found channel %#v; verifying PermissionManageWebhooks", channel) | ||||
| 			perms, permsErr := b.c.State.UserChannelPermissions(userinfo.ID, channel.ID) | ||||
| 			manageWebhooks := discordgo.PermissionManageWebhooks | ||||
| 			if permsErr != nil || perms&manageWebhooks != manageWebhooks { | ||||
| 				b.Log.Warnf("Can't manage webhooks in channel \"%s\"", channel.Name) | ||||
| 				b.canEditWebhooks = false | ||||
| 			} | ||||
| 		} | ||||
| 		if b.canEditWebhooks { | ||||
| 			b.Log.Info("Can manage webhooks; will edit channel for global webhook on send") | ||||
| 		} else { | ||||
| 			b.Log.Warn("Can't manage webhooks; won't edit channel for global webhook on send") | ||||
| 		} | ||||
| 	} | ||||
| 	b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	// Obtaining guild members and initializing nickname mapping. | ||||
| 	b.membersMutex.Lock() | ||||
| 	defer b.membersMutex.Unlock() | ||||
| 	members, err := b.c.GuildMembers(b.guildID, "", 1000) | ||||
| 	if err != nil { | ||||
| 		b.Log.Error("Error obtaining server members: ", err) | ||||
| 		return err | ||||
| 	} | ||||
| 	for _, member := range members { | ||||
| 		if member == nil { | ||||
| 			b.Log.Warnf("Skipping missing information for a user.") | ||||
| 			continue | ||||
| 		} | ||||
| 		b.userMemberMap[member.User.ID] = member | ||||
| 		b.nickMemberMap[member.User.Username] = member | ||||
| 		if member.Nick != "" { | ||||
| 			b.nickMemberMap[member.Nick] = member | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @@ -94,10 +168,13 @@ func (b *Bdiscord) Disconnect() error { | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	b.channelsMutex.Lock() | ||||
| 	defer b.channelsMutex.Unlock() | ||||
|  | ||||
| 	b.channelInfoMap[channel.ID] = &channel | ||||
| 	idcheck := strings.Split(channel.Name, "ID:") | ||||
| 	if len(idcheck) > 1 { | ||||
| 		b.UseChannelID = true | ||||
| 		b.useChannelID = true | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @@ -111,32 +188,42 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EVENT_USER_ACTION { | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "_" + msg.Text + "_" | ||||
| 	} | ||||
|  | ||||
| 	// use initial webhook | ||||
| 	// use initial webhook configured for the entire Discord account | ||||
| 	isGlobalWebhook := true | ||||
| 	wID := b.webhookID | ||||
| 	wToken := b.webhookToken | ||||
|  | ||||
| 	// check if have a channel specific webhook | ||||
| 	b.channelsMutex.RLock() | ||||
| 	if ci, ok := b.channelInfoMap[msg.Channel+b.Account]; ok { | ||||
| 		if ci.Options.WebhookURL != "" { | ||||
| 			wID, wToken = b.splitURL(ci.Options.WebhookURL) | ||||
| 			isGlobalWebhook = false | ||||
| 		} | ||||
| 	} | ||||
| 	b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	// Use webhook to send the message | ||||
| 	if wID != "" { | ||||
| 		// skip events | ||||
| 		if msg.Event != "" && msg.Event != config.EVENT_JOIN_LEAVE && msg.Event != config.EVENT_TOPIC_CHANGE { | ||||
| 		if msg.Event != "" && msg.Event != config.EventJoinLeave && msg.Event != config.EventTopicChange { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		b.Log.Debugf("Broadcasting using Webhook") | ||||
| 		for _, f := range msg.Extra["file"] { | ||||
| 			fi := f.(config.FileInfo) | ||||
| 			if fi.Comment != "" { | ||||
| 				msg.Text += fi.Comment + ": " | ||||
| 			} | ||||
| 			if fi.URL != "" { | ||||
| 				msg.Text += " " + fi.URL | ||||
| 				msg.Text = fi.URL | ||||
| 				if fi.Comment != "" { | ||||
| 					msg.Text = fi.Comment + ": " + fi.URL | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		// skip empty messages | ||||
| @@ -145,6 +232,24 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 		} | ||||
|  | ||||
| 		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, | ||||
| @@ -160,7 +265,7 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("Broadcasting using token (API)") | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| @@ -172,7 +277,9 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			rmsg.Text = helper.ClipMessage(rmsg.Text, MessageLength) | ||||
| 			b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text) | ||||
| 			if _, err := b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text); err != nil { | ||||
| 				b.Log.Errorf("Could not send message %#v: %v", rmsg, err) | ||||
| 			} | ||||
| 		} | ||||
| 		// check if we have files to upload (from slack, telegram or mattermost) | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| @@ -181,6 +288,8 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	msg.Text = helper.ClipMessage(msg.Text, MessageLength) | ||||
| 	msg.Text = b.replaceUserMentions(msg.Text) | ||||
|  | ||||
| 	// Edit message | ||||
| 	if msg.ID != "" { | ||||
| 		_, err := b.c.ChannelMessageEdit(channelID, msg.ID, msg.Username+msg.Text) | ||||
| @@ -195,202 +304,15 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 	return res.ID, err | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageDelete(s *discordgo.Session, m *discordgo.MessageDelete) { | ||||
| 	rmsg := config.Message{Account: b.Account, ID: m.ID, Event: config.EVENT_MSG_DELETE, Text: config.EVENT_MSG_DELETE} | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.UseChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
| 	b.Log.Debugf("<= Sending message from %s to gateway", b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | ||||
| 	if b.GetBool("EditDisable") { | ||||
| 		return | ||||
| 	} | ||||
| 	// only when message is actually edited | ||||
| 	if m.Message.EditedTimestamp != "" { | ||||
| 		b.Log.Debugf("Sending edit message") | ||||
| 		m.Content = m.Content + b.GetString("EditSuffix") | ||||
| 		b.messageCreate(s, (*discordgo.MessageCreate)(m)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { | ||||
| 	var err error | ||||
|  | ||||
| 	// not relay our own messages | ||||
| 	if m.Author.Username == b.Nick { | ||||
| 		return | ||||
| 	} | ||||
| 	// if using webhooks, do not relay if it's ours | ||||
| 	if b.useWebhook() && m.Author.Bot && b.isWebhookID(m.Author.ID) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// add the url of the attachments to content | ||||
| 	if len(m.Attachments) > 0 { | ||||
| 		for _, attach := range m.Attachments { | ||||
| 			m.Content = m.Content + "\n" + attach.URL | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", UserID: m.Author.ID, ID: m.ID} | ||||
|  | ||||
| 	if m.Content != "" { | ||||
| 		b.Log.Debugf("== Receiving event %#v", m.Message) | ||||
| 		m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||
| 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | ||||
| 		rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("ContentWithMoreMentionsReplaced failed: %s", err) | ||||
| 			rmsg.Text = m.ContentWithMentionsReplaced() | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// set channel name | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.UseChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
|  | ||||
| 	// set username | ||||
| 	if !b.GetBool("UseUserName") { | ||||
| 		rmsg.Username = b.getNick(m.Author) | ||||
| 	} else { | ||||
| 		rmsg.Username = m.Author.Username | ||||
| 	} | ||||
|  | ||||
| 	// if we have embedded content add it to text | ||||
| 	if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil { | ||||
| 		for _, embed := range m.Message.Embeds { | ||||
| 			rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// no empty messages | ||||
| 	if rmsg.Text == "" { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// do we have a /me action | ||||
| 	var ok bool | ||||
| 	rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||
| 	if ok { | ||||
| 		rmsg.Event = config.EVENT_USER_ACTION | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||
| 	b.Lock() | ||||
| 	if _, ok := b.userMemberMap[m.Member.User.ID]; ok { | ||||
| 		b.Log.Debugf("%s: memberupdate: user %s (nick %s) changes nick to %s", b.Account, m.Member.User.Username, b.userMemberMap[m.Member.User.ID].Nick, m.Member.Nick) | ||||
| 	} | ||||
| 	b.userMemberMap[m.Member.User.ID] = m.Member | ||||
| 	b.Unlock() | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getNick(user *discordgo.User) string { | ||||
| 	var err error | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
| 	if _, ok := b.userMemberMap[user.ID]; ok { | ||||
| 		if b.userMemberMap[user.ID] != nil { | ||||
| 			if b.userMemberMap[user.ID].Nick != "" { | ||||
| 				// only return if nick is set | ||||
| 				return b.userMemberMap[user.ID].Nick | ||||
| 			} | ||||
| 			// otherwise return username | ||||
| 			return user.Username | ||||
| 		} | ||||
| 	} | ||||
| 	// if we didn't find nick, search for it | ||||
| 	member, err := b.c.GuildMember(b.guildID, user.ID) | ||||
| 	if err != nil { | ||||
| 		return user.Username | ||||
| 	} | ||||
| 	b.userMemberMap[user.ID] = member | ||||
| 	// only return if nick is set | ||||
| 	if b.userMemberMap[user.ID].Nick != "" { | ||||
| 		return b.userMemberMap[user.ID].Nick | ||||
| 	} | ||||
| 	return user.Username | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelID(name string) string { | ||||
| 	idcheck := strings.Split(name, "ID:") | ||||
| 	if len(idcheck) > 1 { | ||||
| 		return idcheck[1] | ||||
| 	} | ||||
| 	for _, channel := range b.Channels { | ||||
| 		if channel.Name == name { | ||||
| 			return channel.ID | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelName(id string) string { | ||||
| 	for _, channel := range b.Channels { | ||||
| 		if channel.ID == id { | ||||
| 			return channel.Name | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceChannelMentions(text string) string { | ||||
| 	var err error | ||||
| 	re := regexp.MustCompile("<#[0-9]+>") | ||||
| 	text = re.ReplaceAllStringFunc(text, func(m string) string { | ||||
| 		channel := b.getChannelName(m[2 : len(m)-1]) | ||||
| 		// if at first don't succeed, try again | ||||
| 		if channel == "" { | ||||
| 			b.Channels, err = b.c.GuildChannels(b.guildID) | ||||
| 			if err != nil { | ||||
| 				return "#unknownchannel" | ||||
| 			} | ||||
| 			channel = b.getChannelName(m[2 : len(m)-1]) | ||||
| 			return "#" + channel | ||||
| 		} | ||||
| 		return "#" + channel | ||||
| 	}) | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||
| 		return strings.Replace(text, "_", "", -1), true | ||||
| 	} | ||||
| 	return text, false | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) stripCustomoji(text string) string { | ||||
| 	// <:doge:302803592035958784> | ||||
| 	re := regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||
| 	return re.ReplaceAllString(text, `$1`) | ||||
| } | ||||
|  | ||||
| // splitURL splits a webhookURL and returns the id and token | ||||
| func (b *Bdiscord) splitURL(url string) (string, string) { | ||||
| 	webhookURLSplit := strings.Split(url, "/") | ||||
| 	if len(webhookURLSplit) != 7 { | ||||
| 		b.Log.Fatalf("%s is no correct discord WebhookURL", url) | ||||
| 	} | ||||
| 	return webhookURLSplit[len(webhookURLSplit)-2], webhookURLSplit[len(webhookURLSplit)-1] | ||||
| } | ||||
|  | ||||
| // useWebhook returns true if we have a webhook defined somewhere | ||||
| func (b *Bdiscord) useWebhook() bool { | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, channel := range b.channelInfoMap { | ||||
| 		if channel.Options.WebhookURL != "" { | ||||
| 			return true | ||||
| @@ -407,6 +329,10 @@ func (b *Bdiscord) isWebhookID(id string) bool { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, channel := range b.channelInfoMap { | ||||
| 		if channel.Options.WebhookURL != "" { | ||||
| 			wID, _ := b.splitURL(channel.Options.WebhookURL) | ||||
| @@ -423,9 +349,16 @@ func (b *Bdiscord) handleUploadFile(msg *config.Message, channelID string) (stri | ||||
| 	var err error | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		files := []*discordgo.File{} | ||||
| 		files = append(files, &discordgo.File{fi.Name, "", bytes.NewReader(*fi.Data)}) | ||||
| 		_, err = b.c.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{Content: msg.Username + fi.Comment, Files: files}) | ||||
| 		file := discordgo.File{ | ||||
| 			Name:        fi.Name, | ||||
| 			ContentType: "", | ||||
| 			Reader:      bytes.NewReader(*fi.Data), | ||||
| 		} | ||||
| 		m := discordgo.MessageSend{ | ||||
| 			Content: msg.Username + fi.Comment, | ||||
| 			Files:   []*discordgo.File{&file}, | ||||
| 		} | ||||
| 		_, err = b.c.ChannelMessageSendComplex(channelID, &m) | ||||
| 		if err != nil { | ||||
| 			return "", fmt.Errorf("file upload failed: %#v", err) | ||||
| 		} | ||||
|   | ||||
							
								
								
									
										170
									
								
								bridge/discord/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,170 @@ | ||||
| package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| 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.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.useChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
| 	b.Log.Debugf("<= Sending message from %s to gateway", b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { //nolint:unparam | ||||
| 	if b.GetBool("EditDisable") { | ||||
| 		return | ||||
| 	} | ||||
| 	// only when message is actually edited | ||||
| 	if m.Message.EditedTimestamp != "" { | ||||
| 		b.Log.Debugf("Sending edit message") | ||||
| 		m.Content += b.GetString("EditSuffix") | ||||
| 		b.messageCreate(s, (*discordgo.MessageCreate)(m)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) { //nolint:unparam | ||||
| 	var err error | ||||
|  | ||||
| 	// not relay our own messages | ||||
| 	if m.Author.Username == b.nick { | ||||
| 		return | ||||
| 	} | ||||
| 	// if using webhooks, do not relay if it's ours | ||||
| 	if b.useWebhook() && m.Author.Bot && b.isWebhookID(m.Author.ID) { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// add the url of the attachments to content | ||||
| 	if len(m.Attachments) > 0 { | ||||
| 		for _, attach := range m.Attachments { | ||||
| 			m.Content = m.Content + "\n" + attach.URL | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg", UserID: m.Author.ID, ID: m.ID} | ||||
|  | ||||
| 	if m.Content != "" { | ||||
| 		b.Log.Debugf("== Receiving event %#v", m.Message) | ||||
| 		m.Message.Content = b.stripCustomoji(m.Message.Content) | ||||
| 		m.Message.Content = b.replaceChannelMentions(m.Message.Content) | ||||
| 		rmsg.Text, err = m.ContentWithMoreMentionsReplaced(b.c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("ContentWithMoreMentionsReplaced failed: %s", err) | ||||
| 			rmsg.Text = m.ContentWithMentionsReplaced() | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// set channel name | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.useChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
|  | ||||
| 	// set username | ||||
| 	if !b.GetBool("UseUserName") { | ||||
| 		rmsg.Username = b.getNick(m.Author) | ||||
| 	} else { | ||||
| 		rmsg.Username = m.Author.Username | ||||
| 		if b.GetBool("UseDiscriminator") { | ||||
| 			rmsg.Username += "#" + m.Author.Discriminator | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// if we have embedded content add it to text | ||||
| 	if b.GetBool("ShowEmbeds") && m.Message.Embeds != nil { | ||||
| 		for _, embed := range m.Message.Embeds { | ||||
| 			rmsg.Text = rmsg.Text + "embed: " + embed.Title + " - " + embed.Description + " - " + embed.URL + "\n" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// no empty messages | ||||
| 	if rmsg.Text == "" { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// do we have a /me action | ||||
| 	var ok bool | ||||
| 	rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||
| 	if ok { | ||||
| 		rmsg.Event = config.EventUserAction | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", m.Author.Username, b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) memberUpdate(s *discordgo.Session, m *discordgo.GuildMemberUpdate) { | ||||
| 	if m.Member == nil { | ||||
| 		b.Log.Warnf("Received member update with no member information: %#v", m) | ||||
| 	} | ||||
|  | ||||
| 	b.membersMutex.Lock() | ||||
| 	defer b.membersMutex.Unlock() | ||||
|  | ||||
| 	if currMember, ok := b.userMemberMap[m.Member.User.ID]; ok { | ||||
| 		b.Log.Debugf( | ||||
| 			"%s: memberupdate: user %s (nick %s) changes nick to %s", | ||||
| 			b.Account, | ||||
| 			m.Member.User.Username, | ||||
| 			b.userMemberMap[m.Member.User.ID].Nick, | ||||
| 			m.Member.Nick, | ||||
| 		) | ||||
| 		delete(b.nickMemberMap, currMember.User.Username) | ||||
| 		delete(b.nickMemberMap, currMember.Nick) | ||||
| 		delete(b.userMemberMap, m.Member.User.ID) | ||||
| 	} | ||||
| 	b.userMemberMap[m.Member.User.ID] = m.Member | ||||
| 	b.nickMemberMap[m.Member.User.Username] = m.Member | ||||
| 	if m.Member.Nick != "" { | ||||
| 		b.nickMemberMap[m.Member.Nick] = m.Member | ||||
| 	} | ||||
| } | ||||
|  | ||||
| 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 | ||||
| } | ||||
							
								
								
									
										189
									
								
								bridge/discord/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,189 @@ | ||||
| package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"unicode" | ||||
|  | ||||
| 	"github.com/bwmarrin/discordgo" | ||||
| ) | ||||
|  | ||||
| func (b *Bdiscord) getNick(user *discordgo.User) string { | ||||
| 	b.membersMutex.RLock() | ||||
| 	defer b.membersMutex.RUnlock() | ||||
|  | ||||
| 	if member, ok := b.userMemberMap[user.ID]; ok { | ||||
| 		if member.Nick != "" { | ||||
| 			// Only return if nick is set. | ||||
| 			return member.Nick | ||||
| 		} | ||||
| 		// Otherwise return username. | ||||
| 		return user.Username | ||||
| 	} | ||||
|  | ||||
| 	// If we didn't find nick, search for it. | ||||
| 	member, err := b.c.GuildMember(b.guildID, user.ID) | ||||
| 	if err != nil { | ||||
| 		b.Log.Warnf("Failed to fetch information for member %#v: %#v", user, err) | ||||
| 		return user.Username | ||||
| 	} else if member == nil { | ||||
| 		b.Log.Warnf("Got no information for member %#v", user) | ||||
| 		return user.Username | ||||
| 	} | ||||
| 	b.userMemberMap[user.ID] = member | ||||
| 	b.nickMemberMap[member.User.Username] = member | ||||
| 	if member.Nick != "" { | ||||
| 		b.nickMemberMap[member.Nick] = member | ||||
| 		return member.Nick | ||||
| 	} | ||||
| 	return user.Username | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getGuildMemberByNick(nick string) (*discordgo.Member, error) { | ||||
| 	b.membersMutex.RLock() | ||||
| 	defer b.membersMutex.RUnlock() | ||||
|  | ||||
| 	if member, ok := b.nickMemberMap[nick]; ok { | ||||
| 		return member, nil | ||||
| 	} | ||||
| 	return nil, errors.New("Couldn't find guild member with nick " + nick) // This will most likely get ignored by the caller | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelID(name string) string { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	idcheck := strings.Split(name, "ID:") | ||||
| 	if len(idcheck) > 1 { | ||||
| 		return idcheck[1] | ||||
| 	} | ||||
| 	for _, channel := range b.channels { | ||||
| 		if channel.Name == name { | ||||
| 			return channel.ID | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelName(id string) string { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	for _, channel := range b.channels { | ||||
| 		if channel.ID == id { | ||||
| 			return channel.Name | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	// See https://discordapp.com/developers/docs/reference#message-formatting. | ||||
| 	channelMentionRE = regexp.MustCompile("<#[0-9]+>") | ||||
| 	emojiRE          = regexp.MustCompile("<(:.*?:)[0-9]+>") | ||||
| 	userMentionRE    = regexp.MustCompile("@[^@\n]{1,32}") | ||||
| ) | ||||
|  | ||||
| func (b *Bdiscord) replaceChannelMentions(text string) string { | ||||
| 	replaceChannelMentionFunc := func(match string) string { | ||||
| 		var err error | ||||
| 		channelID := match[2 : len(match)-1] | ||||
|  | ||||
| 		channelName := b.getChannelName(channelID) | ||||
| 		// If we don't have the channel refresh our list. | ||||
| 		if channelName == "" { | ||||
| 			b.channels, err = b.c.GuildChannels(b.guildID) | ||||
| 			if err != nil { | ||||
| 				return "#unknownchannel" | ||||
| 			} | ||||
| 			channelName = b.getChannelName(channelID) | ||||
| 		} | ||||
| 		return "#" + channelName | ||||
| 	} | ||||
| 	return channelMentionRE.ReplaceAllStringFunc(text, replaceChannelMentionFunc) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceUserMentions(text string) string { | ||||
| 	replaceUserMentionFunc := func(match string) string { | ||||
| 		var ( | ||||
| 			err      error | ||||
| 			member   *discordgo.Member | ||||
| 			username string | ||||
| 		) | ||||
|  | ||||
| 		usernames := enumerateUsernames(match[1:]) | ||||
| 		for _, username = range usernames { | ||||
| 			b.Log.Debugf("Testing mention: '%s'", username) | ||||
| 			member, err = b.getGuildMemberByNick(username) | ||||
| 			if err == nil { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		if member == nil { | ||||
| 			return match | ||||
| 		} | ||||
| 		return strings.Replace(match, "@"+username, member.User.Mention(), 1) | ||||
| 	} | ||||
| 	return userMentionRE.ReplaceAllStringFunc(text, replaceUserMentionFunc) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) stripCustomoji(text string) string { | ||||
| 	return emojiRE.ReplaceAllString(text, `$1`) | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||
| 		return text[1:], true | ||||
| 	} | ||||
| 	return text, false | ||||
| } | ||||
|  | ||||
| // splitURL splits a webhookURL and returns the ID and token. | ||||
| func (b *Bdiscord) splitURL(url string) (string, string) { | ||||
| 	const ( | ||||
| 		expectedWebhookSplitCount = 7 | ||||
| 		webhookIdxID              = 5 | ||||
| 		webhookIdxToken           = 6 | ||||
| 	) | ||||
| 	webhookURLSplit := strings.Split(url, "/") | ||||
| 	if len(webhookURLSplit) != expectedWebhookSplitCount { | ||||
| 		b.Log.Fatalf("%s is no correct discord WebhookURL", url) | ||||
| 	} | ||||
| 	return webhookURLSplit[webhookIdxID], webhookURLSplit[webhookIdxToken] | ||||
| } | ||||
|  | ||||
| func enumerateUsernames(s string) []string { | ||||
| 	onlySpace := true | ||||
| 	for _, r := range s { | ||||
| 		if !unicode.IsSpace(r) { | ||||
| 			onlySpace = false | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	if onlySpace { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	var username, endSpace string | ||||
| 	var usernames []string | ||||
| 	skippingSpace := true | ||||
| 	for _, r := range s { | ||||
| 		if unicode.IsSpace(r) { | ||||
| 			if !skippingSpace { | ||||
| 				usernames = append(usernames, username) | ||||
| 				skippingSpace = true | ||||
| 			} | ||||
| 			endSpace += string(r) | ||||
| 			username += string(r) | ||||
| 		} else { | ||||
| 			endSpace = "" | ||||
| 			username += string(r) | ||||
| 			skippingSpace = false | ||||
| 		} | ||||
| 	} | ||||
| 	if endSpace == "" { | ||||
| 		usernames = append(usernames, username) | ||||
| 	} | ||||
| 	return usernames | ||||
| } | ||||
							
								
								
									
										46
									
								
								bridge/discord/helpers_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,46 @@ | ||||
| package bdiscord | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/stretchr/testify/assert" | ||||
| ) | ||||
|  | ||||
| func TestEnumerateUsernames(t *testing.T) { | ||||
| 	testcases := map[string]struct { | ||||
| 		match             string | ||||
| 		expectedUsernames []string | ||||
| 	}{ | ||||
| 		"only space": { | ||||
| 			match:             "  \t\n \t", | ||||
| 			expectedUsernames: nil, | ||||
| 		}, | ||||
| 		"single word": { | ||||
| 			match:             "veni", | ||||
| 			expectedUsernames: []string{"veni"}, | ||||
| 		}, | ||||
| 		"single word with preceeding space": { | ||||
| 			match:             " vidi", | ||||
| 			expectedUsernames: []string{" vidi"}, | ||||
| 		}, | ||||
| 		"single word with suffixed space": { | ||||
| 			match:             "vici ", | ||||
| 			expectedUsernames: []string{"vici"}, | ||||
| 		}, | ||||
| 		"multi-word with varying whitespace": { | ||||
| 			match: "just me  and\tmy friends \t", | ||||
| 			expectedUsernames: []string{ | ||||
| 				"just", | ||||
| 				"just me", | ||||
| 				"just me  and", | ||||
| 				"just me  and\tmy", | ||||
| 				"just me  and\tmy friends", | ||||
| 			}, | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	for testname, testcase := range testcases { | ||||
| 		foundUsernames := enumerateUsernames(testcase.match) | ||||
| 		assert.Equalf(t, testcase.expectedUsernames, foundUsernames, "Should have found the expected usernames for testcase %s", testname) | ||||
| 	} | ||||
| } | ||||
| @@ -77,7 +77,7 @@ func (b *Bgitter) JoinChannel(channel config.ChannelInfo) error { | ||||
| 						Account: b.Account, Avatar: b.getAvatar(ev.Message.From.Username), UserID: ev.Message.From.ID, | ||||
| 						ID: ev.Message.ID} | ||||
| 					if strings.HasPrefix(ev.Message.Text, "@"+ev.Message.From.Username) { | ||||
| 						rmsg.Event = config.EVENT_USER_ACTION | ||||
| 						rmsg.Event = config.EventUserAction | ||||
| 						rmsg.Text = strings.Replace(rmsg.Text, "@"+ev.Message.From.Username+" ", "", -1) | ||||
| 					} | ||||
| 					b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| @@ -100,7 +100,7 @@ func (b *Bgitter) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
|   | ||||
| @@ -11,13 +11,16 @@ import ( | ||||
| 	"unicode/utf8" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	log "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) { | ||||
| 	return DownloadFileAuth(url, "") | ||||
| } | ||||
|  | ||||
| // DownloadFileAuth downloads the given URL using the specified authentication token. | ||||
| func DownloadFileAuth(url string, auth string) (*[]byte, error) { | ||||
| 	var buf bytes.Buffer | ||||
| 	client := &http.Client{ | ||||
| @@ -40,33 +43,62 @@ func DownloadFileAuth(url string, auth string) (*[]byte, error) { | ||||
| 	return &data, nil | ||||
| } | ||||
|  | ||||
| func SplitStringLength(input string, length int) string { | ||||
| 	a := []rune(input) | ||||
| 	str := "" | ||||
| 	for i, r := range a { | ||||
| 		str = str + string(r) | ||||
| 		if i > 0 && (i+1)%length == 0 { | ||||
| 			str += "\n" | ||||
| // GetSubLines splits messages in newline-delimited lines. If maxLineLength is | ||||
| // specified as non-zero GetSubLines will also clip long lines to the maximum | ||||
| // length and insert a warning marker that the line was clipped. | ||||
| // | ||||
| // TODO: The current implementation has the inconvenient that it disregards | ||||
| // word boundaries when splitting but this is hard to solve without potentially | ||||
| // breaking formatting and other stylistic effects. | ||||
| func GetSubLines(message string, maxLineLength int) []string { | ||||
| 	const clippingMessage = " <clipped message>" | ||||
|  | ||||
| 	var lines []string | ||||
| 	for _, line := range strings.Split(strings.TrimSpace(message), "\n") { | ||||
| 		if maxLineLength == 0 || len([]byte(line)) <= maxLineLength { | ||||
| 			lines = append(lines, line) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// !!! WARNING !!! | ||||
| 		// Before touching the splitting logic below please ensure that you PROPERLY | ||||
| 		// understand how strings, runes and range loops over strings work in Go. | ||||
| 		// A good place to start is to read https://blog.golang.org/strings. :-) | ||||
| 		var splitStart int | ||||
| 		var startOfPreviousRune int | ||||
| 		for i := range line { | ||||
| 			if i-splitStart > maxLineLength-len([]byte(clippingMessage)) { | ||||
| 				lines = append(lines, line[splitStart:startOfPreviousRune]+clippingMessage) | ||||
| 				splitStart = startOfPreviousRune | ||||
| 			} | ||||
| 			startOfPreviousRune = i | ||||
| 		} | ||||
| 		// This last append is safe to do without looking at the remaining byte-length | ||||
| 		// as we assume that the byte-length of the last rune will never exceed that of | ||||
| 		// the byte-length of the clipping message. | ||||
| 		lines = append(lines, line[splitStart:]) | ||||
| 	} | ||||
| 	return str | ||||
| 	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 { | ||||
| 	extra := msg.Extra | ||||
| 	rmsg := []config.Message{} | ||||
| 	if len(extra[config.EVENT_FILE_FAILURE_SIZE]) > 0 { | ||||
| 		for _, f := range extra[config.EVENT_FILE_FAILURE_SIZE] { | ||||
| 			fi := f.(config.FileInfo) | ||||
| 			text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize) | ||||
| 			rmsg = append(rmsg, config.Message{Text: text, Username: "<system> ", Channel: msg.Channel, Account: msg.Account}) | ||||
| 		} | ||||
| 		return rmsg | ||||
| 	for _, f := range extra[config.EventFileFailureSize] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize) | ||||
| 		rmsg = append(rmsg, config.Message{ | ||||
| 			Text:     text, | ||||
| 			Username: "<system> ", | ||||
| 			Channel:  msg.Channel, | ||||
| 			Account:  msg.Account, | ||||
| 		}) | ||||
| 	} | ||||
| 	return rmsg | ||||
| } | ||||
|  | ||||
| // 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 sha, ok := av[userid]; ok { | ||||
| 		return general.MediaServerDownload + "/" + sha + "/" + userid + ".png" | ||||
| @@ -74,13 +106,15 @@ func GetAvatar(av map[string]string, userid string, general *config.Protocol) st | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func HandleDownloadSize(flog *log.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 | ||||
| 	for _, entry := range general.MediaDownloadBlackList { | ||||
| 		if entry != "" { | ||||
| 			re, err := regexp.Compile(entry) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||
| 				logger.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||
| 				continue | ||||
| 			} | ||||
| 			if re.MatchString(name) { | ||||
| @@ -88,43 +122,58 @@ func HandleDownloadSize(flog *log.Entry, msg *config.Message, name string, size | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	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 { | ||||
| 		msg.Event = config.EVENT_FILE_FAILURE_SIZE | ||||
| 		msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: name, Comment: msg.Text, Size: size}) | ||||
| 		msg.Event = config.EventFileFailureSize | ||||
| 		msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{ | ||||
| 			Name:    name, | ||||
| 			Comment: msg.Text, | ||||
| 			Size:    size, | ||||
| 		}) | ||||
| 		return fmt.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, general.MediaDownloadSize) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func HandleDownloadData(flog *log.Entry, msg *config.Message, name, comment, url string, data *[]byte, general *config.Protocol) { | ||||
| // 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 | ||||
| 	flog.Debugf("Download OK %#v %#v", name, len(*data)) | ||||
| 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||
| 	logger.Debugf("Download OK %#v %#v", name, len(*data)) | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		avatar = true | ||||
| 	} | ||||
| 	msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, URL: url, Comment: comment, Avatar: avatar}) | ||||
| 	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 { | ||||
| 	lines := "" | ||||
| 	for _, line := range strings.Split(msg, "\n") { | ||||
| 		if line != "" { | ||||
| 			lines += line + "\n" | ||||
| 		} | ||||
| 	} | ||||
| 	lines = strings.TrimRight(lines, "\n") | ||||
| 	return lines | ||||
| 	return emptyLineMatcher.ReplaceAllString(strings.Trim(msg, "\n"), "\n") | ||||
| } | ||||
|  | ||||
| // ClipMessage trims a message to the specified length if it exceeds it and adds a warning | ||||
| // to the message in case it does so. | ||||
| func ClipMessage(text string, length int) string { | ||||
| 	// clip too long messages | ||||
| 	const clippingMessage = " <clipped message>" | ||||
| 	if len(text) > length { | ||||
| 		text = text[:length-len(" *message clipped*")] | ||||
| 		text = text[:length-len(clippingMessage)] | ||||
| 		if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||
| 			text = text[:len(text)-size] | ||||
| 		} | ||||
| 		text += " *message clipped*" | ||||
| 		text += clippingMessage | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func ParseMarkdown(input string) string { | ||||
| 	md := markdown.New(markdown.XHTMLOutput(true), markdown.Breaks(true)) | ||||
| 	return (md.RenderToString([]byte(input))) | ||||
| } | ||||
|   | ||||
							
								
								
									
										105
									
								
								bridge/helper/helper_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,105 @@ | ||||
| package helper | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/stretchr/testify/assert" | ||||
| ) | ||||
|  | ||||
| const testLineLength = 64 | ||||
|  | ||||
| var ( | ||||
| 	lineSplittingTestCases = map[string]struct { | ||||
| 		input          string | ||||
| 		splitOutput    []string | ||||
| 		nonSplitOutput []string | ||||
| 	}{ | ||||
| 		"Short single-line message": { | ||||
| 			input:          "short", | ||||
| 			splitOutput:    []string{"short"}, | ||||
| 			nonSplitOutput: []string{"short"}, | ||||
| 		}, | ||||
| 		"Long single-line message": { | ||||
| 			input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 			splitOutput: []string{ | ||||
| 				"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>", | ||||
| 				"cing elit, sed do eiusmod tempor incididunt ut <clipped message>", | ||||
| 				" labore et dolore magna aliqua.", | ||||
| 			}, | ||||
| 			nonSplitOutput: []string{"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."}, | ||||
| 		}, | ||||
| 		"Short multi-line message": { | ||||
| 			input: "I\ncan't\nget\nno\nsatisfaction!", | ||||
| 			splitOutput: []string{ | ||||
| 				"I", | ||||
| 				"can't", | ||||
| 				"get", | ||||
| 				"no", | ||||
| 				"satisfaction!", | ||||
| 			}, | ||||
| 			nonSplitOutput: []string{ | ||||
| 				"I", | ||||
| 				"can't", | ||||
| 				"get", | ||||
| 				"no", | ||||
| 				"satisfaction!", | ||||
| 			}, | ||||
| 		}, | ||||
| 		"Long multi-line message": { | ||||
| 			input: "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.\n" + | ||||
| 				"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.\n" + | ||||
| 				"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.\n" + | ||||
| 				"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.", | ||||
| 			splitOutput: []string{ | ||||
| 				"Lorem ipsum dolor sit amet, consectetur adipis <clipped message>", | ||||
| 				"cing elit, sed do eiusmod tempor incididunt ut <clipped message>", | ||||
| 				" labore et dolore magna aliqua.", | ||||
| 				"Ut enim ad minim veniam, quis nostrud exercita <clipped message>", | ||||
| 				"tion ullamco laboris nisi ut aliquip ex ea com <clipped message>", | ||||
| 				"modo consequat.", | ||||
| 				"Duis aute irure dolor in reprehenderit in volu <clipped message>", | ||||
| 				"ptate velit esse cillum dolore eu fugiat nulla <clipped message>", | ||||
| 				" pariatur.", | ||||
| 				"Excepteur sint occaecat cupidatat non proident <clipped message>", | ||||
| 				", sunt in culpa qui officia deserunt mollit an <clipped message>", | ||||
| 				"im id est laborum.", | ||||
| 			}, | ||||
| 			nonSplitOutput: []string{ | ||||
| 				"Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", | ||||
| 				"Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.", | ||||
| 				"Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.", | ||||
| 				"Excepteur sint occaecat cupidatat non proident, sunt in culpa qui officia deserunt mollit anim id est laborum.", | ||||
| 			}, | ||||
| 		}, | ||||
| 		"Message ending with new-line.": { | ||||
| 			input:          "Newline ending\n", | ||||
| 			splitOutput:    []string{"Newline ending"}, | ||||
| 			nonSplitOutput: []string{"Newline ending"}, | ||||
| 		}, | ||||
| 		"Long message containing UTF-8 multi-byte runes": { | ||||
| 			input: "不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說", | ||||
| 			splitOutput: []string{ | ||||
| 				"不布人個我此而及單石業喜資富下 <clipped message>", | ||||
| 				"我河下日沒一我臺空達的常景便物 <clipped message>", | ||||
| 				"沒為……子大我別名解成?生賣的 <clipped message>", | ||||
| 				"全直黑,我自我結毛分洲了世當, <clipped message>", | ||||
| 				"是政福那是東;斯說", | ||||
| 			}, | ||||
| 			nonSplitOutput: []string{"不布人個我此而及單石業喜資富下我河下日沒一我臺空達的常景便物沒為……子大我別名解成?生賣的全直黑,我自我結毛分洲了世當,是政福那是東;斯說"}, | ||||
| 		}, | ||||
| 	} | ||||
| ) | ||||
|  | ||||
| func TestGetSubLines(t *testing.T) { | ||||
| 	for testname, testcase := range lineSplittingTestCases { | ||||
| 		splitLines := GetSubLines(testcase.input, testLineLength) | ||||
| 		assert.Equalf(t, testcase.splitOutput, splitLines, "'%s' testcase should give expected lines with splitting.", testname) | ||||
| 		for _, splitLine := range splitLines { | ||||
| 			byteLength := len([]byte(splitLine)) | ||||
| 			assert.True(t, byteLength <= testLineLength, "Splitted line '%s' of testcase '%s' should not exceed the maximum byte-length (%d vs. %d).", splitLine, testcase, byteLength, testLineLength) | ||||
| 		} | ||||
|  | ||||
| 		nonSplitLines := GetSubLines(testcase.input, 0) | ||||
| 		assert.Equalf(t, testcase.nonSplitOutput, nonSplitLines, "'%s' testcase should give expected lines without splitting.", testname) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										239
									
								
								bridge/irc/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,239 @@ | ||||
| package birc | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/dfordsoft/golib/ic" | ||||
| 	"github.com/lrstanley/girc" | ||||
| 	"github.com/paulrosania/go-charset/charset" | ||||
| 	"github.com/saintfish/chardet" | ||||
|  | ||||
| 	// 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" | ||||
| ) | ||||
|  | ||||
| func (b *Birc) handleCharset(msg *config.Message) error { | ||||
| 	if b.GetString("Charset") != "" { | ||||
| 		switch b.GetString("Charset") { | ||||
| 		case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 			msg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), msg.Text) | ||||
| 		default: | ||||
| 			buf := new(bytes.Buffer) | ||||
| 			w, err := charset.NewWriter(b.GetString("Charset"), buf) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("charset from utf-8 conversion failed: %s", err) | ||||
| 				return err | ||||
| 			} | ||||
| 			fmt.Fprint(w, msg.Text) | ||||
| 			w.Close() | ||||
| 			msg.Text = buf.String() | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleFiles returns true if we have handled the files, otherwise return false | ||||
| func (b *Birc) handleFiles(msg *config.Message) bool { | ||||
| 	if msg.Extra == nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	for _, rmsg := range helper.HandleExtra(msg, b.General) { | ||||
| 		b.Local <- rmsg | ||||
| 	} | ||||
| 	if len(msg.Extra["file"]) == 0 { | ||||
| 		return false | ||||
| 	} | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		if fi.Comment != "" { | ||||
| 			msg.Text += fi.Comment + ": " | ||||
| 		} | ||||
| 		if fi.URL != "" { | ||||
| 			msg.Text = fi.URL | ||||
| 			if fi.Comment != "" { | ||||
| 				msg.Text = fi.Comment + ": " + fi.URL | ||||
| 			} | ||||
| 		} | ||||
| 		b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||
| 	} | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||
| 	if len(event.Params) == 0 { | ||||
| 		b.Log.Debugf("handleJoinPart: empty Params? %#v", event) | ||||
| 		return | ||||
| 	} | ||||
| 	channel := strings.ToLower(event.Params[0]) | ||||
| 	if event.Command == "KICK" && event.Params[1] == b.Nick { | ||||
| 		b.Log.Infof("Got kicked from %s by %s", channel, event.Source.Name) | ||||
| 		time.Sleep(time.Duration(b.GetInt("RejoinDelay")) * time.Second) | ||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EventRejoinChannels} | ||||
| 		return | ||||
| 	} | ||||
| 	if event.Command == "QUIT" { | ||||
| 		if event.Source.Name == b.Nick && strings.Contains(event.Last(), "Ping timeout") { | ||||
| 			b.Log.Infof("%s reconnecting ..", b.Account) | ||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EventFailure} | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	if event.Source.Name != b.Nick { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		// QUIT isn't channel bound, happens for all channels on the bridge | ||||
| 		if event.Command == "QUIT" { | ||||
| 			channel = "" | ||||
| 		} | ||||
| 		msg := config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EventJoinLeave} | ||||
| 		b.Log.Debugf("<= Message is %#v", msg) | ||||
| 		b.Remote <- msg | ||||
| 		return | ||||
| 	} | ||||
| 	b.Log.Debugf("handle %#v", event) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||
| 	b.Log.Debug("Registering callbacks") | ||||
| 	i := b.i | ||||
| 	b.Nick = event.Params[0] | ||||
|  | ||||
| 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||
| 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||
| 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||
| 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||
| 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||
| 	i.Handlers.Add("PART", b.handleJoinPart) | ||||
| 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||
| 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNickServ() { | ||||
| 	if !b.GetBool("UseSASL") && b.GetString("NickServNick") != "" && b.GetString("NickServPassword") != "" { | ||||
| 		b.Log.Debugf("Sending identify to nickserv %s", b.GetString("NickServNick")) | ||||
| 		b.i.Cmd.Message(b.GetString("NickServNick"), "IDENTIFY "+b.GetString("NickServPassword")) | ||||
| 	} | ||||
| 	if strings.EqualFold(b.GetString("NickServNick"), "Q@CServe.quakenet.org") { | ||||
| 		b.Log.Debugf("Authenticating %s against %s", b.GetString("NickServUsername"), b.GetString("NickServNick")) | ||||
| 		b.i.Cmd.Message(b.GetString("NickServNick"), "AUTH "+b.GetString("NickServUsername")+" "+b.GetString("NickServPassword")) | ||||
| 	} | ||||
| 	// give nickserv some slack | ||||
| 	time.Sleep(time.Second * 5) | ||||
| 	b.authDone = true | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNotice(client *girc.Client, event girc.Event) { | ||||
| 	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.GetString("NickServNick") { | ||||
| 		b.handleNickServ() | ||||
| 	} else { | ||||
| 		b.handlePrivMsg(client, event) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleOther(client *girc.Client, event girc.Event) { | ||||
| 	if b.GetInt("DebugLevel") == 1 { | ||||
| 		if event.Command != "CLIENT_STATE_UPDATED" && | ||||
| 			event.Command != "CLIENT_GENERAL_UPDATED" { | ||||
| 			b.Log.Debugf("%#v", event.String()) | ||||
| 		} | ||||
| 		return | ||||
| 	} | ||||
| 	switch event.Command { | ||||
| 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | ||||
| 		return | ||||
| 	} | ||||
| 	b.Log.Debugf("%#v", event.String()) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) { | ||||
| 	b.handleNickServ() | ||||
| 	b.handleRunCommands() | ||||
| 	// we are now fully connected | ||||
| 	b.connected <- nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||
| 	if b.skipPrivMsg(event) { | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||
| 	b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Last(), event) | ||||
|  | ||||
| 	// set action event | ||||
| 	if event.IsAction() { | ||||
| 		rmsg.Event = config.EventUserAction | ||||
| 	} | ||||
|  | ||||
| 	// strip action, we made an event if it was an action | ||||
| 	rmsg.Text += event.StripAction() | ||||
|  | ||||
| 	// strip IRC colors | ||||
| 	re := regexp.MustCompile(`\x03(?:\d{1,2}(?:,\d{1,2})?)?|[[:cntrl:]]`) | ||||
| 	rmsg.Text = re.ReplaceAllString(rmsg.Text, "") | ||||
|  | ||||
| 	// start detecting the charset | ||||
| 	mycharset := b.GetString("Charset") | ||||
| 	if mycharset == "" { | ||||
| 		// detect what were sending so that we convert it to utf-8 | ||||
| 		detector := chardet.NewTextDetector() | ||||
| 		result, err := detector.DetectBest([]byte(rmsg.Text)) | ||||
| 		if err != nil { | ||||
| 			b.Log.Infof("detection failed for rmsg.Text: %#v", rmsg.Text) | ||||
| 			return | ||||
| 		} | ||||
| 		b.Log.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) | ||||
| 		mycharset = result.Charset | ||||
| 		// if we're not sure, just pick ISO-8859-1 | ||||
| 		if result.Confidence < 80 { | ||||
| 			mycharset = "ISO-8859-1" | ||||
| 		} | ||||
| 	} | ||||
| 	switch mycharset { | ||||
| 	case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 		rmsg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), rmsg.Text) | ||||
| 	default: | ||||
| 		r, err := charset.NewReader(mycharset, strings.NewReader(rmsg.Text)) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("charset to utf-8 conversion failed: %s", err) | ||||
| 			return | ||||
| 		} | ||||
| 		output, _ := ioutil.ReadAll(r) | ||||
| 		rmsg.Text = string(output) | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", event.Params[0], b.Account) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleRunCommands() { | ||||
| 	for _, cmd := range b.GetStringSlice("RunCommands") { | ||||
| 		if err := b.i.Cmd.SendRaw(cmd); err != nil { | ||||
| 			b.Log.Errorf("RunCommands %s failed: %s", cmd, err) | ||||
| 		} | ||||
| 		time.Sleep(time.Second) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) { | ||||
| 	parts := strings.Split(event.Params[2], "!") | ||||
| 	t, err := strconv.ParseInt(event.Params[3], 10, 64) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Invalid time stamp: %s", event.Params[3]) | ||||
| 	} | ||||
| 	user := parts[0] | ||||
| 	if len(parts) > 1 { | ||||
| 		user += " [" + parts[1] + "]" | ||||
| 	} | ||||
| 	b.Log.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0)) | ||||
| } | ||||
| @@ -1,61 +0,0 @@ | ||||
| package birc | ||||
|  | ||||
| import ( | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| /* | ||||
| func tableformatter(nicks []string, nicksPerRow int, continued bool) string { | ||||
| 	result := "|IRC users" | ||||
| 	if continued { | ||||
| 		result = "|(continued)" | ||||
| 	} | ||||
| 	for i := 0; i < 2; i++ { | ||||
| 		for j := 1; j <= nicksPerRow && j <= len(nicks); j++ { | ||||
| 			if i == 0 { | ||||
| 				result += "|" | ||||
| 			} else { | ||||
| 				result += ":-|" | ||||
| 			} | ||||
| 		} | ||||
| 		result += "\r\n|" | ||||
| 	} | ||||
| 	result += nicks[0] + "|" | ||||
| 	for i := 1; i < len(nicks); i++ { | ||||
| 		if i%nicksPerRow == 0 { | ||||
| 			result += "\r\n|" + nicks[i] + "|" | ||||
| 		} else { | ||||
| 			result += nicks[i] + "|" | ||||
| 		} | ||||
| 	} | ||||
| 	return result | ||||
| } | ||||
| */ | ||||
|  | ||||
| func plainformatter(nicks []string, nicksPerRow int) string { | ||||
| 	return strings.Join(nicks, ", ") + " currently on IRC" | ||||
| } | ||||
|  | ||||
| func IsMarkup(message string) bool { | ||||
| 	switch message[0] { | ||||
| 	case '|': | ||||
| 		fallthrough | ||||
| 	case '#': | ||||
| 		fallthrough | ||||
| 	case '_': | ||||
| 		fallthrough | ||||
| 	case '*': | ||||
| 		fallthrough | ||||
| 	case '~': | ||||
| 		fallthrough | ||||
| 	case '-': | ||||
| 		fallthrough | ||||
| 	case ':': | ||||
| 		fallthrough | ||||
| 	case '>': | ||||
| 		fallthrough | ||||
| 	case '=': | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
| @@ -1,37 +1,32 @@ | ||||
| package birc | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/tls" | ||||
| 	"fmt" | ||||
| 	"hash/crc32" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"net" | ||||
| 	"regexp" | ||||
| 	"sort" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| 	"unicode/utf8" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/dfordsoft/golib/ic" | ||||
| 	"github.com/lrstanley/girc" | ||||
| 	"github.com/paulrosania/go-charset/charset" | ||||
|  | ||||
| 	// 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" | ||||
| 	"github.com/saintfish/chardet" | ||||
| ) | ||||
|  | ||||
| type Birc struct { | ||||
| 	i                                         *girc.Client | ||||
| 	Nick                                      string | ||||
| 	names                                     map[string][]string | ||||
| 	connected                                 chan struct{} | ||||
| 	connected                                 chan error | ||||
| 	Local                                     chan config.Message // local queue for flood control | ||||
| 	FirstConnection                           bool | ||||
| 	FirstConnection, authDone                 bool | ||||
| 	MessageDelay, MessageQueue, MessageLength int | ||||
|  | ||||
| 	*bridge.Config | ||||
| @@ -42,7 +37,7 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b.Config = cfg | ||||
| 	b.Nick = b.GetString("Nick") | ||||
| 	b.names = make(map[string][]string) | ||||
| 	b.connected = make(chan struct{}) | ||||
| 	b.connected = make(chan error) | ||||
| 	if b.GetInt("MessageDelay") == 0 { | ||||
| 		b.MessageDelay = 1300 | ||||
| 	} else { | ||||
| @@ -63,11 +58,10 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| } | ||||
|  | ||||
| func (b *Birc) Command(msg *config.Message) string { | ||||
| 	switch msg.Text { | ||||
| 	case "!users": | ||||
| 	if msg.Text == "!users" { | ||||
| 		b.i.Handlers.Add(girc.RPL_NAMREPLY, b.storeNames) | ||||
| 		b.i.Handlers.Add(girc.RPL_ENDOFNAMES, b.endNames) | ||||
| 		b.i.Cmd.SendRaw("NAMES " + msg.Channel) | ||||
| 		b.i.Cmd.SendRaw("NAMES " + msg.Channel) //nolint:errcheck | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
| @@ -75,13 +69,164 @@ func (b *Birc) Command(msg *config.Message) string { | ||||
| func (b *Birc) Connect() error { | ||||
| 	b.Local = make(chan config.Message, b.MessageQueue+10) | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
| 	server, portstr, err := net.SplitHostPort(b.GetString("Server")) | ||||
|  | ||||
| 	i, err := b.getClient() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("UseSASL") { | ||||
| 		i.Config.SASL = &girc.SASLPlain{ | ||||
| 			User: b.GetString("NickServNick"), | ||||
| 			Pass: b.GetString("NickServPassword"), | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	i.Handlers.Add(girc.RPL_WELCOME, b.handleNewConnection) | ||||
| 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||
| 	i.Handlers.Add(girc.ERR_NOMOTD, b.handleOtherAuth) | ||||
| 	i.Handlers.Add(girc.ALL_EVENTS, b.handleOther) | ||||
| 	b.i = i | ||||
|  | ||||
| 	go b.doConnect() | ||||
|  | ||||
| 	err = <-b.connected | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("connection failed %s", err) | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.FirstConnection = false | ||||
| 	if b.GetInt("DebugLevel") == 0 { | ||||
| 		i.Handlers.Clear(girc.ALL_EVENTS) | ||||
| 	} | ||||
| 	go b.doSend() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) Disconnect() error { | ||||
| 	b.i.Close() | ||||
| 	close(b.Local) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	// need to check if we have nickserv auth done before joining channels | ||||
| 	for { | ||||
| 		if b.authDone { | ||||
| 			break | ||||
| 		} | ||||
| 		time.Sleep(time.Second) | ||||
| 	} | ||||
| 	if channel.Options.Key != "" { | ||||
| 		b.Log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||
| 		b.i.Cmd.JoinKey(channel.Name, channel.Options.Key) | ||||
| 	} else { | ||||
| 		b.i.Cmd.Join(channel.Name) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// we can be in between reconnects #385 | ||||
| 	if !b.i.IsConnected() { | ||||
| 		b.Log.Error("Not connected to server, dropping message") | ||||
| 	} | ||||
|  | ||||
| 	// Execute a command | ||||
| 	if strings.HasPrefix(msg.Text, "!") { | ||||
| 		b.Command(&msg) | ||||
| 	} | ||||
|  | ||||
| 	// convert to specified charset | ||||
| 	if err := b.handleCharset(&msg); err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// handle files, return if we're done here | ||||
| 	if ok := b.handleFiles(&msg); ok { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	var msgLines []string | ||||
| 	if b.GetBool("MessageSplit") { | ||||
| 		msgLines = helper.GetSubLines(msg.Text, b.MessageLength) | ||||
| 	} else { | ||||
| 		msgLines = helper.GetSubLines(msg.Text, 0) | ||||
| 	} | ||||
| 	for i := range msgLines { | ||||
| 		if len(b.Local) >= b.MessageQueue { | ||||
| 			b.Log.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		b.Local <- config.Message{ | ||||
| 			Text:     msgLines[i], | ||||
| 			Username: msg.Username, | ||||
| 			Channel:  msg.Channel, | ||||
| 			Event:    msg.Event, | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) doConnect() { | ||||
| 	for { | ||||
| 		if err := b.i.Connect(); err != nil { | ||||
| 			b.Log.Errorf("disconnect: error: %s", err) | ||||
| 			if b.FirstConnection { | ||||
| 				b.connected <- err | ||||
| 				return | ||||
| 			} | ||||
| 		} else { | ||||
| 			b.Log.Info("disconnect: client requested quit") | ||||
| 		} | ||||
| 		b.Log.Info("reconnecting in 30 seconds...") | ||||
| 		time.Sleep(30 * time.Second) | ||||
| 		b.i.Handlers.Clear(girc.RPL_WELCOME) | ||||
| 		b.i.Handlers.Add(girc.RPL_WELCOME, func(client *girc.Client, event girc.Event) { | ||||
| 			b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EventRejoinChannels} | ||||
| 			// set our correct nick on reconnect if necessary | ||||
| 			b.Nick = event.Source.Name | ||||
| 		}) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) doSend() { | ||||
| 	rate := time.Millisecond * time.Duration(b.MessageDelay) | ||||
| 	throttle := time.NewTicker(rate) | ||||
| 	for msg := range b.Local { | ||||
| 		<-throttle.C | ||||
| 		username := msg.Username | ||||
| 		if b.GetBool("Colornicks") { | ||||
| 			checksum := crc32.ChecksumIEEE([]byte(msg.Username)) | ||||
| 			colorCode := checksum%14 + 2 // quick fix - prevent white or black color codes | ||||
| 			username = fmt.Sprintf("\x03%02d%s\x0F", colorCode, msg.Username) | ||||
| 		} | ||||
| 		if msg.Event == config.EventUserAction { | ||||
| 			b.i.Cmd.Action(msg.Channel, username+msg.Text) | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Sending to channel %s", msg.Channel) | ||||
| 			b.i.Cmd.Message(msg.Channel, username+msg.Text) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // validateInput validates the server/port/nick configuration. Returns a *girc.Client if successful | ||||
| func (b *Birc) getClient() (*girc.Client, error) { | ||||
| 	server, portstr, err := net.SplitHostPort(b.GetString("Server")) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	port, err := strconv.Atoi(portstr) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	// fix strict user handling of girc | ||||
| 	user := b.GetString("Nick") | ||||
| @@ -101,268 +246,28 @@ func (b *Birc) Connect() error { | ||||
| 		User:       user, | ||||
| 		Name:       b.GetString("Nick"), | ||||
| 		SSL:        b.GetBool("UseTLS"), | ||||
| 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, | ||||
| 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, //nolint:gosec | ||||
| 		PingDelay:  time.Minute, | ||||
| 	}) | ||||
|  | ||||
| 	if b.GetBool("UseSASL") { | ||||
| 		i.Config.SASL = &girc.SASLPlain{b.GetString("NickServNick"), b.GetString("NickServPassword")} | ||||
| 	} | ||||
|  | ||||
| 	i.Handlers.Add(girc.RPL_WELCOME, b.handleNewConnection) | ||||
| 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||
| 	i.Handlers.Add(girc.ALL_EVENTS, b.handleOther) | ||||
| 	go func() { | ||||
| 		for { | ||||
| 			if err := i.Connect(); err != nil { | ||||
| 				b.Log.Errorf("disconnect: error: %s", err) | ||||
| 			} else { | ||||
| 				b.Log.Info("disconnect: client requested quit") | ||||
| 			} | ||||
|  | ||||
| 			b.Log.Info("reconnecting in 30 seconds...") | ||||
| 			time.Sleep(30 * time.Second) | ||||
| 			i.Handlers.Clear(girc.RPL_WELCOME) | ||||
| 			i.Handlers.Add(girc.RPL_WELCOME, func(client *girc.Client, event girc.Event) { | ||||
| 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||
| 				// set our correct nick on reconnect if necessary | ||||
| 				b.Nick = event.Source.Name | ||||
| 			}) | ||||
| 		} | ||||
| 	}() | ||||
| 	b.i = i | ||||
| 	select { | ||||
| 	case <-b.connected: | ||||
| 		b.Log.Info("Connection succeeded") | ||||
| 	case <-time.After(time.Second * 30): | ||||
| 		return fmt.Errorf("connection timed out") | ||||
| 	} | ||||
| 	//i.Debug = false | ||||
| 	if b.GetInt("DebugLevel") == 0 { | ||||
| 		i.Handlers.Clear(girc.ALL_EVENTS) | ||||
| 	} | ||||
| 	go b.doSend() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) Disconnect() error { | ||||
| 	b.i.Close() | ||||
| 	close(b.Local) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if channel.Options.Key != "" { | ||||
| 		b.Log.Debugf("using key %s for channel %s", channel.Options.Key, channel.Name) | ||||
| 		b.i.Cmd.JoinKey(channel.Name, channel.Options.Key) | ||||
| 	} else { | ||||
| 		b.i.Cmd.Join(channel.Name) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// we can be in between reconnects #385 | ||||
| 	if !b.i.IsConnected() { | ||||
| 		b.Log.Error("Not connected to server, dropping message") | ||||
| 	} | ||||
|  | ||||
| 	// Execute a command | ||||
| 	if strings.HasPrefix(msg.Text, "!") { | ||||
| 		b.Command(&msg) | ||||
| 	} | ||||
|  | ||||
| 	// convert to specified charset | ||||
| 	if b.GetString("Charset") != "" { | ||||
| 		switch b.GetString("Charset") { | ||||
| 		case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 			msg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), msg.Text) | ||||
| 		default: | ||||
| 			buf := new(bytes.Buffer) | ||||
| 			w, err := charset.NewWriter(b.GetString("Charset"), buf) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("charset from utf-8 conversion failed: %s", err) | ||||
| 				return "", err | ||||
| 			} | ||||
| 			fmt.Fprint(w, msg.Text) | ||||
| 			w.Close() | ||||
| 			msg.Text = buf.String() | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Handle files | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.Local <- rmsg | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.Comment != "" { | ||||
| 					msg.Text += fi.Comment + ": " | ||||
| 				} | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text = fi.URL | ||||
| 					if fi.Comment != "" { | ||||
| 						msg.Text = fi.Comment + ": " + fi.URL | ||||
| 					} | ||||
| 				} | ||||
| 				b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||
| 			} | ||||
| 			return "", nil | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// split long messages on messageLength, to avoid clipped messages #281 | ||||
| 	if b.GetBool("MessageSplit") { | ||||
| 		msg.Text = helper.SplitStringLength(msg.Text, b.MessageLength) | ||||
| 	} | ||||
| 	for _, text := range strings.Split(msg.Text, "\n") { | ||||
| 		if len(text) > b.MessageLength { | ||||
| 			text = text[:b.MessageLength-len(" <message clipped>")] | ||||
| 			if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError { | ||||
| 				text = text[:len(text)-size] | ||||
| 			} | ||||
| 			text += " <message clipped>" | ||||
| 		} | ||||
| 		if len(b.Local) < b.MessageQueue { | ||||
| 			if len(b.Local) == b.MessageQueue-1 { | ||||
| 				text = text + " <message clipped>" | ||||
| 			} | ||||
| 			b.Local <- config.Message{Text: text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event} | ||||
| 		} else { | ||||
| 			b.Log.Debugf("flooding, dropping message (queue at %d)", len(b.Local)) | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) doSend() { | ||||
| 	rate := time.Millisecond * time.Duration(b.MessageDelay) | ||||
| 	throttle := time.NewTicker(rate) | ||||
| 	for msg := range b.Local { | ||||
| 		<-throttle.C | ||||
| 		username := msg.Username | ||||
| 		if b.GetBool("Colornicks") { | ||||
| 			checksum := crc32.ChecksumIEEE([]byte(msg.Username)) | ||||
| 			colorCode := checksum%14 + 2 // quick fix - prevent white or black color codes | ||||
| 			username = fmt.Sprintf("\x03%02d%s\x0F", colorCode, msg.Username) | ||||
| 		} | ||||
| 		if msg.Event == config.EVENT_USER_ACTION { | ||||
| 			b.i.Cmd.Action(msg.Channel, username+msg.Text) | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Sending to channel %s", msg.Channel) | ||||
| 			b.i.Cmd.Message(msg.Channel, username+msg.Text) | ||||
| 		} | ||||
| 	} | ||||
| 	return i, nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||
| 	channel := event.Params[1] | ||||
| 	sort.Strings(b.names[channel]) | ||||
| 	maxNamesPerPost := (300 / b.nicksPerRow()) * b.nicksPerRow() | ||||
| 	continued := false | ||||
| 	for len(b.names[channel]) > maxNamesPerPost { | ||||
| 		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost], continued), | ||||
| 		b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel][0:maxNamesPerPost]), | ||||
| 			Channel: channel, Account: b.Account} | ||||
| 		b.names[channel] = b.names[channel][maxNamesPerPost:] | ||||
| 		continued = true | ||||
| 	} | ||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel], continued), | ||||
| 	b.Remote <- config.Message{Username: b.Nick, Text: b.formatnicks(b.names[channel]), | ||||
| 		Channel: channel, Account: b.Account} | ||||
| 	b.names[channel] = nil | ||||
| 	b.i.Handlers.Clear(girc.RPL_NAMREPLY) | ||||
| 	b.i.Handlers.Clear(girc.RPL_ENDOFNAMES) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNewConnection(client *girc.Client, event girc.Event) { | ||||
| 	b.Log.Debug("Registering callbacks") | ||||
| 	i := b.i | ||||
| 	b.Nick = event.Params[0] | ||||
|  | ||||
| 	i.Handlers.Add(girc.RPL_ENDOFMOTD, b.handleOtherAuth) | ||||
| 	i.Handlers.Add("PRIVMSG", b.handlePrivMsg) | ||||
| 	i.Handlers.Add("CTCP_ACTION", b.handlePrivMsg) | ||||
| 	i.Handlers.Add(girc.RPL_TOPICWHOTIME, b.handleTopicWhoTime) | ||||
| 	i.Handlers.Add(girc.NOTICE, b.handleNotice) | ||||
| 	i.Handlers.Add("JOIN", b.handleJoinPart) | ||||
| 	i.Handlers.Add("PART", b.handleJoinPart) | ||||
| 	i.Handlers.Add("QUIT", b.handleJoinPart) | ||||
| 	i.Handlers.Add("KICK", b.handleJoinPart) | ||||
| 	// we are now fully connected | ||||
| 	b.connected <- struct{}{} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) { | ||||
| 	if len(event.Params) == 0 { | ||||
| 		b.Log.Debugf("handleJoinPart: empty Params? %#v", event) | ||||
| 		return | ||||
| 	} | ||||
| 	channel := strings.ToLower(event.Params[0]) | ||||
| 	if event.Command == "KICK" && event.Params[1] == b.Nick { | ||||
| 		b.Log.Infof("Got kicked from %s by %s", channel, event.Source.Name) | ||||
| 		time.Sleep(time.Duration(b.GetInt("RejoinDelay")) * time.Second) | ||||
| 		b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||
| 		return | ||||
| 	} | ||||
| 	if event.Command == "QUIT" { | ||||
| 		if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") { | ||||
| 			b.Log.Infof("%s reconnecting ..", b.Account) | ||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EVENT_FAILURE} | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	if event.Source.Name != b.Nick { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		msg := config.Message{Username: "system", Text: event.Source.Name + " " + strings.ToLower(event.Command) + "s", Channel: channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||
| 		b.Log.Debugf("<= Message is %#v", msg) | ||||
| 		b.Remote <- msg | ||||
| 		return | ||||
| 	} | ||||
| 	b.Log.Debugf("handle %#v", event) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleNotice(client *girc.Client, event girc.Event) { | ||||
| 	if strings.Contains(event.String(), "This nickname is registered") && event.Source.Name == b.GetString("NickServNick") { | ||||
| 		b.Log.Debugf("Sending identify to nickserv %s", b.GetString("NickServNick")) | ||||
| 		b.i.Cmd.Message(b.GetString("NickServNick"), "IDENTIFY "+b.GetString("NickServPassword")) | ||||
| 	} else { | ||||
| 		b.handlePrivMsg(client, event) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleOther(client *girc.Client, event girc.Event) { | ||||
| 	if b.GetInt("DebugLevel") == 1 { | ||||
| 		if event.Command != "CLIENT_STATE_UPDATED" && | ||||
| 			event.Command != "CLIENT_GENERAL_UPDATED" { | ||||
| 			b.Log.Debugf("%#v", event.String()) | ||||
| 		} | ||||
| 		return | ||||
| 	} | ||||
| 	switch event.Command { | ||||
| 	case "372", "375", "376", "250", "251", "252", "253", "254", "255", "265", "266", "002", "003", "004", "005": | ||||
| 		return | ||||
| 	} | ||||
| 	b.Log.Debugf("%#v", event.String()) | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleOtherAuth(client *girc.Client, event girc.Event) { | ||||
| 	if strings.EqualFold(b.GetString("NickServNick"), "Q@CServe.quakenet.org") { | ||||
| 		b.Log.Debugf("Authenticating %s against %s", b.GetString("NickServUsername"), b.GetString("NickServNick")) | ||||
| 		b.i.Cmd.Message(b.GetString("NickServNick"), "AUTH "+b.GetString("NickServUsername")+" "+b.GetString("NickServPassword")) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Birc) skipPrivMsg(event girc.Event) bool { | ||||
| 	// Our nick can be changed | ||||
| 	b.Nick = b.i.GetNick() | ||||
| @@ -382,74 +287,6 @@ func (b *Birc) skipPrivMsg(event girc.Event) bool { | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) { | ||||
| 	if b.skipPrivMsg(event) { | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host} | ||||
| 	b.Log.Debugf("== Receiving PRIVMSG: %s %s %#v", event.Source.Name, event.Trailing, event) | ||||
|  | ||||
| 	// set action event | ||||
| 	if event.IsAction() { | ||||
| 		rmsg.Event = config.EVENT_USER_ACTION | ||||
| 	} | ||||
|  | ||||
| 	// strip action, we made an event if it was an action | ||||
| 	rmsg.Text += event.StripAction() | ||||
|  | ||||
| 	// strip IRC colors | ||||
| 	re := regexp.MustCompile(`\x03(?:\d{1,2}(?:,\d{1,2})?)?|[[:cntrl:]]`) | ||||
| 	rmsg.Text = re.ReplaceAllString(rmsg.Text, "") | ||||
|  | ||||
| 	// start detecting the charset | ||||
| 	var r io.Reader | ||||
| 	var err error | ||||
| 	mycharset := b.GetString("Charset") | ||||
| 	if mycharset == "" { | ||||
| 		// detect what were sending so that we convert it to utf-8 | ||||
| 		detector := chardet.NewTextDetector() | ||||
| 		result, err := detector.DetectBest([]byte(rmsg.Text)) | ||||
| 		if err != nil { | ||||
| 			b.Log.Infof("detection failed for rmsg.Text: %#v", rmsg.Text) | ||||
| 			return | ||||
| 		} | ||||
| 		b.Log.Debugf("detected %s confidence %#v", result.Charset, result.Confidence) | ||||
| 		mycharset = result.Charset | ||||
| 		// if we're not sure, just pick ISO-8859-1 | ||||
| 		if result.Confidence < 80 { | ||||
| 			mycharset = "ISO-8859-1" | ||||
| 		} | ||||
| 	} | ||||
| 	switch mycharset { | ||||
| 	case "gbk", "gb18030", "gb2312", "big5", "euc-kr", "euc-jp", "shift-jis", "iso-2022-jp": | ||||
| 		rmsg.Text = ic.ConvertString("utf-8", b.GetString("Charset"), rmsg.Text) | ||||
| 	default: | ||||
| 		r, err = charset.NewReader(mycharset, strings.NewReader(rmsg.Text)) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("charset to utf-8 conversion failed: %s", err) | ||||
| 			return | ||||
| 		} | ||||
| 		output, _ := ioutil.ReadAll(r) | ||||
| 		rmsg.Text = string(output) | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", event.Params[0], b.Account) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Birc) handleTopicWhoTime(client *girc.Client, event girc.Event) { | ||||
| 	parts := strings.Split(event.Params[2], "!") | ||||
| 	t, err := strconv.ParseInt(event.Params[3], 10, 64) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("Invalid time stamp: %s", event.Params[3]) | ||||
| 	} | ||||
| 	user := parts[0] | ||||
| 	if len(parts) > 1 { | ||||
| 		user += " [" + parts[1] + "]" | ||||
| 	} | ||||
| 	b.Log.Debugf("%s: Topic set by %s [%s]", event.Command, user, time.Unix(t, 0)) | ||||
| } | ||||
|  | ||||
| func (b *Birc) nicksPerRow() int { | ||||
| 	return 4 | ||||
| } | ||||
| @@ -458,9 +295,9 @@ func (b *Birc) storeNames(client *girc.Client, event girc.Event) { | ||||
| 	channel := event.Params[2] | ||||
| 	b.names[channel] = append( | ||||
| 		b.names[channel], | ||||
| 		strings.Split(strings.TrimSpace(event.Trailing), " ")...) | ||||
| 		strings.Split(strings.TrimSpace(event.Last()), " ")...) | ||||
| } | ||||
|  | ||||
| func (b *Birc) formatnicks(nicks []string, continued bool) string { | ||||
| 	return plainformatter(nicks, b.nicksPerRow()) | ||||
| func (b *Birc) formatnicks(nicks []string) string { | ||||
| 	return strings.Join(nicks, ", ") + " currently on IRC" | ||||
| } | ||||
|   | ||||
| @@ -3,6 +3,7 @@ package bmatrix | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"mime" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| @@ -19,11 +20,13 @@ type Bmatrix struct { | ||||
| 	UserID  string | ||||
| 	RoomMap map[string]string | ||||
| 	sync.RWMutex | ||||
| 	htmlTag *regexp.Regexp | ||||
| 	*bridge.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bmatrix{Config: cfg} | ||||
| 	b.htmlTag = regexp.MustCompile("</.*?>") | ||||
| 	b.RoomMap = make(map[string]string) | ||||
| 	return b | ||||
| } | ||||
| @@ -72,9 +75,12 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel) | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EVENT_USER_ACTION { | ||||
| 		resp, err := b.mc.SendMessageEvent(channel, "m.room.message", | ||||
| 			matrix.TextMessage{"m.emote", msg.Username + msg.Text}) | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		m := matrix.TextMessage{ | ||||
| 			MsgType: "m.emote", | ||||
| 			Body:    msg.Username + msg.Text, | ||||
| 		} | ||||
| 		resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| @@ -82,7 +88,7 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| @@ -96,19 +102,35 @@ func (b *Bmatrix) Send(msg config.Message) (string, error) { | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.mc.SendText(channel, rmsg.Username+rmsg.Text) | ||||
| 			if _, err := b.mc.SendText(channel, rmsg.Username+rmsg.Text); err != nil { | ||||
| 				b.Log.Errorf("sendText failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
| 		// check if we have files to upload (from slack, telegram or mattermost) | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			return b.handleUploadFile(&msg, channel) | ||||
| 			return b.handleUploadFiles(&msg, channel) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Edit message if we have an ID | ||||
| 	// matrix has no editing support | ||||
|  | ||||
| 	// Post normal message | ||||
| 	resp, err := b.mc.SendText(channel, msg.Username+msg.Text) | ||||
| 	// Use notices to send join/leave events | ||||
| 	if msg.Event == config.EventJoinLeave { | ||||
| 		resp, err := b.mc.SendNotice(channel, msg.Username+msg.Text) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return resp.EventID, err | ||||
| 	} | ||||
|  | ||||
| 	username := html.EscapeString(msg.Username) | ||||
| 	// check if we have a </tag>. if we have, we don't escape HTML. #696 | ||||
| 	if b.htmlTag.MatchString(msg.Username) { | ||||
| 		username = msg.Username | ||||
| 	} | ||||
| 	// Post normal message with HTML support (eg riot.im) | ||||
| 	resp, err := b.mc.SendHTML(channel, msg.Username+msg.Text, username+helper.ParseMarkdown(msg.Text)) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| @@ -126,7 +148,7 @@ func (b *Bmatrix) getRoomID(channel string) string { | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handlematrix() error { | ||||
| func (b *Bmatrix) handlematrix() { | ||||
| 	syncer := b.mc.Syncer.(*matrix.DefaultSyncer) | ||||
| 	syncer.OnEventType("m.room.redaction", b.handleEvent) | ||||
| 	syncer.OnEventType("m.room.message", b.handleEvent) | ||||
| @@ -137,7 +159,6 @@ func (b *Bmatrix) handlematrix() error { | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
| @@ -158,7 +179,8 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
|  | ||||
| 		// Text must be a string | ||||
| 		if rmsg.Text, ok = ev.Content["body"].(string); !ok { | ||||
| 			b.Log.Errorf("Content[body] wasn't a %T ?", rmsg.Text) | ||||
| 			b.Log.Errorf("Content[body] is not a string: %T\n%#v", | ||||
| 				ev.Content["body"], ev.Content) | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| @@ -170,16 +192,16 @@ func (b *Bmatrix) handleEvent(ev *matrix.Event) { | ||||
|  | ||||
| 		// Delete event | ||||
| 		if ev.Type == "m.room.redaction" { | ||||
| 			rmsg.Event = config.EVENT_MSG_DELETE | ||||
| 			rmsg.Event = config.EventMsgDelete | ||||
| 			rmsg.ID = ev.Redacts | ||||
| 			rmsg.Text = config.EVENT_MSG_DELETE | ||||
| 			rmsg.Text = config.EventMsgDelete | ||||
| 			b.Remote <- rmsg | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		// Do we have a /me action | ||||
| 		if ev.Content["msgtype"].(string) == "m.emote" { | ||||
| 			rmsg.Event = config.EVENT_USER_ACTION | ||||
| 			rmsg.Event = config.EventUserAction | ||||
| 		} | ||||
|  | ||||
| 		// Do we have attachments | ||||
| @@ -231,11 +253,11 @@ func (b *Bmatrix) handleDownloadFile(rmsg *config.Message, content map[string]in | ||||
| 		if msgtype == "m.image" { | ||||
| 			mext, _ := mime.ExtensionsByType(mtype) | ||||
| 			if len(mext) > 0 { | ||||
| 				name = name + mext[0] | ||||
| 				name += mext[0] | ||||
| 			} | ||||
| 		} else { | ||||
| 			// just a default .png extension if we don't have mime info | ||||
| 			name = name + ".png" | ||||
| 			name += ".png" | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| @@ -254,47 +276,60 @@ func (b *Bmatrix) handleDownloadFile(rmsg *config.Message, content map[string]in | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string) (string, error) { | ||||
| // handleUploadFiles handles native upload of files. | ||||
| func (b *Bmatrix) handleUploadFiles(msg *config.Message, channel string) (string, error) { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		content := bytes.NewReader(*fi.Data) | ||||
| 		sp := strings.Split(fi.Name, ".") | ||||
| 		mtype := mime.TypeByExtension("." + sp[len(sp)-1]) | ||||
| 		if strings.Contains(mtype, "image") || | ||||
| 			strings.Contains(mtype, "video") { | ||||
| 			if fi.Comment != "" { | ||||
| 				_, err := b.mc.SendText(channel, msg.Username+fi.Comment) | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("file comment failed: %#v", err) | ||||
| 				} | ||||
| 			} | ||||
| 			b.Log.Debugf("uploading file: %s %s", fi.Name, mtype) | ||||
| 			res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("file upload failed: %#v", err) | ||||
| 				continue | ||||
| 			} | ||||
| 			if strings.Contains(mtype, "video") { | ||||
| 				b.Log.Debugf("sendVideo %s", res.ContentURI) | ||||
| 				_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("sendVideo failed: %#v", err) | ||||
| 				} | ||||
| 			} | ||||
| 			if strings.Contains(mtype, "image") { | ||||
| 				b.Log.Debugf("sendImage %s", res.ContentURI) | ||||
| 				_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("sendImage failed: %#v", err) | ||||
| 				} | ||||
| 			} | ||||
| 			b.Log.Debugf("result: %#v", res) | ||||
| 		if fi, ok := f.(config.FileInfo); ok { | ||||
| 			b.handleUploadFile(msg, channel, &fi) | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of a file. | ||||
| func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) { | ||||
| 	content := bytes.NewReader(*fi.Data) | ||||
| 	sp := strings.Split(fi.Name, ".") | ||||
| 	mtype := mime.TypeByExtension("." + sp[len(sp)-1]) | ||||
| 	if !strings.Contains(mtype, "image") && !strings.Contains(mtype, "video") { | ||||
| 		return | ||||
| 	} | ||||
| 	if fi.Comment != "" { | ||||
| 		_, err := b.mc.SendText(channel, msg.Username+fi.Comment) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file comment failed: %#v", err) | ||||
| 		} | ||||
| 	} else { | ||||
| 		// image and video uploads send no username, we have to do this ourself here #715 | ||||
| 		_, err := b.mc.SendText(channel, msg.Username) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file comment failed: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Debugf("uploading file: %s %s", fi.Name, mtype) | ||||
| 	res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) | ||||
| 	if err != nil { | ||||
| 		b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	switch { | ||||
| 	case strings.Contains(mtype, "video"): | ||||
| 		b.Log.Debugf("sendVideo %s", res.ContentURI) | ||||
| 		_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendVideo failed: %#v", err) | ||||
| 		} | ||||
| 	case strings.Contains(mtype, "image"): | ||||
| 		b.Log.Debugf("sendImage %s", res.ContentURI) | ||||
| 		_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("sendImage failed: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Debugf("result: %#v", res) | ||||
| } | ||||
|  | ||||
| // skipMessages returns true if this message should not be handled | ||||
| func (b *Bmatrix) containsAttachment(content map[string]interface{}) bool { | ||||
| 	// Skip empty messages | ||||
|   | ||||
							
								
								
									
										195
									
								
								bridge/mattermost/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,195 @@ | ||||
| package bmattermost | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||
| // logs an error message if it fails | ||||
| func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||
| 	rmsg := config.Message{ | ||||
| 		Username: "system", | ||||
| 		Text:     "avatar", | ||||
| 		Channel:  channel, | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   userid, | ||||
| 		Event:    config.EventAvatarDownload, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 	} | ||||
| 	if _, ok := b.avatarMap[userid]; !ok { | ||||
| 		data, resp := b.mc.Client.GetProfileImage(userid, "") | ||||
| 		if resp.Error != nil { | ||||
| 			b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error) | ||||
| 			return | ||||
| 		} | ||||
| 		err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) | ||||
| 		if err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 			return | ||||
| 		} | ||||
| 		helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General) | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error { | ||||
| 	url, _ := b.mc.Client.GetFileLink(id) | ||||
| 	finfo, resp := b.mc.Client.GetFileInfo(id) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, resp := b.mc.Client.DownloadFile(id, true) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatter() { | ||||
| 	messages := make(chan *config.Message) | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		b.Log.Debugf("Choosing webhooks based receiving") | ||||
| 		go b.handleMatterHook(messages) | ||||
| 	} else { | ||||
| 		if b.GetString("Token") != "" { | ||||
| 			b.Log.Debugf("Choosing token based receiving") | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Choosing login/password based receiving") | ||||
| 		} | ||||
| 		go b.handleMatterClient(messages) | ||||
| 	} | ||||
| 	var ok bool | ||||
| 	for message := range messages { | ||||
| 		message.Avatar = helper.GetAvatar(b.avatarMap, message.UserID, b.General) | ||||
| 		message.Account = b.Account | ||||
| 		message.Text, ok = b.replaceAction(message.Text) | ||||
| 		if ok { | ||||
| 			message.Event = config.EventUserAction | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account) | ||||
| 		b.Log.Debugf("<= Message is %#v", message) | ||||
| 		b.Remote <- *message | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 	for message := range b.mc.MessageChan { | ||||
| 		b.Log.Debugf("%#v", message.Raw.Data) | ||||
|  | ||||
| 		if b.skipMessage(message) { | ||||
| 			b.Log.Debugf("Skipped message: %#v", message) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 		if b.General.MediaServerUpload != "" || b.General.MediaDownloadPath != "" { | ||||
| 			b.handleDownloadAvatar(message.UserID, message.Channel) | ||||
| 		} | ||||
|  | ||||
| 		b.Log.Debugf("== Receiving event %#v", message) | ||||
|  | ||||
| 		rmsg := &config.Message{ | ||||
| 			Username: message.Username, | ||||
| 			UserID:   message.UserID, | ||||
| 			Channel:  message.Channel, | ||||
| 			Text:     message.Text, | ||||
| 			ID:       message.Post.Id, | ||||
| 			ParentID: message.Post.ParentId, | ||||
| 			Extra:    make(map[string][]interface{}), | ||||
| 		} | ||||
|  | ||||
| 		// handle mattermost post properties (override username and attachments) | ||||
| 		b.handleProps(rmsg, message) | ||||
|  | ||||
| 		// create a text for bridges that don't support native editing | ||||
| 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") { | ||||
| 			rmsg.Text = message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED { | ||||
| 			rmsg.Event = config.EventMsgDelete | ||||
| 		} | ||||
|  | ||||
| 		for _, id := range message.Post.FileIds { | ||||
| 			err := b.handleDownloadFile(rmsg, id) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("download failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Use nickname instead of username if defined | ||||
| 		if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||
| 			rmsg.Username = nick | ||||
| 		} | ||||
|  | ||||
| 		messages <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatterHook(messages chan *config.Message) { | ||||
| 	for { | ||||
| 		message := b.mh.Receive() | ||||
| 		b.Log.Debugf("Receiving from matterhook %#v", message) | ||||
| 		messages <- &config.Message{ | ||||
| 			UserID:   message.UserID, | ||||
| 			Username: message.UserName, | ||||
| 			Text:     message.Text, | ||||
| 			Channel:  message.ChannelName, | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	var err error | ||||
| 	var res, id string | ||||
| 	channelID := b.mc.GetChannelId(msg.Channel, b.TeamID) | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		msg.Text = fi.Comment | ||||
| 		if b.GetBool("PrefixMessagesWithNick") { | ||||
| 			msg.Text = msg.Username + msg.Text | ||||
| 		} | ||||
| 		res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, msg.ParentID, []string{id}) | ||||
| 	} | ||||
| 	return res, err | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleProps(rmsg *config.Message, message *matterclient.Message) { | ||||
| 	props := message.Post.Props | ||||
| 	if props == nil { | ||||
| 		return | ||||
| 	} | ||||
| 	if _, ok := props["override_username"].(string); ok { | ||||
| 		rmsg.Username = props["override_username"].(string) | ||||
| 	} | ||||
| 	if _, ok := props["attachments"].([]interface{}); ok { | ||||
| 		rmsg.Extra["attachments"] = props["attachments"].([]interface{}) | ||||
| 		if rmsg.Text == "" { | ||||
| 			for _, attachment := range rmsg.Extra["attachments"] { | ||||
| 				attach := attachment.(map[string]interface{}) | ||||
| 				if attach["text"].(string) != "" { | ||||
| 					rmsg.Text += attach["text"].(string) | ||||
| 					continue | ||||
| 				} | ||||
| 				if attach["fallback"].(string) != "" { | ||||
| 					rmsg.Text += attach["fallback"].(string) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										218
									
								
								bridge/mattermost/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,218 @@ | ||||
| package bmattermost | ||||
|  | ||||
| import ( | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| func (b *Bmattermost) 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"), | ||||
| 				BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 	case b.GetString("Token") != "": | ||||
| 		b.Log.Info("Connecting using token (sending)") | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	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.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 			matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 				BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) doConnectWebhookURL() error { | ||||
| 	b.Log.Info("Connecting using webhookurl (sending)") | ||||
| 	b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 		matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 			DisableServer: true}) | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		b.Log.Info("Connecting using token (receiving)") | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} else if b.GetString("Login") != "" { | ||||
| 		b.Log.Info("Connecting using login/password (receiving)") | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) apiLogin() error { | ||||
| 	password := b.GetString("Password") | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		password = "token=" + b.GetString("Token") | ||||
| 	} | ||||
|  | ||||
| 	b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server")) | ||||
| 	if b.GetBool("debug") { | ||||
| 		b.mc.SetLogLevel("debug") | ||||
| 	} | ||||
| 	b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify") | ||||
| 	b.mc.NoTLS = b.GetBool("NoTLS") | ||||
| 	b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server")) | ||||
| 	err := b.mc.Login() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.TeamID = b.mc.GetTeamId() | ||||
| 	go b.mc.WsReceiver() | ||||
| 	go b.mc.StatusLoop() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // replaceAction replace the message with the correct action (/me) code | ||||
| func (b *Bmattermost) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | ||||
| 		return strings.Replace(text, "*", "", -1), true | ||||
| 	} | ||||
| 	return text, false | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) cacheAvatar(msg *config.Message) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
| 	/* if we have a sha we have successfully uploaded the file to the media server, | ||||
| 	so we can now cache the sha */ | ||||
| 	if fi.SHA != "" { | ||||
| 		b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID) | ||||
| 		b.avatarMap[msg.UserID] = fi.SHA | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // sendWebhook uses the configured WebhookURL to send the message | ||||
| func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| 	// skip events | ||||
| 	if msg.Event != "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("PrefixMessagesWithNick") { | ||||
| 		msg.Text = msg.Username + msg.Text | ||||
| 	} | ||||
| 	if msg.Extra != nil { | ||||
| 		// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			rmsg := rmsg // scopelint | ||||
| 			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl")) | ||||
| 			matterMessage := matterhook.OMessage{ | ||||
| 				IconURL:  iconURL, | ||||
| 				Channel:  rmsg.Channel, | ||||
| 				UserName: rmsg.Username, | ||||
| 				Text:     rmsg.Text, | ||||
| 				Props:    make(map[string]interface{}), | ||||
| 			} | ||||
| 			matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 			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, | ||||
| 		Props:    make(map[string]interface{}), | ||||
| 	} | ||||
| 	if msg.Avatar != "" { | ||||
| 		matterMessage.IconURL = msg.Avatar | ||||
| 	} | ||||
| 	matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 	err := b.mh.Send(matterMessage) | ||||
| 	if err != nil { | ||||
| 		b.Log.Info(err) | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // skipMessages returns true if this message should not be handled | ||||
| func (b *Bmattermost) skipMessage(message *matterclient.Message) bool { | ||||
| 	// Handle join/leave | ||||
| 	if message.Type == "system_join_leave" || | ||||
| 		message.Type == "system_join_channel" || | ||||
| 		message.Type == "system_leave_channel" { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return true | ||||
| 		} | ||||
| 		b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		b.Remote <- config.Message{ | ||||
| 			Username: "system", | ||||
| 			Text:     message.Text, | ||||
| 			Channel:  message.Channel, | ||||
| 			Account:  b.Account, | ||||
| 			Event:    config.EventJoinLeave, | ||||
| 		} | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Handle edited messages | ||||
| 	if (message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED) && b.GetBool("EditDisable") { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from matterbridge | ||||
| 	if message.Post.Props != nil { | ||||
| 		if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok { | ||||
| 			b.Log.Debugf("sent by matterbridge, ignoring") | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from a user logged in as the bot | ||||
| 	if b.mc.User.Username == message.Username { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// if the message has reactions don't repost it (for now, until we can correlate reaction with message) | ||||
| 	if message.Post.HasReactions { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// ignore messages from other teams than ours | ||||
| 	if message.Raw.Data["team_id"].(string) != b.TeamID { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// only handle posted, edited or deleted events | ||||
| 	if !(message.Raw.Event == "posted" || message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED || | ||||
| 		message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED) { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
| @@ -3,7 +3,6 @@ package bmattermost | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| @@ -22,6 +21,8 @@ type Bmattermost struct { | ||||
| 	avatarMap map[string]string | ||||
| } | ||||
|  | ||||
| const mattermostPlugin = "mattermost.plugin" | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	b := &Bmattermost{Config: cfg, avatarMap: make(map[string]string)} | ||||
| 	b.uuid = xid.New().String() | ||||
| @@ -33,62 +34,31 @@ func (b *Bmattermost) Command(cmd string) string { | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) Connect() error { | ||||
| 	if b.Account == mattermostPlugin { | ||||
| 		return nil | ||||
| 	} | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		if b.GetString("WebhookURL") != "" { | ||||
| 			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||
| 			b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 					BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 		} else if b.GetString("Token") != "" { | ||||
| 			b.Log.Info("Connecting using token (sending)") | ||||
| 			err := b.apiLogin() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} else if b.GetString("Login") != "" { | ||||
| 			b.Log.Info("Connecting using login/password (sending)") | ||||
| 			err := b.apiLogin() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} else { | ||||
| 			b.Log.Info("Connecting using webhookbindaddress (receiving)") | ||||
| 			b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 				matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 					BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 		if err := b.doConnectWebhookBind(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 		return nil | ||||
| 	} | ||||
| 	if b.GetString("WebhookURL") != "" { | ||||
| 		b.Log.Info("Connecting using webhookurl (sending)") | ||||
| 		b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 			matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 				DisableServer: true}) | ||||
| 		if b.GetString("Token") != "" { | ||||
| 			b.Log.Info("Connecting using token (receiving)") | ||||
| 			err := b.apiLogin() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			go b.handleMatter() | ||||
| 		} else if b.GetString("Login") != "" { | ||||
| 			b.Log.Info("Connecting using login/password (receiving)") | ||||
| 			err := b.apiLogin() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 			go b.handleMatter() | ||||
| 	switch { | ||||
| 	case b.GetString("WebhookURL") != "": | ||||
| 		if err := b.doConnectWebhookURL(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 		return nil | ||||
| 	} else if b.GetString("Token") != "" { | ||||
| 	case b.GetString("Token") != "": | ||||
| 		b.Log.Info("Connecting using token (sending and receiving)") | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 	} else if b.GetString("Login") != "" { | ||||
| 	case b.GetString("Login") != "": | ||||
| 		b.Log.Info("Connecting using login/password (sending and receiving)") | ||||
| 		err := b.apiLogin() | ||||
| 		if err != nil { | ||||
| @@ -96,7 +66,8 @@ func (b *Bmattermost) Connect() error { | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 	} | ||||
| 	if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && b.GetString("Login") == "" && b.GetString("Token") == "" { | ||||
| 	if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && | ||||
| 		b.GetString("Login") == "" && b.GetString("Token") == "" { | ||||
| 		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured") | ||||
| 	} | ||||
| 	return nil | ||||
| @@ -107,9 +78,12 @@ func (b *Bmattermost) Disconnect() error { | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if b.Account == mattermostPlugin { | ||||
| 		return nil | ||||
| 	} | ||||
| 	// we can only join channels using the API | ||||
| 	if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" { | ||||
| 		id := b.mc.GetChannelId(channel.Name, "") | ||||
| 		id := b.mc.GetChannelId(channel.Name, b.TeamID) | ||||
| 		if id == "" { | ||||
| 			return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) | ||||
| 		} | ||||
| @@ -119,15 +93,18 @@ func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	if b.Account == mattermostPlugin { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EVENT_USER_ACTION { | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "*" + msg.Text + "*" | ||||
| 	} | ||||
|  | ||||
| 	// map the file SHA to our user (caches the avatar) | ||||
| 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		return b.cacheAvatar(&msg) | ||||
| 	} | ||||
|  | ||||
| @@ -137,7 +114,7 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| @@ -147,7 +124,9 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, ""), rmsg.Username+rmsg.Text) | ||||
| 			if _, err := b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, b.TeamID), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { | ||||
| 				b.Log.Errorf("PostMessage failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			return b.handleUploadFile(&msg) | ||||
| @@ -165,303 +144,5 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message | ||||
| 	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, ""), msg.Text) | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatter() { | ||||
| 	messages := make(chan *config.Message) | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		b.Log.Debugf("Choosing webhooks based receiving") | ||||
| 		go b.handleMatterHook(messages) | ||||
| 	} else { | ||||
| 		if b.GetString("Token") != "" { | ||||
| 			b.Log.Debugf("Choosing token based receiving") | ||||
| 		} else { | ||||
| 			b.Log.Debugf("Choosing login/password based receiving") | ||||
| 		} | ||||
| 		go b.handleMatterClient(messages) | ||||
| 	} | ||||
| 	var ok bool | ||||
| 	for message := range messages { | ||||
| 		message.Avatar = helper.GetAvatar(b.avatarMap, message.UserID, b.General) | ||||
| 		message.Account = b.Account | ||||
| 		message.Text, ok = b.replaceAction(message.Text) | ||||
| 		if ok { | ||||
| 			message.Event = config.EVENT_USER_ACTION | ||||
| 		} | ||||
| 		b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account) | ||||
| 		b.Log.Debugf("<= Message is %#v", message) | ||||
| 		b.Remote <- *message | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { | ||||
| 	for message := range b.mc.MessageChan { | ||||
| 		b.Log.Debugf("%#v", message.Raw.Data) | ||||
|  | ||||
| 		if b.skipMessage(message) { | ||||
| 			b.Log.Debugf("Skipped message: %#v", message) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 		if b.General.MediaServerUpload != "" || b.General.MediaDownloadPath != "" { | ||||
| 			b.handleDownloadAvatar(message.UserID, message.Channel) | ||||
| 		} | ||||
|  | ||||
| 		b.Log.Debugf("== Receiving event %#v", message) | ||||
|  | ||||
| 		rmsg := &config.Message{Username: message.Username, UserID: message.UserID, Channel: message.Channel, Text: message.Text, ID: message.Post.Id, Extra: make(map[string][]interface{})} | ||||
|  | ||||
| 		// handle mattermost post properties (override username and attachments) | ||||
| 		props := message.Post.Props | ||||
| 		if props != nil { | ||||
| 			if _, ok := props["override_username"].(string); ok { | ||||
| 				rmsg.Username = props["override_username"].(string) | ||||
| 			} | ||||
| 			if _, ok := props["attachments"].([]interface{}); ok { | ||||
| 				rmsg.Extra["attachments"] = props["attachments"].([]interface{}) | ||||
| 				if rmsg.Text == "" { | ||||
| 					for _, attachment := range rmsg.Extra["attachments"] { | ||||
| 						attach := attachment.(map[string]interface{}) | ||||
| 						if attach["text"].(string) != "" { | ||||
| 							rmsg.Text += attach["text"].(string) | ||||
| 							continue | ||||
| 						} | ||||
| 						if attach["fallback"].(string) != "" { | ||||
| 							rmsg.Text += attach["fallback"].(string) | ||||
| 						} | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// create a text for bridges that don't support native editing | ||||
| 		if message.Raw.Event == "post_edited" && !b.GetBool("EditDisable") { | ||||
| 			rmsg.Text = message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		if message.Raw.Event == "post_deleted" { | ||||
| 			rmsg.Event = config.EVENT_MSG_DELETE | ||||
| 		} | ||||
|  | ||||
| 		if len(message.Post.FileIds) > 0 { | ||||
| 			for _, id := range message.Post.FileIds { | ||||
| 				err := b.handleDownloadFile(rmsg, id) | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("download failed: %s", err) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		// Use nickname instead of username if defined | ||||
| 		if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||
| 			rmsg.Username = nick | ||||
| 		} | ||||
|  | ||||
| 		messages <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatterHook(messages chan *config.Message) { | ||||
| 	for { | ||||
| 		message := b.mh.Receive() | ||||
| 		b.Log.Debugf("Receiving from matterhook %#v", message) | ||||
| 		messages <- &config.Message{UserID: message.UserID, Username: message.UserName, Text: message.Text, Channel: message.ChannelName} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) apiLogin() error { | ||||
| 	password := b.GetString("Password") | ||||
| 	if b.GetString("Token") != "" { | ||||
| 		password = "MMAUTHTOKEN=" + b.GetString("Token") | ||||
| 	} | ||||
|  | ||||
| 	b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server")) | ||||
| 	if b.GetBool("debug") { | ||||
| 		b.mc.SetLogLevel("debug") | ||||
| 	} | ||||
| 	b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify") | ||||
| 	b.mc.NoTLS = b.GetBool("NoTLS") | ||||
| 	b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server")) | ||||
| 	err := b.mc.Login() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.TeamID = b.mc.GetTeamId() | ||||
| 	go b.mc.WsReceiver() | ||||
| 	go b.mc.StatusLoop() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // replaceAction replace the message with the correct action (/me) code | ||||
| func (b *Bmattermost) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") { | ||||
| 		return strings.Replace(text, "*", "", -1), true | ||||
| 	} | ||||
| 	return text, false | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) cacheAvatar(msg *config.Message) (string, error) { | ||||
| 	fi := msg.Extra["file"][0].(config.FileInfo) | ||||
| 	/* if we have a sha we have successfully uploaded the file to the media server, | ||||
| 	so we can now cache the sha */ | ||||
| 	if fi.SHA != "" { | ||||
| 		b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID) | ||||
| 		b.avatarMap[msg.UserID] = fi.SHA | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||
| // logs an error message if it fails | ||||
| func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||
| 	rmsg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: userid, Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})} | ||||
| 	if _, ok := b.avatarMap[userid]; !ok { | ||||
| 		data, resp := b.mc.Client.GetProfileImage(userid, "") | ||||
| 		if resp.Error != nil { | ||||
| 			b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error) | ||||
| 			return | ||||
| 		} | ||||
| 		err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) | ||||
| 		if err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 			return | ||||
| 		} | ||||
| 		helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General) | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error { | ||||
| 	url, _ := b.mc.Client.GetFileLink(id) | ||||
| 	finfo, resp := b.mc.Client.GetFileInfo(id) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, resp := b.mc.Client.DownloadFile(id, true) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	var err error | ||||
| 	var res, id string | ||||
| 	channelID := b.mc.GetChannelId(msg.Channel, "") | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		msg.Text = fi.Comment | ||||
| 		if b.GetBool("PrefixMessagesWithNick") { | ||||
| 			msg.Text = msg.Username + msg.Text | ||||
| 		} | ||||
| 		res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, []string{id}) | ||||
| 	} | ||||
| 	return res, err | ||||
| } | ||||
|  | ||||
| // sendWebhook uses the configured WebhookURL to send the message | ||||
| func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| 	// skip events | ||||
| 	if msg.Event != "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("PrefixMessagesWithNick") { | ||||
| 		msg.Text = msg.Username + msg.Text | ||||
| 	} | ||||
| 	if msg.Extra != nil { | ||||
| 		// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl")) | ||||
| 			matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text, Props: make(map[string]interface{})} | ||||
| 			matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 			b.mh.Send(matterMessage) | ||||
| 		} | ||||
|  | ||||
| 		// webhook doesn't support file uploads, so we add the url manually | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text += fi.URL | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	iconURL := config.GetIconURL(&msg, b.GetString("iconurl")) | ||||
| 	matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: msg.Channel, UserName: msg.Username, Text: msg.Text, Props: make(map[string]interface{})} | ||||
| 	if msg.Avatar != "" { | ||||
| 		matterMessage.IconURL = msg.Avatar | ||||
| 	} | ||||
| 	matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 	err := b.mh.Send(matterMessage) | ||||
| 	if err != nil { | ||||
| 		b.Log.Info(err) | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // skipMessages returns true if this message should not be handled | ||||
| func (b *Bmattermost) skipMessage(message *matterclient.Message) bool { | ||||
| 	// Handle join/leave | ||||
| 	if message.Type == "system_join_leave" || | ||||
| 		message.Type == "system_join_channel" || | ||||
| 		message.Type == "system_leave_channel" { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return true | ||||
| 		} | ||||
| 		b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EVENT_JOIN_LEAVE} | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Handle edited messages | ||||
| 	if (message.Raw.Event == "post_edited") && b.GetBool("EditDisable") { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from matterbridge | ||||
| 	if message.Post.Props != nil { | ||||
| 		if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok { | ||||
| 			b.Log.Debugf("sent by matterbridge, ignoring") | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from a user logged in as the bot | ||||
| 	if b.mc.User.Username == message.Username { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// if the message has reactions don't repost it (for now, until we can correlate reaction with message) | ||||
| 	if message.Post.HasReactions { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// ignore messages from other teams than ours | ||||
| 	if message.Raw.Data["team_id"].(string) != b.TeamID { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// only handle posted, edited or deleted events | ||||
| 	if !(message.Raw.Event == "posted" || message.Raw.Event == "post_edited" || message.Raw.Event == "post_deleted") { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| 	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text, msg.ParentID) | ||||
| } | ||||
|   | ||||
							
								
								
									
										69
									
								
								bridge/rocketchat/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,69 @@ | ||||
| package brocketchat | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| ) | ||||
|  | ||||
| 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) handleRocketClient(messages chan *config.Message) { | ||||
| 	for message := range b.messageChan { | ||||
| 		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, | ||||
| 		} | ||||
| 		messages <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| 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 | ||||
| } | ||||
							
								
								
									
										198
									
								
								bridge/rocketchat/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,198 @@ | ||||
| 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")} | ||||
| 	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 { | ||||
| 			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,25 +1,37 @@ | ||||
| package brocketchat | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"sync" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/hook/rockethook" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"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 MMhook struct { | ||||
| type Brocketchat struct { | ||||
| 	mh *matterhook.Client | ||||
| 	rh *rockethook.Client | ||||
| } | ||||
|  | ||||
| type Brocketchat struct { | ||||
| 	MMhook | ||||
| 	c  *realtime.Client | ||||
| 	r  *rest.Client | ||||
| 	*bridge.Config | ||||
| 	messageChan chan models.Message | ||||
| 	channelMap  map[string]string | ||||
| 	user        *models.User | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Brocketchat{Config: cfg} | ||||
| 	b := &Brocketchat{Config: cfg} | ||||
| 	b.messageChan = make(chan models.Message) | ||||
| 	b.channelMap = make(map[string]string) | ||||
| 	b.Log.Debugf("enabling rocketchat") | ||||
| 	return b | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) Command(cmd string) string { | ||||
| @@ -27,69 +39,118 @@ func (b *Brocketchat) Command(cmd string) string { | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) Connect() error { | ||||
| 	b.Log.Info("Connecting webhooks") | ||||
| 	b.mh = matterhook.New(b.GetString("WebhookURL"), | ||||
| 		matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), | ||||
| 			DisableServer: true}) | ||||
| 	b.rh = rockethook.New(b.GetString("WebhookURL"), rockethook.Config{BindAddress: b.GetString("WebhookBindAddress")}) | ||||
| 	go b.handleRocketHook() | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		if err := b.doConnectWebhookBind(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleRocket() | ||||
| 		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 | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) Disconnect() error { | ||||
| 	return nil | ||||
|  | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	if b.c == nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	id, err := b.c.GetChannelId(channel.Name) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Lock() | ||||
| 	b.channelMap[id] = channel.Name | ||||
| 	b.Unlock() | ||||
| 	mychannel := &models.Channel{ID: id, Name: 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 | ||||
| } | ||||
|  | ||||
| func (b *Brocketchat) Send(msg config.Message) (string, error) { | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 		return "", nil | ||||
| 	channel := &models.Channel{ID: b.getChannelID(msg.Channel), Name: msg.Channel} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		return msg.ID, b.c.DeleteMessage(&models.Message{ID: msg.ID}) | ||||
| 	} | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// 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 { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl")) | ||||
| 			matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text} | ||||
| 			b.mh.Send(matterMessage) | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text += fi.URL | ||||
| 				} | ||||
| 			smsg := &models.Message{ | ||||
| 				RoomID: b.getChannelID(rmsg.Channel), | ||||
| 				Msg:    rmsg.Username + rmsg.Text, | ||||
| 				PostMessage: models.PostMessage{ | ||||
| 					Avatar: rmsg.Avatar, | ||||
| 					Alias:  rmsg.Username, | ||||
| 				}, | ||||
| 			} | ||||
| 			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")) | ||||
| 	matterMessage := matterhook.OMessage{IconURL: iconURL} | ||||
| 	matterMessage.Channel = msg.Channel | ||||
| 	matterMessage.UserName = msg.Username | ||||
| 	matterMessage.Type = "" | ||||
| 	matterMessage.Text = msg.Text | ||||
| 	err := b.mh.Send(matterMessage) | ||||
| 	if err != nil { | ||||
| 		b.Log.Info(err) | ||||
| 	smsg := &models.Message{ | ||||
| 		RoomID: channel.ID, | ||||
| 		Msg:    msg.Text, | ||||
| 		PostMessage: models.PostMessage{ | ||||
| 			Avatar: msg.Avatar, | ||||
| 			Alias:  msg.Username, | ||||
| 		}, | ||||
| 	} | ||||
|  | ||||
| 	rmsg, err := b.c.SendMessage(smsg) | ||||
| 	if rmsg == nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| 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} | ||||
| 	} | ||||
| 	return rmsg.ID, err | ||||
| } | ||||
|   | ||||
							
								
								
									
										379
									
								
								bridge/slack/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,379 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/nlopes/slack" | ||||
| ) | ||||
|  | ||||
| func (b *Bslack) handleSlack() { | ||||
| 	messages := make(chan *config.Message) | ||||
| 	if b.GetString(incomingWebhookConfig) != "" { | ||||
| 		b.Log.Debugf("Choosing webhooks based receiving") | ||||
| 		go b.handleMatterHook(messages) | ||||
| 	} else { | ||||
| 		b.Log.Debugf("Choosing token based receiving") | ||||
| 		go b.handleSlackClient(messages) | ||||
| 	} | ||||
| 	time.Sleep(time.Second) | ||||
| 	b.Log.Debug("Start listening for Slack messages") | ||||
| 	for message := range messages { | ||||
| 		if message.Event != config.EventUserTyping { | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account) | ||||
| 		} | ||||
|  | ||||
| 		// cleanup the message | ||||
| 		message.Text = b.replaceMention(message.Text) | ||||
| 		message.Text = b.replaceVariable(message.Text) | ||||
| 		message.Text = b.replaceChannel(message.Text) | ||||
| 		message.Text = b.replaceURL(message.Text) | ||||
| 		message.Text = html.UnescapeString(message.Text) | ||||
|  | ||||
| 		// Add the avatar | ||||
| 		message.Avatar = b.getAvatar(message.UserID) | ||||
|  | ||||
| 		b.Log.Debugf("<= Message is %#v", message) | ||||
| 		b.Remote <- *message | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||
| 	for msg := range b.rtm.IncomingEvents { | ||||
| 		if msg.Type != sUserTyping && msg.Type != sLatencyReport { | ||||
| 			b.Log.Debugf("== Receiving event %#v", msg.Data) | ||||
| 		} | ||||
| 		switch ev := msg.Data.(type) { | ||||
| 		case *slack.UserTypingEvent: | ||||
| 			if !b.GetBool("ShowUserTyping") { | ||||
| 				continue | ||||
| 			} | ||||
| 			rmsg, err := b.handleTypingEvent(ev) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("%#v", err) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			messages <- rmsg | ||||
| 		case *slack.MessageEvent: | ||||
| 			if b.skipMessageEvent(ev) { | ||||
| 				b.Log.Debugf("Skipped message: %#v", ev) | ||||
| 				continue | ||||
| 			} | ||||
| 			rmsg, err := b.handleMessageEvent(ev) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("%#v", err) | ||||
| 				continue | ||||
| 			} | ||||
| 			messages <- rmsg | ||||
| 		case *slack.OutgoingErrorEvent: | ||||
| 			b.Log.Debugf("%#v", ev.Error()) | ||||
| 		case *slack.ChannelJoinedEvent: | ||||
| 			// When we join a channel we update the full list of users as | ||||
| 			// well as the information for the channel that we joined as this | ||||
| 			// should now tell that we are a member of it. | ||||
| 			b.channelsMutex.Lock() | ||||
| 			b.channelsByID[ev.Channel.ID] = &ev.Channel | ||||
| 			b.channelsByName[ev.Channel.Name] = &ev.Channel | ||||
| 			b.channelsMutex.Unlock() | ||||
| 		case *slack.ConnectedEvent: | ||||
| 			b.si = ev.Info | ||||
| 			b.populateChannels(true) | ||||
| 			b.populateUsers(true) | ||||
| 		case *slack.InvalidAuthEvent: | ||||
| 			b.Log.Fatalf("Invalid Token %#v", ev) | ||||
| 		case *slack.ConnectionErrorEvent: | ||||
| 			b.Log.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj) | ||||
| 		case *slack.MemberJoinedChannelEvent: | ||||
| 			b.populateUser(ev.User) | ||||
| 		case *slack.LatencyReport: | ||||
| 			continue | ||||
| 		default: | ||||
| 			b.Log.Debugf("Unhandled incoming event: %T", ev) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleMatterHook(messages chan *config.Message) { | ||||
| 	for { | ||||
| 		message := b.mh.Receive() | ||||
| 		b.Log.Debugf("receiving from matterhook (slack) %#v", message) | ||||
| 		if message.UserName == "slackbot" { | ||||
| 			continue | ||||
| 		} | ||||
| 		messages <- &config.Message{ | ||||
| 			Username: message.UserName, | ||||
| 			Text:     message.Text, | ||||
| 			Channel:  message.ChannelName, | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // skipMessageEvent skips event that need to be skipped :-) | ||||
| func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool { | ||||
| 	switch ev.SubType { | ||||
| 	case sChannelLeave, sChannelJoin: | ||||
| 		return b.GetBool(noSendJoinConfig) | ||||
| 	case sPinnedItem, sUnpinnedItem: | ||||
| 		return true | ||||
| 	case sChannelTopic, sChannelPurpose: | ||||
| 		// Skip the event if our bot/user account changed the topic/purpose | ||||
| 		if ev.User == b.si.User.ID { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Skip any messages that we made ourselves or from 'slackbot' (see #527). | ||||
| 	if ev.Username == sSlackBotUser || | ||||
| 		(b.rtm != nil && ev.Username == b.si.User.Name) || | ||||
| 		(len(ev.Attachments) > 0 && ev.Attachments[0].CallbackID == "matterbridge_"+b.uuid) { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// It seems ev.SubMessage.Edited == nil when slack unfurls. | ||||
| 	// Do not forward these messages. See Github issue #266. | ||||
| 	if ev.SubMessage != nil && | ||||
| 		ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp && | ||||
| 		ev.SubMessage.Edited == nil { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	if len(ev.Files) > 0 { | ||||
| 		return b.filesCached(ev.Files) | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (b *Bslack) filesCached(files []slack.File) bool { | ||||
| 	for i := range files { | ||||
| 		if !b.fileCached(&files[i]) { | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| // handleMessageEvent handles the message events. Together with any called sub-methods, | ||||
| // this method implements the following event processing pipeline: | ||||
| // | ||||
| // 1. Check if the message should be ignored. | ||||
| //    NOTE: This is not actually part of the method below but is done just before it | ||||
| //          is called via the 'skipMessageEvent()' method. | ||||
| // 2. Populate the Matterbridge message that will be sent to the router based on the | ||||
| //    received event and logic that is common to all events that are not skipped. | ||||
| // 3. Detect and handle any message that is "status" related (think join channel, etc.). | ||||
| //    This might result in an early exit from the pipeline and passing of the | ||||
| //    pre-populated message to the Matterbridge router. | ||||
| // 4. Handle the specific case of messages that edit existing messages depending on | ||||
| //    configuration. | ||||
| // 5. Handle any attachments of the received event. | ||||
| // 6. Check that the Matterbridge message that we end up with after at the end of the | ||||
| //    pipeline is valid before sending it to the Matterbridge router. | ||||
| func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, error) { | ||||
| 	rmsg, err := b.populateReceivedMessage(ev) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	// Handle some message types early. | ||||
| 	if b.handleStatusEvent(ev, rmsg) { | ||||
| 		return rmsg, nil | ||||
| 	} | ||||
|  | ||||
| 	b.handleAttachments(ev, rmsg) | ||||
|  | ||||
| 	// Verify that we have the right information and the message | ||||
| 	// is well-formed before sending it out to the router. | ||||
| 	if len(ev.Files) == 0 && (rmsg.Text == "" || rmsg.Username == "") { | ||||
| 		if ev.BotID != "" { | ||||
| 			// This is probably a webhook we couldn't resolve. | ||||
| 			return nil, fmt.Errorf("message handling resulted in an empty bot message (probably an incoming webhook we couldn't resolve): %#v", ev) | ||||
| 		} | ||||
| 		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 rmsg, nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) bool { | ||||
| 	switch ev.SubType { | ||||
| 	case sChannelJoined, sMemberJoined: | ||||
| 		// There's no further processing needed on channel events | ||||
| 		// so we return 'true'. | ||||
| 		return true | ||||
| 	case sChannelJoin, sChannelLeave: | ||||
| 		rmsg.Username = sSystemUser | ||||
| 		rmsg.Event = config.EventJoinLeave | ||||
| 	case sChannelTopic, sChannelPurpose: | ||||
| 		b.populateChannels(false) | ||||
| 		rmsg.Event = config.EventTopicChange | ||||
| 	case sMessageChanged: | ||||
| 		rmsg.Text = ev.SubMessage.Text | ||||
| 		// handle deleted thread starting messages | ||||
| 		if ev.SubMessage.Text == "This message was deleted." { | ||||
| 			rmsg.Event = config.EventMsgDelete | ||||
| 			return true | ||||
| 		} | ||||
| 	case sMessageDeleted: | ||||
| 		rmsg.Text = config.EventMsgDelete | ||||
| 		rmsg.Event = config.EventMsgDelete | ||||
| 		rmsg.ID = ev.DeletedTimestamp | ||||
| 		// If a message is being deleted we do not need to process | ||||
| 		// the event any further so we return 'true'. | ||||
| 		return true | ||||
| 	case sMeMessage: | ||||
| 		rmsg.Event = config.EventUserAction | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) { | ||||
| 	// File comments are set by the system (because there is no username given). | ||||
| 	if ev.SubType == sFileComment { | ||||
| 		rmsg.Username = sSystemUser | ||||
| 	} | ||||
|  | ||||
| 	// See if we have some text in the attachments. | ||||
| 	if rmsg.Text == "" { | ||||
| 		for _, attach := range ev.Attachments { | ||||
| 			if attach.Text != "" { | ||||
| 				if attach.Title != "" { | ||||
| 					rmsg.Text = attach.Title + "\n" | ||||
| 				} | ||||
| 				rmsg.Text += attach.Text | ||||
| 			} else { | ||||
| 				rmsg.Text = attach.Fallback | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Save the attachments, so that we can send them to other slack (compatible) bridges. | ||||
| 	if len(ev.Attachments) > 0 { | ||||
| 		rmsg.Extra[sSlackAttachment] = append(rmsg.Extra[sSlackAttachment], ev.Attachments) | ||||
| 	} | ||||
|  | ||||
| 	// If we have files attached, download them (in memory) and put a pointer to it in msg.Extra. | ||||
| 	for i := range ev.Files { | ||||
| 		if err := b.handleDownloadFile(rmsg, &ev.Files[i], false); err != nil { | ||||
| 			b.Log.Errorf("Could not download incoming file: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message, error) { | ||||
| 	channelInfo, err := b.getChannelByID(ev.Channel) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return &config.Message{ | ||||
| 		Channel: channelInfo.Name, | ||||
| 		Account: b.Account, | ||||
| 		Event:   config.EventUserTyping, | ||||
| 	}, nil | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File, retry bool) error { | ||||
| 	if b.fileCached(file) { | ||||
| 		return nil | ||||
| 	} | ||||
| 	// Check that the file is neither too large nor blacklisted. | ||||
| 	if err := helper.HandleDownloadSize(b.Log, rmsg, file.Name, int64(file.Size), b.General); err != nil { | ||||
| 		b.Log.WithError(err).Infof("Skipping download of incoming file.") | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// Actually download the file. | ||||
| 	data, err := helper.DownloadFileAuth(file.URLPrivateDownload, "Bearer "+b.GetString(tokenConfig)) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("download %s failed %#v", file.URLPrivateDownload, err) | ||||
| 	} | ||||
|  | ||||
| 	if len(*data) != file.Size && !retry { | ||||
| 		b.Log.Debugf("Data size (%d) is not equal to size declared (%d)\n", len(*data), file.Size) | ||||
| 		time.Sleep(1 * time.Second) | ||||
| 		return b.handleDownloadFile(rmsg, file, true) | ||||
| 	} | ||||
|  | ||||
| 	// If a comment is attached to the file(s) it is in the 'Text' field of the Slack messge event | ||||
| 	// and should be added as comment to only one of the files. We reset the 'Text' field to ensure | ||||
| 	// that the comment is not duplicated. | ||||
| 	comment := rmsg.Text | ||||
| 	rmsg.Text = "" | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, file.Name, comment, file.URLPrivateDownload, data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleGetChannelMembers handles messages containing the GetChannelMembers event | ||||
| // Sends a message to the router containing *config.ChannelMembers | ||||
| func (b *Bslack) handleGetChannelMembers(rmsg *config.Message) bool { | ||||
| 	if rmsg.Event != config.EventGetChannelMembers { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	cMembers := config.ChannelMembers{} | ||||
|  | ||||
| 	b.channelMembersMutex.RLock() | ||||
|  | ||||
| 	for channelID, members := range b.channelMembers { | ||||
| 		for _, member := range members { | ||||
| 			channelName := "" | ||||
| 			userName := "" | ||||
| 			userNick := "" | ||||
| 			user := b.getUser(member) | ||||
| 			if user != nil { | ||||
| 				userName = user.Name | ||||
| 				userNick = user.Profile.DisplayName | ||||
| 			} | ||||
| 			channel, _ := b.getChannelByID(channelID) | ||||
| 			if channel != nil { | ||||
| 				channelName = channel.Name | ||||
| 			} | ||||
| 			cMember := config.ChannelMember{ | ||||
| 				Username:    userName, | ||||
| 				Nick:        userNick, | ||||
| 				UserID:      member, | ||||
| 				ChannelID:   channelID, | ||||
| 				ChannelName: channelName, | ||||
| 			} | ||||
| 			cMembers = append(cMembers, cMember) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.channelMembersMutex.RUnlock() | ||||
|  | ||||
| 	extra := make(map[string][]interface{}) | ||||
| 	extra[config.EventGetChannelMembers] = append(extra[config.EventGetChannelMembers], cMembers) | ||||
| 	msg := config.Message{ | ||||
| 		Extra:   extra, | ||||
| 		Event:   config.EventGetChannelMembers, | ||||
| 		Account: b.Account, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("sending msg to remote %#v", msg) | ||||
| 	b.Remote <- msg | ||||
|  | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| // fileCached implements Matterbridge's caching logic for files | ||||
| // shared via Slack. | ||||
| // | ||||
| // We consider that a file was cached if its ID was added in the last minute or | ||||
| // it's name was registered in the last 10 seconds. This ensures that an | ||||
| // identically named file but with different content will be uploaded correctly | ||||
| // (the assumption is that such name collisions will not occur within the given | ||||
| // timeframes). | ||||
| func (b *Bslack) fileCached(file *slack.File) bool { | ||||
| 	if ts, ok := b.cache.Get("file" + file.ID); ok && time.Since(ts.(time.Time)) < time.Minute { | ||||
| 		return true | ||||
| 	} else if ts, ok = b.cache.Get("filename" + file.Name); ok && time.Since(ts.(time.Time)) < 10*time.Second { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
							
								
								
									
										441
									
								
								bridge/slack/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,441 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
| 	"fmt" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/nlopes/slack" | ||||
| ) | ||||
|  | ||||
| func (b *Bslack) getUser(id string) *slack.User { | ||||
| 	b.usersMutex.RLock() | ||||
| 	user, ok := b.users[id] | ||||
| 	b.usersMutex.RUnlock() | ||||
| 	if ok { | ||||
| 		return user | ||||
| 	} | ||||
| 	b.populateUser(id) | ||||
| 	b.usersMutex.RLock() | ||||
| 	defer b.usersMutex.RUnlock() | ||||
|  | ||||
| 	return b.users[id] | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getUsername(id string) string { | ||||
| 	if user := b.getUser(id); user != nil { | ||||
| 		if user.Profile.DisplayName != "" { | ||||
| 			return user.Profile.DisplayName | ||||
| 		} | ||||
| 		return user.Name | ||||
| 	} | ||||
| 	b.Log.Warnf("Could not find user with ID '%s'", id) | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getAvatar(id string) string { | ||||
| 	if user := b.getUser(id); user != nil { | ||||
| 		return user.Profile.Image48 | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannel(channel string) (*slack.Channel, error) { | ||||
| 	if strings.HasPrefix(channel, "ID:") { | ||||
| 		return b.getChannelByID(strings.TrimPrefix(channel, "ID:")) | ||||
| 	} | ||||
| 	return b.getChannelByName(channel) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelByName(name string) (*slack.Channel, error) { | ||||
| 	return b.getChannelBy(name, b.channelsByName) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelByID(ID string) (*slack.Channel, error) { | ||||
| 	return b.getChannelBy(ID, b.channelsByID) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelBy(lookupKey string, lookupMap map[string]*slack.Channel) (*slack.Channel, error) { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	if channel, ok := lookupMap[lookupKey]; ok { | ||||
| 		return channel, nil | ||||
| 	} | ||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, lookupKey) | ||||
| } | ||||
|  | ||||
| const minimumRefreshInterval = 10 * time.Second | ||||
|  | ||||
| func (b *Bslack) populateUser(userID string) { | ||||
| 	b.usersMutex.RLock() | ||||
| 	_, exists := b.users[userID] | ||||
| 	b.usersMutex.RUnlock() | ||||
| 	if exists { | ||||
| 		// already in cache | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	user, err := b.sc.GetUserInfo(userID) | ||||
| 	if err != nil { | ||||
| 		b.Log.Debugf("GetUserInfo failed for %v: %v", userID, err) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	b.usersMutex.Lock() | ||||
| 	b.users[userID] = user | ||||
| 	b.usersMutex.Unlock() | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateUsers(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (time.Now().Before(b.earliestUserRefresh) || b.refreshInProgress) { | ||||
| 		b.Log.Debugf("Not refreshing user list as it was done less than %v ago.", | ||||
| 			minimumRefreshInterval) | ||||
| 		b.refreshMutex.Unlock() | ||||
|  | ||||
| 		return | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newUsers := map[string]*slack.User{} | ||||
| 	pagination := b.sc.GetUsersPaginated(slack.GetUsersOptionLimit(200)) | ||||
| 	count := 0 | ||||
| 	for { | ||||
| 		var err error | ||||
| 		pagination, err = pagination.Next(context.Background()) | ||||
| 		time.Sleep(time.Second) | ||||
| 		if err != nil { | ||||
| 			if pagination.Done(err) { | ||||
| 				break | ||||
| 			} | ||||
|  | ||||
| 			if err = b.handleRateLimit(err); err != nil { | ||||
| 				b.Log.Errorf("Could not retrieve users: %#v", err) | ||||
| 				return | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		for i := range pagination.Users { | ||||
| 			newUsers[pagination.Users[i].ID] = &pagination.Users[i] | ||||
| 		} | ||||
| 		b.Log.Debugf("getting %d users", len(pagination.Users)) | ||||
| 		count++ | ||||
| 		// more > 2000 users, slack will complain and ratelimit. break | ||||
| 		if count > 10 { | ||||
| 			b.Log.Info("Large slack detected > 2000 users, skipping loading complete userlist.") | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.usersMutex.Lock() | ||||
| 	defer b.usersMutex.Unlock() | ||||
| 	b.users = newUsers | ||||
|  | ||||
| 	b.refreshMutex.Lock() | ||||
| 	defer b.refreshMutex.Unlock() | ||||
| 	b.earliestUserRefresh = time.Now().Add(minimumRefreshInterval) | ||||
| 	b.refreshInProgress = false | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateChannels(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if !wait && (time.Now().Before(b.earliestChannelRefresh) || b.refreshInProgress) { | ||||
| 		b.Log.Debugf("Not refreshing channel list as it was done less than %v seconds ago.", | ||||
| 			minimumRefreshInterval) | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		return | ||||
| 	} | ||||
| 	for b.refreshInProgress { | ||||
| 		b.refreshMutex.Unlock() | ||||
| 		time.Sleep(time.Second) | ||||
| 		b.refreshMutex.Lock() | ||||
| 	} | ||||
| 	b.refreshInProgress = true | ||||
| 	b.refreshMutex.Unlock() | ||||
|  | ||||
| 	newChannelsByID := map[string]*slack.Channel{} | ||||
| 	newChannelsByName := map[string]*slack.Channel{} | ||||
| 	newChannelMembers := make(map[string][]string) | ||||
|  | ||||
| 	// We only retrieve public and private channels, not IMs | ||||
| 	// and MPIMs as those do not have a channel name. | ||||
| 	queryParams := &slack.GetConversationsParameters{ | ||||
| 		ExcludeArchived: "true", | ||||
| 		Types:           []string{"public_channel,private_channel"}, | ||||
| 	} | ||||
| 	for { | ||||
| 		channels, nextCursor, err := b.sc.GetConversations(queryParams) | ||||
| 		if err != nil { | ||||
| 			if err = b.handleRateLimit(err); err != nil { | ||||
| 				b.Log.Errorf("Could not retrieve channels: %#v", err) | ||||
| 				return | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		for i := range channels { | ||||
| 			newChannelsByID[channels[i].ID] = &channels[i] | ||||
| 			newChannelsByName[channels[i].Name] = &channels[i] | ||||
| 			// also find all the members in every channel | ||||
| 			// comment for now, issues on big slacks | ||||
| 			/* | ||||
| 				members, err := b.getUsersInConversation(channels[i].ID) | ||||
| 				if err != nil { | ||||
| 					if err = b.handleRateLimit(err); err != nil { | ||||
| 						b.Log.Errorf("Could not retrieve channel members: %#v", err) | ||||
| 						return | ||||
| 					} | ||||
| 					continue | ||||
| 				} | ||||
| 				newChannelMembers[channels[i].ID] = members | ||||
| 			*/ | ||||
| 		} | ||||
|  | ||||
| 		if nextCursor == "" { | ||||
| 			break | ||||
| 		} | ||||
| 		queryParams.Cursor = nextCursor | ||||
| 	} | ||||
|  | ||||
| 	b.channelsMutex.Lock() | ||||
| 	defer b.channelsMutex.Unlock() | ||||
| 	b.channelsByID = newChannelsByID | ||||
| 	b.channelsByName = newChannelsByName | ||||
|  | ||||
| 	b.channelMembersMutex.Lock() | ||||
| 	defer b.channelMembersMutex.Unlock() | ||||
| 	b.channelMembers = newChannelMembers | ||||
|  | ||||
| 	b.refreshMutex.Lock() | ||||
| 	defer b.refreshMutex.Unlock() | ||||
| 	b.earliestChannelRefresh = time.Now().Add(minimumRefreshInterval) | ||||
| 	b.refreshInProgress = false | ||||
| } | ||||
|  | ||||
| // populateReceivedMessage shapes the initial Matterbridge message that we will forward to the | ||||
| // router before we apply message-dependent modifications. | ||||
| func (b *Bslack) populateReceivedMessage(ev *slack.MessageEvent) (*config.Message, error) { | ||||
| 	// Use our own func because rtm.GetChannelInfo doesn't work for private channels. | ||||
| 	channel, err := b.getChannelByID(ev.Channel) | ||||
| 	if err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	rmsg := &config.Message{ | ||||
| 		Text:     ev.Text, | ||||
| 		Channel:  channel.Name, | ||||
| 		Account:  b.Account, | ||||
| 		ID:       ev.Timestamp, | ||||
| 		Extra:    make(map[string][]interface{}), | ||||
| 		ParentID: ev.ThreadTimestamp, | ||||
| 		Protocol: b.Protocol, | ||||
| 	} | ||||
| 	if b.useChannelID { | ||||
| 		rmsg.Channel = "ID:" + channel.ID | ||||
| 	} | ||||
|  | ||||
| 	// Handle 'edit' messages. | ||||
| 	if ev.SubMessage != nil && !b.GetBool(editDisableConfig) { | ||||
| 		rmsg.ID = ev.SubMessage.Timestamp | ||||
| 		if ev.SubMessage.ThreadTimestamp != ev.SubMessage.Timestamp { | ||||
| 			b.Log.Debugf("SubMessage %#v", ev.SubMessage) | ||||
| 			rmsg.Text = ev.SubMessage.Text + b.GetString(editSuffixConfig) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// 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 { | ||||
| 		return nil, err | ||||
| 	} | ||||
| 	return rmsg, err | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateMessageWithUserInfo(ev *slack.MessageEvent, rmsg *config.Message) error { | ||||
| 	if ev.SubType == sMessageDeleted || ev.SubType == sFileComment { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// First, deal with bot-originating messages but only do so when not using webhooks: we | ||||
| 	// would not be able to distinguish which bot would be sending them. | ||||
| 	if err := b.populateMessageWithBotInfo(ev, rmsg); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	// Second, deal with "real" users if we have the necessary information. | ||||
| 	var userID string | ||||
| 	switch { | ||||
| 	case ev.User != "": | ||||
| 		userID = ev.User | ||||
| 	case ev.SubMessage != nil && ev.SubMessage.User != "": | ||||
| 		userID = ev.SubMessage.User | ||||
| 	default: | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	user := b.getUser(userID) | ||||
| 	if user == nil { | ||||
| 		return fmt.Errorf("could not find information for user with id %s", ev.User) | ||||
| 	} | ||||
|  | ||||
| 	rmsg.UserID = user.ID | ||||
| 	rmsg.Username = user.Name | ||||
| 	if user.Profile.DisplayName != "" { | ||||
| 		rmsg.Username = user.Profile.DisplayName | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config.Message) error { | ||||
| 	if ev.BotID == "" || b.GetString(outgoingWebhookConfig) != "" { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	var err error | ||||
| 	var bot *slack.Bot | ||||
| 	for { | ||||
| 		bot, err = b.rtm.GetBotInfo(ev.BotID) | ||||
| 		if err == nil { | ||||
| 			break | ||||
| 		} | ||||
|  | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 			b.Log.Errorf("Could not retrieve bot information: %#v", err) | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Debugf("Found bot %#v", bot) | ||||
|  | ||||
| 	if bot.Name != "" { | ||||
| 		rmsg.Username = bot.Name | ||||
| 		if ev.Username != "" { | ||||
| 			rmsg.Username = ev.Username | ||||
| 		} | ||||
| 		rmsg.UserID = bot.ID | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	mentionRE        = regexp.MustCompile(`<@([a-zA-Z0-9]+)>`) | ||||
| 	channelRE        = regexp.MustCompile(`<#[a-zA-Z0-9]+\|(.+?)>`) | ||||
| 	variableRE       = regexp.MustCompile(`<!((?:subteam\^)?[a-zA-Z0-9]+)(?:\|@?(.+?))?>`) | ||||
| 	urlRE            = regexp.MustCompile(`<(.*?)(\|.*?)?>`) | ||||
| 	codeFenceRE      = regexp.MustCompile(`(?m)^` + "```" + `\w+$`) | ||||
| 	topicOrPurposeRE = regexp.MustCompile(`(?s)(@.+) (cleared|set)(?: the)? channel (topic|purpose)(?:: (.*))?`) | ||||
| ) | ||||
|  | ||||
| func (b *Bslack) extractTopicOrPurpose(text string) (string, string) { | ||||
| 	r := topicOrPurposeRE.FindStringSubmatch(text) | ||||
| 	if len(r) == 5 { | ||||
| 		action, updateType, extracted := r[2], r[3], r[4] | ||||
| 		switch action { | ||||
| 		case "set": | ||||
| 			return updateType, extracted | ||||
| 		case "cleared": | ||||
| 			return updateType, "" | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Warnf("Encountered channel topic or purpose change message with unexpected format: %s", text) | ||||
| 	return "unknown", "" | ||||
| } | ||||
|  | ||||
| // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||
| func (b *Bslack) replaceMention(text string) string { | ||||
| 	replaceFunc := func(match string) string { | ||||
| 		userID := strings.Trim(match, "@<>") | ||||
| 		if username := b.getUsername(userID); userID != "" { | ||||
| 			return "@" + username | ||||
| 		} | ||||
| 		return match | ||||
| 	} | ||||
| 	return mentionRE.ReplaceAllStringFunc(text, replaceFunc) | ||||
| } | ||||
|  | ||||
| // @see https://api.slack.com/docs/message-formatting#linking_to_channels_and_users | ||||
| func (b *Bslack) replaceChannel(text string) string { | ||||
| 	for _, r := range channelRE.FindAllStringSubmatch(text, -1) { | ||||
| 		text = strings.Replace(text, r[0], "#"+r[1], 1) | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| // @see https://api.slack.com/docs/message-formatting#variables | ||||
| func (b *Bslack) replaceVariable(text string) string { | ||||
| 	for _, r := range variableRE.FindAllStringSubmatch(text, -1) { | ||||
| 		if r[2] != "" { | ||||
| 			text = strings.Replace(text, r[0], "@"+r[2], 1) | ||||
| 		} else { | ||||
| 			text = strings.Replace(text, r[0], "@"+r[1], 1) | ||||
| 		} | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| // @see https://api.slack.com/docs/message-formatting#linking_to_urls | ||||
| func (b *Bslack) replaceURL(text string) string { | ||||
| 	for _, r := range urlRE.FindAllStringSubmatch(text, -1) { | ||||
| 		if len(strings.TrimSpace(r[2])) == 1 { // A display text separator was found, but the text was blank | ||||
| 			text = strings.Replace(text, r[0], "", 1) | ||||
| 		} else { | ||||
| 			text = strings.Replace(text, r[0], r[1], 1) | ||||
| 		} | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func (b *Bslack) replaceCodeFence(text string) string { | ||||
| 	return codeFenceRE.ReplaceAllString(text, "```") | ||||
| } | ||||
|  | ||||
| func (b *Bslack) handleRateLimit(err error) error { | ||||
| 	rateLimit, ok := err.(*slack.RateLimitedError) | ||||
| 	if !ok { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Log.Infof("Rate-limited by Slack. Sleeping for %v", rateLimit.RetryAfter) | ||||
| 	time.Sleep(rateLimit.RetryAfter) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // getUsersInConversation returns an array of userIDs that are members of channelID | ||||
| func (b *Bslack) getUsersInConversation(channelID string) ([]string, error) { | ||||
| 	channelMembers := []string{} | ||||
| 	for { | ||||
| 		queryParams := &slack.GetUsersInConversationParameters{ | ||||
| 			ChannelID: channelID, | ||||
| 		} | ||||
|  | ||||
| 		members, nextCursor, err := b.sc.GetUsersInConversation(queryParams) | ||||
| 		if err != nil { | ||||
| 			if err = b.handleRateLimit(err); err != nil { | ||||
| 				return channelMembers, fmt.Errorf("Could not retrieve users in channels: %#v", err) | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		channelMembers = append(channelMembers, members...) | ||||
|  | ||||
| 		if nextCursor == "" { | ||||
| 			break | ||||
| 		} | ||||
| 		queryParams.Cursor = nextCursor | ||||
| 	} | ||||
| 	return channelMembers, nil | ||||
| } | ||||
							
								
								
									
										36
									
								
								bridge/slack/helpers_test.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,36 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"io/ioutil" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
| ) | ||||
|  | ||||
| func TestExtractTopicOrPurpose(t *testing.T) { | ||||
| 	testcases := map[string]struct { | ||||
| 		input          string | ||||
| 		wantChangeType string | ||||
| 		wantOutput     string | ||||
| 	}{ | ||||
| 		"success - topic type":   {"@someone set channel topic: foo bar", "topic", "foo bar"}, | ||||
| 		"success - purpose type": {"@someone set channel purpose: foo bar", "purpose", "foo bar"}, | ||||
| 		"success - one line":     {"@someone set channel topic: foo bar", "topic", "foo bar"}, | ||||
| 		"success - multi-line":   {"@someone set channel topic: foo\nbar", "topic", "foo\nbar"}, | ||||
| 		"success - cleared":      {"@someone cleared channel topic", "topic", ""}, | ||||
| 		"error - unhandled":      {"some unmatched message", "unknown", ""}, | ||||
| 	} | ||||
|  | ||||
| 	logger := logrus.New() | ||||
| 	logger.SetOutput(ioutil.Discard) | ||||
| 	cfg := &bridge.Config{Bridge: &bridge.Bridge{Log: logrus.NewEntry(logger)}} | ||||
| 	b := newBridge(cfg) | ||||
| 	for name, tc := range testcases { | ||||
| 		gotChangeType, gotOutput := b.extractTopicOrPurpose(tc.input) | ||||
|  | ||||
| 		assert.Equalf(t, tc.wantChangeType, gotChangeType, "This testcase failed: %s", name) | ||||
| 		assert.Equalf(t, tc.wantOutput, gotOutput, "This testcase failed: %s", name) | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										74
									
								
								bridge/slack/legacy.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,74 @@ | ||||
| package bslack | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/nlopes/slack" | ||||
| ) | ||||
|  | ||||
| type BLegacy struct { | ||||
| 	*Bslack | ||||
| } | ||||
|  | ||||
| func NewLegacy(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &BLegacy{Bslack: newBridge(cfg)} | ||||
| } | ||||
|  | ||||
| func (b *BLegacy) Connect() error { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	if b.GetString(incomingWebhookConfig) != "" { | ||||
| 		switch { | ||||
| 		case b.GetString(outgoingWebhookConfig) != "": | ||||
| 			b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)") | ||||
| 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||
| 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||
| 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||
| 			}) | ||||
| 		case b.GetString(tokenConfig) != "": | ||||
| 			b.Log.Info("Connecting using token (sending)") | ||||
| 			b.sc = slack.New(b.GetString(tokenConfig)) | ||||
| 			b.rtm = b.sc.NewRTM() | ||||
| 			go b.rtm.ManageConnection() | ||||
| 			b.Log.Info("Connecting using webhookbindaddress (receiving)") | ||||
| 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||
| 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||
| 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||
| 			}) | ||||
| 		default: | ||||
| 			b.Log.Info("Connecting using webhookbindaddress (receiving)") | ||||
| 			b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||
| 				InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||
| 				BindAddress:        b.GetString(incomingWebhookConfig), | ||||
| 			}) | ||||
| 		} | ||||
| 		go b.handleSlack() | ||||
| 		return nil | ||||
| 	} | ||||
| 	if b.GetString(outgoingWebhookConfig) != "" { | ||||
| 		b.Log.Info("Connecting using webhookurl (sending)") | ||||
| 		b.mh = matterhook.New(b.GetString(outgoingWebhookConfig), matterhook.Config{ | ||||
| 			InsecureSkipVerify: b.GetBool(skipTLSConfig), | ||||
| 			DisableServer:      true, | ||||
| 		}) | ||||
| 		if b.GetString(tokenConfig) != "" { | ||||
| 			b.Log.Info("Connecting using token (receiving)") | ||||
| 			b.sc = slack.New(b.GetString(tokenConfig)) | ||||
| 			b.rtm = b.sc.NewRTM() | ||||
| 			go b.rtm.ManageConnection() | ||||
| 			go b.handleSlack() | ||||
| 		} | ||||
| 	} else if b.GetString(tokenConfig) != "" { | ||||
| 		b.Log.Info("Connecting using token (sending and receiving)") | ||||
| 		b.sc = slack.New(b.GetString(tokenConfig)) | ||||
| 		b.rtm = b.sc.NewRTM() | ||||
| 		go b.rtm.ManageConnection() | ||||
| 		go b.handleSlack() | ||||
| 	} | ||||
| 	if b.GetString(incomingWebhookConfig) == "" && b.GetString(outgoingWebhookConfig) == "" && b.GetString(tokenConfig) == "" { | ||||
| 		return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token configured") | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @@ -9,7 +9,6 @@ import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/shazow/ssh-chat/sshd" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Bsshchat struct { | ||||
| @@ -23,21 +22,35 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| } | ||||
|  | ||||
| func (b *Bsshchat) Connect() error { | ||||
| 	var err error | ||||
| 	b.Log.Infof("Connecting %s", b.GetString("Server")) | ||||
|  | ||||
| 	// connHandler will be called by 'sshd.ConnectShell()' below | ||||
| 	// once the connection is established in order to handle it. | ||||
| 	connErr := make(chan error, 1) // Needs to be buffered. | ||||
| 	connSignal := make(chan struct{}) | ||||
| 	connHandler := func(r io.Reader, w io.WriteCloser) error { | ||||
| 		b.r = bufio.NewScanner(r) | ||||
| 		b.r.Scan() | ||||
| 		b.w = w | ||||
| 		if _, err := b.w.Write([]byte("/theme mono\r\n/quiet\r\n")); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		close(connSignal) // Connection is established so we can signal the success. | ||||
| 		return b.handleSSHChat() | ||||
| 	} | ||||
|  | ||||
| 	go func() { | ||||
| 		err = sshd.ConnectShell(b.GetString("Server"), b.GetString("Nick"), func(r io.Reader, w io.WriteCloser) error { | ||||
| 			b.r = bufio.NewScanner(r) | ||||
| 			b.w = w | ||||
| 			b.r.Scan() | ||||
| 			w.Write([]byte("/theme mono\r\n")) | ||||
| 			b.handleSshChat() | ||||
| 			return nil | ||||
| 		}) | ||||
| 		// As a successful connection will result in this returning after the Connection | ||||
| 		// method has already returned point we NEED to have a buffered channel to still | ||||
| 		// be able to write. | ||||
| 		connErr <- sshd.ConnectShell(b.GetString("Server"), b.GetString("Nick"), connHandler) | ||||
| 	}() | ||||
| 	if err != nil { | ||||
| 		b.Log.Debugf("%#v", err) | ||||
|  | ||||
| 	select { | ||||
| 	case err := <-connErr: | ||||
| 		b.Log.Error("Connection failed") | ||||
| 		return err | ||||
| 	case <-connSignal: | ||||
| 	} | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	return nil | ||||
| @@ -53,33 +66,22 @@ func (b *Bsshchat) JoinChannel(channel config.ChannelInfo) error { | ||||
|  | ||||
| func (b *Bsshchat) Send(msg config.Message) (string, error) { | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.w.Write([]byte(rmsg.Username + rmsg.Text + "\r\n")) | ||||
| 			if _, err := b.w.Write([]byte(rmsg.Username + rmsg.Text + "\r\n")); err != nil { | ||||
| 				b.Log.Errorf("Could not send extra message: %#v", err) | ||||
| 			} | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.Comment != "" { | ||||
| 					msg.Text += fi.Comment + ": " | ||||
| 				} | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text = fi.URL | ||||
| 					if fi.Comment != "" { | ||||
| 						msg.Text = fi.Comment + ": " + fi.URL | ||||
| 					} | ||||
| 				} | ||||
| 				b.w.Write([]byte(msg.Username + msg.Text)) | ||||
| 			} | ||||
| 			return "", nil | ||||
| 			return b.handleUploadFile(&msg) | ||||
| 		} | ||||
| 	} | ||||
| 	b.w.Write([]byte(msg.Username + msg.Text + "\r\n")) | ||||
| 	return "", nil | ||||
| 	_, err := b.w.Write([]byte(msg.Username + msg.Text + "\r\n")) | ||||
| 	return "", err | ||||
| } | ||||
|  | ||||
| /* | ||||
| @@ -113,7 +115,7 @@ func stripPrompt(s string) string { | ||||
| 	return s[pos+3:] | ||||
| } | ||||
|  | ||||
| func (b *Bsshchat) handleSshChat() error { | ||||
| func (b *Bsshchat) handleSSHChat() error { | ||||
| 	/* | ||||
| 		done := b.sshchatKeepAlive() | ||||
| 		defer close(done) | ||||
| @@ -125,17 +127,39 @@ func (b *Bsshchat) handleSshChat() error { | ||||
| 			if !strings.Contains(b.r.Text(), "\033[K") { | ||||
| 				continue | ||||
| 			} | ||||
| 			if strings.Contains(b.r.Text(), "Rate limiting is in effect") { | ||||
| 				continue | ||||
| 			} | ||||
| 			res := strings.Split(stripPrompt(b.r.Text()), ":") | ||||
| 			if res[0] == "-> Set theme" { | ||||
| 				wait = false | ||||
| 				log.Debugf("mono found, allowing") | ||||
| 				b.Log.Debugf("mono found, allowing") | ||||
| 				continue | ||||
| 			} | ||||
| 			if !wait { | ||||
| 				b.Log.Debugf("<= Message %#v", res) | ||||
| 				rmsg := config.Message{Username: res[0], Text: strings.Join(res[1:], ":"), Channel: "sshchat", Account: b.Account, UserID: "nick"} | ||||
| 				rmsg := config.Message{Username: res[0], Text: strings.TrimSpace(strings.Join(res[1:], ":")), Channel: "sshchat", Account: b.Account, UserID: "nick"} | ||||
| 				b.Remote <- rmsg | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bsshchat) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		if fi.Comment != "" { | ||||
| 			msg.Text += fi.Comment + ": " | ||||
| 		} | ||||
| 		if fi.URL != "" { | ||||
| 			msg.Text = fi.URL | ||||
| 			if fi.Comment != "" { | ||||
| 				msg.Text = fi.Comment + ": " + fi.URL | ||||
| 			} | ||||
| 		} | ||||
| 		if _, err := b.w.Write([]byte(msg.Username + msg.Text + "\r\n")); err != nil { | ||||
| 			b.Log.Errorf("Could not send file message: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|   | ||||
							
								
								
									
										126
									
								
								bridge/steam/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,126 @@ | ||||
| package bsteam | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/Philipp15b/go-steam" | ||||
| 	"github.com/Philipp15b/go-steam/protocol/steamlang" | ||||
| ) | ||||
|  | ||||
| func (b *Bsteam) handleChatMsg(e *steam.ChatMsgEvent) { | ||||
| 	b.Log.Debugf("Receiving ChatMsgEvent: %#v", e) | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | ||||
| 	var channel int64 | ||||
| 	if e.ChatRoomId == 0 { | ||||
| 		channel = int64(e.ChatterId) | ||||
| 	} else { | ||||
| 		// for some reason we have to remove 0x18000000000000 | ||||
| 		// TODO | ||||
| 		// https://github.com/42wim/matterbridge/pull/630#discussion_r238102751 | ||||
| 		// channel = int64(e.ChatRoomId) & 0xfffffffffffff | ||||
| 		channel = int64(e.ChatRoomId) - 0x18000000000000 | ||||
| 	} | ||||
| 	msg := config.Message{ | ||||
| 		Username: b.getNick(e.ChatterId), | ||||
| 		Text:     e.Message, | ||||
| 		Channel:  strconv.FormatInt(channel, 10), | ||||
| 		Account:  b.Account, | ||||
| 		UserID:   strconv.FormatInt(int64(e.ChatterId), 10), | ||||
| 	} | ||||
| 	b.Remote <- msg | ||||
| } | ||||
|  | ||||
| func (b *Bsteam) handleEvents() { | ||||
| 	myLoginInfo := &steam.LogOnDetails{ | ||||
| 		Username: b.GetString("Login"), | ||||
| 		Password: b.GetString("Password"), | ||||
| 		AuthCode: b.GetString("AuthCode"), | ||||
| 	} | ||||
| 	// TODO Attempt to read existing auth hash to avoid steam guard. | ||||
| 	// Maybe works | ||||
| 	//myLoginInfo.SentryFileHash, _ = ioutil.ReadFile("sentry") | ||||
| 	for event := range b.c.Events() { | ||||
| 		switch e := event.(type) { | ||||
| 		case *steam.ChatMsgEvent: | ||||
| 			b.handleChatMsg(e) | ||||
| 		case *steam.PersonaStateEvent: | ||||
| 			b.Log.Debugf("PersonaStateEvent: %#v\n", e) | ||||
| 			b.Lock() | ||||
| 			b.userMap[e.FriendId] = e.Name | ||||
| 			b.Unlock() | ||||
| 		case *steam.ConnectedEvent: | ||||
| 			b.c.Auth.LogOn(myLoginInfo) | ||||
| 		case *steam.MachineAuthUpdateEvent: | ||||
| 		// TODO sentry files for 2 auth | ||||
| 		/* | ||||
| 			b.Log.Info("authupdate", e) | ||||
| 			b.Log.Info("hash", e.Hash) | ||||
| 			ioutil.WriteFile("sentry", e.Hash, 0666) | ||||
| 		*/ | ||||
| 		case *steam.LogOnFailedEvent: | ||||
| 			b.Log.Info("Logon failed", e) | ||||
| 			err := b.handleLogOnFailed(e, myLoginInfo) | ||||
| 			if err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 				return | ||||
| 			} | ||||
| 		case *steam.LoggedOnEvent: | ||||
| 			b.Log.Debugf("LoggedOnEvent: %#v", e) | ||||
| 			b.connected <- struct{}{} | ||||
| 			b.Log.Debugf("setting online") | ||||
| 			b.c.Social.SetPersonaState(steamlang.EPersonaState_Online) | ||||
| 		case *steam.DisconnectedEvent: | ||||
| 			b.Log.Info("Disconnected") | ||||
| 			b.Log.Info("Attempting to reconnect...") | ||||
| 			b.c.Connect() | ||||
| 		case steam.FatalErrorEvent: | ||||
| 			b.Log.Errorf("steam FatalErrorEvent: %#v", e) | ||||
| 		default: | ||||
| 			b.Log.Debugf("unknown event %#v", e) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bsteam) handleLogOnFailed(e *steam.LogOnFailedEvent, myLoginInfo *steam.LogOnDetails) error { | ||||
| 	switch e.Result { | ||||
| 	case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode: | ||||
| 		b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:") | ||||
| 		var code string | ||||
| 		fmt.Scanf("%s", &code) | ||||
| 		// TODO https://github.com/42wim/matterbridge/pull/630#discussion_r238103978 | ||||
| 		myLoginInfo.TwoFactorCode = code | ||||
| 	case steamlang.EResult_AccountLogonDenied: | ||||
| 		b.Log.Info("Steam guard isn't letting me in! Enter auth code:") | ||||
| 		var code string | ||||
| 		fmt.Scanf("%s", &code) | ||||
| 		// TODO https://github.com/42wim/matterbridge/pull/630#discussion_r238103978 | ||||
| 		myLoginInfo.AuthCode = code | ||||
| 	case steamlang.EResult_InvalidLoginAuthCode: | ||||
| 		return fmt.Errorf("Steam guard: invalid login auth code: %#v ", e.Result) | ||||
| 	default: | ||||
| 		return fmt.Errorf("LogOnFailedEvent: %#v ", e.Result) | ||||
| 		// TODO: Handle EResult_InvalidLoginAuthCode | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleFileInfo handles config.FileInfo and adds correct file comment or URL to msg.Text. | ||||
| // Returns error if cast fails. | ||||
| func (b *Bsteam) handleFileInfo(msg *config.Message, f interface{}) error { | ||||
| 	if _, ok := f.(config.FileInfo); !ok { | ||||
| 		return fmt.Errorf("handleFileInfo cast failed %#v", f) | ||||
| 	} | ||||
| 	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 | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
| @@ -2,6 +2,8 @@ package bsteam | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| @@ -9,10 +11,6 @@ import ( | ||||
| 	"github.com/Philipp15b/go-steam" | ||||
| 	"github.com/Philipp15b/go-steam/protocol/steamlang" | ||||
| 	"github.com/Philipp15b/go-steam/steamid" | ||||
| 	//"io/ioutil" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| type Bsteam struct { | ||||
| @@ -61,7 +59,7 @@ func (b *Bsteam) JoinChannel(channel config.ChannelInfo) error { | ||||
|  | ||||
| func (b *Bsteam) Send(msg config.Message) (string, error) { | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	id, err := steamid.NewId(msg.Channel) | ||||
| @@ -74,22 +72,13 @@ func (b *Bsteam) Send(msg config.Message) (string, error) { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, rmsg.Username+rmsg.Text) | ||||
| 		} | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.Comment != "" { | ||||
| 					msg.Text += fi.Comment + ": " | ||||
| 				} | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text = fi.URL | ||||
| 					if fi.Comment != "" { | ||||
| 						msg.Text = fi.Comment + ": " + fi.URL | ||||
| 					} | ||||
| 				} | ||||
| 				b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||
| 		for i := range msg.Extra["file"] { | ||||
| 			if err := b.handleFileInfo(&msg, msg.Extra["file"][i]); err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 			} | ||||
| 			return "", nil | ||||
| 			b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||
| 		} | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	b.c.Social.SendMessage(id, steamlang.EChatEntryType_ChatMsg, msg.Username+msg.Text) | ||||
| @@ -104,80 +93,3 @@ func (b *Bsteam) getNick(id steamid.SteamId) string { | ||||
| 	} | ||||
| 	return "unknown" | ||||
| } | ||||
|  | ||||
| func (b *Bsteam) handleEvents() { | ||||
| 	myLoginInfo := new(steam.LogOnDetails) | ||||
| 	myLoginInfo.Username = b.GetString("Login") | ||||
| 	myLoginInfo.Password = b.GetString("Password") | ||||
| 	myLoginInfo.AuthCode = b.GetString("AuthCode") | ||||
| 	// Attempt to read existing auth hash to avoid steam guard. | ||||
| 	// Maybe works | ||||
| 	//myLoginInfo.SentryFileHash, _ = ioutil.ReadFile("sentry") | ||||
| 	for event := range b.c.Events() { | ||||
| 		//b.Log.Info(event) | ||||
| 		switch e := event.(type) { | ||||
| 		case *steam.ChatMsgEvent: | ||||
| 			b.Log.Debugf("Receiving ChatMsgEvent: %#v", e) | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", b.getNick(e.ChatterId), b.Account) | ||||
| 			var channel int64 | ||||
| 			if e.ChatRoomId == 0 { | ||||
| 				channel = int64(e.ChatterId) | ||||
| 			} else { | ||||
| 				// for some reason we have to remove 0x18000000000000 | ||||
| 				channel = int64(e.ChatRoomId) - 0x18000000000000 | ||||
| 			} | ||||
| 			msg := config.Message{Username: b.getNick(e.ChatterId), Text: e.Message, Channel: strconv.FormatInt(channel, 10), Account: b.Account, UserID: strconv.FormatInt(int64(e.ChatterId), 10)} | ||||
| 			b.Remote <- msg | ||||
| 		case *steam.PersonaStateEvent: | ||||
| 			b.Log.Debugf("PersonaStateEvent: %#v\n", e) | ||||
| 			b.Lock() | ||||
| 			b.userMap[e.FriendId] = e.Name | ||||
| 			b.Unlock() | ||||
| 		case *steam.ConnectedEvent: | ||||
| 			b.c.Auth.LogOn(myLoginInfo) | ||||
| 		case *steam.MachineAuthUpdateEvent: | ||||
| 			/* | ||||
| 				b.Log.Info("authupdate", e) | ||||
| 				b.Log.Info("hash", e.Hash) | ||||
| 				ioutil.WriteFile("sentry", e.Hash, 0666) | ||||
| 			*/ | ||||
| 		case *steam.LogOnFailedEvent: | ||||
| 			b.Log.Info("Logon failed", e) | ||||
| 			switch e.Result { | ||||
| 			case steamlang.EResult_AccountLogonDeniedNeedTwoFactorCode: | ||||
| 				{ | ||||
| 					b.Log.Info("Steam guard isn't letting me in! Enter 2FA code:") | ||||
| 					var code string | ||||
| 					fmt.Scanf("%s", &code) | ||||
| 					myLoginInfo.TwoFactorCode = code | ||||
| 				} | ||||
| 			case steamlang.EResult_AccountLogonDenied: | ||||
| 				{ | ||||
| 					b.Log.Info("Steam guard isn't letting me in! Enter auth code:") | ||||
| 					var code string | ||||
| 					fmt.Scanf("%s", &code) | ||||
| 					myLoginInfo.AuthCode = code | ||||
| 				} | ||||
| 			default: | ||||
| 				b.Log.Errorf("LogOnFailedEvent: %#v ", e.Result) | ||||
| 				// TODO: Handle EResult_InvalidLoginAuthCode | ||||
| 				return | ||||
| 			} | ||||
| 		case *steam.LoggedOnEvent: | ||||
| 			b.Log.Debugf("LoggedOnEvent: %#v", e) | ||||
| 			b.connected <- struct{}{} | ||||
| 			b.Log.Debugf("setting online") | ||||
| 			b.c.Social.SetPersonaState(steamlang.EPersonaState_Online) | ||||
| 		case *steam.DisconnectedEvent: | ||||
| 			b.Log.Info("Disconnected") | ||||
| 			b.Log.Info("Attempting to reconnect...") | ||||
| 			b.c.Connect() | ||||
| 		case steam.FatalErrorEvent: | ||||
| 			b.Log.Error(e) | ||||
| 		case error: | ||||
| 			b.Log.Error(e) | ||||
| 		default: | ||||
| 			b.Log.Debugf("unknown event %#v", e) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										346
									
								
								bridge/telegram/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,346 @@ | ||||
| package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"html" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| ) | ||||
|  | ||||
| func (b *Btelegram) handleUpdate(rmsg *config.Message, message, posted, edited *tgbotapi.Message) *tgbotapi.Message { | ||||
| 	// handle channels | ||||
| 	if posted != nil { | ||||
| 		message = posted | ||||
| 		rmsg.Text = message.Text | ||||
| 	} | ||||
|  | ||||
| 	// edited channel message | ||||
| 	if edited != nil && !b.GetBool("EditDisable") { | ||||
| 		message = edited | ||||
| 		rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix") | ||||
| 	} | ||||
| 	return message | ||||
| } | ||||
|  | ||||
| // handleChannels checks if it's a channel message and if the message is a new or edited messages | ||||
| func (b *Btelegram) handleChannels(rmsg *config.Message, message *tgbotapi.Message, update tgbotapi.Update) *tgbotapi.Message { | ||||
| 	return b.handleUpdate(rmsg, message, update.ChannelPost, update.EditedChannelPost) | ||||
| } | ||||
|  | ||||
| // handleGroups checks if it's a group message and if the message is a new or edited messages | ||||
| func (b *Btelegram) handleGroups(rmsg *config.Message, message *tgbotapi.Message, update tgbotapi.Update) *tgbotapi.Message { | ||||
| 	return b.handleUpdate(rmsg, message, update.Message, update.EditedMessage) | ||||
| } | ||||
|  | ||||
| // handleForwarded handles forwarded messages | ||||
| func (b *Btelegram) handleForwarded(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.ForwardFrom != nil { | ||||
| 		usernameForward := "" | ||||
| 		if b.GetBool("UseFirstName") { | ||||
| 			usernameForward = message.ForwardFrom.FirstName | ||||
| 		} | ||||
| 		if usernameForward == "" { | ||||
| 			usernameForward = message.ForwardFrom.UserName | ||||
| 			if usernameForward == "" { | ||||
| 				usernameForward = message.ForwardFrom.FirstName | ||||
| 			} | ||||
| 		} | ||||
| 		if usernameForward == "" { | ||||
| 			usernameForward = unknownUser | ||||
| 		} | ||||
| 		rmsg.Text = "Forwarded from " + usernameForward + ": " + rmsg.Text | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleQuoting handles quoting of previous messages | ||||
| func (b *Btelegram) handleQuoting(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.ReplyToMessage != nil { | ||||
| 		usernameReply := "" | ||||
| 		if message.ReplyToMessage.From != nil { | ||||
| 			if b.GetBool("UseFirstName") { | ||||
| 				usernameReply = message.ReplyToMessage.From.FirstName | ||||
| 			} | ||||
| 			if usernameReply == "" { | ||||
| 				usernameReply = message.ReplyToMessage.From.UserName | ||||
| 				if usernameReply == "" { | ||||
| 					usernameReply = message.ReplyToMessage.From.FirstName | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		if usernameReply == "" { | ||||
| 			usernameReply = unknownUser | ||||
| 		} | ||||
| 		if !b.GetBool("QuoteDisable") { | ||||
| 			rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, message.ReplyToMessage.Text) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleUsername handles the correct setting of the username | ||||
| func (b *Btelegram) handleUsername(rmsg *config.Message, message *tgbotapi.Message) { | ||||
| 	if message.From != nil { | ||||
| 		rmsg.UserID = strconv.Itoa(message.From.ID) | ||||
| 		if b.GetBool("UseFirstName") { | ||||
| 			rmsg.Username = message.From.FirstName | ||||
| 		} | ||||
| 		if rmsg.Username == "" { | ||||
| 			rmsg.Username = message.From.UserName | ||||
| 			if rmsg.Username == "" { | ||||
| 				rmsg.Username = message.From.FirstName | ||||
| 			} | ||||
| 		} | ||||
| 		// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 		if b.General.MediaServerUpload != "" { | ||||
| 			b.handleDownloadAvatar(message.From.ID, rmsg.Channel) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// if we really didn't find a username, set it to unknown | ||||
| 	if rmsg.Username == "" { | ||||
| 		rmsg.Username = unknownUser | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
| 	for update := range updates { | ||||
| 		b.Log.Debugf("== Receiving event: %#v", update.Message) | ||||
|  | ||||
| 		if update.Message == nil && update.ChannelPost == nil && | ||||
| 			update.EditedMessage == nil && update.EditedChannelPost == nil { | ||||
| 			b.Log.Error("Getting nil messages, this shouldn't happen.") | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		var message *tgbotapi.Message | ||||
|  | ||||
| 		rmsg := config.Message{Account: b.Account, Extra: make(map[string][]interface{})} | ||||
|  | ||||
| 		// handle channels | ||||
| 		message = b.handleChannels(&rmsg, message, update) | ||||
|  | ||||
| 		// handle groups | ||||
| 		message = b.handleGroups(&rmsg, message, update) | ||||
|  | ||||
| 		// set the ID's from the channel or group message | ||||
| 		rmsg.ID = strconv.Itoa(message.MessageID) | ||||
| 		rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10) | ||||
|  | ||||
| 		// handle username | ||||
| 		b.handleUsername(&rmsg, message) | ||||
|  | ||||
| 		// handle any downloads | ||||
| 		err := b.handleDownload(&rmsg, message) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("download failed: %s", err) | ||||
| 		} | ||||
|  | ||||
| 		// handle forwarded messages | ||||
| 		b.handleForwarded(&rmsg, message) | ||||
|  | ||||
| 		// quote the previous message | ||||
| 		b.handleQuoting(&rmsg, message) | ||||
|  | ||||
| 		if rmsg.Text != "" || len(rmsg.Extra) > 0 { | ||||
| 			rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text) | ||||
| 			// channels don't have (always?) user information. see #410 | ||||
| 			if message.From != nil { | ||||
| 				rmsg.Avatar = helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General) | ||||
| 			} | ||||
|  | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account) | ||||
| 			b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||
| // logs an error message if it fails | ||||
| func (b *Btelegram) handleDownloadAvatar(userid int, channel string) { | ||||
| 	rmsg := config.Message{Username: "system", | ||||
| 		Text:    "avatar", | ||||
| 		Channel: channel, | ||||
| 		Account: b.Account, | ||||
| 		UserID:  strconv.Itoa(userid), | ||||
| 		Event:   config.EventAvatarDownload, | ||||
| 		Extra:   make(map[string][]interface{})} | ||||
|  | ||||
| 	if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok { | ||||
| 		photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Userprofile download failed for %#v %s", userid, err) | ||||
| 		} | ||||
|  | ||||
| 		if len(photos.Photos) > 0 { | ||||
| 			photo := photos.Photos[0][0] | ||||
| 			url := b.getFileDirectURL(photo.FileID) | ||||
| 			name := strconv.Itoa(userid) + ".png" | ||||
| 			b.Log.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize) | ||||
|  | ||||
| 			err := helper.HandleDownloadSize(b.Log, &rmsg, name, int64(photo.FileSize), b.General) | ||||
| 			if err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 				return | ||||
| 			} | ||||
| 			data, err := helper.DownloadFile(url) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("download %s failed %#v", url, err) | ||||
| 				return | ||||
| 			} | ||||
| 			helper.HandleDownloadData(b.Log, &rmsg, name, rmsg.Text, "", data, b.General) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Btelegram) handleDownload(rmsg *config.Message, message *tgbotapi.Message) error { | ||||
| 	size := 0 | ||||
| 	var url, name, text string | ||||
| 	switch { | ||||
| 	case message.Sticker != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Sticker.FileID, ".webp", true) | ||||
| 		size = message.Sticker.FileSize | ||||
| 	case message.Voice != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Voice.FileID, ".ogg", true) | ||||
| 		size = message.Voice.FileSize | ||||
| 	case message.Video != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Video.FileID, "", true) | ||||
| 		size = message.Video.FileSize | ||||
| 	case message.Audio != nil: | ||||
| 		text, name, url = b.getDownloadInfo(message.Audio.FileID, "", true) | ||||
| 		size = message.Audio.FileSize | ||||
| 	case message.Document != nil: | ||||
| 		_, _, url = b.getDownloadInfo(message.Document.FileID, "", false) | ||||
| 		size = message.Document.FileSize | ||||
| 		name = message.Document.FileName | ||||
| 		text = " " + message.Document.FileName + " : " + url | ||||
| 	case message.Photo != nil: | ||||
| 		photos := *message.Photo | ||||
| 		size = photos[len(photos)-1].FileSize | ||||
| 		text, name, url = b.getDownloadInfo(photos[len(photos)-1].FileID, "", true) | ||||
| 	} | ||||
|  | ||||
| 	// if name is empty we didn't match a thing to download | ||||
| 	if name == "" { | ||||
| 		return nil | ||||
| 	} | ||||
| 	// use the URL instead of native upload | ||||
| 	if b.GetBool("UseInsecureURL") { | ||||
| 		b.Log.Debugf("Setting message text to :%s", text) | ||||
| 		rmsg.Text += text | ||||
| 		return nil | ||||
| 	} | ||||
| 	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, err := helper.DownloadFile(url) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) getDownloadInfo(id string, suffix string, urlpart bool) (string, string, string) { | ||||
| 	url := b.getFileDirectURL(id) | ||||
| 	name := "" | ||||
| 	if urlpart { | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 	} | ||||
| 	if suffix != "" && !strings.HasSuffix(name, suffix) { | ||||
| 		name += suffix | ||||
| 	} | ||||
| 	text := " " + url | ||||
| 	return text, name, url | ||||
| } | ||||
|  | ||||
| // handleDelete handles message deleting | ||||
| func (b *Btelegram) handleDelete(msg *config.Message, chatid int64) (string, error) { | ||||
| 	if msg.ID == "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	msgid, err := strconv.Atoi(msg.ID) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	_, err = b.c.DeleteMessage(tgbotapi.DeleteMessageConfig{ChatID: chatid, MessageID: msgid}) | ||||
| 	return "", err | ||||
| } | ||||
|  | ||||
| // handleEdit handles message editing. | ||||
| func (b *Btelegram) handleEdit(msg *config.Message, chatid int64) (string, error) { | ||||
| 	msgid, err := strconv.Atoi(msg.ID) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||
| 		b.Log.Debug("Using mode HTML - nick only") | ||||
| 		msg.Text = html.EscapeString(msg.Text) | ||||
| 	} | ||||
| 	m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text) | ||||
| 	switch b.GetString("MessageFormat") { | ||||
| 	case HTMLFormat: | ||||
| 		b.Log.Debug("Using mode HTML") | ||||
| 		m.ParseMode = tgbotapi.ModeHTML | ||||
| 	case "Markdown": | ||||
| 		b.Log.Debug("Using mode markdown") | ||||
| 		m.ParseMode = tgbotapi.ModeMarkdown | ||||
| 	} | ||||
| 	if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||
| 		b.Log.Debug("Using mode HTML - nick only") | ||||
| 		m.ParseMode = tgbotapi.ModeHTML | ||||
| 	} | ||||
| 	_, err = b.c.Send(m) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) string { | ||||
| 	var c tgbotapi.Chattable | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		file := tgbotapi.FileBytes{ | ||||
| 			Name:  fi.Name, | ||||
| 			Bytes: *fi.Data, | ||||
| 		} | ||||
| 		re := regexp.MustCompile(".(jpg|png)$") | ||||
| 		if re.MatchString(fi.Name) { | ||||
| 			c = tgbotapi.NewPhotoUpload(chatid, file) | ||||
| 		} else { | ||||
| 			c = tgbotapi.NewDocumentUpload(chatid, file) | ||||
| 		} | ||||
| 		_, err := b.c.Send(c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		} | ||||
| 		if fi.Comment != "" { | ||||
| 			if _, err := b.sendMessage(chatid, msg.Username, fi.Comment); err != nil { | ||||
| 				b.Log.Errorf("posting file comment %s failed: %s", fi.Comment, err) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string { | ||||
| 	format := b.GetString("quoteformat") | ||||
| 	if format == "" { | ||||
| 		format = "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})" | ||||
| 	} | ||||
| 	format = strings.Replace(format, "{MESSAGE}", message, -1) | ||||
| 	format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1) | ||||
| 	format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1) | ||||
| 	return format | ||||
| } | ||||
| @@ -3,6 +3,7 @@ package btelegram | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"html" | ||||
| 	"io" | ||||
|  | ||||
| 	"github.com/russross/blackfriday" | ||||
| ) | ||||
| @@ -32,8 +33,8 @@ func (options *customHTML) Header(out *bytes.Buffer, text func() bool, level int | ||||
| 	options.Paragraph(out, text) | ||||
| } | ||||
|  | ||||
| func (options *customHTML) HRule(out *bytes.Buffer) { | ||||
| 	out.WriteByte('\n') | ||||
| func (options *customHTML) HRule(out io.ByteWriter) { | ||||
| 	out.WriteByte('\n') //nolint:errcheck | ||||
| } | ||||
|  | ||||
| func (options *customHTML) BlockQuote(out *bytes.Buffer, text []byte) { | ||||
| @@ -53,13 +54,16 @@ func (options *customHTML) ListItem(out *bytes.Buffer, text []byte, flags int) { | ||||
| } | ||||
|  | ||||
| func makeHTML(input string) string { | ||||
| 	return string(blackfriday.Markdown([]byte(input), | ||||
| 		&customHTML{blackfriday.HtmlRenderer(blackfriday.HTML_USE_XHTML|blackfriday.HTML_SKIP_IMAGES, "", "")}, | ||||
| 		blackfriday.EXTENSION_NO_INTRA_EMPHASIS| | ||||
| 			blackfriday.EXTENSION_FENCED_CODE| | ||||
| 			blackfriday.EXTENSION_AUTOLINK| | ||||
| 			blackfriday.EXTENSION_SPACE_HEADERS| | ||||
| 			blackfriday.EXTENSION_HEADER_IDS| | ||||
| 			blackfriday.EXTENSION_BACKSLASH_LINE_BREAK| | ||||
| 			blackfriday.EXTENSION_DEFINITION_LISTS)) | ||||
| 	extensions := blackfriday.NoIntraEmphasis | | ||||
| 		blackfriday.FencedCode | | ||||
| 		blackfriday.Autolink | | ||||
| 		blackfriday.SpaceHeadings | | ||||
| 		blackfriday.HeadingIDs | | ||||
| 		blackfriday.BackslashLineBreak | | ||||
| 		blackfriday.DefinitionLists | ||||
|  | ||||
| 	renderer := &customHTML{blackfriday.NewHTMLRenderer(blackfriday.HTMLRendererParameters{ | ||||
| 		Flags: blackfriday.UseXHTML | blackfriday.SkipImages, | ||||
| 	})} | ||||
| 	return string(blackfriday.Run([]byte(input), blackfriday.WithExtensions(extensions), blackfriday.WithRenderer(renderer))) | ||||
| } | ||||
|   | ||||
| @@ -2,7 +2,6 @@ package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"html" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| @@ -12,6 +11,12 @@ import ( | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	unknownUser = "unknown" | ||||
| 	HTMLFormat  = "HTML" | ||||
| 	HTMLNick    = "htmlnick" | ||||
| ) | ||||
|  | ||||
| type Btelegram struct { | ||||
| 	c *tgbotapi.BotAPI | ||||
| 	*bridge.Config | ||||
| @@ -60,31 +65,25 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// map the file SHA to our user (caches the avatar) | ||||
| 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		return b.cacheAvatar(&msg) | ||||
| 	} | ||||
|  | ||||
| 	if b.GetString("MessageFormat") == "HTML" { | ||||
| 	if b.GetString("MessageFormat") == HTMLFormat { | ||||
| 		msg.Text = makeHTML(msg.Text) | ||||
| 	} | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		msgid, err := strconv.Atoi(msg.ID) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		_, err = b.c.DeleteMessage(tgbotapi.DeleteMessageConfig{ChatID: chatid, MessageID: msgid}) | ||||
| 		return "", err | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return b.handleDelete(&msg, chatid) | ||||
| 	} | ||||
|  | ||||
| 	// Upload a file if it exists | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			b.sendMessage(chatid, rmsg.Username, rmsg.Text) | ||||
| 			if _, err := b.sendMessage(chatid, rmsg.Username, rmsg.Text); err != nil { | ||||
| 				b.Log.Errorf("sendMessage failed: %s", err) | ||||
| 			} | ||||
| 		} | ||||
| 		// check if we have files to upload (from slack, telegram or mattermost) | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| @@ -94,162 +93,13 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
|  | ||||
| 	// edit the message if we have a msg ID | ||||
| 	if msg.ID != "" { | ||||
| 		msgid, err := strconv.Atoi(msg.ID) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		if strings.ToLower(b.GetString("MessageFormat")) == "htmlnick" { | ||||
| 			b.Log.Debug("Using mode HTML - nick only") | ||||
| 			msg.Text = html.EscapeString(msg.Text) | ||||
| 		} | ||||
| 		m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text) | ||||
| 		if b.GetString("MessageFormat") == "HTML" { | ||||
| 			b.Log.Debug("Using mode HTML") | ||||
| 			m.ParseMode = tgbotapi.ModeHTML | ||||
| 		} | ||||
| 		if b.GetString("MessageFormat") == "Markdown" { | ||||
| 			b.Log.Debug("Using mode markdown") | ||||
| 			m.ParseMode = tgbotapi.ModeMarkdown | ||||
| 		} | ||||
| 		if strings.ToLower(b.GetString("MessageFormat")) == "htmlnick" { | ||||
| 			b.Log.Debug("Using mode HTML - nick only") | ||||
| 			m.ParseMode = tgbotapi.ModeHTML | ||||
| 		} | ||||
| 		_, err = b.c.Send(m) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return "", nil | ||||
| 		return b.handleEdit(&msg, chatid) | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message | ||||
| 	return b.sendMessage(chatid, msg.Username, msg.Text) | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) { | ||||
| 	for update := range updates { | ||||
| 		b.Log.Debugf("== Receiving event: %#v", update.Message) | ||||
|  | ||||
| 		if update.Message == nil && update.ChannelPost == nil && update.EditedMessage == nil && update.EditedChannelPost == nil { | ||||
| 			b.Log.Error("Getting nil messages, this shouldn't happen.") | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		var message *tgbotapi.Message | ||||
|  | ||||
| 		rmsg := config.Message{Account: b.Account, Extra: make(map[string][]interface{})} | ||||
|  | ||||
| 		// handle channels | ||||
| 		if update.ChannelPost != nil { | ||||
| 			message = update.ChannelPost | ||||
| 			rmsg.Text = message.Text | ||||
| 		} | ||||
|  | ||||
| 		// edited channel message | ||||
| 		if update.EditedChannelPost != nil && !b.GetBool("EditDisable") { | ||||
| 			message = update.EditedChannelPost | ||||
| 			rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		// handle groups | ||||
| 		if update.Message != nil { | ||||
| 			message = update.Message | ||||
| 			rmsg.Text = message.Text | ||||
| 		} | ||||
|  | ||||
| 		// edited group message | ||||
| 		if update.EditedMessage != nil && !b.GetBool("EditDisable") { | ||||
| 			message = update.EditedMessage | ||||
| 			rmsg.Text = rmsg.Text + message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		// set the ID's from the channel or group message | ||||
| 		rmsg.ID = strconv.Itoa(message.MessageID) | ||||
| 		rmsg.Channel = strconv.FormatInt(message.Chat.ID, 10) | ||||
|  | ||||
| 		// handle username | ||||
| 		if message.From != nil { | ||||
| 			rmsg.UserID = strconv.Itoa(message.From.ID) | ||||
| 			if b.GetBool("UseFirstName") { | ||||
| 				rmsg.Username = message.From.FirstName | ||||
| 			} | ||||
| 			if rmsg.Username == "" { | ||||
| 				rmsg.Username = message.From.UserName | ||||
| 				if rmsg.Username == "" { | ||||
| 					rmsg.Username = message.From.FirstName | ||||
| 				} | ||||
| 			} | ||||
| 			// only download avatars if we have a place to upload them (configured mediaserver) | ||||
| 			if b.General.MediaServerUpload != "" { | ||||
| 				b.handleDownloadAvatar(message.From.ID, rmsg.Channel) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// if we really didn't find a username, set it to unknown | ||||
| 		if rmsg.Username == "" { | ||||
| 			rmsg.Username = "unknown" | ||||
| 		} | ||||
|  | ||||
| 		// handle any downloads | ||||
| 		err := b.handleDownload(message, &rmsg) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("download failed: %s", err) | ||||
| 		} | ||||
|  | ||||
| 		// handle forwarded messages | ||||
| 		if message.ForwardFrom != nil { | ||||
| 			usernameForward := "" | ||||
| 			if b.GetBool("UseFirstName") { | ||||
| 				usernameForward = message.ForwardFrom.FirstName | ||||
| 			} | ||||
| 			if usernameForward == "" { | ||||
| 				usernameForward = message.ForwardFrom.UserName | ||||
| 				if usernameForward == "" { | ||||
| 					usernameForward = message.ForwardFrom.FirstName | ||||
| 				} | ||||
| 			} | ||||
| 			if usernameForward == "" { | ||||
| 				usernameForward = "unknown" | ||||
| 			} | ||||
| 			rmsg.Text = "Forwarded from " + usernameForward + ": " + rmsg.Text | ||||
| 		} | ||||
|  | ||||
| 		// quote the previous message | ||||
| 		if message.ReplyToMessage != nil { | ||||
| 			usernameReply := "" | ||||
| 			if message.ReplyToMessage.From != nil { | ||||
| 				if b.GetBool("UseFirstName") { | ||||
| 					usernameReply = message.ReplyToMessage.From.FirstName | ||||
| 				} | ||||
| 				if usernameReply == "" { | ||||
| 					usernameReply = message.ReplyToMessage.From.UserName | ||||
| 					if usernameReply == "" { | ||||
| 						usernameReply = message.ReplyToMessage.From.FirstName | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 			if usernameReply == "" { | ||||
| 				usernameReply = "unknown" | ||||
| 			} | ||||
| 			if !b.GetBool("QuoteDisable") { | ||||
| 				rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, message.ReplyToMessage.Text) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		if rmsg.Text != "" || len(rmsg.Extra) > 0 { | ||||
| 			rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text) | ||||
| 			// channels don't have (always?) user information. see #410 | ||||
| 			if message.From != nil { | ||||
| 				rmsg.Avatar = helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General) | ||||
| 			} | ||||
|  | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account) | ||||
| 			b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) getFileDirectURL(id string) string { | ||||
| 	res, err := b.c.GetFileDirectURL(id) | ||||
| 	if err != nil { | ||||
| @@ -258,147 +108,10 @@ func (b *Btelegram) getFileDirectURL(id string) string { | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||
| // logs an error message if it fails | ||||
| func (b *Btelegram) handleDownloadAvatar(userid int, channel string) { | ||||
| 	rmsg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: strconv.Itoa(userid), Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})} | ||||
| 	if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok { | ||||
| 		photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Userprofile download failed for %#v %s", userid, err) | ||||
| 		} | ||||
|  | ||||
| 		if len(photos.Photos) > 0 { | ||||
| 			photo := photos.Photos[0][0] | ||||
| 			url := b.getFileDirectURL(photo.FileID) | ||||
| 			name := strconv.Itoa(userid) + ".png" | ||||
| 			b.Log.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize) | ||||
|  | ||||
| 			err := helper.HandleDownloadSize(b.Log, &rmsg, name, int64(photo.FileSize), b.General) | ||||
| 			if err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 				return | ||||
| 			} | ||||
| 			data, err := helper.DownloadFile(url) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("download %s failed %#v", url, err) | ||||
| 				return | ||||
| 			} | ||||
| 			helper.HandleDownloadData(b.Log, &rmsg, name, rmsg.Text, "", data, b.General) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Btelegram) handleDownload(message *tgbotapi.Message, rmsg *config.Message) error { | ||||
| 	size := 0 | ||||
| 	var url, name, text string | ||||
|  | ||||
| 	if message.Sticker != nil { | ||||
| 		v := message.Sticker | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		if !strings.HasSuffix(name, ".webp") { | ||||
| 			name = name + ".webp" | ||||
| 		} | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Video != nil { | ||||
| 		v := message.Video | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Photo != nil { | ||||
| 		photos := *message.Photo | ||||
| 		size = photos[len(photos)-1].FileSize | ||||
| 		url = b.getFileDirectURL(photos[len(photos)-1].FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Document != nil { | ||||
| 		v := message.Document | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		name = v.FileName | ||||
| 		text = " " + v.FileName + " : " + url | ||||
| 	} | ||||
| 	if message.Voice != nil { | ||||
| 		v := message.Voice | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 		if !strings.HasSuffix(name, ".ogg") { | ||||
| 			name = name + ".ogg" | ||||
| 		} | ||||
| 	} | ||||
| 	if message.Audio != nil { | ||||
| 		v := message.Audio | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	// if name is empty we didn't match a thing to download | ||||
| 	if name == "" { | ||||
| 		return nil | ||||
| 	} | ||||
| 	// use the URL instead of native upload | ||||
| 	if b.GetBool("UseInsecureURL") { | ||||
| 		b.Log.Debugf("Setting message text to :%s", text) | ||||
| 		rmsg.Text = rmsg.Text + text | ||||
| 		return nil | ||||
| 	} | ||||
| 	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, err := helper.DownloadFile(url) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) (string, error) { | ||||
| 	var c tgbotapi.Chattable | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		file := tgbotapi.FileBytes{fi.Name, *fi.Data} | ||||
| 		re := regexp.MustCompile(".(jpg|png)$") | ||||
| 		if re.MatchString(fi.Name) { | ||||
| 			c = tgbotapi.NewPhotoUpload(chatid, file) | ||||
| 		} else { | ||||
| 			c = tgbotapi.NewDocumentUpload(chatid, file) | ||||
| 		} | ||||
| 		_, err := b.c.Send(c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		} | ||||
| 		if fi.Comment != "" { | ||||
| 			b.sendMessage(chatid, msg.Username, fi.Comment) | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) { | ||||
| 	m := tgbotapi.NewMessage(chatid, "") | ||||
| 	m.Text = username + text | ||||
| 	if b.GetString("MessageFormat") == "HTML" { | ||||
| 	if b.GetString("MessageFormat") == HTMLFormat { | ||||
| 		b.Log.Debug("Using mode HTML") | ||||
| 		m.ParseMode = tgbotapi.ModeHTML | ||||
| 	} | ||||
| @@ -406,7 +119,7 @@ func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, er | ||||
| 		b.Log.Debug("Using mode markdown") | ||||
| 		m.ParseMode = tgbotapi.ModeMarkdown | ||||
| 	} | ||||
| 	if strings.ToLower(b.GetString("MessageFormat")) == "htmlnick" { | ||||
| 	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 | ||||
| @@ -428,14 +141,3 @@ func (b *Btelegram) cacheAvatar(msg *config.Message) (string, error) { | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) handleQuote(message, quoteNick, quoteMessage string) string { | ||||
| 	format := b.GetString("quoteformat") | ||||
| 	if format == "" { | ||||
| 		format = "{MESSAGE} (re @{QUOTENICK}: {QUOTEMESSAGE})" | ||||
| 	} | ||||
| 	format = strings.Replace(format, "{MESSAGE}", message, -1) | ||||
| 	format = strings.Replace(format, "{QUOTENICK}", quoteNick, -1) | ||||
| 	format = strings.Replace(format, "{QUOTEMESSAGE}", quoteMessage, -1) | ||||
| 	return format | ||||
| } | ||||
|   | ||||
							
								
								
									
										104
									
								
								bridge/whatsapp/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,104 @@ | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
|  | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
|  | ||||
| 	whatsappExt "maunium.net/go/mautrix-whatsapp/whatsapp-ext" | ||||
| ) | ||||
|  | ||||
| /* | ||||
| 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) { | ||||
| 	b.Log.Errorf("%v", err) // TODO implement proper handling? at least respond to different error types | ||||
| } | ||||
|  | ||||
| // HandleTextMessage sent from WhatsApp, relay it to the brige | ||||
| func (b *Bwhatsapp) HandleTextMessage(message whatsapp.TextMessage) { | ||||
| 	if message.Info.FromMe { // || !strings.Contains(strings.ToLower(message.Text), "@echo") { | ||||
| 		return | ||||
| 	} | ||||
| 	// whatsapp sends last messages to show context , cut them | ||||
| 	if message.Info.Timestamp < b.startedAt { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	messageTime := time.Unix(int64(message.Info.Timestamp), 0) // TODO check how behaves between timezones | ||||
| 	groupJid := message.Info.RemoteJid | ||||
|  | ||||
| 	senderJid := message.Info.SenderJid | ||||
| 	if len(senderJid) == 0 { | ||||
| 		// TODO workaround till https://github.com/Rhymen/go-whatsapp/issues/86 resolved | ||||
| 		senderJid = *message.Info.Source.Participant | ||||
| 	} | ||||
|  | ||||
| 	// 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] + whatsappExt.NewUserSuffix) | ||||
| 			if mention == "" { | ||||
| 				mention = "someone" | ||||
| 			} | ||||
| 			message.Text = strings.Replace(message.Text, "@"+numberAndSuffix[0], "@"+mention, 1) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Sending message from %s on %s to gateway", senderJid, b.Account) | ||||
| 	rmsg := config.Message{ | ||||
| 		UserID:    senderJid, | ||||
| 		Username:  senderName, | ||||
| 		Text:      message.Text, | ||||
| 		Timestamp: messageTime, | ||||
| 		Channel:   groupJid, | ||||
| 		Account:   b.Account, | ||||
| 		Protocol:  b.Protocol, | ||||
| 		Extra:     make(map[string][]interface{}), | ||||
| 		//		ParentID: TODO, // TODO handle thread replies  // map from Info.QuotedMessageID string | ||||
| 		//	Event     string    `json:"event"` | ||||
| 		//	Gateway   string  // will be added during message processing | ||||
| 		ID: message.Info.Id} | ||||
|  | ||||
| 	if avatarURL, exists := b.userAvatars[senderJid]; exists { | ||||
| 		rmsg.Avatar = avatarURL | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleImageMessage(message whatsapp.ImageMessage) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleVideoMessage(message whatsapp.VideoMessage) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // | ||||
| //func (b *Bwhatsapp) HandleJsonMessage(message string) { | ||||
| //	fmt.Println(message) // TODO implement | ||||
| //} | ||||
| // TODO HandleRawMessage | ||||
| // TODO HandleAudioMessage | ||||
							
								
								
									
										84
									
								
								bridge/whatsapp/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,84 @@ | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"encoding/gob" | ||||
| 	"errors" | ||||
| 	"os" | ||||
|  | ||||
| 	qrcodeTerminal "github.com/Baozisoftware/qrcode-terminal-go" | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
| ) | ||||
|  | ||||
| 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) | ||||
| 	err = decoder.Decode(&session) | ||||
| 	if err != nil { | ||||
| 		return session, err | ||||
| 	} | ||||
| 	return session, nil | ||||
| } | ||||
|  | ||||
| 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) | ||||
| 	err = encoder.Encode(session) | ||||
|  | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| 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 | ||||
| 		return sender.Notify | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (b *Bwhatsapp) getSenderNotify(senderJid string) string { | ||||
| 	if sender, exists := b.users[senderJid]; exists { | ||||
| 		return sender.Notify | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
							
								
								
									
										305
									
								
								bridge/whatsapp/whatsapp.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,305 @@ | ||||
| package bwhatsapp | ||||
|  | ||||
| import ( | ||||
| 	"crypto/rand" | ||||
| 	"encoding/hex" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
|  | ||||
| 	"github.com/Rhymen/go-whatsapp" | ||||
|  | ||||
| 	whatsappExt "maunium.net/go/mautrix-whatsapp/whatsapp-ext" | ||||
| ) | ||||
|  | ||||
| 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 | ||||
|  | ||||
| 	// https://github.com/Rhymen/go-whatsapp/blob/c31092027237441cffba1b9cb148eadf7c83c3d2/session.go#L18-L21 | ||||
| 	session *whatsapp.Session | ||||
| 	conn    *whatsapp.Conn | ||||
| 	// https://github.com/tulir/mautrix-whatsapp/blob/master/whatsapp-ext/whatsapp.go | ||||
| 	connExt   *whatsappExt.ExtendedConn | ||||
| 	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 | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) Connect() error { | ||||
| 	b.RLock() // TODO do we need locking for Whatsapp? | ||||
| 	defer b.RUnlock() | ||||
|  | ||||
| 	number := b.GetString(cfgNumber) | ||||
| 	if number == "" { | ||||
| 		return errors.New("WhatsApp's telephone Number need to be configured") | ||||
| 	} | ||||
|  | ||||
| 	// https://github.com/Rhymen/go-whatsapp#creating-a-connection | ||||
| 	b.Log.Debugln("Connecting to WhatsApp..") | ||||
| 	conn, err := whatsapp.NewConn(20 * time.Second) | ||||
| 	if err != nil { | ||||
| 		return errors.New("failed to connect to WhatsApp: " + err.Error()) | ||||
| 	} | ||||
|  | ||||
| 	b.conn = conn | ||||
| 	b.connExt = whatsappExt.ExtendConn(b.conn) | ||||
| 	// TODO do we want to use it? b.connExt.SetClientName("Matterbridge WhatsApp bridge", "mb-wa") | ||||
|  | ||||
| 	b.conn.AddHandler(b) | ||||
| 	b.Log.Debugln("WhatsApp connection successful") | ||||
|  | ||||
| 	// load existing session in order to keep it between restarts | ||||
| 	if b.session == nil { | ||||
| 		var session whatsapp.Session | ||||
| 		session, err = b.readSession() | ||||
|  | ||||
| 		if err == nil { | ||||
| 			b.Log.Debugln("Restoring WhatsApp session..") | ||||
|  | ||||
| 			// https://github.com/Rhymen/go-whatsapp#restore | ||||
| 			session, err = b.conn.RestoreSession(session) | ||||
| 			if err != nil { | ||||
| 				// TODO return or continue to normal login? | ||||
| 				// restore session connection timed out (I couldn't get over it without logging in again) | ||||
| 				return errors.New("failed to restore session: " + err.Error()) | ||||
| 			} | ||||
|  | ||||
| 			b.session = &session | ||||
| 			b.Log.Debugln("Session restored successfully!") | ||||
| 		} else { | ||||
| 			b.Log.Warn(err.Error()) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// login to a new session | ||||
| 	if b.session == nil { | ||||
| 		err = b.Login() | ||||
| 		if 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) | ||||
| 	} | ||||
|  | ||||
| 	// 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.connExt.GetProfilePicThumb(jid) | ||||
| 			if err != nil { | ||||
| 				b.Log.Warnf("Could not get profile photo of %s: %v", jid, err) | ||||
|  | ||||
| 			} else { | ||||
| 				// TODO any race conditions here? | ||||
| 				b.userAvatars[jid] = info.URL | ||||
| 			} | ||||
| 		} | ||||
| 		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) | ||||
| 	} | ||||
|  | ||||
| 	// TODO change connection strings to configured ones longClientName:"github.com/rhymen/go-whatsapp", shortClientName:"go-whatsapp"}" prefix=whatsapp | ||||
| 	// TODO get also a nice logo | ||||
|  | ||||
| 	// TODO notification about unplugged and dead battery | ||||
| 	// conn.Info: Wid, Pushname, Connected, Battery, Plugged | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Disconnect is called while reconnecting to the bridge | ||||
| // TODO 42wim Documentation would be helpful on when reconnects happen and what should be done in this function | ||||
| // Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) Disconnect() error { | ||||
| 	// We could Logout, but that would close the session completely and would require a new QR code scan | ||||
| 	// https://github.com/Rhymen/go-whatsapp/blob/c31092027237441cffba1b9cb148eadf7c83c3d2/session.go#L377-L381 | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func isGroupJid(identifier string) bool { | ||||
| 	return strings.HasSuffix(identifier, "@g.us") || strings.HasSuffix(identifier, "@temp") | ||||
| } | ||||
|  | ||||
| // JoinChannel Join a WhatsApp group specified in gateway config as channel='number-id@g.us' or channel='Channel name' | ||||
| // Required implementation of the Bridger interface | ||||
| // https://github.com/42wim/matterbridge/blob/2cfd880cdb0df29771bf8f31df8d990ab897889d/bridge/bridge.go#L11-L16 | ||||
| func (b *Bwhatsapp) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	byJid := isGroupJid(channel.Name) | ||||
|  | ||||
| 	// verify if we are member of the given group | ||||
| 	if byJid { | ||||
| 		// channel.Name specifies static group jID, not the name | ||||
| 		if _, exists := b.conn.Store.Contacts[channel.Name]; !exists { | ||||
| 			return fmt.Errorf("account doesn't belong to group with jid %s", channel.Name) | ||||
| 		} | ||||
| 	} else { | ||||
| 		// channel.Name specifies group name that might change, warn about it | ||||
| 		var jids []string | ||||
| 		for id, contact := range b.conn.Store.Contacts { | ||||
| 			if isGroupJid(id) && contact.Name == channel.Name { | ||||
| 				jids = append(jids, id) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		switch len(jids) { | ||||
| 		case 0: | ||||
| 			// didn't match any group - print out possibilites | ||||
| 			// TODO sort | ||||
| 			// copy b; | ||||
| 			//sort.Slice(people, func(i, j int) bool { | ||||
| 			//	return people[i].Age > people[j].Age | ||||
| 			//}) | ||||
| 			for id, contact := range b.conn.Store.Contacts { | ||||
| 				if isGroupJid(id) { | ||||
| 					b.Log.Infof("%s %s", contact.Jid, contact.Name) | ||||
| 				} | ||||
| 			} | ||||
| 			return fmt.Errorf("please specify group's JID from the list above instead of the name '%s'", channel.Name) | ||||
|  | ||||
| 		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) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // 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 | ||||
|  | ||||
| 			// TODO 42wim Doesn't the app get clogged with a ton of IDs after some time of running? | ||||
| 			// WhatsApp allows to set any ID so in that case we could use external IDs and don't do mapping | ||||
| 			// but external IDs are not set | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		// TODO delete message on WhatsApp https://github.com/Rhymen/go-whatsapp/issues/100 | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Edit message | ||||
| 	if msg.ID != "" { | ||||
| 		b.Log.Debugf("updating message with id %s", msg.ID) | ||||
|  | ||||
| 		msg.Text += " (edited)" | ||||
| 		// TODO handle edit as a message reply with updated text | ||||
| 	} | ||||
|  | ||||
| 	//// TODO Handle Upload a file | ||||
| 	//if msg.Extra != nil { | ||||
| 	//	for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 	//		b.c.SendMessage(roomID, rmsg.Username+rmsg.Text) | ||||
| 	//	} | ||||
| 	//	if len(msg.Extra["file"]) > 0 { | ||||
| 	//		return b.handleUploadFile(&msg, roomID) | ||||
| 	//	} | ||||
| 	//} | ||||
|  | ||||
| 	// Post text message | ||||
| 	text := whatsapp.TextMessage{ | ||||
| 		Info: whatsapp.MessageInfo{ | ||||
| 			RemoteJid: msg.Channel, // which equals to group id | ||||
| 		}, | ||||
| 		Text: msg.Username + msg.Text, | ||||
| 	} | ||||
|  | ||||
| 	b.Log.Debugf("=> Sending %#v", msg) | ||||
|  | ||||
| 	// create message ID | ||||
| 	// TODO follow and act if https://github.com/Rhymen/go-whatsapp/issues/101 implemented | ||||
| 	bytes := make([]byte, 10) | ||||
| 	if _, err := rand.Read(bytes); err != nil { | ||||
| 		b.Log.Warn(err.Error()) | ||||
| 	} | ||||
| 	text.Info.Id = strings.ToUpper(hex.EncodeToString(bytes)) | ||||
|  | ||||
| 	err := b.conn.Send(text) | ||||
|  | ||||
| 	return text.Info.Id, err | ||||
| } | ||||
|  | ||||
| // 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 "" | ||||
| //} | ||||
| @@ -17,6 +17,7 @@ type Bxmpp struct { | ||||
| 	xc      *xmpp.Client | ||||
| 	xmppMap map[string]string | ||||
| 	*bridge.Config | ||||
| 	startTime time.Time | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| @@ -51,7 +52,7 @@ func (b *Bxmpp) Connect() error { | ||||
| 			time.Sleep(d) | ||||
| 			b.xc, err = b.createXMPP() | ||||
| 			if err == nil { | ||||
| 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS} | ||||
| 				b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: "", Account: b.Account, Event: config.EventRejoinChannels} | ||||
| 				b.handleXMPP() | ||||
| 				bf.Reset() | ||||
| 			} | ||||
| @@ -75,10 +76,8 @@ func (b *Bxmpp) JoinChannel(channel config.ChannelInfo) error { | ||||
| } | ||||
|  | ||||
| func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||
| 	var msgid = "" | ||||
| 	var msgreplaceid = "" | ||||
| 	// ignore delete messages | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
| @@ -93,7 +92,8 @@ func (b *Bxmpp) Send(msg config.Message) (string, error) { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	msgid = xid.New().String() | ||||
| 	var msgreplaceid string | ||||
| 	msgid := xid.New().String() | ||||
| 	if msg.ID != "" { | ||||
| 		msgid = msg.ID | ||||
| 		msgreplaceid = msg.ID | ||||
| @@ -154,6 +154,7 @@ func (b *Bxmpp) xmppKeepAlive() chan bool { | ||||
| func (b *Bxmpp) handleXMPP() error { | ||||
| 	var ok bool | ||||
| 	var msgid string | ||||
| 	b.startTime = time.Now() | ||||
| 	done := b.xmppKeepAlive() | ||||
| 	defer close(done) | ||||
| 	for { | ||||
| @@ -165,20 +166,32 @@ func (b *Bxmpp) handleXMPP() error { | ||||
| 		case xmpp.Chat: | ||||
| 			if v.Type == "groupchat" { | ||||
| 				b.Log.Debugf("== Receiving %#v", v) | ||||
| 				event := "" | ||||
| 				// skip invalid messages | ||||
| 				if b.skipMessage(v) { | ||||
| 					continue | ||||
| 				} | ||||
| 				if strings.Contains(v.Text, "has set the subject to:") { | ||||
| 					event = config.EventTopicChange | ||||
| 				} | ||||
| 				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} | ||||
| 				rmsg := config.Message{ | ||||
| 					Username: b.parseNick(v.Remote), | ||||
| 					Text:     v.Text, | ||||
| 					Channel:  b.parseChannel(v.Remote), | ||||
| 					Account:  b.Account, | ||||
| 					UserID:   v.Remote, | ||||
| 					ID:       msgid, | ||||
| 					Event:    event, | ||||
| 				} | ||||
|  | ||||
| 				// check if we have an action event | ||||
| 				rmsg.Text, ok = b.replaceAction(rmsg.Text) | ||||
| 				if ok { | ||||
| 					rmsg.Event = config.EVENT_USER_ACTION | ||||
| 					rmsg.Event = config.EventUserAction | ||||
| 				} | ||||
| 				b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account) | ||||
| 				b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| @@ -260,6 +273,11 @@ func (b *Bxmpp) skipMessage(message xmpp.Chat) bool { | ||||
| 		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 | ||||
| 	} | ||||
|  | ||||
| 	// skip delayed messages | ||||
| 	t := time.Time{} | ||||
| 	return message.Stamp != t | ||||
|   | ||||
| @@ -4,6 +4,7 @@ import ( | ||||
| 	"encoding/json" | ||||
| 	"io/ioutil" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| @@ -17,10 +18,12 @@ type Bzulip struct { | ||||
| 	bot     *gzb.Bot | ||||
| 	streams map[int]string | ||||
| 	*bridge.Config | ||||
| 	channelToTopic map[string]string | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	return &Bzulip{Config: cfg, streams: make(map[int]string)} | ||||
| 	return &Bzulip{Config: cfg, streams: make(map[int]string), channelToTopic: make(map[string]string)} | ||||
| } | ||||
|  | ||||
| func (b *Bzulip) Connect() error { | ||||
| @@ -45,6 +48,9 @@ func (b *Bzulip) Disconnect() error { | ||||
| } | ||||
|  | ||||
| func (b *Bzulip) JoinChannel(channel config.ChannelInfo) error { | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
| 	b.channelToTopic[channel.Name] = channel.Options.Topic | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| @@ -52,7 +58,7 @@ func (b *Bzulip) Send(msg config.Message) (string, error) { | ||||
| 	b.Log.Debugf("=> Receiving %#v", msg) | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EVENT_MSG_DELETE { | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| @@ -100,7 +106,30 @@ func (b *Bzulip) getChannel(id int) string { | ||||
|  | ||||
| func (b *Bzulip) handleQueue() error { | ||||
| 	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 | ||||
| 			b.q, err = b.bot.RegisterAll() | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("reconnecting failed: %s. Sleeping 10 seconds", err) | ||||
| 				time.Sleep(time.Second * 10) | ||||
| 				continue | ||||
| 			} | ||||
| 		case gzb.HeartbeatError: | ||||
| 			b.Log.Debug("heartbeat received.") | ||||
| 		default: | ||||
| 			b.Log.Debugf("receiving error: %#v", err) | ||||
| 		} | ||||
| 		if err != nil { | ||||
| 			continue | ||||
| 		} | ||||
| 		for _, m := range messages { | ||||
| 			b.Log.Debugf("== Receiving %#v", m) | ||||
| 			// ignore our own messages | ||||
| @@ -122,6 +151,9 @@ func (b *Bzulip) sendMessage(msg config.Message) (string, error) { | ||||
| 	if b.GetString("topic") != "" { | ||||
| 		topic = b.GetString("topic") | ||||
| 	} | ||||
| 	if res := b.getTopic(msg.Channel); res != "" { | ||||
| 		topic = res | ||||
| 	} | ||||
| 	m := gzb.Message{ | ||||
| 		Stream:  msg.Channel, | ||||
| 		Topic:   topic, | ||||
| @@ -168,3 +200,9 @@ func (b *Bzulip) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Bzulip) getTopic(channel string) string { | ||||
| 	b.RLock() | ||||
| 	defer b.RUnlock() | ||||
| 	return b.channelToTopic[channel] | ||||
| } | ||||
|   | ||||
							
								
								
									
										124
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						| @@ -1,3 +1,127 @@ | ||||
| # v1.14.0-rc1 | ||||
|  | ||||
| ## New features | ||||
| * whatsapp: new protocol added. Add initial WhatsApp support (#711) Thanks to @KrzysztofMadejski | ||||
| * facebook messenger: new protocol via matterbridge api. See https://github.com/VictorNine/fbridge/ for more information. | ||||
| * general: Add scripting (tengo) support for every incoming message (#731). See `TengoModifyMessage` | ||||
| * general: Allow regexs in ignoreNicks. Closes #690 (#720) | ||||
| * general: Support rewriting messages from relaybots using ExtractNicks. Fixes #466 (#730). See `ExtractNicks` in matterbridge.toml.sample | ||||
| * general: refactor Make all loggers derive from non-default instance (#728). Thanks to @Helcaraxan | ||||
| * rocketchat: add support for the rocketchat API. Sending to rocketchat now supports uploading of files, editing and deleting of messages. | ||||
| * discord: Support join/leaves from discord. Closes #654 (#721) | ||||
| * discord: Allow sending discriminator with Discord username (#726). See `UseDiscriminator` in matterbridge.toml.sample | ||||
| * zulip: Allow zulip bridge to specify topic per channel. Closes #701 (#723). See `Topic` in matterbridge.toml.sample | ||||
| * slack: Add extra debug option (slack). See `Debug` in the slack section in matterbridge.toml.sample | ||||
|  | ||||
| ## Bugfix | ||||
| * zulip: Fix error handling on bad event queue id (zulip). Closes #694 | ||||
| * irc: add support for (older) unrealircd versions. #708 | ||||
| * irc: Support quits from irc correctly. Fixes #722 (#724) | ||||
| * matrix: Send username when uploading video/images (matrix). Fixes #715 (#717) | ||||
| * matrix: Send notices on join/parts (matrix). Fixes #712 (#716) | ||||
| * matrix: Detect html nicks in RemoteNickFormat (matrix). Fixes #696 (#719) | ||||
| * slack: Hint at thread replies when messages are unthreaded (slack) (#684) | ||||
| * xmpp: Do not send topic changes on connect (xmpp). Fixes #732 (#733) | ||||
|  | ||||
| ## Contributors | ||||
| This release couldn't exist without the following contributors: | ||||
| @Helcaraxan, @KrzysztofMadejski, @AJolly, @DeclanHoare | ||||
|  | ||||
| # v1.13.1 | ||||
|  | ||||
| This release fixes go modules issues because of https://github.com/labstack/echo/issues/1272 | ||||
|  | ||||
| ## Bugfix | ||||
| * general: fixes Unable to build 1.13.0 #698 | ||||
| * api: move to labstack/echo/v4 fixes #698 | ||||
|  | ||||
| # v1.13.0 | ||||
|  | ||||
| ## New features | ||||
| * general: refactors of telegram, irc, mattermost, matrix, discord, sshchat bridges and the gateway. | ||||
| * irc: Add option to send RAW commands after connection (irc) #490. See `RunCommands` in matterbridge.toml.sample | ||||
| * mattermost: 3.x support dropped | ||||
| * mattermost: Add support for mattermost threading (#627) | ||||
| * slack: Sync channel topics between Slack bridges #585. See `SyncTopic` in matterbridge.toml.sample | ||||
| * matrix: Add support for markdown to HTML conversion (matrix). Closes #663 (#670) | ||||
| * discord: Improve error reporting on failure to join Discord. Fixes #672 (#680) | ||||
| * discord: Use only one webhook if possible (discord) (#681) | ||||
| * discord: Allow to bridge non-bot Discord users (discord) (#689) If you prefix a token with `User ` it'll treat is as a user token. | ||||
|  | ||||
| ## Bugfix | ||||
| * slack: Try downloading files again if slack is too slow (slack). Closes #655 (#656) | ||||
| * slack: Ignore LatencyReport event (slack) | ||||
| * slack: Fix #668 strip lang in code fences sent to Slack (#673) | ||||
| * sshchat: Fix sshchat connection logic (#661) | ||||
| * sshchat: set quiet mode to filter joins/quits | ||||
| * sshchat: Trim newlines in the end of relayed messages | ||||
| * sshchat: fix media links | ||||
| * sshchat: do not relay "Rate limiting is in effect" message | ||||
| * mattermost: Fail if channel starts with hashtag (mattermost). Closes #625 | ||||
| * discord: Add file comment to webhook messages (discord). Fixes #358 | ||||
| * matrix: Fix displaying usernames for plain text clients. (matrix) (#685) | ||||
| * irc: Fix possible data race (irc). Closes #693 | ||||
| * irc: Handle servers without MOTD (irc). Closes #692 | ||||
|  | ||||
| # v1.12.3 | ||||
| ## Bugfix | ||||
| * slack: Fix bot (legacy token) messages not being send. Closes #571 | ||||
| * slack: Populate user on channel join (slack) (#644) | ||||
| * slack: Add wait option for populateUsers/Channels (slack) Fixes #579 (#653) | ||||
|  | ||||
| # v1.12.2 | ||||
|  | ||||
| ## Bugfix | ||||
| * irc: Fix multiple channel join regression. Closes #639 | ||||
| * slack: Make slack-legacy change less restrictive (#626) | ||||
|  | ||||
| # v1.12.1 | ||||
|  | ||||
| ## Bugfix | ||||
| * discord: fix regression on server ID connection #619 #617 | ||||
| * discord: Limit discord username via webhook to 32 chars | ||||
| * slack: Make sure threaded files stay in thread (slack). Fixes #590 | ||||
| * slack: Do not post empty messages (slack). Fixes #574 | ||||
| * slack: Handle deleted/edited thread starting messages (slack). Fixes #600 (#605) | ||||
| * irc: Rework connection logic (irc) | ||||
| * irc: Fix Nickserv logic (irc) #602 | ||||
|  | ||||
| # v1.12.0 | ||||
|  | ||||
| ## Breaking changes | ||||
| The slack bridge has been split in a `slack-legacy` and `slack` bridge. | ||||
| If you're still using `legacy tokens` and want to keep using them you'll have to rename `slack` to `slack-legacy` in your configuration. See [wiki](https://github.com/42wim/matterbridge/wiki/Section-Slack-(basic)#legacy-configuration) for more information. | ||||
|  | ||||
| To migrate to the new bot-token based setup you can follow the instructions [here](https://github.com/42wim/matterbridge/wiki/Slack-bot-setup). | ||||
|  | ||||
| Slack legacy tokens may be deprecated by Slack at short notice, so it is STRONGLY recommended to use a proper bot-token instead. | ||||
|  | ||||
| ## New features | ||||
| * general: New {GATEWAY} variable for `RemoteNickFormat` #501. See `RemoteNickFormat` in matterbridge.toml.sample. | ||||
| * general: New {CHANNEL} variable for `RemoteNickFormat` #515. See `RemoteNickFormat` in matterbridge.toml.sample. | ||||
| * general: Remove hyphens when auto-loading envvars from viper config #545 | ||||
| * discord: You can mention discord-users from other bridges. | ||||
| * slack: Preserve threading between Slack instances #529. See `PreserveThreading` in matterbridge.toml.sample. | ||||
| * slack: Add ability to show when user is typing across Slack bridges #559 | ||||
| * slack: Add rate-limiting | ||||
| * mattermost: Add support for mattermost [matterbridge plugin](https://github.com/matterbridge/mattermost-plugin) | ||||
| * api: Respond with message on connect. #550 | ||||
| * api: Add a health endpoint to API #554 | ||||
|  | ||||
| ## Bugfix | ||||
| * slack: Refactoring and making it better. | ||||
| * slack: Restore file comments coming from Slack. #583 | ||||
| * irc: Fix IRC line splitting. #587 | ||||
| * mattermost: Fix cookie and personal token behaviour. #530 | ||||
| * mattermost: Check for expiring sessions and reconnect. | ||||
|  | ||||
|  | ||||
| ## Contributors | ||||
| This release couldn't exist without the following contributors: | ||||
| @jheiselman, @NikkyAI, @dajohi, @NetwideRogue, @patcon and @Helcaraxan | ||||
|  | ||||
| Special thanks to @Helcaraxan and @patcon for their work on improving/refactoring slack. | ||||
|  | ||||
| # v1.11.3 | ||||
|  | ||||
| ## Bugfix | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| #!/bin/bash | ||||
| go version |grep go1.10 || exit | ||||
| 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 | ||||
|   | ||||
							
								
								
									
										210
									
								
								contrib/api.yaml
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,210 @@ | ||||
| openapi: 3.0.0 | ||||
| info: | ||||
|   contact: {} | ||||
|   description: A read/write API for the Matterbridge chat bridge. | ||||
|   license: | ||||
|     name: Apache 2.0 | ||||
|     url: 'https://github.com/42wim/matterbridge/blob/master/LICENSE' | ||||
|   title: Matterbridge API | ||||
|   version: "0.1.0-oas3" | ||||
| paths: | ||||
|   /health: | ||||
|     get: | ||||
|       responses: | ||||
|         '200': | ||||
|           description: OK | ||||
|           content: | ||||
|             '*/*': | ||||
|               schema: | ||||
|                 type: string | ||||
|       summary: Checks if the server is alive. | ||||
|   /message: | ||||
|     post: | ||||
|       responses: | ||||
|         '200': | ||||
|           description: OK | ||||
|           content: | ||||
|             application/json: | ||||
|               schema: | ||||
|                 $ref: '#/components/schemas/config.OutgoingMessageResponse' | ||||
|       summary: Create a message | ||||
|       requestBody: | ||||
|         content: | ||||
|           application/json: | ||||
|             schema: | ||||
|               $ref: '#/components/schemas/config.OutgoingMessage' | ||||
|         description: Message object to create | ||||
|         required: true | ||||
|   /messages: | ||||
|     get: | ||||
|       responses: | ||||
|         '200': | ||||
|           description: OK | ||||
|           content: | ||||
|             application/json: | ||||
|               schema: | ||||
|                 items: | ||||
|                   $ref: '#/components/schemas/config.IncomingMessage' | ||||
|                 type: array | ||||
|       security: | ||||
|         - ApiKeyAuth: [] | ||||
|       summary: List new messages | ||||
|   /stream: | ||||
|     get: | ||||
|       responses: | ||||
|         '200': | ||||
|           description: OK | ||||
|           content: | ||||
|             application/x-json-stream: | ||||
|               schema: | ||||
|                 $ref: '#/components/schemas/config.IncomingMessage' | ||||
|       summary: Stream realtime messages | ||||
| servers: | ||||
|   - url: /api | ||||
| components: | ||||
|   securitySchemes: | ||||
|     bearerAuth: | ||||
|       type: http | ||||
|       scheme: bearer | ||||
|   schemas: | ||||
|     config.IncomingMessage: | ||||
|       properties: | ||||
|         avatar: | ||||
|           description: URL to an avatar image | ||||
|           example: >- | ||||
|             https://secure.gravatar.com/avatar/1234567890abcdef1234567890abcdef.jpg | ||||
|           type: string | ||||
|         event: | ||||
|           description: >- | ||||
|             A specific matterbridge event. (see | ||||
|             https://github.com/42wim/matterbridge/blob/master/bridge/config/config.go#L16) | ||||
|           type: string | ||||
|         gateway: | ||||
|           description: Name of the gateway as configured in matterbridge.toml | ||||
|           example: mygateway | ||||
|           type: string | ||||
|         text: | ||||
|           description: Content of the message | ||||
|           example: 'Testing, testing, 1-2-3.' | ||||
|           type: string | ||||
|         username: | ||||
|           description: Human-readable username | ||||
|           example: alice | ||||
|           type: string | ||||
|         account: | ||||
|           description: Unique account name of format "[protocol].[slug]" as defined in matterbridge.toml  | ||||
|           example: slack.myteam | ||||
|           type: string | ||||
|         channel: | ||||
|           description: Human-readable channel name of sending bridge | ||||
|           example: test-channel | ||||
|           type: string | ||||
|         id: | ||||
|           description: Unique ID of message on the gateway | ||||
|           example: slack 1541361213.030700 | ||||
|           type: string | ||||
|         parent_id: | ||||
|           description: Unique ID of a parent message, if threaded | ||||
|           example: slack 1541361213.030700 | ||||
|           type: string | ||||
|         protocol: | ||||
|           description: Chat protocol of the sending bridge | ||||
|           example: slack | ||||
|           type: string | ||||
|         timestamp: | ||||
|           description: Timestamp of the message | ||||
|           example: "1541361213.030700" | ||||
|           type: string | ||||
|         userid: | ||||
|           description: Userid on the sending bridge | ||||
|           example: U4MCXJKNC | ||||
|           type: string | ||||
|         extra: | ||||
|           description: Extra data that doesn't fit in other fields (eg base64 encoded files) | ||||
|           type: object | ||||
|     config.OutgoingMessage: | ||||
|       properties: | ||||
|         avatar: | ||||
|           description: URL to an avatar image | ||||
|           example: >- | ||||
|             https://secure.gravatar.com/avatar/1234567890abcdef1234567890abcdef.jpg | ||||
|           type: string | ||||
|         event: | ||||
|           description: >- | ||||
|             A specific matterbridge event. (see | ||||
|             https://github.com/42wim/matterbridge/blob/master/bridge/config/config.go#L16) | ||||
|           example: "" | ||||
|           type: string | ||||
|         gateway: | ||||
|           description: Name of the gateway as configured in matterbridge.toml | ||||
|           example: mygateway | ||||
|           type: string | ||||
|         text: | ||||
|           description: Content of the message | ||||
|           example: 'Testing, testing, 1-2-3.' | ||||
|           type: string | ||||
|         username: | ||||
|           description: Human-readable username | ||||
|           example: alice | ||||
|           type: string | ||||
|       type: object | ||||
|       required: | ||||
|         - gateway | ||||
|         - text | ||||
|         - username | ||||
|     config.OutgoingMessageResponse: | ||||
|       properties: | ||||
|         avatar: | ||||
|           description: URL to an avatar image | ||||
|           example: >- | ||||
|             https://secure.gravatar.com/avatar/1234567890abcdef1234567890abcdef.jpg | ||||
|           type: string | ||||
|         event: | ||||
|           description: >- | ||||
|             A specific matterbridge event. (see | ||||
|             https://github.com/42wim/matterbridge/blob/master/bridge/config/config.go#L16) | ||||
|           example: "" | ||||
|           type: string | ||||
|         gateway: | ||||
|           description: Name of the gateway as configured in matterbridge.toml | ||||
|           example: mygateway | ||||
|           type: string | ||||
|         text: | ||||
|           description: Content of the message | ||||
|           example: 'Testing, testing, 1-2-3.' | ||||
|           type: string | ||||
|         username: | ||||
|           description: Human-readable username | ||||
|           example: alice | ||||
|           type: string | ||||
|         account: | ||||
|           description: fixed api account  | ||||
|           example: api.local | ||||
|           type: string | ||||
|         channel: | ||||
|           description: fixed api channel  | ||||
|           example: api | ||||
|           type: string | ||||
|         id: | ||||
|           example: "" | ||||
|           type: string | ||||
|         parent_id: | ||||
|           example: "" | ||||
|           type: string | ||||
|         protocol: | ||||
|           description: fixed api protocol | ||||
|           example: api | ||||
|           type: string | ||||
|         timestamp: | ||||
|           description: Timestamp of the message | ||||
|           example: "1541361213.030700" | ||||
|           type: string | ||||
|         userid: | ||||
|           example: "" | ||||
|           type: string | ||||
|         extra: | ||||
|           example: null | ||||
|           type: object | ||||
|       type: object | ||||
| security: | ||||
|   - bearerAuth: [] | ||||
							
								
								
									
										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)") | ||||
| @@ -1,11 +1,9 @@ | ||||
| FROM cmosh/alpine-arm:edge | ||||
| ENTRYPOINT ["/bin/matterbridge"] | ||||
| FROM alpine:edge as certs | ||||
| RUN apk --update add ca-certificates | ||||
|  | ||||
| COPY . /go/src/github.com/42wim/matterbridge | ||||
| RUN apk update && apk add go git gcc musl-dev ca-certificates \ | ||||
|         && cd /go/src/github.com/42wim/matterbridge \ | ||||
|         && export GOPATH=/go \ | ||||
|         && go get \ | ||||
|         && go build -x -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge \ | ||||
|         && rm -rf /go \ | ||||
|         && apk del --purge git go gcc musl-dev | ||||
| FROM scratch | ||||
| ARG VERSION=1.12.3 | ||||
| COPY --from=certs /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt | ||||
| ADD https://github.com/42wim/matterbridge/releases/download/v${VERSION}/matterbridge-linux-arm /bin/matterbridge | ||||
| RUN chmod +x /bin/matterbridge | ||||
| ENTRYPOINT ["/bin/matterbridge"] | ||||
|   | ||||
							
								
								
									
										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" | ||||
| } | ||||
							
								
								
									
										37
									
								
								gateway/bridgemap/bridgemap.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,37 @@ | ||||
| package bridgemap | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/api" | ||||
| 	"github.com/42wim/matterbridge/bridge/discord" | ||||
| 	"github.com/42wim/matterbridge/bridge/gitter" | ||||
| 	"github.com/42wim/matterbridge/bridge/irc" | ||||
| 	"github.com/42wim/matterbridge/bridge/matrix" | ||||
| 	"github.com/42wim/matterbridge/bridge/mattermost" | ||||
| 	"github.com/42wim/matterbridge/bridge/rocketchat" | ||||
| 	"github.com/42wim/matterbridge/bridge/slack" | ||||
| 	"github.com/42wim/matterbridge/bridge/sshchat" | ||||
| 	"github.com/42wim/matterbridge/bridge/steam" | ||||
| 	"github.com/42wim/matterbridge/bridge/telegram" | ||||
| 	"github.com/42wim/matterbridge/bridge/whatsapp" | ||||
| 	"github.com/42wim/matterbridge/bridge/xmpp" | ||||
| 	"github.com/42wim/matterbridge/bridge/zulip" | ||||
| ) | ||||
|  | ||||
| var FullMap = map[string]bridge.Factory{ | ||||
| 	"api":          api.New, | ||||
| 	"discord":      bdiscord.New, | ||||
| 	"gitter":       bgitter.New, | ||||
| 	"irc":          birc.New, | ||||
| 	"mattermost":   bmattermost.New, | ||||
| 	"matrix":       bmatrix.New, | ||||
| 	"rocketchat":   brocketchat.New, | ||||
| 	"slack-legacy": bslack.NewLegacy, | ||||
| 	"slack":        bslack.New, | ||||
| 	"sshchat":      bsshchat.New, | ||||
| 	"steam":        bsteam.New, | ||||
| 	"telegram":     btelegram.New, | ||||
| 	"whatsapp":     bwhatsapp.New, | ||||
| 	"xmpp":         bxmpp.New, | ||||
| 	"zulip":        bzulip.New, | ||||
| } | ||||
| @@ -1,41 +1,23 @@ | ||||
| package gateway | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 	"os" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/api" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	bdiscord "github.com/42wim/matterbridge/bridge/discord" | ||||
| 	bgitter "github.com/42wim/matterbridge/bridge/gitter" | ||||
| 	birc "github.com/42wim/matterbridge/bridge/irc" | ||||
| 	bmatrix "github.com/42wim/matterbridge/bridge/matrix" | ||||
| 	bmattermost "github.com/42wim/matterbridge/bridge/mattermost" | ||||
| 	brocketchat "github.com/42wim/matterbridge/bridge/rocketchat" | ||||
| 	bslack "github.com/42wim/matterbridge/bridge/slack" | ||||
| 	bsshchat "github.com/42wim/matterbridge/bridge/sshchat" | ||||
| 	bsteam "github.com/42wim/matterbridge/bridge/steam" | ||||
| 	btelegram "github.com/42wim/matterbridge/bridge/telegram" | ||||
| 	bxmpp "github.com/42wim/matterbridge/bridge/xmpp" | ||||
| 	bzulip "github.com/42wim/matterbridge/bridge/zulip" | ||||
| 	"github.com/hashicorp/golang-lru" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	//	"github.com/davecgh/go-spew/spew" | ||||
| 	"crypto/sha1" | ||||
| 	"path/filepath" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/d5/tengo/script" | ||||
| 	lru "github.com/hashicorp/golang-lru" | ||||
| 	"github.com/peterhellberg/emojilib" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Gateway struct { | ||||
| 	*config.Config | ||||
| 	config.Config | ||||
|  | ||||
| 	Router         *Router | ||||
| 	MyConfig       *config.Gateway | ||||
| 	Bridges        map[string]*bridge.Bridge | ||||
| @@ -44,6 +26,8 @@ type Gateway struct { | ||||
| 	Message        chan config.Message | ||||
| 	Name           string | ||||
| 	Messages       *lru.Cache | ||||
|  | ||||
| 	logger *logrus.Entry | ||||
| } | ||||
|  | ||||
| type BrMsgID struct { | ||||
| @@ -52,59 +36,78 @@ type BrMsgID struct { | ||||
| 	ChannelID string | ||||
| } | ||||
|  | ||||
| var flog *log.Entry | ||||
| const apiProtocol = "api" | ||||
|  | ||||
| var bridgeMap = map[string]bridge.Factory{ | ||||
| 	"api":        api.New, | ||||
| 	"discord":    bdiscord.New, | ||||
| 	"gitter":     bgitter.New, | ||||
| 	"irc":        birc.New, | ||||
| 	"mattermost": bmattermost.New, | ||||
| 	"matrix":     bmatrix.New, | ||||
| 	"rocketchat": brocketchat.New, | ||||
| 	"slack":      bslack.New, | ||||
| 	"sshchat":    bsshchat.New, | ||||
| 	"steam":      bsteam.New, | ||||
| 	"telegram":   btelegram.New, | ||||
| 	"xmpp":       bxmpp.New, | ||||
| 	"zulip":      bzulip.New, | ||||
| } | ||||
| // New creates a new Gateway object associated with the specified router and | ||||
| // following the given configuration. | ||||
| func New(rootLogger *logrus.Logger, cfg *config.Gateway, r *Router) *Gateway { | ||||
| 	logger := rootLogger.WithFields(logrus.Fields{"prefix": "gateway"}) | ||||
|  | ||||
| func init() { | ||||
| 	flog = log.WithFields(log.Fields{"prefix": "gateway"}) | ||||
| } | ||||
|  | ||||
| func New(cfg config.Gateway, r *Router) *Gateway { | ||||
| 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | ||||
| 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | ||||
| 	cache, _ := lru.New(5000) | ||||
| 	gw.Messages = cache | ||||
| 	gw.AddConfig(&cfg) | ||||
| 	gw := &Gateway{ | ||||
| 		Channels: make(map[string]*config.ChannelInfo), | ||||
| 		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 | ||||
| } | ||||
|  | ||||
| // FindCanonicalMsgID returns the ID under which a message was stored in the cache. | ||||
| func (gw *Gateway) FindCanonicalMsgID(protocol string, mID string) string { | ||||
| 	ID := protocol + " " + mID | ||||
| 	if gw.Messages.Contains(ID) { | ||||
| 		return mID | ||||
| 	} | ||||
|  | ||||
| 	// If not keyed, iterate through cache for downstream, and infer upstream. | ||||
| 	for _, mid := range gw.Messages.Keys() { | ||||
| 		v, _ := gw.Messages.Peek(mid) | ||||
| 		ids := v.([]*BrMsgID) | ||||
| 		for _, downstreamMsgObj := range ids { | ||||
| 			if ID == downstreamMsgObj.ID { | ||||
| 				return strings.Replace(mid.(string), protocol+" ", "", 1) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| // AddBridge sets up a new bridge in the gateway object with the specified configuration. | ||||
| func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||
| 	br := gw.Router.getBridge(cfg.Account) | ||||
| 	if br == nil { | ||||
| 		br = bridge.New(cfg) | ||||
| 		br.Config = gw.Router.Config | ||||
| 		br.General = &gw.General | ||||
| 		// set logging | ||||
| 		br.Log = log.WithFields(log.Fields{"prefix": "bridge"}) | ||||
| 		brconfig := &bridge.Config{Remote: gw.Message, Log: log.WithFields(log.Fields{"prefix": br.Protocol}), Bridge: br} | ||||
| 		br.General = &gw.BridgeValues().General | ||||
| 		br.Log = gw.logger.WithFields(logrus.Fields{"prefix": br.Protocol}) | ||||
| 		brconfig := &bridge.Config{ | ||||
| 			Remote: gw.Message, | ||||
| 			Bridge: br, | ||||
| 		} | ||||
| 		// add the actual bridger for this protocol to this bridge using the bridgeMap | ||||
| 		br.Bridger = bridgeMap[br.Protocol](brconfig) | ||||
| 		br.Bridger = gw.Router.BridgeMap[br.Protocol](brconfig) | ||||
| 	} | ||||
| 	gw.mapChannelsToBridge(br) | ||||
| 	gw.Bridges[cfg.Account] = br | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // AddConfig associates a new configuration with the gateway object. | ||||
| func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | ||||
| 	gw.Name = cfg.Name | ||||
| 	gw.MyConfig = cfg | ||||
| 	gw.mapChannels() | ||||
| 	if err := gw.mapChannels(); err != nil { | ||||
| 		gw.logger.Errorf("mapChannels() failed: %s", err) | ||||
| 	} | ||||
| 	for _, br := range append(gw.MyConfig.In, append(gw.MyConfig.InOut, gw.MyConfig.Out...)...) { | ||||
| 		br := br //scopelint | ||||
| 		err := gw.AddBridge(&br) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| @@ -122,33 +125,47 @@ func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) { | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | ||||
| 	br.Disconnect() | ||||
| 	if err := br.Disconnect(); err != nil { | ||||
| 		gw.logger.Errorf("Disconnect() %s failed: %s", br.Account, err) | ||||
| 	} | ||||
| 	time.Sleep(time.Second * 5) | ||||
| RECONNECT: | ||||
| 	flog.Infof("Reconnecting %s", br.Account) | ||||
| 	gw.logger.Infof("Reconnecting %s", br.Account) | ||||
| 	err := br.Connect() | ||||
| 	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) | ||||
| 		goto RECONNECT | ||||
| 	} | ||||
| 	br.Joined = make(map[string]bool) | ||||
| 	br.JoinChannels() | ||||
| 	if err := br.JoinChannels(); err != nil { | ||||
| 		gw.logger.Errorf("JoinChannels() %s failed: %s", br.Account, err) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | ||||
| 	for _, br := range cfg { | ||||
| 		if isApi(br.Account) { | ||||
| 			br.Channel = "api" | ||||
| 		if isAPI(br.Account) { | ||||
| 			br.Channel = apiProtocol | ||||
| 		} | ||||
| 		// make sure to lowercase irc channels in config #348 | ||||
| 		if strings.HasPrefix(br.Account, "irc.") { | ||||
| 			br.Channel = strings.ToLower(br.Channel) | ||||
| 		} | ||||
| 		if strings.HasPrefix(br.Account, "mattermost.") && strings.HasPrefix(br.Channel, "#") { | ||||
| 			gw.logger.Errorf("Mattermost channels do not start with a #: remove the # in %s", br.Channel) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
| 		ID := br.Channel + br.Account | ||||
| 		if _, ok := gw.Channels[ID]; !ok { | ||||
| 			channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account, | ||||
| 				SameChannel: make(map[string]bool)} | ||||
| 			channel := &config.ChannelInfo{ | ||||
| 				Name:        br.Channel, | ||||
| 				Direction:   direction, | ||||
| 				ID:          ID, | ||||
| 				Options:     br.Options, | ||||
| 				Account:     br.Account, | ||||
| 				SameChannel: make(map[string]bool), | ||||
| 			} | ||||
| 			channel.SameChannel[gw.Name] = br.SameChannel | ||||
| 			gw.Channels[channel.ID] = channel | ||||
| 		} else { | ||||
| @@ -172,14 +189,42 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con | ||||
| 	var channels []config.ChannelInfo | ||||
|  | ||||
| 	// for messages received from the api check that the gateway is the specified one | ||||
| 	if msg.Protocol == "api" && gw.Name != msg.Gateway { | ||||
| 	if msg.Protocol == apiProtocol && gw.Name != msg.Gateway { | ||||
| 		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 | ||||
| 	} | ||||
|  | ||||
| 	// irc quit is for the whole bridge, isn't a per channel quit. | ||||
| 	// channel is empty when we quit | ||||
| 	if msg.Event == config.EventJoinLeave && getProtocol(msg) == "irc" && msg.Channel == "" { | ||||
| 		// if we only have one channel on this irc bridge it's got to be the sending one. | ||||
| 		// don't send it back | ||||
| 		if dest.Account == msg.Account && len(dest.Channels) == 1 && dest.Protocol == "irc" { | ||||
| 			return channels | ||||
| 		} | ||||
| 		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 | ||||
| 	for _, channel := range gw.Channels { | ||||
| 		// 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 | ||||
| 			if !strings.Contains(channel.Direction, "in") { | ||||
| 				return channels | ||||
| @@ -188,105 +233,55 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con | ||||
| 		} | ||||
| 	} | ||||
| 	for _, channel := range gw.Channels { | ||||
| 		if _, ok := gw.Channels[getChannelID(*msg)]; !ok { | ||||
| 		if _, ok := gw.Channels[getChannelID(msg)]; !ok { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// do samechannelgateway flogic | ||||
| 		// do samechannelgateway logic | ||||
| 		if channel.SameChannel[msg.Gateway] { | ||||
| 			if msg.Channel == channel.Name && msg.Account != dest.Account { | ||||
| 				channels = append(channels, *channel) | ||||
| 			} | ||||
| 			continue | ||||
| 		} | ||||
| 		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg, channel) { | ||||
| 		if strings.Contains(channel.Direction, "out") && channel.Account == dest.Account && gw.validGatewayDest(msg) { | ||||
| 			channels = append(channels, *channel) | ||||
| 		} | ||||
| 	} | ||||
| 	return channels | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||
| 	var brMsgIDs []*BrMsgID | ||||
|  | ||||
| 	// if we have an attached file, or other info | ||||
| 	if msg.Extra != nil { | ||||
| 		if len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) != 0 { | ||||
| 			if msg.Text == "" { | ||||
| 				return brMsgIDs | ||||
| func (gw *Gateway) getDestMsgID(msgID string, dest *bridge.Bridge, channel *config.ChannelInfo) string { | ||||
| 	if res, ok := gw.Messages.Get(msgID); ok { | ||||
| 		IDs := res.([]*BrMsgID) | ||||
| 		for _, id := range IDs { | ||||
| 			// check protocol, bridge name and channelname | ||||
| 			// for people that reuse the same bridge multiple times. see #342 | ||||
| 			if dest.Protocol == id.br.Protocol && dest.Name == id.br.Name && channel.ID == id.ChannelID { | ||||
| 				return strings.Replace(id.ID, dest.Protocol+" ", "", 1) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| 	// Avatar downloads are only relevant for telegram and mattermost for now | ||||
| 	if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||
| 		if dest.Protocol != "mattermost" && | ||||
| 			dest.Protocol != "telegram" { | ||||
| 			return brMsgIDs | ||||
| 		} | ||||
| // ignoreTextEmpty returns true if we need to ignore a message with an empty text. | ||||
| func (gw *Gateway) ignoreTextEmpty(msg *config.Message) bool { | ||||
| 	if msg.Text != "" { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	// only relay join/part when configured | ||||
| 	if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].GetBool("ShowJoinPart") { | ||||
| 		return brMsgIDs | ||||
| 	if msg.Event == config.EventUserTyping { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	// only relay topic change when configured | ||||
| 	if msg.Event == config.EVENT_TOPIC_CHANGE && !gw.Bridges[dest.Account].GetBool("ShowTopicChange") { | ||||
| 		return brMsgIDs | ||||
| 	// we have an attachment or actual bytes, do not ignore | ||||
| 	if msg.Extra != nil && | ||||
| 		(msg.Extra["attachments"] != nil || | ||||
| 			len(msg.Extra["file"]) > 0 || | ||||
| 			len(msg.Extra[config.EventFileFailureSize]) > 0) { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	// broadcast to every out channel (irc QUIT) | ||||
| 	if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE { | ||||
| 		flog.Debug("empty channel") | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	originchannel := msg.Channel | ||||
| 	origmsg := msg | ||||
| 	channels := gw.getDestChannel(&msg, *dest) | ||||
| 	for _, channel := range channels { | ||||
| 		// Only send the avatar download event to ourselves. | ||||
| 		if msg.Event == config.EVENT_AVATAR_DOWNLOAD { | ||||
| 			if channel.ID != getChannelID(origmsg) { | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// do not send to ourself for any other event | ||||
| 			if channel.ID == getChannelID(origmsg) { | ||||
| 				continue | ||||
| 			} | ||||
| 		} | ||||
| 		flog.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name) | ||||
| 		msg.Channel = channel.Name | ||||
| 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||
| 		msg.Username = gw.modifyUsername(origmsg, dest) | ||||
| 		msg.ID = "" | ||||
| 		if res, ok := gw.Messages.Get(origmsg.ID); ok { | ||||
| 			IDs := res.([]*BrMsgID) | ||||
| 			for _, id := range IDs { | ||||
| 				// check protocol, bridge name and channelname | ||||
| 				// for people that reuse the same bridge multiple times. see #342 | ||||
| 				if dest.Protocol == id.br.Protocol && dest.Name == id.br.Name && channel.ID == id.ChannelID { | ||||
| 					msg.ID = id.ID | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		// for api we need originchannel as channel | ||||
| 		if dest.Protocol == "api" { | ||||
| 			msg.Channel = originchannel | ||||
| 		} | ||||
| 		mID, err := dest.Send(msg) | ||||
| 		if err != nil { | ||||
| 			flog.Error(err) | ||||
| 		} | ||||
| 		// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||
| 		if mID != "" { | ||||
| 			flog.Debugf("mID %s: %s", dest.Account, mID) | ||||
| 			brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID, channel.ID}) | ||||
| 		} | ||||
| 	} | ||||
| 	return brMsgIDs | ||||
| 	gw.logger.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||
| @@ -295,56 +290,23 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// check if we need to ignore a empty message | ||||
| 	if msg.Text == "" { | ||||
| 		// we have an attachment or actual bytes, do not ignore | ||||
| 		if msg.Extra != nil && | ||||
| 			(msg.Extra["attachments"] != nil || | ||||
| 				len(msg.Extra["file"]) > 0 || | ||||
| 				len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) > 0) { | ||||
| 			return false | ||||
| 		} | ||||
| 		flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||
| 	igNicks := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks")) | ||||
| 	igMessages := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages")) | ||||
| 	if gw.ignoreTextEmpty(msg) || gw.ignoreText(msg.Username, igNicks) || gw.ignoreText(msg.Text, igMessages) { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// is the username in IgnoreNicks field | ||||
| 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks")) { | ||||
| 		if msg.Username == entry { | ||||
| 			flog.Debugf("ignoring %s from %s", msg.Username, msg.Account) | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// does the message match regex in IgnoreMessages field | ||||
| 	// TODO do not compile regexps everytime | ||||
| 	for _, entry := range strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages")) { | ||||
| 		if entry != "" { | ||||
| 			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 | ||||
| } | ||||
|  | ||||
| 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 gw.Config.General.StripNick || dest.GetBool("StripNick") { | ||||
| 	if dest.GetBool("StripNick") { | ||||
| 		re := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||
| 		msg.Username = re.ReplaceAllString(msg.Username, "") | ||||
| 	} | ||||
| 	nick := dest.GetString("RemoteNickFormat") | ||||
| 	if nick == "" { | ||||
| 		nick = gw.Config.General.RemoteNickFormat | ||||
| 	} | ||||
|  | ||||
| 	// loop to replace nicks | ||||
| 	for _, outer := range br.GetStringSlice2D("ReplaceNicks") { | ||||
| @@ -353,7 +315,7 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin | ||||
| 		// TODO move compile to bridge init somewhere | ||||
| 		re, err := regexp.Compile(search) | ||||
| 		if err != nil { | ||||
| 			flog.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||
| 			gw.logger.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||
| 			break | ||||
| 		} | ||||
| 		msg.Username = re.ReplaceAllString(msg.Username, replace) | ||||
| @@ -374,16 +336,15 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin | ||||
|  | ||||
| 	nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1) | ||||
| 	nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1) | ||||
| 	nick = strings.Replace(nick, "{GATEWAY}", gw.Name, -1) | ||||
| 	nick = strings.Replace(nick, "{LABEL}", br.GetString("Label"), -1) | ||||
| 	nick = strings.Replace(nick, "{NICK}", msg.Username, -1) | ||||
| 	nick = strings.Replace(nick, "{CHANNEL}", msg.Channel, -1) | ||||
| 	return nick | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string { | ||||
| 	iconurl := gw.Config.General.IconURL | ||||
| 	if iconurl == "" { | ||||
| 		iconurl = dest.GetString("IconURL") | ||||
| 	} | ||||
| func (gw *Gateway) modifyAvatar(msg *config.Message, dest *bridge.Bridge) string { | ||||
| 	iconurl := dest.GetString("IconURL") | ||||
| 	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1) | ||||
| 	if msg.Avatar == "" { | ||||
| 		msg.Avatar = iconurl | ||||
| @@ -392,6 +353,10 @@ func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||
| 	if err := modifyMessageTengo(gw.BridgeValues().General.TengoModifyMessage, msg); err != nil { | ||||
| 		gw.logger.Errorf("TengoModifyMessage failed: %s", err) | ||||
| 	} | ||||
|  | ||||
| 	// replace :emoji: to unicode | ||||
| 	msg.Text = emojilib.Replace(msg.Text) | ||||
|  | ||||
| @@ -403,106 +368,146 @@ func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||
| 		// TODO move compile to bridge init somewhere | ||||
| 		re, err := regexp.Compile(search) | ||||
| 		if err != nil { | ||||
| 			flog.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||
| 			gw.logger.Errorf("regexp in %s failed: %s", msg.Account, err) | ||||
| 			break | ||||
| 		} | ||||
| 		msg.Text = re.ReplaceAllString(msg.Text, replace) | ||||
| 	} | ||||
|  | ||||
| 	gw.handleExtractNicks(msg) | ||||
|  | ||||
| 	// messages from api have Gateway specified, don't overwrite | ||||
| 	if msg.Protocol != "api" { | ||||
| 	if msg.Protocol != apiProtocol { | ||||
| 		msg.Gateway = gw.Name | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleFiles uploads or places all files on the given msg to the MediaServer and | ||||
| // adds the new URL of the file on the MediaServer onto the given msg. | ||||
| func (gw *Gateway) handleFiles(msg *config.Message) { | ||||
| 	reg := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||
|  | ||||
| 	// If we don't have a attachfield or we don't have a mediaserver configured return | ||||
| 	if msg.Extra == nil || (gw.Config.General.MediaServerUpload == "" && gw.Config.General.MediaDownloadPath == "") { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// If we don't have files, nothing to upload. | ||||
| 	if len(msg.Extra["file"]) == 0 { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	client := &http.Client{ | ||||
| 		Timeout: time.Second * 5, | ||||
| 	} | ||||
|  | ||||
| 	for i, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		ext := filepath.Ext(fi.Name) | ||||
| 		fi.Name = fi.Name[0 : len(fi.Name)-len(ext)] | ||||
| 		fi.Name = reg.ReplaceAllString(fi.Name, "_") | ||||
| 		fi.Name = fi.Name + ext | ||||
|  | ||||
| 		sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] | ||||
|  | ||||
| 		if gw.Config.General.MediaServerUpload != "" { | ||||
| 			// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth. | ||||
|  | ||||
| 			url := gw.Config.General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 			req, err := http.NewRequest("PUT", url, bytes.NewReader(*fi.Data)) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver upload failed, could not create request: %#v", err) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			flog.Debugf("mediaserver upload url: %s", url) | ||||
|  | ||||
| 			req.Header.Set("Content-Type", "binary/octet-stream") | ||||
| 			_, err = client.Do(req) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver upload failed, could not Do request: %#v", err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// Use MediaServerPath. Place the file on the current filesystem. | ||||
|  | ||||
| 			dir := gw.Config.General.MediaDownloadPath + "/" + sha1sum | ||||
| 			err := os.Mkdir(dir, os.ModePerm) | ||||
| 			if err != nil && !os.IsExist(err) { | ||||
| 				flog.Errorf("mediaserver path failed, could not mkdir: %s %#v", err, err) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			path := dir + "/" + fi.Name | ||||
| 			flog.Debugf("mediaserver path placing file: %s", path) | ||||
|  | ||||
| 			err = ioutil.WriteFile(path, *fi.Data, os.ModePerm) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver path failed, could not writefile: %s %#v", err, err) | ||||
| 				continue | ||||
| 			} | ||||
| // SendMessage sends a message (with specified parentID) to the channel on the selected | ||||
| // destination bridge and returns a message ID or an error. | ||||
| func (gw *Gateway) SendMessage( | ||||
| 	rmsg *config.Message, | ||||
| 	dest *bridge.Bridge, | ||||
| 	channel *config.ChannelInfo, | ||||
| 	canonicalParentMsgID string, | ||||
| ) (string, error) { | ||||
| 	msg := *rmsg | ||||
| 	// Only send the avatar download event to ourselves. | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		if channel.ID != getChannelID(rmsg) { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 	} else { | ||||
| 		// do not send to ourself for any other event | ||||
| 		if channel.ID == getChannelID(rmsg) { | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		// Download URL. | ||||
| 		durl := gw.Config.General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 		flog.Debugf("mediaserver download URL = %s", durl) | ||||
|  | ||||
| 		// We uploaded/placed the file successfully. Add the SHA and URL. | ||||
| 		extra := msg.Extra["file"][i].(config.FileInfo) | ||||
| 		extra.URL = durl | ||||
| 		extra.SHA = sha1sum | ||||
| 		msg.Extra["file"][i] = extra | ||||
| 	} | ||||
|  | ||||
| 	// Too noisy to log like other events | ||||
| 	if msg.Event != config.EventUserTyping { | ||||
| 		gw.logger.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, rmsg.Channel, dest.Account, channel.Name) | ||||
| 	} | ||||
|  | ||||
| 	msg.Channel = channel.Name | ||||
| 	msg.Avatar = gw.modifyAvatar(rmsg, dest) | ||||
| 	msg.Username = gw.modifyUsername(rmsg, dest) | ||||
|  | ||||
| 	msg.ID = gw.getDestMsgID(rmsg.Protocol+" "+rmsg.ID, dest, channel) | ||||
|  | ||||
| 	// for api we need originchannel as channel | ||||
| 	if dest.Protocol == apiProtocol { | ||||
| 		msg.Channel = rmsg.Channel | ||||
| 	} | ||||
|  | ||||
| 	msg.ParentID = gw.getDestMsgID(rmsg.Protocol+" "+canonicalParentMsgID, dest, channel) | ||||
| 	if msg.ParentID == "" { | ||||
| 		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 "msg-parent-not-found" | ||||
| 	if msg.ParentID == "" && rmsg.ParentID != "" { | ||||
| 		msg.ParentID = "msg-parent-not-found" | ||||
| 	} | ||||
|  | ||||
| 	// if we are using mattermost plugin account, send messages to MattermostPlugin channel | ||||
| 	// that can be picked up by the mattermost matterbridge plugin | ||||
| 	if dest.Account == "mattermost.plugin" { | ||||
| 		gw.Router.MattermostPlugin <- msg | ||||
| 	} | ||||
|  | ||||
| 	mID, err := dest.Send(msg) | ||||
| 	if err != nil { | ||||
| 		return mID, err | ||||
| 	} | ||||
|  | ||||
| 	// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||
| 	if mID != "" { | ||||
| 		gw.logger.Debugf("mID %s: %s", dest.Account, mID) | ||||
| 		return mID, nil | ||||
| 		//brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + mID, channel.ID}) | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) validGatewayDest(msg *config.Message, channel *config.ChannelInfo) bool { | ||||
| func (gw *Gateway) validGatewayDest(msg *config.Message) bool { | ||||
| 	return msg.Gateway == gw.Name | ||||
| } | ||||
|  | ||||
| func getChannelID(msg config.Message) string { | ||||
| func getChannelID(msg *config.Message) string { | ||||
| 	return msg.Channel + msg.Account | ||||
| } | ||||
|  | ||||
| func isApi(account string) bool { | ||||
| func isAPI(account string) bool { | ||||
| 	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 modifyMessageTengo(filename string, msg *config.Message) error { | ||||
| 	if filename == "" { | ||||
| 		return nil | ||||
| 	} | ||||
| 	res, err := ioutil.ReadFile(filename) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	s := script.New(res) | ||||
| 	_ = s.Add("msgText", msg.Text) | ||||
| 	_ = s.Add("msgUsername", msg.Username) | ||||
| 	_ = 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 | ||||
| } | ||||
|   | ||||
| @@ -2,12 +2,15 @@ package gateway | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"strconv" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/gateway/bridgemap" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
|  | ||||
| 	"testing" | ||||
| 	"github.com/stretchr/testify/suite" | ||||
| ) | ||||
|  | ||||
| var testconfig = []byte(` | ||||
| @@ -152,9 +155,17 @@ enable=true | ||||
|     channel="--333333333333" | ||||
| `) | ||||
|  | ||||
| const ( | ||||
| 	ircTestAccount   = "irc.zzz" | ||||
| 	tgTestAccount    = "telegram.zzz" | ||||
| 	slackTestAccount = "slack.zzz" | ||||
| ) | ||||
|  | ||||
| func maketestRouter(input []byte) *Router { | ||||
| 	cfg := config.NewConfigFromString(input) | ||||
| 	r, err := NewRouter(cfg) | ||||
| 	logger := logrus.New() | ||||
| 	logger.SetOutput(ioutil.Discard) | ||||
| 	cfg := config.NewConfigFromString(logger, input) | ||||
| 	r, err := NewRouter(logger, cfg, bridgemap.FullMap) | ||||
| 	if err != nil { | ||||
| 		fmt.Println(err) | ||||
| 	} | ||||
| @@ -172,18 +183,27 @@ func TestNewRouter(t *testing.T) { | ||||
| 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Bridges)) | ||||
| 	assert.Equal(t, 4, len(r.Gateways["bridge1"].Channels)) | ||||
| 	assert.Equal(t, 3, len(r.Gateways["bridge2"].Channels)) | ||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "out", | ||||
| 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||
| 		SameChannel: map[string]bool{"bridge2": false}}, | ||||
| 		r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | ||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "42wim/testroom", Direction: "in", | ||||
| 		ID: "42wim/testroomgitter.42wim", Account: "gitter.42wim", | ||||
| 		SameChannel: map[string]bool{"bridge1": false}}, | ||||
| 		r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) | ||||
| 	assert.Equal(t, &config.ChannelInfo{Name: "general", Direction: "inout", | ||||
| 		ID: "generaldiscord.test", Account: "discord.test", | ||||
| 		SameChannel: map[string]bool{"bridge1": false}}, | ||||
| 		r.Gateways["bridge1"].Channels["generaldiscord.test"]) | ||||
| 	assert.Equal(t, &config.ChannelInfo{ | ||||
| 		Name:        "42wim/testroom", | ||||
| 		Direction:   "out", | ||||
| 		ID:          "42wim/testroomgitter.42wim", | ||||
| 		Account:     "gitter.42wim", | ||||
| 		SameChannel: map[string]bool{"bridge2": false}, | ||||
| 	}, r.Gateways["bridge2"].Channels["42wim/testroomgitter.42wim"]) | ||||
| 	assert.Equal(t, &config.ChannelInfo{ | ||||
| 		Name:        "42wim/testroom", | ||||
| 		Direction:   "in", | ||||
| 		ID:          "42wim/testroomgitter.42wim", | ||||
| 		Account:     "gitter.42wim", | ||||
| 		SameChannel: map[string]bool{"bridge1": false}, | ||||
| 	}, r.Gateways["bridge1"].Channels["42wim/testroomgitter.42wim"]) | ||||
| 	assert.Equal(t, &config.ChannelInfo{ | ||||
| 		Name:        "general", | ||||
| 		Direction:   "inout", | ||||
| 		ID:          "generaldiscord.test", | ||||
| 		Account:     "discord.test", | ||||
| 		SameChannel: map[string]bool{"bridge1": false}, | ||||
| 	}, r.Gateways["bridge1"].Channels["generaldiscord.test"]) | ||||
| } | ||||
|  | ||||
| func TestGetDestChannel(t *testing.T) { | ||||
| @@ -192,11 +212,23 @@ func TestGetDestChannel(t *testing.T) { | ||||
| 	for _, br := range r.Gateways["bridge1"].Bridges { | ||||
| 		switch br.Account { | ||||
| 		case "discord.test": | ||||
| 			assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "discord.test", Direction: "inout", ID: "generaldiscord.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||
| 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||
| 			assert.Equal(t, []config.ChannelInfo{{ | ||||
| 				Name:        "general", | ||||
| 				Account:     "discord.test", | ||||
| 				Direction:   "inout", | ||||
| 				ID:          "generaldiscord.test", | ||||
| 				SameChannel: map[string]bool{"bridge1": false}, | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 			}}, r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||
| 		case "slack.test": | ||||
| 			assert.Equal(t, []config.ChannelInfo{{Name: "testing", Account: "slack.test", Direction: "out", ID: "testingslack.test", SameChannel: map[string]bool{"bridge1": false}, Options: config.ChannelOptions{Key: ""}}}, | ||||
| 				r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||
| 			assert.Equal(t, []config.ChannelInfo{{ | ||||
| 				Name:        "testing", | ||||
| 				Account:     "slack.test", | ||||
| 				Direction:   "out", | ||||
| 				ID:          "testingslack.test", | ||||
| 				SameChannel: map[string]bool{"bridge1": false}, | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 			}}, r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||
| 		case "gitter.42wim": | ||||
| 			assert.Equal(t, []config.ChannelInfo(nil), r.Gateways["bridge1"].getDestChannel(msg, *br)) | ||||
| 		case "irc.freenode": | ||||
| @@ -226,35 +258,87 @@ func TestGetDestChannelAdvanced(t *testing.T) { | ||||
| 				} | ||||
| 				switch gw.Name { | ||||
| 				case "bridge": | ||||
| 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz" || msg.Account == "slack.zzz") { | ||||
| 					if (msg.Channel == "#main" || msg.Channel == "-1111111111111" || msg.Channel == "irc") && | ||||
| 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount || msg.Account == slackTestAccount) { | ||||
| 						hits[gw.Name]++ | ||||
| 						switch br.Account { | ||||
| 						case "irc.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "inout", ID: "#mainirc.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case "telegram.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "-1111111111111", Account: "telegram.zzz", Direction: "inout", ID: "-1111111111111telegram.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case "slack.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "irc", Account: "slack.zzz", Direction: "inout", ID: "ircslack.zzz", SameChannel: map[string]bool{"bridge": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case ircTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "#main", | ||||
| 								Account:     ircTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "#mainirc.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						case tgTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "-1111111111111", | ||||
| 								Account:     tgTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "-1111111111111telegram.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						case slackTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "irc", | ||||
| 								Account:     slackTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "ircslack.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						} | ||||
| 					} | ||||
| 				case "bridge2": | ||||
| 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||
| 					if (msg.Channel == "#main-help" || msg.Channel == "--444444444444") && | ||||
| 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount) { | ||||
| 						hits[gw.Name]++ | ||||
| 						switch br.Account { | ||||
| 						case "irc.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-help", Account: "irc.zzz", Direction: "inout", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case "telegram.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "--444444444444", Account: "telegram.zzz", Direction: "inout", ID: "--444444444444telegram.zzz", SameChannel: map[string]bool{"bridge2": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case ircTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "#main-help", | ||||
| 								Account:     ircTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "#main-helpirc.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge2": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						case tgTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "--444444444444", | ||||
| 								Account:     tgTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "--444444444444telegram.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge2": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						} | ||||
| 					} | ||||
| 				case "bridge3": | ||||
| 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && (msg.Account == "irc.zzz" || msg.Account == "telegram.zzz") { | ||||
| 					if (msg.Channel == "#main-telegram" || msg.Channel == "--333333333333") && | ||||
| 						(msg.Account == ircTestAccount || msg.Account == tgTestAccount) { | ||||
| 						hits[gw.Name]++ | ||||
| 						switch br.Account { | ||||
| 						case "irc.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "#main-telegram", Account: "irc.zzz", Direction: "inout", ID: "#main-telegramirc.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case "telegram.zzz": | ||||
| 							assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "inout", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"bridge3": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 						case ircTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "#main-telegram", | ||||
| 								Account:     ircTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "#main-telegramirc.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge3": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						case tgTestAccount: | ||||
| 							assert.Equal(t, []config.ChannelInfo{{ | ||||
| 								Name:        "--333333333333", | ||||
| 								Account:     tgTestAccount, | ||||
| 								Direction:   "inout", | ||||
| 								ID:          "--333333333333telegram.zzz", | ||||
| 								SameChannel: map[string]bool{"bridge3": false}, | ||||
| 								Options:     config.ChannelOptions{Key: ""}, | ||||
| 							}}, channels) | ||||
| 						} | ||||
| 					} | ||||
| 				case "announcements": | ||||
| @@ -264,12 +348,42 @@ func TestGetDestChannelAdvanced(t *testing.T) { | ||||
| 					} | ||||
| 					hits[gw.Name]++ | ||||
| 					switch br.Account { | ||||
| 					case "irc.zzz": | ||||
| 						assert.Equal(t, []config.ChannelInfo{{Name: "#main", Account: "irc.zzz", Direction: "out", ID: "#mainirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}, {Name: "#main-help", Account: "irc.zzz", Direction: "out", ID: "#main-helpirc.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 					case "slack.zzz": | ||||
| 						assert.Equal(t, []config.ChannelInfo{{Name: "general", Account: "slack.zzz", Direction: "out", ID: "generalslack.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 					case "telegram.zzz": | ||||
| 						assert.Equal(t, []config.ChannelInfo{{Name: "--333333333333", Account: "telegram.zzz", Direction: "out", ID: "--333333333333telegram.zzz", SameChannel: map[string]bool{"announcements": false}, Options: config.ChannelOptions{Key: ""}}}, channels) | ||||
| 					case ircTestAccount: | ||||
| 						assert.Len(t, channels, 2) | ||||
| 						assert.Contains(t, channels, config.ChannelInfo{ | ||||
| 							Name:        "#main", | ||||
| 							Account:     ircTestAccount, | ||||
| 							Direction:   "out", | ||||
| 							ID:          "#mainirc.zzz", | ||||
| 							SameChannel: map[string]bool{"announcements": false}, | ||||
| 							Options:     config.ChannelOptions{Key: ""}, | ||||
| 						}) | ||||
| 						assert.Contains(t, channels, config.ChannelInfo{ | ||||
| 							Name:        "#main-help", | ||||
| 							Account:     ircTestAccount, | ||||
| 							Direction:   "out", | ||||
| 							ID:          "#main-helpirc.zzz", | ||||
| 							SameChannel: map[string]bool{"announcements": false}, | ||||
| 							Options:     config.ChannelOptions{Key: ""}, | ||||
| 						}) | ||||
| 					case slackTestAccount: | ||||
| 						assert.Equal(t, []config.ChannelInfo{{ | ||||
| 							Name:        "general", | ||||
| 							Account:     slackTestAccount, | ||||
| 							Direction:   "out", | ||||
| 							ID:          "generalslack.zzz", | ||||
| 							SameChannel: map[string]bool{"announcements": false}, | ||||
| 							Options:     config.ChannelOptions{Key: ""}, | ||||
| 						}}, channels) | ||||
| 					case tgTestAccount: | ||||
| 						assert.Equal(t, []config.ChannelInfo{{ | ||||
| 							Name:        "--333333333333", | ||||
| 							Account:     tgTestAccount, | ||||
| 							Direction:   "out", | ||||
| 							ID:          "--333333333333telegram.zzz", | ||||
| 							SameChannel: map[string]bool{"announcements": false}, | ||||
| 							Options:     config.ChannelOptions{Key: ""}, | ||||
| 						}}, channels) | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| @@ -277,3 +391,139 @@ func TestGetDestChannelAdvanced(t *testing.T) { | ||||
| 	} | ||||
| 	assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits) | ||||
| } | ||||
|  | ||||
| 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{}) | ||||
| 	extraAttach := make(map[string][]interface{}) | ||||
| 	extraFailure := make(map[string][]interface{}) | ||||
| 	extraFile["file"] = append(extraFile["file"], config.FileInfo{}) | ||||
| 	extraAttach["attachments"] = append(extraAttach["attachments"], []string{}) | ||||
| 	extraFailure[config.EventFileFailureSize] = append(extraFailure[config.EventFileFailureSize], config.FileInfo{}) | ||||
|  | ||||
| 	msgTests := map[string]struct { | ||||
| 		input  *config.Message | ||||
| 		output bool | ||||
| 	}{ | ||||
| 		"usertyping": { | ||||
| 			input:  &config.Message{Event: config.EventUserTyping}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"file attach": { | ||||
| 			input:  &config.Message{Extra: extraFile}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"attachments": { | ||||
| 			input:  &config.Message{Extra: extraAttach}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		config.EventFileFailureSize: { | ||||
| 			input:  &config.Message{Extra: extraFailure}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"nil extra": { | ||||
| 			input:  &config.Message{Extra: nil}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"empty": { | ||||
| 			input:  &config.Message{}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 	} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := s.gw.ignoreTextEmpty(testcase.input) | ||||
| 		s.Assert().Equalf(testcase.output, output, "case '%s' failed", testname) | ||||
| 	} | ||||
|  | ||||
| } | ||||
|  | ||||
| func (s *ignoreTestSuite) TestIgnoreTexts() { | ||||
| 	msgTests := map[string]struct { | ||||
| 		input  string | ||||
| 		re     []string | ||||
| 		output bool | ||||
| 	}{ | ||||
| 		"no regex": { | ||||
| 			input:  "a text message", | ||||
| 			re:     []string{}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"simple regex": { | ||||
| 			input:  "a text message", | ||||
| 			re:     []string{"text"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple regex fail": { | ||||
| 			input:  "a text message", | ||||
| 			re:     []string{"abc", "123$"}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"multiple regex pass": { | ||||
| 			input:  "a text message", | ||||
| 			re:     []string{"lala", "sage$"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 	} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := s.gw.ignoreText(testcase.input, testcase.re) | ||||
| 		s.Assert().Equalf(testcase.output, output, "case '%s' failed", testname) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (s *ignoreTestSuite) TestIgnoreNicks() { | ||||
| 	msgTests := map[string]struct { | ||||
| 		input  string | ||||
| 		re     []string | ||||
| 		output bool | ||||
| 	}{ | ||||
| 		"no entry": { | ||||
| 			input:  "user", | ||||
| 			re:     []string{}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"one entry": { | ||||
| 			input:  "user", | ||||
| 			re:     []string{"user"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple entries": { | ||||
| 			input:  "user", | ||||
| 			re:     []string{"abc", "user"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple entries fail": { | ||||
| 			input:  "user", | ||||
| 			re:     []string{"abc", "def"}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 	} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := s.gw.ignoreText(testcase.input, testcase.re) | ||||
| 		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 := modifyMessageTengo("bench.tengo", msg) | ||||
| 		if err != nil { | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										266
									
								
								gateway/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,266 @@ | ||||
| package gateway | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/sha1" //nolint:gosec | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| ) | ||||
|  | ||||
| // handleEventFailure handles failures and reconnects bridges. | ||||
| func (r *Router) handleEventFailure(msg *config.Message) { | ||||
| 	if msg.Event != config.EventFailure { | ||||
| 		return | ||||
| 	} | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		for _, br := range gw.Bridges { | ||||
| 			if msg.Account == br.Account { | ||||
| 				go gw.reconnectBridge(br) | ||||
| 				return | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleEventGetChannelMembers handles channel members | ||||
| func (r *Router) handleEventGetChannelMembers(msg *config.Message) { | ||||
| 	if msg.Event != config.EventGetChannelMembers { | ||||
| 		return | ||||
| 	} | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		for _, br := range gw.Bridges { | ||||
| 			if msg.Account == br.Account { | ||||
| 				cMembers := msg.Extra[config.EventGetChannelMembers][0].(config.ChannelMembers) | ||||
| 				r.logger.Debugf("Syncing channelmembers from %s", msg.Account) | ||||
| 				br.SetChannelMembers(&cMembers) | ||||
| 				return | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleEventRejoinChannels handles rejoining of channels. | ||||
| func (r *Router) handleEventRejoinChannels(msg *config.Message) { | ||||
| 	if msg.Event != config.EventRejoinChannels { | ||||
| 		return | ||||
| 	} | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		for _, br := range gw.Bridges { | ||||
| 			if msg.Account == br.Account { | ||||
| 				br.Joined = make(map[string]bool) | ||||
| 				if err := br.JoinChannels(); err != nil { | ||||
| 					r.logger.Errorf("channel join failed for %s: %s", msg.Account, err) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleFiles uploads or places all files on the given msg to the MediaServer and | ||||
| // adds the new URL of the file on the MediaServer onto the given msg. | ||||
| func (gw *Gateway) handleFiles(msg *config.Message) { | ||||
| 	reg := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||
|  | ||||
| 	// If we don't have a attachfield or we don't have a mediaserver configured return | ||||
| 	if msg.Extra == nil || | ||||
| 		(gw.BridgeValues().General.MediaServerUpload == "" && | ||||
| 			gw.BridgeValues().General.MediaDownloadPath == "") { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// If we don't have files, nothing to upload. | ||||
| 	if len(msg.Extra["file"]) == 0 { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	for i, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		ext := filepath.Ext(fi.Name) | ||||
| 		fi.Name = fi.Name[0 : len(fi.Name)-len(ext)] | ||||
| 		fi.Name = reg.ReplaceAllString(fi.Name, "_") | ||||
| 		fi.Name += ext | ||||
|  | ||||
| 		sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] //nolint:gosec | ||||
|  | ||||
| 		if gw.BridgeValues().General.MediaServerUpload != "" { | ||||
| 			// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth. | ||||
| 			if err := gw.handleFilesUpload(&fi); err != nil { | ||||
| 				gw.logger.Error(err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// Use MediaServerPath. Place the file on the current filesystem. | ||||
| 			if err := gw.handleFilesLocal(&fi); err != nil { | ||||
| 				gw.logger.Error(err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Download URL. | ||||
| 		durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 		gw.logger.Debugf("mediaserver download URL = %s", durl) | ||||
|  | ||||
| 		// We uploaded/placed the file successfully. Add the SHA and URL. | ||||
| 		extra := msg.Extra["file"][i].(config.FileInfo) | ||||
| 		extra.URL = durl | ||||
| 		extra.SHA = sha1sum | ||||
| 		msg.Extra["file"][i] = extra | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleFilesUpload uses MediaServerUpload configuration to upload the file. | ||||
| // Returns error on failure. | ||||
| func (gw *Gateway) handleFilesUpload(fi *config.FileInfo) error { | ||||
| 	client := &http.Client{ | ||||
| 		Timeout: time.Second * 5, | ||||
| 	} | ||||
| 	// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth. | ||||
| 	sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] //nolint:gosec | ||||
| 	url := gw.BridgeValues().General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 	req, err := http.NewRequest("PUT", url, bytes.NewReader(*fi.Data)) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("mediaserver upload failed, could not create request: %#v", err) | ||||
| 	} | ||||
|  | ||||
| 	gw.logger.Debugf("mediaserver upload url: %s", url) | ||||
|  | ||||
| 	req.Header.Set("Content-Type", "binary/octet-stream") | ||||
| 	_, err = client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("mediaserver upload failed, could not Do request: %#v", err) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleFilesLocal use MediaServerPath configuration, places the file on the current filesystem. | ||||
| // Returns error on failure. | ||||
| func (gw *Gateway) handleFilesLocal(fi *config.FileInfo) error { | ||||
| 	sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] //nolint:gosec | ||||
| 	dir := gw.BridgeValues().General.MediaDownloadPath + "/" + sha1sum | ||||
| 	err := os.Mkdir(dir, os.ModePerm) | ||||
| 	if err != nil && !os.IsExist(err) { | ||||
| 		return fmt.Errorf("mediaserver path failed, could not mkdir: %s %#v", err, err) | ||||
| 	} | ||||
|  | ||||
| 	path := dir + "/" + fi.Name | ||||
| 	gw.logger.Debugf("mediaserver path placing file: %s", path) | ||||
|  | ||||
| 	err = ioutil.WriteFile(path, *fi.Data, os.ModePerm) | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("mediaserver path failed, could not writefile: %s %#v", err, err) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // ignoreEvent returns true if we need to ignore this event for the specified destination bridge. | ||||
| func (gw *Gateway) ignoreEvent(event string, dest *bridge.Bridge) bool { | ||||
| 	switch event { | ||||
| 	case config.EventAvatarDownload: | ||||
| 		// Avatar downloads are only relevant for telegram and mattermost for now | ||||
| 		if dest.Protocol != "mattermost" && dest.Protocol != "telegram" { | ||||
| 			return true | ||||
| 		} | ||||
| 	case config.EventJoinLeave: | ||||
| 		// only relay join/part when configured | ||||
| 		if !dest.GetBool("ShowJoinPart") { | ||||
| 			return true | ||||
| 		} | ||||
| 	case config.EventTopicChange: | ||||
| 		// only relay topic change when used in some way on other side | ||||
| 		if dest.GetBool("ShowTopicChange") && dest.GetBool("SyncTopic") { | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| // handleMessage makes sure the message get sent to the correct bridge/channels. | ||||
| // Returns an array of msg ID's | ||||
| func (gw *Gateway) handleMessage(rmsg *config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||
| 	var brMsgIDs []*BrMsgID | ||||
|  | ||||
| 	// if we have an attached file, or other info | ||||
| 	if rmsg.Extra != nil && len(rmsg.Extra[config.EventFileFailureSize]) != 0 && rmsg.Text == "" { | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	if gw.ignoreEvent(rmsg.Event, dest) { | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// broadcast to every out channel (irc QUIT) | ||||
| 	if rmsg.Channel == "" && rmsg.Event != config.EventJoinLeave { | ||||
| 		gw.logger.Debug("empty channel") | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// Get the ID of the parent message in thread | ||||
| 	var canonicalParentMsgID string | ||||
| 	if rmsg.ParentID != "" && dest.GetBool("PreserveThreading") { | ||||
| 		canonicalParentMsgID = gw.FindCanonicalMsgID(rmsg.Protocol, rmsg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	channels := gw.getDestChannel(rmsg, *dest) | ||||
| 	for idx := range channels { | ||||
| 		channel := &channels[idx] | ||||
| 		msgID, err := gw.SendMessage(rmsg, dest, channel, canonicalParentMsgID) | ||||
| 		if err != nil { | ||||
| 			gw.logger.Errorf("SendMessage failed: %s", err) | ||||
| 			continue | ||||
| 		} | ||||
| 		if msgID == "" { | ||||
| 			continue | ||||
| 		} | ||||
| 		brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + msgID, channel.ID}) | ||||
| 	} | ||||
| 	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
									
								
							
							
						
						| @@ -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) | ||||
| 	} | ||||
|  | ||||
| } | ||||
| @@ -2,26 +2,45 @@ package gateway | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	samechannelgateway "github.com/42wim/matterbridge/gateway/samechannel" | ||||
| 	//	"github.com/davecgh/go-spew/spew" | ||||
| 	"time" | ||||
| 	"github.com/42wim/matterbridge/gateway/samechannel" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Router struct { | ||||
| 	Gateways map[string]*Gateway | ||||
| 	Message  chan config.Message | ||||
| 	*config.Config | ||||
| 	config.Config | ||||
| 	sync.RWMutex | ||||
|  | ||||
| 	BridgeMap        map[string]bridge.Factory | ||||
| 	Gateways         map[string]*Gateway | ||||
| 	Message          chan config.Message | ||||
| 	MattermostPlugin chan config.Message | ||||
|  | ||||
| 	logger *logrus.Entry | ||||
| } | ||||
|  | ||||
| func NewRouter(cfg *config.Config) (*Router, error) { | ||||
| 	r := &Router{Message: make(chan config.Message), Gateways: make(map[string]*Gateway), Config: cfg} | ||||
| 	sgw := samechannelgateway.New(cfg) | ||||
| 	gwconfigs := sgw.GetConfig() | ||||
| // 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"}) | ||||
|  | ||||
| 	for _, entry := range append(gwconfigs, cfg.Gateway...) { | ||||
| 	r := &Router{ | ||||
| 		Config:           cfg, | ||||
| 		BridgeMap:        bridgeMap, | ||||
| 		Message:          make(chan config.Message), | ||||
| 		MattermostPlugin: make(chan config.Message), | ||||
| 		Gateways:         make(map[string]*Gateway), | ||||
| 		logger:           logger, | ||||
| 	} | ||||
| 	sgw := samechannel.New(cfg) | ||||
| 	gwconfigs := append(sgw.GetConfig(), cfg.BridgeValues().Gateway...) | ||||
|  | ||||
| 	for idx := range gwconfigs { | ||||
| 		entry := &gwconfigs[idx] | ||||
| 		if !entry.Enable { | ||||
| 			continue | ||||
| 		} | ||||
| @@ -31,34 +50,66 @@ func NewRouter(cfg *config.Config) (*Router, error) { | ||||
| 		if _, ok := r.Gateways[entry.Name]; ok { | ||||
| 			return nil, fmt.Errorf("Gateway with name %s already exists", entry.Name) | ||||
| 		} | ||||
| 		r.Gateways[entry.Name] = New(entry, r) | ||||
| 		r.Gateways[entry.Name] = New(rootLogger, entry, r) | ||||
| 	} | ||||
| 	return r, nil | ||||
| } | ||||
|  | ||||
| // Start will connect all gateways belonging to this router and subsequently route messages | ||||
| // between them. | ||||
| func (r *Router) Start() error { | ||||
| 	m := make(map[string]*bridge.Bridge) | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		flog.Infof("Parsing gateway %s", gw.Name) | ||||
| 		r.logger.Infof("Parsing gateway %s", gw.Name) | ||||
| 		for _, br := range gw.Bridges { | ||||
| 			m[br.Account] = br | ||||
| 		} | ||||
| 	} | ||||
| 	for _, br := range m { | ||||
| 		flog.Infof("Starting bridge: %s ", br.Account) | ||||
| 		r.logger.Infof("Starting bridge: %s ", br.Account) | ||||
| 		err := br.Connect() | ||||
| 		if err != nil { | ||||
| 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||
| 			e := fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||
| 			if r.disableBridge(br, e) { | ||||
| 				continue | ||||
| 			} | ||||
| 			return e | ||||
| 		} | ||||
| 		err = br.JoinChannels() | ||||
| 		if err != nil { | ||||
| 			return fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||
| 			e := fmt.Errorf("Bridge %s failed to join channel: %v", br.Account, err) | ||||
| 			if r.disableBridge(br, e) { | ||||
| 				continue | ||||
| 			} | ||||
| 			return e | ||||
| 		} | ||||
| 	} | ||||
| 	// remove unused bridges | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		for i, br := range gw.Bridges { | ||||
| 			if br.Bridger == nil { | ||||
| 				r.logger.Errorf("removing failed bridge %s", i) | ||||
| 				delete(gw.Bridges, i) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	go r.handleReceive() | ||||
| 	//go r.updateChannelMembers() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // disableBridge returns true and empties a bridge if we have IgnoreFailureOnStart configured | ||||
| // otherwise returns false | ||||
| func (r *Router) disableBridge(br *bridge.Bridge, err error) bool { | ||||
| 	if r.BridgeValues().General.IgnoreFailureOnStart { | ||||
| 		r.logger.Error(err) | ||||
| 		// setting this bridge empty | ||||
| 		*br = bridge.Bridge{} | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (r *Router) getBridge(account string) *bridge.Bridge { | ||||
| 	for _, gw := range r.Gateways { | ||||
| 		if br, ok := gw.Bridges[account]; ok { | ||||
| @@ -70,42 +121,48 @@ func (r *Router) getBridge(account string) *bridge.Bridge { | ||||
|  | ||||
| func (r *Router) handleReceive() { | ||||
| 	for msg := range r.Message { | ||||
| 		if msg.Event == config.EVENT_FAILURE { | ||||
| 		Loop: | ||||
| 			for _, gw := range r.Gateways { | ||||
| 				for _, br := range gw.Bridges { | ||||
| 					if msg.Account == br.Account { | ||||
| 						go gw.reconnectBridge(br) | ||||
| 						break Loop | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		if msg.Event == config.EVENT_REJOIN_CHANNELS { | ||||
| 			for _, gw := range r.Gateways { | ||||
| 				for _, br := range gw.Bridges { | ||||
| 					if msg.Account == br.Account { | ||||
| 						br.Joined = make(map[string]bool) | ||||
| 						br.JoinChannels() | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		msg := msg // scopelint | ||||
| 		r.handleEventGetChannelMembers(&msg) | ||||
| 		r.handleEventFailure(&msg) | ||||
| 		r.handleEventRejoinChannels(&msg) | ||||
| 		for _, gw := range r.Gateways { | ||||
| 			// record all the message ID's of the different bridges | ||||
| 			var msgIDs []*BrMsgID | ||||
| 			if !gw.ignoreMessage(&msg) { | ||||
| 				msg.Timestamp = time.Now() | ||||
| 				gw.modifyMessage(&msg) | ||||
| 				gw.handleFiles(&msg) | ||||
| 				for _, br := range gw.Bridges { | ||||
| 					msgIDs = append(msgIDs, gw.handleMessage(msg, br)...) | ||||
| 				} | ||||
| 				// only add the message ID if it doesn't already exists | ||||
| 				if _, ok := gw.Messages.Get(msg.ID); !ok && msg.ID != "" { | ||||
| 					gw.Messages.Add(msg.ID, msgIDs) | ||||
| 				} | ||||
| 			if gw.ignoreMessage(&msg) { | ||||
| 				continue | ||||
| 			} | ||||
| 			msg.Timestamp = time.Now() | ||||
| 			gw.modifyMessage(&msg) | ||||
| 			gw.handleFiles(&msg) | ||||
| 			for _, br := range gw.Bridges { | ||||
| 				msgIDs = append(msgIDs, gw.handleMessage(&msg, br)...) | ||||
| 			} | ||||
| 			// only add the message ID if it doesn't already exists | ||||
| 			if _, ok := gw.Messages.Get(msg.Protocol + " " + msg.ID); !ok && msg.ID != "" { | ||||
| 				gw.Messages.Add(msg.Protocol+" "+msg.ID, msgIDs) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // updateChannelMembers sends every minute an GetChannelMembers event to all bridges. | ||||
| func (r *Router) updateChannelMembers() { | ||||
| 	// TODO sleep a minute because slack can take a while | ||||
| 	// fix this by having actually connectionDone events send to the router | ||||
| 	time.Sleep(time.Minute) | ||||
| 	for { | ||||
| 		for _, gw := range r.Gateways { | ||||
| 			for _, br := range gw.Bridges { | ||||
| 				// only for slack now | ||||
| 				if br.Protocol != "slack" { | ||||
| 					continue | ||||
| 				} | ||||
| 				r.logger.Debugf("sending %s to %s", config.EventGetChannelMembers, br.Account) | ||||
| 				if _, err := br.Send(config.Message{Event: config.EventGetChannelMembers}); err != nil { | ||||
| 					r.logger.Errorf("updateChannelMembers: %s", err) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		time.Sleep(time.Minute) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,21 +1,21 @@ | ||||
| package samechannelgateway | ||||
| package samechannel | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| ) | ||||
|  | ||||
| type SameChannelGateway struct { | ||||
| 	*config.Config | ||||
| 	config.Config | ||||
| } | ||||
|  | ||||
| func New(cfg *config.Config) *SameChannelGateway { | ||||
| func New(cfg config.Config) *SameChannelGateway { | ||||
| 	return &SameChannelGateway{Config: cfg} | ||||
| } | ||||
|  | ||||
| func (sgw *SameChannelGateway) GetConfig() []config.Gateway { | ||||
| 	var gwconfigs []config.Gateway | ||||
| 	cfg := sgw.Config | ||||
| 	for _, gw := range cfg.SameChannelGateway { | ||||
| 	for _, gw := range cfg.BridgeValues().SameChannelGateway { | ||||
| 		gwconfig := config.Gateway{Name: gw.Name, Enable: gw.Enable} | ||||
| 		for _, account := range gw.Accounts { | ||||
| 			for _, channel := range gw.Channels { | ||||
|   | ||||
| @@ -1,16 +1,15 @@ | ||||
| package samechannelgateway | ||||
| package samechannel | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/BurntSushi/toml" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
|  | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| var testconfig = ` | ||||
| const testConfig = ` | ||||
| [mattermost.test] | ||||
| [slack.test] | ||||
|  | ||||
| @@ -21,12 +20,58 @@ var testconfig = ` | ||||
|       channels = [ "testing","testing2","testing10"] | ||||
| ` | ||||
|  | ||||
| func TestGetConfig(t *testing.T) { | ||||
| 	var cfg *config.Config | ||||
| 	if _, err := toml.Decode(testconfig, &cfg); err != nil { | ||||
| 		fmt.Println(err) | ||||
| var ( | ||||
| 	expectedConfig = config.Gateway{ | ||||
| 		Name:   "blah", | ||||
| 		Enable: true, | ||||
| 		In:     []config.Bridge(nil), | ||||
| 		Out:    []config.Bridge(nil), | ||||
| 		InOut: []config.Bridge{ | ||||
| 			{ | ||||
| 				Account:     "mattermost.test", | ||||
| 				Channel:     "testing", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 			{ | ||||
| 				Account:     "mattermost.test", | ||||
| 				Channel:     "testing2", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 			{ | ||||
| 				Account:     "mattermost.test", | ||||
| 				Channel:     "testing10", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 			{ | ||||
| 				Account:     "slack.test", | ||||
| 				Channel:     "testing", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 			{ | ||||
| 				Account:     "slack.test", | ||||
| 				Channel:     "testing2", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 			{ | ||||
| 				Account:     "slack.test", | ||||
| 				Channel:     "testing10", | ||||
| 				Options:     config.ChannelOptions{Key: ""}, | ||||
| 				SameChannel: true, | ||||
| 			}, | ||||
| 		}, | ||||
| 	} | ||||
| ) | ||||
|  | ||||
| func TestGetConfig(t *testing.T) { | ||||
| 	logger := logrus.New() | ||||
| 	logger.SetOutput(ioutil.Discard) | ||||
| 	cfg := config.NewConfigFromString(logger, []byte(testConfig)) | ||||
| 	sgw := New(cfg) | ||||
| 	configs := sgw.GetConfig() | ||||
| 	assert.Equal(t, []config.Gateway{{Name: "blah", Enable: true, In: []config.Bridge(nil), Out: []config.Bridge(nil), InOut: []config.Bridge{{Account: "mattermost.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "mattermost.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing2", Options: config.ChannelOptions{Key: ""}, SameChannel: true}, {Account: "slack.test", Channel: "testing10", Options: config.ChannelOptions{Key: ""}, SameChannel: true}}}}, configs) | ||||
| 	assert.Equal(t, []config.Gateway{expectedConfig}, configs) | ||||
| } | ||||
|   | ||||
							
								
								
									
										100
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						| @@ -2,82 +2,72 @@ module github.com/42wim/matterbridge | ||||
|  | ||||
| require ( | ||||
| 	github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 | ||||
| 	github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 | ||||
| 	github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 | ||||
| 	github.com/Sirupsen/logrus v1.0.6 // indirect | ||||
| 	github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b // indirect | ||||
| 	github.com/bwmarrin/discordgo v0.0.0-20180201002541-8d5ab59c63e5 // indirect | ||||
| 	github.com/davecgh/go-spew v1.1.0 // indirect | ||||
| 	github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d | ||||
| 	github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a // indirect | ||||
| 	github.com/Baozisoftware/qrcode-terminal-go v0.0.0-20170407111555-c0650d8dff0f | ||||
| 	github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 // indirect | ||||
| 	github.com/Jeffail/gabs v1.1.1 // indirect | ||||
| 	github.com/Philipp15b/go-steam v1.0.1-0.20180818081528-681bd9573329 | ||||
| 	github.com/Rhymen/go-whatsapp v0.0.0-20190208184307-c9a81e957884 | ||||
| 	github.com/bwmarrin/discordgo v0.19.0 | ||||
| 	github.com/d5/tengo v1.9.2 | ||||
| 	github.com/dfordsoft/golib v0.0.0-20180902042739-76ee6ab99bec | ||||
| 	github.com/fsnotify/fsnotify v1.4.7 | ||||
| 	github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c | ||||
| 	github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc // indirect | ||||
| 	github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c | ||||
| 	github.com/go-telegram-bot-api/telegram-bot-api v4.6.5-0.20181225215658-ec221ba9ea45+incompatible | ||||
| 	github.com/google/gops v0.3.5 | ||||
| 	github.com/gopackage/ddp v0.0.0-20170117053602-652027933df4 // indirect | ||||
| 	github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f // indirect | ||||
| 	github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c | ||||
| 	github.com/gorilla/websocket v0.0.0-20170319172727-a91eba7f9777 | ||||
| 	github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad | ||||
| 	github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb // indirect | ||||
| 	github.com/gorilla/schema v1.0.2 | ||||
| 	github.com/gorilla/websocket v1.4.0 | ||||
| 	github.com/hashicorp/golang-lru v0.5.0 | ||||
| 	github.com/hpcloud/tail v1.0.0 // indirect | ||||
| 	github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d | ||||
| 	github.com/jpillora/backoff v0.0.0-20180909062703-3050d21c67d7 | ||||
| 	github.com/jtolds/gls v4.2.1+incompatible // indirect | ||||
| 	github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462 // indirect | ||||
| 	github.com/kardianos/osext v0.0.0-20170510131534-ae77be60afb1 // indirect | ||||
| 	github.com/kr/pretty v0.1.0 // indirect | ||||
| 	github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1 | ||||
| 	github.com/labstack/gommon v0.2.1 // indirect | ||||
| 	github.com/lrstanley/girc v0.0.0-20180427160007-102f17f86306 | ||||
| 	github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 // indirect | ||||
| 	github.com/matterbridge/discordgo v0.0.0-20180806170629-ef40ff5ba64f | ||||
| 	github.com/labstack/echo/v4 v4.0.0 | ||||
| 	github.com/lrstanley/girc v0.0.0-20190210212025-51b8e096d398 | ||||
| 	github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 // indirect | ||||
| 	github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 // indirect | ||||
| 	github.com/matterbridge/Rocket.Chat.Go.SDK v0.0.0-20190210153444-cc9d05784d5d | ||||
| 	github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 | ||||
| 	github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f | ||||
| 	github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544 | ||||
| 	github.com/matterbridge/gomatrix v0.0.0-20190102230110-6f9631ca6dea | ||||
| 	github.com/matterbridge/gozulipbot v0.0.0-20190212232658-7aa251978a18 | ||||
| 	github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 | ||||
| 	github.com/mattermost/platform v4.6.2+incompatible | ||||
| 	github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597 // indirect | ||||
| 	github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc // indirect | ||||
| 	github.com/mattermost/mattermost-server v5.5.0+incompatible | ||||
| 	github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect | ||||
| 	github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 // indirect | ||||
| 	github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 // indirect | ||||
| 	github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff // indirect | ||||
| 	github.com/nelsonken/gomf v0.0.0-20180504123937-a9dd2f9deae9 | ||||
| 	github.com/nicksnyder/go-i18n v1.4.0 // indirect | ||||
| 	github.com/nlopes/slack v0.3.1-0.20180805133408-21749ab136a8 | ||||
| 	github.com/nlopes/slack v0.5.0 | ||||
| 	github.com/onsi/ginkgo v1.6.0 // indirect | ||||
| 	github.com/onsi/gomega v1.4.1 // indirect | ||||
| 	github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83 | ||||
| 	github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606 // indirect | ||||
| 	github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e // indirect | ||||
| 	github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271 | ||||
| 	github.com/peterhellberg/emojilib v0.0.0-20190124112554-c18758d55320 | ||||
| 	github.com/pkg/errors v0.8.0 // indirect | ||||
| 	github.com/pmezard/go-difflib v1.0.0 // indirect | ||||
| 	github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a | ||||
| 	github.com/russross/blackfriday v1.5.1 | ||||
| 	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/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1 // indirect | ||||
| 	github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991 | ||||
| 	github.com/sirupsen/logrus v0.0.0-20180213143110-8c0189d9f6bb | ||||
| 	github.com/shazow/ssh-chat v0.0.0-20190125184227-81d7e1686296 | ||||
| 	github.com/sirupsen/logrus v1.3.0 | ||||
| 	github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 // indirect | ||||
| 	github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a // indirect | ||||
| 	github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff // indirect | ||||
| 	github.com/spf13/cast v1.2.0 // indirect | ||||
| 	github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec // indirect | ||||
| 	github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac // indirect | ||||
| 	github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7 | ||||
| 	github.com/stretchr/testify v0.0.0-20170714215325-05e8a0eda380 | ||||
| 	github.com/spf13/viper v1.3.1 | ||||
| 	github.com/stretchr/testify v1.3.0 | ||||
| 	github.com/technoweenie/multipartstreamer v1.0.1 // indirect | ||||
| 	github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a // indirect | ||||
| 	github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4 // indirect | ||||
| 	github.com/x-cray/logrus-prefixed-formatter v0.5.2 // indirect | ||||
| 	github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6 | ||||
| 	golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88 // indirect | ||||
| 	golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 // indirect | ||||
| 	golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f // indirect | ||||
| 	golang.org/x/sys v0.0.0-20171130163741-8b4580aae2a0 // indirect | ||||
| 	golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 // indirect | ||||
| 	gopkg.in/airbrake/gobrake.v2 v2.0.9 // indirect | ||||
| 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect | ||||
| 	github.com/zfjagann/golang-ring v0.0.0-20190106091943-a88bb6aef447 | ||||
| 	gitlab.com/golang-commonmark/html v0.0.0-20180917080848-cfaf75183c4a // indirect | ||||
| 	gitlab.com/golang-commonmark/linkify v0.0.0-20180917065525-c22b7bdb1179 // indirect | ||||
| 	gitlab.com/golang-commonmark/markdown v0.0.0-20181102083822-772775880e1f | ||||
| 	gitlab.com/golang-commonmark/mdurl v0.0.0-20180912090424-e5bce34c34f2 // indirect | ||||
| 	gitlab.com/golang-commonmark/puny v0.0.0-20180912090636-2cd490539afe // indirect | ||||
| 	gitlab.com/opennota/wd v0.0.0-20180912061657-c5d65f63c638 // indirect | ||||
| 	go.uber.org/atomic v1.3.2 // indirect | ||||
| 	go.uber.org/multierr v1.1.0 // indirect | ||||
| 	go.uber.org/zap v1.9.1 // indirect | ||||
| 	gopkg.in/fsnotify.v1 v1.4.7 // indirect | ||||
| 	gopkg.in/gemnasium/logrus-airbrake-hook.v2 v2.1.1 // indirect | ||||
| 	gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect | ||||
| 	gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 // indirect | ||||
| 	gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129 // indirect | ||||
| 	maunium.net/go/mautrix-whatsapp v0.0.0-20190127121751-281b3e8f77f3 | ||||
| ) | ||||
|   | ||||
							
								
								
									
										276
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						| @@ -1,90 +1,116 @@ | ||||
| github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 h1:IZtuWGfzQnKnCSu+vl8WGLhpVQ5Uvy3rlSwqXSg+sQg= | ||||
| github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557/go.mod h1:jL0YSXMs/txjtGJ4PWrmETOk6KUHMDPMshgQZlTeB3Y= | ||||
| github.com/Baozisoftware/qrcode-terminal-go v0.0.0-20170407111555-c0650d8dff0f h1:2dk3eOnYllh+wUOuDhOoC2vUVoJF/5z478ryJ+wzEII= | ||||
| github.com/Baozisoftware/qrcode-terminal-go v0.0.0-20170407111555-c0650d8dff0f/go.mod h1:4a58ifQTEe2uwwsaqbh3i2un5/CBPg+At/qHpt18Tmk= | ||||
| github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 h1:v/zr4ns/4sSahF9KBm4Uc933bLsEEv7LuT63CJ019yo= | ||||
| github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= | ||||
| github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 h1:V4+1E1SRYUySqwOoI3ZphFADtabbF568zTHa5ix/zU0= | ||||
| github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3/go.mod h1:HuVM+sZFzumUdKPWiz+IlCMb4RdsKdT3T+nQBKL+sYg= | ||||
| github.com/Sirupsen/logrus v1.0.6 h1:HCAGQRk48dRVPA5Y+Yh0qdCSTzPOyU1tBJ7Q9YzotII= | ||||
| github.com/Sirupsen/logrus v1.0.6/go.mod h1:rmk17hk6i8ZSAJkSDa7nOxamrG+SP4P0mm+DAvExv4U= | ||||
| github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b h1:1OpGXps6UOY5HtQaQcLowsV1qMWCNBzhFvK7q4fgXtc= | ||||
| github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b/go.mod h1:iCVmQ9g4TfaRX5m5jq5sXY7RXYWPv9/PynM/GocbG3w= | ||||
| github.com/bwmarrin/discordgo v0.0.0-20180201002541-8d5ab59c63e5 h1:M7u44DKGpA5goDIBf0zRMYhT1Sp2Rd7hiTzXfeuw1UY= | ||||
| github.com/bwmarrin/discordgo v0.0.0-20180201002541-8d5ab59c63e5/go.mod h1:5NIvFv5Z7HddYuXbuQegZ684DleQaCFqChP2iuBivJ8= | ||||
| github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8= | ||||
| github.com/Jeffail/gabs v1.1.1 h1:V0uzR08Hj22EX8+8QMhyI9sX2hwRu+/RJhJUmnwda/E= | ||||
| github.com/Jeffail/gabs v1.1.1/go.mod h1:6xMvQMK4k33lb7GUUpaAPh6nKMmemQeg5d4gn7/bOXc= | ||||
| github.com/Philipp15b/go-steam v1.0.1-0.20180818081528-681bd9573329 h1:xZBoq249G9MSt+XuY7sVQzcfONJ6IQuwpCK+KAaOpnY= | ||||
| github.com/Philipp15b/go-steam v1.0.1-0.20180818081528-681bd9573329/go.mod h1:HuVM+sZFzumUdKPWiz+IlCMb4RdsKdT3T+nQBKL+sYg= | ||||
| github.com/Rhymen/go-whatsapp v0.0.0-20181218094654-2ca6af00572c h1:ldRXgMEfKmzBomrZusl3edG9AGEeztA7jovLEQy62us= | ||||
| github.com/Rhymen/go-whatsapp v0.0.0-20181218094654-2ca6af00572c/go.mod h1:MSDmePOOkbFFbVW2WRRppBcbA+aabwpXRgyIIG7jDFQ= | ||||
| github.com/Rhymen/go-whatsapp v0.0.0-20190208184307-c9a81e957884 h1:2AxfzkQi2L4QGBvUCZoWD6hQuUJa5MG54wiYyNqJlf4= | ||||
| github.com/Rhymen/go-whatsapp v0.0.0-20190208184307-c9a81e957884/go.mod h1:rdQr95g2C1xcOfM7QGOhza58HeI3I+tZ/bbluv7VazA= | ||||
| github.com/alexcesaro/log v0.0.0-20150915221235-61e686294e58 h1:MkpmYfld/S8kXqTYI68DfL8/hHXjHogL120Dy00TIxc= | ||||
| github.com/alexcesaro/log v0.0.0-20150915221235-61e686294e58/go.mod h1:YNfsMyWSs+h+PaYkxGeMVmVCX75Zj/pqdjbu12ciCYE= | ||||
| github.com/armon/consul-api v0.0.0-20180202201655-eb2c6b5be1b6/go.mod h1:grANhF5doyWs3UAsr3K4I6qtAmlQcZDesFNEHPZAzj8= | ||||
| github.com/bwmarrin/discordgo v0.19.0 h1:kMED/DB0NR1QhRcalb85w0Cu3Ep2OrGAqZH1R5awQiY= | ||||
| github.com/bwmarrin/discordgo v0.19.0/go.mod h1:O9S4p+ofTFwB02em7jkpkV8M3R0/PUVOwN61zSZ0r4Q= | ||||
| github.com/coreos/etcd v3.3.10+incompatible/go.mod h1:uF7uidLiAD3TWHmW31ZFd/JWoc32PjwdhPthX9715RE= | ||||
| github.com/coreos/go-etcd v2.0.0+incompatible/go.mod h1:Jez6KQU2B/sWsbdaef3ED8NzMklzPG4d5KIOhIy30Tk= | ||||
| github.com/coreos/go-semver v0.2.0/go.mod h1:nnelYz7RCh+5ahJtPPxZlU+153eP4D4r3EedlOD2RNk= | ||||
| github.com/d5/tengo v1.9.2 h1:UE/X8PYl7bLS4Ww2zGeh91nq5PTnkhe8ncgNeA5PK7k= | ||||
| github.com/d5/tengo v1.9.2/go.mod h1:gsbjo7lBXzBIWBd6NQp1lRKqqiDDANqBOyhW8rTlFsY= | ||||
| github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d h1:rONNnZDE5CYuaSFQk+gP4GEQTXEUcyQ5p6p/dgxIHas= | ||||
| github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d/go.mod h1:UGa5M2Sz/Uh13AMse4+RELKCDw7kqgqlTjeGae+7vUY= | ||||
| github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a h1:MuHMeSsXbNEeUyxjB7T9P8s1+5k8OLTC/M27qsVwixM= | ||||
| github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= | ||||
| github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/dfordsoft/golib v0.0.0-20180902042739-76ee6ab99bec h1:JEUiu7P9smN7zgX87a2zVnnbPPickIM9Gf9OIhsIgWQ= | ||||
| github.com/dfordsoft/golib v0.0.0-20180902042739-76ee6ab99bec/go.mod h1:UGa5M2Sz/Uh13AMse4+RELKCDw7kqgqlTjeGae+7vUY= | ||||
| github.com/dgrijalva/jwt-go v3.2.0+incompatible h1:7qlOGliEKZXTDg6OTjfoBKDXWrumCAMpl/TFQ4/5kLM= | ||||
| github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= | ||||
| github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4= | ||||
| github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I= | ||||
| github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c h1:3gMh737vMGqAkkkSfNbwjO8VRHOSaCjYRG4y9xVMEIQ= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM= | ||||
| github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc h1:wdhDSKrkYy24mcfzuA3oYm58h0QkyXjwERCkzJDP5kA= | ||||
| github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | ||||
| github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c h1:MrMA1vhRTNidtgENqmsmLOIUS6ixMBOU/g10rm7IUe8= | ||||
| github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c/go.mod h1:pMQgrscwEK/aUSW1IFSaBPbJX82FPHWaSoJw1axQfD0= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v4.6.5-0.20181225215658-ec221ba9ea45+incompatible h1:i64CCJcSqkRIkm5OSdZQjZq84/gJsk2zNwHWIRYWlKE= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v4.6.5-0.20181225215658-ec221ba9ea45+incompatible/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM= | ||||
| github.com/golang/protobuf v1.2.0 h1:P3YflyNX/ehuJFLhxviNdFxQPkGK5cDcApsge1SqnvM= | ||||
| github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= | ||||
| github.com/google/gops v0.3.5 h1:SIWvPLiYvy5vMwjxB3rVFTE4QBhUFj2KKWr3Xm7CKhw= | ||||
| github.com/google/gops v0.3.5/go.mod h1:pMQgrscwEK/aUSW1IFSaBPbJX82FPHWaSoJw1axQfD0= | ||||
| github.com/gopackage/ddp v0.0.0-20170117053602-652027933df4 h1:4EZlYQIiyecYJlUbVkFXCXHz1QPhVXcHnQKAzBTPfQo= | ||||
| github.com/gopackage/ddp v0.0.0-20170117053602-652027933df4/go.mod h1:lEO7XoHJ/xNRBCxrn4h/CEB67h0kW1B0t4ooP2yrjUA= | ||||
| github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f h1:FDM3EtwZLyhW48YRiyqjivNlNZjAObv4xt4NnJaU+NQ= | ||||
| github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= | ||||
| github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c h1:mORYpib1aLu3M2Oi50Z1pNTXuDJEHcoLb6oo6VdOutk= | ||||
| github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c/go.mod h1:kgLaKoK1FELgZqMAVxx/5cbj0kT+57qxUrAlIO2eleU= | ||||
| github.com/gorilla/websocket v0.0.0-20170319172727-a91eba7f9777 h1:JIM+OacoOJRU30xpjMf8sulYqjr0ViA3WDrTX6j/yDI= | ||||
| github.com/gorilla/websocket v0.0.0-20170319172727-a91eba7f9777/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ= | ||||
| github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad h1:eMxs9EL0PvIGS9TTtxg4R+JxuPGav82J8rA+GFnY7po= | ||||
| github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= | ||||
| github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb h1:1OvvPvZkn/yCQ3xBcM8y4020wdkMXPHLB4+NfoGWh4U= | ||||
| github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb/go.mod h1:oZtUIOe8dh44I2q6ScRibXws4Ajl+d+nod3AaR9vL5w= | ||||
| github.com/gorilla/mux v1.6.2/go.mod h1:1lud6UwP+6orDFRuTfBEV8e9/aOM/c4fVVCaMa2zaAs= | ||||
| github.com/gorilla/schema v1.0.2 h1:sAgNfOcNYvdDSrzGHVy9nzCQahG+qmsg+nE8dK85QRA= | ||||
| github.com/gorilla/schema v1.0.2/go.mod h1:kgLaKoK1FELgZqMAVxx/5cbj0kT+57qxUrAlIO2eleU= | ||||
| github.com/gorilla/websocket v1.4.0 h1:WDFjx/TMzVgy9VdMMQi2K2Emtwi2QcUQsztZ/zLaH/Q= | ||||
| github.com/gorilla/websocket v1.4.0/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ= | ||||
| github.com/hashicorp/golang-lru v0.5.0 h1:CL2msUPvZTLb5O648aiLNJw3hnBxN2+1Jq8rCOH9wdo= | ||||
| github.com/hashicorp/golang-lru v0.5.0/go.mod h1:/m3WP610KZHVQ1SGc6re/UDhFvYD7pJ4Ao+sR/qLZy8= | ||||
| github.com/hashicorp/hcl v1.0.0 h1:0Anlzjpi4vEasTeNFn2mLJgTSwt0+6sfsiTG8qcWGx4= | ||||
| github.com/hashicorp/hcl v1.0.0/go.mod h1:E5yfLk+7swimpb2L/Alb/PJmXilQ/rhwaUYs4T20WEQ= | ||||
| github.com/howeyc/gopass v0.0.0-20170109162249-bf9dde6d0d2c/go.mod h1:lADxMC39cJJqL93Duh1xhAs4I2Zs8mKS89XWXFGp9cs= | ||||
| github.com/hpcloud/tail v1.0.0 h1:nfCOvKYfkgYP8hkirhJocXT2+zOD8yUNjXaWfTlyFKI= | ||||
| github.com/hpcloud/tail v1.0.0/go.mod h1:ab1qPbhIpdTxEkNHXyeSf5vhxWSCs/tWer42PpOxQnU= | ||||
| github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d h1:ETeT81zgLgSNc4BWdDO2Fg9ekVItYErbNtE8mKD2pJA= | ||||
| github.com/jpillora/backoff v0.0.0-20170222002228-06c7a16c845d/go.mod h1:2iMrUgbbvHEiQClaW2NsSzMyGHqN+rDFqY705q49KG0= | ||||
| github.com/jessevdk/go-flags v1.3.0/go.mod h1:4FA24M0QyGHXBuZZK/XkWh8h0e1EYbRYJSGM75WSRxI= | ||||
| github.com/jpillora/backoff v0.0.0-20180909062703-3050d21c67d7 h1:K//n/AqR5HjG3qxbrBCL4vJPW0MVFSs9CPK1OOJdRME= | ||||
| github.com/jpillora/backoff v0.0.0-20180909062703-3050d21c67d7/go.mod h1:2iMrUgbbvHEiQClaW2NsSzMyGHqN+rDFqY705q49KG0= | ||||
| github.com/jtolds/gls v4.2.1+incompatible h1:fSuqC+Gmlu6l/ZYAoZzx2pyucC8Xza35fpRVWLVmUEE= | ||||
| github.com/jtolds/gls v4.2.1+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= | ||||
| github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462 h1:oSOOTPHkCzMeu1vJ0nHxg5+XZBdMMjNa+6NPnm8arok= | ||||
| github.com/kardianos/osext v0.0.0-20170207191655-9b883c5eb462/go.mod h1:1NbS8ALrpOvjt0rHPNLyCIeMtbizbir8U//inJ+zuB8= | ||||
| github.com/kardianos/osext v0.0.0-20170510131534-ae77be60afb1 h1:PJPDf8OUfOK1bb/NeTKd4f1QXZItOX389VN3B6qC8ro= | ||||
| github.com/kardianos/osext v0.0.0-20170510131534-ae77be60afb1/go.mod h1:1NbS8ALrpOvjt0rHPNLyCIeMtbizbir8U//inJ+zuB8= | ||||
| github.com/konsorten/go-windows-terminal-sequences v1.0.1 h1:mweAR1A6xJ3oS2pRaGiHgQ4OO8tzTaLawm8vnODuwDk= | ||||
| github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= | ||||
| github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI= | ||||
| github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= | ||||
| github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= | ||||
| github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE= | ||||
| github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||
| github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1 h1:cOIt0LZKdfeirAfTP4VtIJuWbjVTGtd1suuPXp/J+dE= | ||||
| github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1/go.mod h1:0INS7j/VjnFxD4E2wkz67b8cVwCLbBmJyDaka6Cmk1s= | ||||
| github.com/labstack/gommon v0.2.1 h1:C+I4NYknueQncqKYZQ34kHsLZJVeB5KwPUhnO0nmbpU= | ||||
| github.com/labstack/gommon v0.2.1/go.mod h1:/tj9csK2iPSBvn+3NLM9e52usepMtrd5ilFYA+wQNJ4= | ||||
| github.com/lrstanley/girc v0.0.0-20180427160007-102f17f86306 h1:IqN61cmi7LM/IaYaP9a/KXFtHRS2a3+WHu8GhAXJT7c= | ||||
| github.com/lrstanley/girc v0.0.0-20180427160007-102f17f86306/go.mod h1:7cRs1SIBfKQ7e3Tam6GKTILSNHzR862JD0JpINaZoJk= | ||||
| github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 h1:HWxJJvF+QceKcql4r9PC93NtMEgEBfBxlQrZPvbcQvs= | ||||
| github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= | ||||
| github.com/matterbridge/discordgo v0.0.0-20180806170629-ef40ff5ba64f h1:9IIOO9Aznn8zJx3nokZ4U6nfuzWw5xAlygPvuRZMisQ= | ||||
| github.com/matterbridge/discordgo v0.0.0-20180806170629-ef40ff5ba64f/go.mod h1:5QtN542bJn9FunZqYlIbleNtToxfLCVV9pW7m7Q42Fc= | ||||
| github.com/labstack/echo/v4 v4.0.0 h1:q1GH+caIXPP7H2StPIdzy/ez9CO0EepqYeUg6vi9SWM= | ||||
| github.com/labstack/echo/v4 v4.0.0/go.mod h1:tZv7nai5buKSg5h/8E6zz4LsD/Dqh9/91Mvs7Z5Zyno= | ||||
| github.com/labstack/gommon v0.2.8 h1:JvRqmeZcfrHC5u6uVleB4NxxNbzx6gpbJiQknDbKQu0= | ||||
| github.com/labstack/gommon v0.2.8/go.mod h1:/tj9csK2iPSBvn+3NLM9e52usepMtrd5ilFYA+wQNJ4= | ||||
| github.com/lrstanley/girc v0.0.0-20190210212025-51b8e096d398 h1:a40kRmhA1p2XFJ6gqXfCExSyuDDCp/U9LA8ZY27u2Lk= | ||||
| github.com/lrstanley/girc v0.0.0-20190210212025-51b8e096d398/go.mod h1:7cRs1SIBfKQ7e3Tam6GKTILSNHzR862JD0JpINaZoJk= | ||||
| github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 h1:AsEBgzv3DhuYHI/GiQh2HxvTP71HCCE9E/tzGUzGdtU= | ||||
| github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5/go.mod h1:c2mYKRyMb1BPkO5St0c/ps62L4S0W2NAkaTXj9qEI+0= | ||||
| github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 h1:iOAVXzZyXtW408TMYejlUPo6BIn92HmOacWtIfNyYns= | ||||
| github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6/go.mod h1:sFlOUpQL1YcjhFVXhg1CG8ZASEs/Mf1oVb6H75JL/zg= | ||||
| github.com/magiconair/properties v1.8.0 h1:LLgXmsheXeRoUOBOjtwPQCWIYqM/LU1ayDtDePerRcY= | ||||
| github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= | ||||
| github.com/matterbridge/Rocket.Chat.Go.SDK v0.0.0-20190210153444-cc9d05784d5d h1:F+Sr+C0ojSlYQ37BLylQtSFmyQULe3jbAygcyXQ9mVs= | ||||
| github.com/matterbridge/Rocket.Chat.Go.SDK v0.0.0-20190210153444-cc9d05784d5d/go.mod h1:c6MxwqHD+0HvtAJjsHMIdPCiAwGiQwPRPTp69ACMg8A= | ||||
| github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 h1:KzDEcy8eDbTx881giW8a6llsAck3e2bJvMyKvh1IK+k= | ||||
| github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91/go.mod h1:ECDRehsR9TYTKCAsRS8/wLeOk6UUqDydw47ln7wG41Q= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f h1:2eKh6Qi/sJ8bXvYMoyVfQxHgR8UcCDWjOmhV1oCstMU= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f/go.mod h1:+jWeaaUtXQbBRdKYWfjW6JDDYiI2XXE+3NnTjW5kg8g= | ||||
| github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544 h1:A8lLG3DAu75B5jITHs9z4JBmU6oCq1WiUNnDAmqKCZc= | ||||
| github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544/go.mod h1:yAjnZ34DuDyPHMPHHjOsTk/FefW4JJjoMMCGt/8uuQA= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20190102230110-6f9631ca6dea h1:kaADGqpK4gGO2BpzEyJrBxq2Jc57Rsar4i2EUxcACUc= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20190102230110-6f9631ca6dea/go.mod h1:+jWeaaUtXQbBRdKYWfjW6JDDYiI2XXE+3NnTjW5kg8g= | ||||
| github.com/matterbridge/gozulipbot v0.0.0-20190212232658-7aa251978a18 h1:fLhwXtWGtfTgZVxHG1lcKjv+re7dRwyyuYFNu69xdho= | ||||
| github.com/matterbridge/gozulipbot v0.0.0-20190212232658-7aa251978a18/go.mod h1:yAjnZ34DuDyPHMPHHjOsTk/FefW4JJjoMMCGt/8uuQA= | ||||
| github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 h1:R/MgM/eUyRBQx2FiH6JVmXck8PaAuKfe2M1tWIzW7nE= | ||||
| github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61/go.mod h1:iXGEotOvwI1R1SjLxRc+BF5rUORTMtE0iMZBT2lxqAU= | ||||
| github.com/mattermost/platform v4.6.2+incompatible h1:9WqKNuJFIp6SDYn5wl1RF5urdhEw8d7o5tAOwT1MW0A= | ||||
| github.com/mattermost/platform v4.6.2+incompatible/go.mod h1:HjGKtkQNu3HXTOykPMQckMnH11WHvNvQqDBNnVXVbfM= | ||||
| github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597 h1:hGizH4aMDFFt1iOA4HNKC13lqIBoCyxIjWcAnWIy7aU= | ||||
| github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= | ||||
| github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc h1:pK7tzC30erKOTfEDCYGvPZQCkmM9X5iSmmAR5m9x3Yc= | ||||
| github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= | ||||
| github.com/mattermost/mattermost-server v5.5.0+incompatible h1:0wcLGgYtd+YImtLDPf2AOfpBHxbU4suATx+6XKw1XbU= | ||||
| github.com/mattermost/mattermost-server v5.5.0+incompatible/go.mod h1:5L6MjAec+XXQwMIt791Ganu45GKsSiM+I0tLR9wUj8Y= | ||||
| github.com/mattn/go-colorable v0.0.9 h1:UVL0vNpWh04HeJXV0KLcaT7r06gOH2l4OW6ddYRUIY4= | ||||
| github.com/mattn/go-colorable v0.0.9/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= | ||||
| github.com/mattn/go-isatty v0.0.4 h1:bnP0vzxcAdeI1zdubAl5PjU6zsERjGZb7raWodagDYs= | ||||
| github.com/mattn/go-isatty v0.0.4/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= | ||||
| github.com/mattn/go-sqlite3 v1.10.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc= | ||||
| github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4= | ||||
| github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE= | ||||
| github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 h1:+MZW2uvHgN8kYvksEN3f7eFL2wpzk0GxmlFsMybWc7E= | ||||
| github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= | ||||
| github.com/mitchellh/mapstructure v1.1.2 h1:fmNYVwqnSfB9mZU6OS2O6GsXM+wcskZDuKQzvN1EDeE= | ||||
| github.com/mitchellh/mapstructure v1.1.2/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= | ||||
| github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 h1:oKIteTqeSpenyTrOVj5zkiyCaflLa8B+CD0324otT+o= | ||||
| github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474/go.mod h1:OQA4XLvDbMgS8P0CevmM4m9Q3Jq4phKUzcocxuGJ5m8= | ||||
| github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff h1:HLGD5/9UxxfEuO9DtP8gnTmNtMxbPyhYltfxsITel8g= | ||||
| github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff/go.mod h1:B8jLfIIPn2sKyWr0D7cL2v7tnrDD5z291s2Zypdu89E= | ||||
| github.com/nelsonken/gomf v0.0.0-20180504123937-a9dd2f9deae9 h1:mp6tU1r0xLostUGLkTspf/9/AiHuVD7ptyXhySkDEsE= | ||||
| github.com/nelsonken/gomf v0.0.0-20180504123937-a9dd2f9deae9/go.mod h1:A5SRAcpTemjGgIuBq6Kic2yHcoeUFWUinOAlMP/i9xo= | ||||
| github.com/nicksnyder/go-i18n v1.4.0 h1:AgLl+Yq7kg5OYlzCgu9cKTZOyI4tD/NgukKqLqC8E+I= | ||||
| github.com/nicksnyder/go-i18n v1.4.0/go.mod h1:HrK7VCrbOvQoUAQ7Vpy7i87N7JZZZ7R2xBGjv0j365Q= | ||||
| github.com/nlopes/slack v0.0.0-20180101221843-107290b5bbaf h1:M+xGhDxie/MqC+tzs+3ZHBSY4Wsv+fEkrpIMCKy8PTg= | ||||
| github.com/nlopes/slack v0.0.0-20180101221843-107290b5bbaf/go.mod h1:jVI4BBK3lSktibKahxBF74txcK2vyvkza1z/+rRnVAM= | ||||
| github.com/nlopes/slack v0.3.1-0.20180805133408-21749ab136a8 h1:PSy8NkmkyldLmPPnNNw7mwfQFOHDqOI6bINpJ+/KV7Y= | ||||
| github.com/nlopes/slack v0.3.1-0.20180805133408-21749ab136a8/go.mod h1:jVI4BBK3lSktibKahxBF74txcK2vyvkza1z/+rRnVAM= | ||||
| github.com/nlopes/slack v0.5.0 h1:NbIae8Kd0NpqaEI3iUrsuS0KbcEDhzhc939jLW5fNm0= | ||||
| github.com/nlopes/slack v0.5.0/go.mod h1:jVI4BBK3lSktibKahxBF74txcK2vyvkza1z/+rRnVAM= | ||||
| github.com/onsi/ginkgo v1.6.0 h1:Ix8l273rp3QzYgXSR+c8d1fTG7UPgYkOSELPhiY/YGw= | ||||
| github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE= | ||||
| github.com/onsi/gomega v1.4.1 h1:PZSj/UFNaVp3KxrzHOcS7oyuWA7LoOY/77yCTEFu21U= | ||||
| @@ -93,71 +119,119 @@ github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83 h1:XQonH5Iv | ||||
| github.com/paulrosania/go-charset v0.0.0-20151028000031-621bb39fcc83/go.mod h1:YnNlZP7l4MhyGQ4CBRwv6ohZTPrUJJZtEv4ZgADkbs4= | ||||
| github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606 h1:/CPgDYrfeK2LMK6xcUhvI17yO9SlpAdDIJGkhDEgO8A= | ||||
| github.com/pborman/uuid v0.0.0-20160216163710-c55201b03606/go.mod h1:VyrYX9gd7irzKovcSS6BIIEwPRkP2Wm2m9ufcdFSJ34= | ||||
| github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e h1:ZW8599OjioQsmBbkGpyruHUlRVQceYFWnJsGr4NCkiA= | ||||
| github.com/pelletier/go-toml v0.0.0-20180228233631-05bcc0fb0d3e/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic= | ||||
| github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271 h1:wQ9lVx75za6AT2kI0S9QID0uWuwTWnvcTfN+uw1F8vg= | ||||
| github.com/peterhellberg/emojilib v0.0.0-20170616163716-41920917e271/go.mod h1:G7LufuPajuIvdt9OitkNt2qh0mmvD4bfRgRM7bhDIOA= | ||||
| github.com/pelletier/go-toml v1.2.0 h1:T5zMGML61Wp+FlcbWjRDT7yAxhJNAiPPLOFECq181zc= | ||||
| github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic= | ||||
| github.com/peterhellberg/emojilib v0.0.0-20190124112554-c18758d55320 h1:YxcQy/DV+48NGv1lxx1vsWBzs6W1f1ogubkuCozxpX0= | ||||
| github.com/peterhellberg/emojilib v0.0.0-20190124112554-c18758d55320/go.mod h1:G7LufuPajuIvdt9OitkNt2qh0mmvD4bfRgRM7bhDIOA= | ||||
| github.com/pkg/errors v0.8.0 h1:WdK/asTD0HN+q6hsWO3/vpuAkAr+tw6aNJNDFFf0+qw= | ||||
| github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= | ||||
| github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= | ||||
| github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= | ||||
| github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a h1:UWKek6MK3K6/TpbsFcv+8rrO6rSc6KKSp2FbMOHWsq4= | ||||
| github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a/go.mod h1:+uKXf+4Djp6Md1KODXJxgGQPKngRmWyn10oCKFzNHOQ= | ||||
| github.com/russross/blackfriday v1.5.1 h1:B8ZN6pD4PVofmlDCDUdELeYrbsVIDM/bpjW3v3zgcRc= | ||||
| github.com/russross/blackfriday v1.5.1/go.mod h1:JO/DiYxRf+HjHt06OyowR9PTA263kcR/rfWxYHBV53g= | ||||
| github.com/rs/xid v1.2.1 h1:mhH9Nq+C1fY2l1XIpgxIiUOfNpRBYH1kKcr+qfKgjRc= | ||||
| github.com/rs/xid v1.2.1/go.mod h1:+uKXf+4Djp6Md1KODXJxgGQPKngRmWyn10oCKFzNHOQ= | ||||
| github.com/russross/blackfriday v2.0.0+incompatible h1:cBXrhZNUf9C+La9/YpS+UHpUT8YD6Td9ZMSU9APFcsk= | ||||
| github.com/russross/blackfriday v2.0.0+incompatible/go.mod h1:JO/DiYxRf+HjHt06OyowR9PTA263kcR/rfWxYHBV53g= | ||||
| github.com/russross/blackfriday/v2 v2.0.1/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= | ||||
| github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca h1:NugYot0LIVPxTvN8n+Kvkn6TrbMyxQiuvKdEwFdR9vI= | ||||
| github.com/saintfish/chardet v0.0.0-20120816061221-3af4cd4741ca/go.mod h1:uugorj2VCxiV1x+LzaIdVa9b4S4qGAcH6cbhh4qVxOU= | ||||
| github.com/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1 h1:Lx3BlDGFElJt4u/zKc9A3BuGYbQAGlEFyPuUA3jeMD0= | ||||
| github.com/shazow/rateio v0.0.0-20150116013248-e8e00881e5c1/go.mod h1:vt2jWY/3Qw1bIzle5thrJWucsLuuX9iUNnp20CqCciI= | ||||
| github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991 h1:PQiUTDzUC5EUh0vNurK7KQS22zlKqLLOFn+K9nJXDQQ= | ||||
| github.com/shazow/ssh-chat v0.0.0-20171012174035-2078e1381991/go.mod h1:KwtnpMClmrXsHCKTbRui5xBUNt17n1GGrGhdiw2KcoY= | ||||
| github.com/sirupsen/logrus v0.0.0-20180213143110-8c0189d9f6bb h1:eKjx20EiekBRT2tjZ0XEdKpftfPJQwiavtFshwTyqf0= | ||||
| github.com/sirupsen/logrus v0.0.0-20180213143110-8c0189d9f6bb/go.mod h1:pMByvHTf9Beacp5x1UXfOR9xyW/9antXMhjMPG0dEzc= | ||||
| github.com/shazow/ssh-chat v0.0.0-20190125184227-81d7e1686296 h1:8RLq547MSVc6vhOuCl4Ca0TsAQknj6NX6ZLSZ3+xmio= | ||||
| github.com/shazow/ssh-chat v0.0.0-20190125184227-81d7e1686296/go.mod h1:1GLXsL4esywkpNId3v4QWuMf3THtWGitWvtQ/L3aSA4= | ||||
| github.com/shurcooL/sanitized_anchor_name v1.0.0 h1:PdmoCO6wvbs+7yrJyMORt4/BmY5IYyJwS/kOiWx8mHo= | ||||
| github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= | ||||
| github.com/sirupsen/logrus v1.3.0 h1:hI/7Q+DtNZ2kINb6qt/lS+IyXnHQe9e90POfeewL/ME= | ||||
| github.com/sirupsen/logrus v1.3.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= | ||||
| github.com/skip2/go-qrcode v0.0.0-20171229120447-cf5f9fa2f0d8/go.mod h1:PLPIyL7ikehBD1OAjmKKiOEhbvWyHGaNDjquXMcYABo= | ||||
| github.com/skip2/go-qrcode v0.0.0-20190110000554-dc11ecdae0a9 h1:lpEzuenPuO1XNTeikEmvqYFcU37GVLl8SRNblzyvGBE= | ||||
| github.com/skip2/go-qrcode v0.0.0-20190110000554-dc11ecdae0a9/go.mod h1:PLPIyL7ikehBD1OAjmKKiOEhbvWyHGaNDjquXMcYABo= | ||||
| github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 h1:lXQ+j+KwZcbwrbgU0Rp4Eglg3EJLHbuZU3BbOqAGBmg= | ||||
| github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= | ||||
| github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a h1:JSvGDIbmil4Ui/dDdFBExb7/cmkNjyX5F97oglmvCDo= | ||||
| github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a/go.mod h1:XDJAKZRPZ1CvBcN2aX5YOUTYGHki24fSF0Iv48Ibg0s= | ||||
| github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff h1:HLvGWId7M56TfuxTeZ6aoiTAcrWO5Mnq/ArwVRgV62I= | ||||
| github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ= | ||||
| github.com/spf13/cast v1.2.0 h1:HHl1DSRbEQN2i8tJmtS6ViPyHx35+p51amrdsiTCrkg= | ||||
| github.com/spf13/cast v1.2.0/go.mod h1:r2rcYCSwa1IExKTDiTfzaxqT2FNHs8hODu4LnUfgKEg= | ||||
| github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec h1:2ZXvIUGghLpdTVHR1UfvfrzoVlZaE/yOWC5LueIHZig= | ||||
| github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo= | ||||
| github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac h1:+uzyQ0TQ3aKorQxsOjcDDgE7CuUXwpkKnK19LULQALQ= | ||||
| github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= | ||||
| github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7 h1:Wj4cg2M6Um7j1N7yD/mxsdy1/wrsdjzVha2eWdOhti8= | ||||
| github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7/go.mod h1:A8kyI5cUJhb8N+3pkfONlcEcZbueH6nhAm0Fq7SrnBM= | ||||
| github.com/stretchr/testify v0.0.0-20170714215325-05e8a0eda380 h1:MsolbevHkd4SpbeG4dHLHj6I9jzoohyNI6EK6JvR5hE= | ||||
| github.com/stretchr/testify v0.0.0-20170714215325-05e8a0eda380/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= | ||||
| github.com/spf13/afero v1.1.2 h1:m8/z1t7/fwjysjQRYbP0RD+bUIF/8tJwPdEZsI83ACI= | ||||
| github.com/spf13/afero v1.1.2/go.mod h1:j4pytiNVoe2o6bmDsKpLACNPDBIoEAkihy7loJ1B0CQ= | ||||
| github.com/spf13/cast v1.3.0 h1:oget//CVOEoFewqQxwr0Ej5yjygnqGkvggSE/gB35Q8= | ||||
| github.com/spf13/cast v1.3.0/go.mod h1:Qx5cxh0v+4UWYiBimWS+eyWzqEqokIECu5etghLkUJE= | ||||
| github.com/spf13/jwalterweatherman v1.0.0 h1:XHEdyB+EcvlqZamSM4ZOMGlc93t6AcsBEu9Gc1vn7yk= | ||||
| github.com/spf13/jwalterweatherman v1.0.0/go.mod h1:cQK4TGJAtQXfYWX+Ddv3mKDzgVb68N+wFjFa4jdeBTo= | ||||
| github.com/spf13/pflag v1.0.3 h1:zPAT6CGy6wXeQ7NtTnaTerfKOsV6V6F8agHXFiazDkg= | ||||
| github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= | ||||
| github.com/spf13/viper v1.3.1 h1:5+8j8FTpnFV4nEImW/ofkzEt8VoOiLXxdYIDsB73T38= | ||||
| github.com/spf13/viper v1.3.1/go.mod h1:ZiWeW+zYFKm7srdB9IoDzzZXaJaI5eL9QjNiN/DMA2s= | ||||
| github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||
| github.com/stretchr/objx v0.1.1 h1:2vfRuCMp5sSVIDSqO8oNnWJq7mPa6KVP3iPIwFBuy8A= | ||||
| github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= | ||||
| github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w= | ||||
| github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= | ||||
| github.com/stretchr/testify v1.3.0 h1:TivCn/peBQ7UY8ooIcPgZFpTNSz0Q2U6UrFlUfqbe0Q= | ||||
| github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= | ||||
| github.com/technoweenie/multipartstreamer v1.0.1 h1:XRztA5MXiR1TIRHxH2uNxXxaIkKQDeX7m2XsSOlQEnM= | ||||
| github.com/technoweenie/multipartstreamer v1.0.1/go.mod h1:jNVxdtShOxzAsukZwTSw6MDx5eUJoiEBsSvzDU9uzog= | ||||
| github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a h1:AOcehBWpFhYPYw0ioDTppQzgI8pAAahVCiMSKTp9rbo= | ||||
| github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= | ||||
| github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0= | ||||
| github.com/valyala/bytebufferpool v1.0.0 h1:GqA5TC/0021Y/b9FG4Oi9Mr3q7XYx6KllzawFIhcdPw= | ||||
| github.com/valyala/bytebufferpool v1.0.0/go.mod h1:6bBcMArwyJ5K/AmCkWv1jt77kVWyCJ6HpOuEn7z0Csc= | ||||
| github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4 h1:gKMu1Bf6QINDnvyZuTaACm9ofY+PRh+5vFz4oxBZeF8= | ||||
| github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4/go.mod h1:50wTf68f99/Zt14pr046Tgt3Lp2vLyFZKzbFXTOabXw= | ||||
| github.com/x-cray/logrus-prefixed-formatter v0.5.2 h1:00txxvfBM9muc0jiLIEAkAcIMJzfthRT6usrui8uGmg= | ||||
| github.com/x-cray/logrus-prefixed-formatter v0.5.2/go.mod h1:2duySbKsL6M18s5GU7VPsoEPHyzalCE06qoARUCeBBE= | ||||
| github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6 h1:/WULP+6asFz569UbOwg87f3iDT7T+GF5/vjLmL51Pdk= | ||||
| github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6/go.mod h1:0MsIttMJIF/8Y7x0XjonJP7K99t3sR6bjj4m5S4JmqU= | ||||
| golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88 h1:jLkAo/qlT9whgCLYC5GAJ9kcKrv3Wj8VCc4N+KJ4wpw= | ||||
| golang.org/x/crypto v0.0.0-20180228161326-91a49db82a88/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 h1:BkNcmLtAVeWe9h5k0jt24CQgaG5vb4x/doFbAiEC/Ho= | ||||
| golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f h1:wMNYb4v58l5UBM7MYRLPG6ZhfOqbKu7X5eyFl8ZhKvA= | ||||
| golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sys v0.0.0-20171130163741-8b4580aae2a0 h1:x4M4WCms+ErQg/4VyECbP2kSNcDJ6nLwqEGov1QPtqk= | ||||
| golang.org/x/sys v0.0.0-20171130163741-8b4580aae2a0/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 h1:WNm0tmiuBMW4FJRuXKWOqaQfmKptHs0n8nTCyG0ayjc= | ||||
| golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||
| gopkg.in/airbrake/gobrake.v2 v2.0.9 h1:7z2uVWwn7oVeeugY1DtlPAy5H+KYgB1KeKTnqjNatLo= | ||||
| gopkg.in/airbrake/gobrake.v2 v2.0.9/go.mod h1:/h5ZAUhDkGaJfjzjKLSjv6zCL6O0LLBxU4K+aSYdM/U= | ||||
| github.com/xordataexchange/crypt v0.0.3-0.20170626215501-b2862e3d0a77/go.mod h1:aYKd//L2LvnjZzWKhF00oedf4jCCReLcmhLdhm1A27Q= | ||||
| github.com/zfjagann/golang-ring v0.0.0-20190106091943-a88bb6aef447 h1:CHgPZh8bFkZmislPrr/0gd7MciDAX+JJB70A2/5Lvmo= | ||||
| github.com/zfjagann/golang-ring v0.0.0-20190106091943-a88bb6aef447/go.mod h1:0MsIttMJIF/8Y7x0XjonJP7K99t3sR6bjj4m5S4JmqU= | ||||
| gitlab.com/golang-commonmark/html v0.0.0-20180917080848-cfaf75183c4a h1:Ax7kdHNICZiIeFpmevmaEWb0Ae3BUj3zCTKhZHZ+zd0= | ||||
| gitlab.com/golang-commonmark/html v0.0.0-20180917080848-cfaf75183c4a/go.mod h1:JT4uoTz0tfPoyVH88GZoWDNm5NHJI2VbUW+eyPClueI= | ||||
| gitlab.com/golang-commonmark/linkify v0.0.0-20180917065525-c22b7bdb1179 h1:rbON2KwBnWuFMlSHM8LELLlwroDRZw6xv0e6il6e5dk= | ||||
| gitlab.com/golang-commonmark/linkify v0.0.0-20180917065525-c22b7bdb1179/go.mod h1:Gn+LZmCrhPECMD3SOKlE+BOHwhOYD9j7WT9NUtkCrC8= | ||||
| gitlab.com/golang-commonmark/markdown v0.0.0-20181102083822-772775880e1f h1:jwXy/CsM4xS2aoiF2fHAlukmInWhd2TlWB+HDCyvzKc= | ||||
| gitlab.com/golang-commonmark/markdown v0.0.0-20181102083822-772775880e1f/go.mod h1:SIHlEr9462fpIfTrVWf3GqQDxnA65Vm3BMMsUtuA6W0= | ||||
| gitlab.com/golang-commonmark/mdurl v0.0.0-20180912090424-e5bce34c34f2 h1:wD/sPUgx2QJFPTyXZpJnLaROolfeKuruh06U4pRV0WY= | ||||
| gitlab.com/golang-commonmark/mdurl v0.0.0-20180912090424-e5bce34c34f2/go.mod h1:wQk4rLkWrdOPjUAtqJRJ10hIlseLSVYWP95PLrjDF9s= | ||||
| gitlab.com/golang-commonmark/puny v0.0.0-20180912090636-2cd490539afe h1:5kUPFAF52umOUPH12MuNUmyVTseJRNBftDl/KfsvX3I= | ||||
| gitlab.com/golang-commonmark/puny v0.0.0-20180912090636-2cd490539afe/go.mod h1:P9LSM1KVzrIstFgUaveuwiAm8PK5VTB3yJEU8kqlbrU= | ||||
| gitlab.com/opennota/wd v0.0.0-20180912061657-c5d65f63c638 h1:uPZaMiz6Sz0PZs3IZJWpU5qHKGNy///1pacZC9txiUI= | ||||
| gitlab.com/opennota/wd v0.0.0-20180912061657-c5d65f63c638/go.mod h1:EGRJaqe2eO9XGmFtQCvV3Lm9NLico3UhFwUpCG/+mVU= | ||||
| go.uber.org/atomic v1.3.2 h1:2Oa65PReHzfn29GpvgsYwloV9AVFHPDk8tYxt2c2tr4= | ||||
| go.uber.org/atomic v1.3.2/go.mod h1:gD2HeocX3+yG+ygLZcrzQJaqmWj9AIm7n08wl/qW/PE= | ||||
| go.uber.org/multierr v1.1.0 h1:HoEmRHQPVSqub6w2z2d2EOVs2fjyFRGyofhKuyDq0QI= | ||||
| go.uber.org/multierr v1.1.0/go.mod h1:wR5kodmAFQ0UK8QlbwjlSNy0Z68gJhDJUG5sjR94q/0= | ||||
| go.uber.org/zap v1.9.1 h1:XCJQEf3W6eZaVwhRBof6ImoYGJSITeKWsyeh3HFu/5o= | ||||
| go.uber.org/zap v1.9.1/go.mod h1:vwi/ZaCAaUcBkycHslxD9B2zi4UTXhF60s6SWpuDF0Q= | ||||
| golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16 h1:y6ce7gCWtnH+m3dCjzQ1PCuwl28DDIc3VNnvY29DlIA= | ||||
| golang.org/x/crypto v0.0.0-20181030102418-4d3f4d9ffa16/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/crypto v0.0.0-20181203042331-505ab145d0a9/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/crypto v0.0.0-20190103213133-ff983b9c42bc/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/crypto v0.0.0-20190130090550-b01c7a725664 h1:YbZJ76lQ1BqNhVe7dKTSB67wDrc2VPRR75IyGyyPDX8= | ||||
| golang.org/x/crypto v0.0.0-20190130090550-b01c7a725664/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613 h1:MQ/ZZiDsUapFFiMS+vzwXkCTeEKaum+Do5rINYJDmxc= | ||||
| golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| golang.org/x/net v0.0.0-20190110200230-915654e7eabc h1:Yx9JGxI1SBhVLFjpAkWMaO1TF+xyqtHLjZpvQboJGiM= | ||||
| golang.org/x/net v0.0.0-20190110200230-915654e7eabc/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= | ||||
| golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4 h1:YUO/7uOKsKeq9UokNS62b8FYywz3ker1l1vDZRCRefw= | ||||
| golang.org/x/sync v0.0.0-20181221193216-37e7f081c4d4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= | ||||
| golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20181205085412-a5c9d58dba9a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20181212120007-b05ddf57801d/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20190109145017-48ac38b7c8cb/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20190129075346-302c3dd5f1cc h1:WiYx1rIFmx8c0mXAFtv5D/mHyKe1+jmuP7PViuwqwuQ= | ||||
| golang.org/x/sys v0.0.0-20190129075346-302c3dd5f1cc/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/text v0.3.0 h1:g61tztE5qeGQ89tm6NTjjM9VPIm088od1l6aSorWRWg= | ||||
| golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY= | ||||
| gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/fsnotify.v1 v1.4.7 h1:xOHLXZwVvI9hhs+cLKq5+I5onOuwQLhQwiu63xxlHs4= | ||||
| gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys= | ||||
| gopkg.in/gemnasium/logrus-airbrake-hook.v2 v2.1.1 h1:4buh9nXkpqc7+GLzDFHei0jwoU9wCQYfVB5Kfo58Yz0= | ||||
| gopkg.in/gemnasium/logrus-airbrake-hook.v2 v2.1.1/go.mod h1:Xk6kEKp8OKb+X14hQBKWaSkCsqBpgog8nAV2xsGOxlo= | ||||
| gopkg.in/natefinch/lumberjack.v2 v2.0.0 h1:1Lc07Kr7qY4U2YPouBjpCLxpiyxIVoxqXgkXLknAOE8= | ||||
| gopkg.in/natefinch/lumberjack.v2 v2.0.0/go.mod h1:l0ndWWf7gzL7RNwBG7wST/UCcT4T24xpD6X8LsfU/+k= | ||||
| gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= | ||||
| gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw= | ||||
| gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129 h1:RBgb9aPUbZ9nu66ecQNIBNsA7j3mB5h8PNDIfhPjaJg= | ||||
| gopkg.in/yaml.v2 v2.0.0-20160301204022-a83829b6f129/go.mod h1:JAlM8MvJe8wmxCU4Bli9HhUf9+ttbYbLASfIpnQbh74= | ||||
| gopkg.in/yaml.v2 v2.2.2 h1:ZCJp+EgiOT7lHqUV2J862kp8Qj64Jo6az82+3Td9dZw= | ||||
| gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
| maunium.net/go/mauflag v1.0.0/go.mod h1:nLivPOpTpHnpzEh8jEdSL9UqO9+/KBJFmNRlwKfkPeA= | ||||
| maunium.net/go/maulogger/v2 v2.0.0/go.mod h1:Hbbkq3NV6jvJodByZu1mgEF3fpT7Kz9z0MjEZ3/BusI= | ||||
| maunium.net/go/mautrix v0.1.0-alpha.3/go.mod h1:GTVu6WDHR+98DKOrYetWsXorvUeKQV3jsSWO6ScbuFI= | ||||
| maunium.net/go/mautrix-appservice v0.1.0-alpha.3/go.mod h1:wOnWOIuprYad7ly12rHIo3JLCPh4jwvx1prVrAB9RhM= | ||||
| maunium.net/go/mautrix-whatsapp v0.0.0-20190127121751-281b3e8f77f3 h1:A18t5Lp7I3aK0V7B7zdpb0hb/PBlu0X/Ai2AyU/XEk4= | ||||
| maunium.net/go/mautrix-whatsapp v0.0.0-20190127121751-281b3e8f77f3/go.mod h1:r5E3J4urDEsjfui9OYZYMLBfCliaAqcCwM2xeczta6k= | ||||
|   | ||||
| @@ -38,7 +38,7 @@ type Config struct { | ||||
| func New(url string, config Config) *Client { | ||||
| 	c := &Client{In: make(chan Message), Config: config} | ||||
| 	tr := &http.Transport{ | ||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, | ||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, //nolint:gosec | ||||
| 	} | ||||
| 	c.httpclient = &http.Client{Transport: tr} | ||||
| 	_, _, err := net.SplitHostPort(c.BindAddress) | ||||
|   | ||||
							
								
								
									
										
											BIN
										
									
								
								img/matterbridge-notext.gif
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 70 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-add-scopes.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 270 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-app-page.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 170 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-create-app.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 282 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-create-bot.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 204 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-finished.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 48 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-install-app.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 73 KiB | 
							
								
								
									
										
											BIN
										
									
								
								img/slack-setup-invite-bot.png
									
									
									
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 62 KiB | 
| @@ -8,51 +8,78 @@ import ( | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/gateway" | ||||
| 	"github.com/42wim/matterbridge/gateway/bridgemap" | ||||
| 	"github.com/google/gops/agent" | ||||
| 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	version = "1.11.3" | ||||
| 	version = "1.14.0-rc1" | ||||
| 	githash string | ||||
|  | ||||
| 	flagConfig  = flag.String("conf", "matterbridge.toml", "config file") | ||||
| 	flagDebug   = flag.Bool("debug", false, "enable debug") | ||||
| 	flagVersion = flag.Bool("version", false, "show version") | ||||
| 	flagGops    = flag.Bool("gops", false, "enable gops agent") | ||||
| ) | ||||
|  | ||||
| func main() { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true}) | ||||
| 	flog := log.WithFields(log.Fields{"prefix": "main"}) | ||||
| 	flagConfig := flag.String("conf", "matterbridge.toml", "config file") | ||||
| 	flagDebug := flag.Bool("debug", false, "enable debug") | ||||
| 	flagVersion := flag.Bool("version", false, "show version") | ||||
| 	flagGops := flag.Bool("gops", false, "enable gops agent") | ||||
| 	flag.Parse() | ||||
| 	if *flagGops { | ||||
| 		agent.Listen(&agent.Options{}) | ||||
| 		defer agent.Close() | ||||
| 	} | ||||
| 	if *flagVersion { | ||||
| 		fmt.Printf("version: %s %s\n", version, githash) | ||||
| 		return | ||||
| 	} | ||||
| 	if *flagDebug || os.Getenv("DEBUG") == "1" { | ||||
| 		log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||
| 		flog.Info("Enabling debug") | ||||
| 		log.SetLevel(log.DebugLevel) | ||||
|  | ||||
| 	rootLogger := setupLogger() | ||||
| 	logger := rootLogger.WithFields(logrus.Fields{"prefix": "main"}) | ||||
|  | ||||
| 	if *flagGops { | ||||
| 		if err := agent.Listen(agent.Options{}); err != nil { | ||||
| 			logger.Errorf("Failed to start gops agent: %#v", err) | ||||
| 		} else { | ||||
| 			defer agent.Close() | ||||
| 		} | ||||
| 	} | ||||
| 	flog.Printf("Running version %s %s", version, githash) | ||||
|  | ||||
| 	logger.Printf("Running version %s %s", version, githash) | ||||
| 	if strings.Contains(version, "-dev") { | ||||
| 		flog.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||
| 		logger.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.") | ||||
| 	} | ||||
| 	cfg := config.NewConfig(*flagConfig) | ||||
| 	cfg.General.Debug = *flagDebug | ||||
| 	r, err := gateway.NewRouter(cfg) | ||||
|  | ||||
| 	cfg := config.NewConfig(rootLogger, *flagConfig) | ||||
| 	cfg.BridgeValues().General.Debug = *flagDebug | ||||
|  | ||||
| 	r, err := gateway.NewRouter(rootLogger, cfg, bridgemap.FullMap) | ||||
| 	if err != nil { | ||||
| 		flog.Fatalf("Starting gateway failed: %s", err) | ||||
| 		logger.Fatalf("Starting gateway failed: %s", err) | ||||
| 	} | ||||
| 	err = r.Start() | ||||
| 	if err != nil { | ||||
| 		flog.Fatalf("Starting gateway failed: %s", err) | ||||
| 	if err = r.Start(); err != nil { | ||||
| 		logger.Fatalf("Starting gateway failed: %s", err) | ||||
| 	} | ||||
| 	flog.Printf("Gateway(s) started succesfully. Now relaying messages") | ||||
| 	logger.Printf("Gateway(s) started succesfully. Now relaying messages") | ||||
| 	select {} | ||||
| } | ||||
|  | ||||
| func setupLogger() *logrus.Logger { | ||||
| 	logger := &logrus.Logger{ | ||||
| 		Out: os.Stdout, | ||||
| 		Formatter: &prefixed.TextFormatter{ | ||||
| 			PrefixPadding: 13, | ||||
| 			DisableColors: true, | ||||
| 			FullTimestamp: true, | ||||
| 		}, | ||||
| 		Level: logrus.InfoLevel, | ||||
| 	} | ||||
| 	if *flagDebug || os.Getenv("DEBUG") == "1" { | ||||
| 		logger.Formatter = &prefixed.TextFormatter{ | ||||
| 			PrefixPadding:   13, | ||||
| 			DisableColors:   true, | ||||
| 			FullTimestamp:   false, | ||||
| 			ForceFormatting: true, | ||||
| 		} | ||||
| 		logger.Level = logrus.DebugLevel | ||||
| 		logger.WithFields(logrus.Fields{"prefix": "main"}).Info("Enabling debug logging.") | ||||
| 	} | ||||
| 	return logger | ||||
| } | ||||
|   | ||||
| @@ -96,7 +96,13 @@ RejoinDelay=0 | ||||
| #Only works in IRC right now. | ||||
| ColorNicks=false | ||||
|  | ||||
| #RunCommands allows you to send RAW irc commands after connection | ||||
| #Array of strings | ||||
| #OPTIONAL (default empty) | ||||
| RunCommands=["PRIVMSG user hello","PRIVMSG chanserv something"] | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -124,21 +130,28 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #See [general] config section for default options | ||||
| #The string "{NOPINGNICK}" (case sensitive) will be replaced by the actual nick / username, but with a ZWSP inside the nick, so the irc user with the same nick won't get pinged. See https://github.com/42wim/matterbridge/issues/175 for more information | ||||
| #OPTIONAL (default empty) | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -195,6 +208,7 @@ SkipTLSVerify=true | ||||
| ## Settings below can be reloaded by editing the file | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -222,20 +236,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #OPTIONAL (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -279,6 +300,7 @@ Nick="yourlogin" | ||||
| ## Settings below can be reloaded by editing the file | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="spammer1 spammer2" | ||||
| @@ -306,20 +328,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -423,6 +452,7 @@ EditDisable=false | ||||
| EditSuffix=" (edited)" | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -450,20 +480,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -502,6 +539,7 @@ Token="Yourtokenhere" | ||||
| ## Settings below can be reloaded by editing the file | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -529,20 +567,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -572,6 +617,10 @@ ShowTopicChange=false | ||||
| #REQUIRED (when not using webhooks) | ||||
| Token="yourslacktoken" | ||||
|  | ||||
| #Extra slack specific debug info, warning this generates a lot of output. | ||||
| #OPTIONAL (default false) | ||||
| Debug="false" | ||||
|  | ||||
| #### Settings for webhook matterbridge. | ||||
| #NOT RECOMMENDED TO USE INCOMING/OUTGOING WEBHOOK. USE SLACK API | ||||
| #AND DEDICATED BOT USER WHEN POSSIBLE! | ||||
| @@ -624,6 +673,7 @@ EditSuffix=" (edited)" | ||||
| PrefixMessagesWithNick=false | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -651,20 +701,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -683,6 +740,20 @@ StripNick=false | ||||
| #OPTIONAL (default false) | ||||
| ShowTopicChange=false | ||||
|  | ||||
| #Opportunistically preserve threaded replies between Slack channels. | ||||
| #This only works if the parent message is still in the cache. | ||||
| #Cache is flushed between restarts. | ||||
| #Note: Not currently working on gateways with mixed bridges of | ||||
| # both slack and slack-legacy type. Context in issue #624. | ||||
| #OPTIONAL (default false) | ||||
| PreserveThreading=false | ||||
|  | ||||
| #Enable showing "user_typing" events from across gateway when available. | ||||
| #Protip: Set your bot/user's "Full Name" to be "Someone (over chat bridge)", | ||||
| #and so the message will say "Someone (over chat bridge) is typing". | ||||
| #OPTIONAL (default false) | ||||
| ShowUserTyping=false | ||||
|  | ||||
| ################################################################### | ||||
| #discord section | ||||
| ################################################################### | ||||
| @@ -709,10 +780,14 @@ Server="yourservername" | ||||
| #OPTIONAL (default false) | ||||
| ShowEmbeds=false | ||||
|  | ||||
| #Shows the username (minus the discriminator) instead of the server nickname | ||||
| #Shows the username instead of the server nickname | ||||
| #OPTIONAL (default false) | ||||
| UseUserName=false | ||||
|  | ||||
| #Show #xxxx discriminator with UseUserName | ||||
| #OPTIONAL (default false) | ||||
| UseDiscriminator=false | ||||
|  | ||||
| #Specify WebhookURL. If given, will relay messages using the Webhook, which gives a better look to messages. | ||||
| #This only works if you have one discord channel, if you have multiple discord channels you'll have to specify it in the gateway config | ||||
| #OPTIONAL (default empty) | ||||
| @@ -727,6 +802,7 @@ EditDisable=false | ||||
| EditSuffix=" (edited)" | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -754,20 +830,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -776,11 +859,16 @@ ShowJoinPart=false | ||||
| #OPTIONAL (default false) | ||||
| StripNick=false | ||||
|  | ||||
| #Enable to show topic changes from other bridges  | ||||
| #Enable to show topic/purpose changes from other bridges | ||||
| #Only works hiding/show topic changes from slack bridge for now | ||||
| #OPTIONAL (default false) | ||||
| ShowTopicChange=false | ||||
|  | ||||
| #Enable to sync topic/purpose changes from other bridges | ||||
| #Only works syncing topic changes from slack bridge for now | ||||
| #OPTIONAL (default false) | ||||
| SyncTopic=false | ||||
|  | ||||
| ################################################################### | ||||
| #telegram section | ||||
| ################################################################### | ||||
| @@ -834,6 +922,7 @@ EditDisable=false | ||||
| EditSuffix=" (edited)" | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="spammer1 spammer2" | ||||
| @@ -861,25 +950,31 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #See [general] config section for default options | ||||
| # | ||||
| #WARNING: if you have set MessageFormat="HTML" be sure that this format matches the guidelines | ||||
| #on https://core.telegram.org/bots/api#html-style otherwise the message will not go through to | ||||
| #telegram! eg <{NICK}> should be <{NICK}> | ||||
| # | ||||
| #OPTIONAL (default empty) | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -902,6 +997,21 @@ ShowTopicChange=false | ||||
| #REQUIRED | ||||
|  | ||||
| [rocketchat.rockme] | ||||
| #The rocketchat hostname. (prefix it with http or https) | ||||
| #REQUIRED (when not using webhooks) | ||||
| Server="https://yourrocketchatserver.domain.com:443" | ||||
|  | ||||
| #login/pass of your bot.  | ||||
| #Use a dedicated user for this and not your own!  | ||||
| #REQUIRED (when not using webhooks) | ||||
| Login="yourlogin" | ||||
| Password="yourpass" | ||||
|  | ||||
| #### Settings for webhook matterbridge. | ||||
| #USE DEDICATED BOT USER WHEN POSSIBLE! This allows you to use advanced features like message editing/deleting and uploads | ||||
| #You don't need to configure this, if you have configured the settings  | ||||
| #above. | ||||
|  | ||||
| #Url is your incoming webhook url as specified in rocketchat | ||||
| #Read #https://rocket.chat/docs/administrator-guides/integrations/#how-to-create-a-new-incoming-webhook | ||||
| #See administration - integrations - new integration - incoming webhook | ||||
| @@ -926,6 +1036,8 @@ NoTLS=false | ||||
| #OPTIONAL (default false) | ||||
| SkipTLSVerify=true | ||||
|  | ||||
| #### End settings for webhook matterbridge. | ||||
|  | ||||
| ## RELOADABLE SETTINGS | ||||
| ## Settings below can be reloaded by editing the file | ||||
|  | ||||
| @@ -937,6 +1049,7 @@ SkipTLSVerify=true | ||||
| PrefixMessagesWithNick=false | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="ircspammer1 ircspammer2" | ||||
| @@ -964,20 +1077,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -1027,6 +1147,7 @@ NoHomeServerSuffix=false | ||||
| PrefixMessagesWithNick=false | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="spammer1 spammer2" | ||||
| @@ -1054,20 +1175,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -1111,6 +1239,7 @@ Authcode="ABCE12" | ||||
| PrefixMessagesWithNick=false | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="spammer1 spammer2" | ||||
| @@ -1138,20 +1267,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -1165,10 +1301,45 @@ StripNick=false | ||||
| #OPTIONAL (default false) | ||||
| ShowTopicChange=false | ||||
|  | ||||
|  | ||||
|  | ||||
| ################################################################### | ||||
| #zulip section | ||||
| # | ||||
| # WhatsApp | ||||
| # | ||||
| ################################################################### | ||||
|  | ||||
| [whatsapp.bridge] | ||||
|  | ||||
| # Number you will use as a relay bot. Tip: Get some disposable sim card, don't rely on your own number. | ||||
| Number="+48111222333" | ||||
|  | ||||
| # First time that you login you will need to scan QR code, then credentials willl be saved in a session file | ||||
| # If you won't set SessionFile then you will need to scan QR code on every restart | ||||
| # optional (by default the session is stored only in memory, till restarting matterbridge) | ||||
| SessionFile="session-48111222333.gob" | ||||
|  | ||||
| # If your terminal is white we need to invert QR code in order for it to be scanned properly | ||||
| # optional (default false) | ||||
| QrOnWhiteTerminal=true | ||||
|  | ||||
| # Messages will be seen by other WhatsApp contacts as coming from the bridge. Original nick will be part of the message. | ||||
| RemoteNickFormat="@{NICK}: " | ||||
|  | ||||
| # extra label that can be used in the RemoteNickFormat | ||||
| # optional (default empty) | ||||
| Label="Organization" | ||||
|  | ||||
|  | ||||
|  | ||||
| ################################################################### | ||||
| # | ||||
| # zulip | ||||
| # | ||||
| ################################################################### | ||||
|  | ||||
| [zulip] | ||||
|  | ||||
| #You can configure multiple servers "[zulip.name]" or "[zulip.name2]" | ||||
| #In this example we use [zulip.streamchat] | ||||
| #REQUIRED | ||||
| @@ -1189,12 +1360,15 @@ Server="https://yourserver.zulipchat.com" | ||||
|  | ||||
| #Topic of the messages matterbridge will use | ||||
| #OPTIONAL (default "matterbridge") | ||||
| #You can specify a specific topic for each channel using [gateway.inout.options] | ||||
| #See more information below at the gateway configuration | ||||
| Topic="matterbridge" | ||||
|  | ||||
| ## RELOADABLE SETTINGS | ||||
| ## Settings below can be reloaded by editing the file | ||||
|  | ||||
| #Nicks you want to ignore.  | ||||
| #Regular expressions supported | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| IgnoreNicks="spammer1 spammer2" | ||||
| @@ -1222,20 +1396,27 @@ ReplaceMessages=[ ["cat","dog"] ] | ||||
| #optional (default empty) | ||||
| ReplaceNicks=[ ["user--","user"] ] | ||||
|  | ||||
| #Extractnicks is used to for example rewrite messages from other relaybots | ||||
| #See https://github.com/42wim/matterbridge/issues/713 and https://github.com/42wim/matterbridge/issues/466 | ||||
| #some examples: | ||||
| #this replaces a message like "Relaybot: <relayeduser> something interesting" to "relayeduser: something interesting" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ] ] | ||||
| #you can use multiple entries for multiplebots | ||||
| #this also replaces a message like "otherbot: (relayeduser) something else" to "relayeduser: something else" | ||||
| #ExtractNicks=[ [ "Relaybot", "<(.*?)>\\s+" ],[ "otherbot","\\((.*?)\\)\\s+" ] | ||||
| #OPTIONAL (default empty) | ||||
| ExtractNicks=[ ["otherbot","<(.*?)>\\s+" ] ] | ||||
|  | ||||
| #extra label that can be used in the RemoteNickFormat | ||||
| #optional (default empty) | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| #Enable to show users joins/parts from other bridges  | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack | ||||
| #Currently works for messages from the following bridges: irc, mattermost, slack, discord | ||||
| #OPTIONAL (default false) | ||||
| ShowJoinPart=false | ||||
|  | ||||
| @@ -1263,6 +1444,7 @@ ShowTopicChange=false | ||||
| BindAddress="127.0.0.1:4242" | ||||
|  | ||||
| #Amount of messages to keep in memory | ||||
| #OPTIONAL (library default 10) | ||||
| Buffer=1000 | ||||
|  | ||||
| #Bearer token used for authentication | ||||
| @@ -1275,11 +1457,7 @@ Token="mytoken" | ||||
| Label="" | ||||
|  | ||||
| #RemoteNickFormat defines how remote users appear on this bridge  | ||||
| #The string "{NICK}" (case sensitive) will be replaced by the actual nick / username. | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| #See [general] config section for default options | ||||
| RemoteNickFormat="{NICK}" | ||||
|  | ||||
|  | ||||
| @@ -1298,6 +1476,8 @@ RemoteNickFormat="{NICK}" | ||||
| #The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge | ||||
| #The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge | ||||
| #The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge | ||||
| #The string "{GATEWAY}" (case sensitive) will be replaced by the origin gateway name that is replicating the message. | ||||
| #The string "{CHANNEL}" (case sensitive) will be replaced by the origin channel name used by the bridge | ||||
| #OPTIONAL (default empty) | ||||
| RemoteNickFormat="[{PROTOCOL}] <{NICK}> " | ||||
|  | ||||
| @@ -1341,6 +1521,35 @@ MediaDownloadSize=1000000 | ||||
| #OPTIONAL (default empty) | ||||
| MediaDownloadBlacklist=[".html$",".htm$"] | ||||
|  | ||||
| #IgnoreFailureOnStart allows you to ignore failing bridges on startup. | ||||
| #Matterbridge will disable the failed bridge and continue with the other ones. | ||||
| #Context: https://github.com/42wim/matterbridge/issues/455 | ||||
| #OPTIONAL (default false) | ||||
| IgnoreFailureOnStart=false | ||||
|  | ||||
|  | ||||
| #TengoModifyMessage allows you to specify the location of a tengo (https://github.com/d5/tengo/) script. | ||||
| #This script will receive every incoming message and can be used to modify the Username and the Text of that message. | ||||
| #The script will have the following global variables: | ||||
| #to modify: msgUsername and msgText | ||||
| #to read: msgChannel and msgAccount | ||||
| # | ||||
| #The script is reloaded on every message, so you can modify the script on the fly. | ||||
| # | ||||
| #Example script can be found in https://github.com/42wim/matterbridge/tree/master/gateway/bench.tengo | ||||
| #and https://github.com/42wim/matterbridge/tree/master/contrib/example.tengo | ||||
| # | ||||
| #The example below will check if the text contains blah and if so, it'll replace the text and the username of that message. | ||||
| #text := import("text") | ||||
| #if text.re_match("blah",msgText) { | ||||
| #    msgText="replaced by this" | ||||
| #    msgUsername="fakeuser" | ||||
| #} | ||||
| #More information about tengo on: https://github.com/d5/tengo/blob/master/docs/tutorial.md and | ||||
| #https://github.com/d5/tengo/blob/master/docs/stdlib.md | ||||
| #OPTIONAL (default empty) | ||||
| TengoModifyMessage="example.tengo" | ||||
|  | ||||
| ################################################################### | ||||
| #Gateway configuration | ||||
| ################################################################### | ||||
| @@ -1362,36 +1571,41 @@ name="gateway1" | ||||
| ##OPTIONAL (default false) | ||||
| enable=true | ||||
|  | ||||
|     #[[gateway.in]] specifies the account and channels we will receive messages from. | ||||
|     #The following example bridges between mattermost and irc | ||||
|     # [[gateway.in]] specifies the account and channels we will receive messages from. | ||||
|     # The following example bridges between mattermost and irc | ||||
|     [[gateway.in]] | ||||
|  | ||||
|     #account specified above | ||||
|     #REQUIRED | ||||
|     # account specified above | ||||
|     # REQUIRED | ||||
|     account="irc.freenode" | ||||
|     #channel to connect on that account | ||||
|     #How to specify them for the different bridges: | ||||
|  | ||||
|     # channel to connect on that account | ||||
|     # How to specify them for the different bridges: | ||||
|     # | ||||
|     #irc        - #channel (# is required) (this needs to be lowercase!) | ||||
|     #mattermost - channel (the channel name as seen in the URL, not the displayname) | ||||
|     #gitter     - username/room  | ||||
|     #xmpp       - channel | ||||
|     #slack      - channel (without the #) | ||||
|     #           - ID:C123456 (where C123456 is the channel ID) does not work with webhook | ||||
|     #discord    - channel (without the #) | ||||
|     #           - ID:123456789 (where 123456789 is the channel ID)  | ||||
|     # irc        - #channel (# is required) (this needs to be lowercase!) | ||||
|     # mattermost - channel (the channel name as seen in the URL, not the displayname) | ||||
|     # gitter     - username/room | ||||
|     # xmpp       - channel | ||||
|     # slack      - channel (without the #) | ||||
|     #            - ID:C123456 (where C123456 is the channel ID) does not work with webhook | ||||
|     # discord    - channel (without the #) | ||||
|     #            - ID:123456789 (where 123456789 is the channel ID) | ||||
|     #               (https://github.com/42wim/matterbridge/issues/57) | ||||
|     #telegram   - chatid (a large negative number, eg -123456789) | ||||
|     # telegram   - chatid (a large negative number, eg -123456789) | ||||
|     #             see (https://www.linkedin.com/pulse/telegram-bots-beginners-marco-frau) | ||||
|     #hipchat    - id_channel (see https://www.hipchat.com/account/xmpp for the correct channel) | ||||
|     #rocketchat - #channel (# is required (also needed for private channels!) | ||||
|     #matrix     - #channel:server (eg #yourchannel:matrix.org)  | ||||
|     #           - encrypted rooms are not supported in matrix | ||||
|     #steam      - chatid (a large number).  | ||||
|     # hipchat    - id_channel (see https://www.hipchat.com/account/xmpp for the correct channel) | ||||
|     # rocketchat - #channel (# is required (also needed for private channels!) | ||||
|     # matrix     - #channel:server (eg #yourchannel:matrix.org) | ||||
|     #            - encrypted rooms are not supported in matrix | ||||
|     # steam      - chatid (a large number). | ||||
|     #             The number in the URL when you click "enter chat room" in the browser | ||||
|     #zulip      - stream (without the #) | ||||
|     # whatsapp   - 48111222333-123455678999@g.us A unique group JID; | ||||
|     #              if you specify an empty string bridge will list all the possibilities | ||||
|     #            - "Group Name" if you specify a group name the bridge will hint its JID to specify | ||||
|     #              as group names might change in time and contain weird emoticons | ||||
|     # zulip      - stream (without the #) | ||||
|     #                   | ||||
|     #REQUIRED | ||||
|     # REQUIRED | ||||
|     channel="#testing" | ||||
|  | ||||
|         #OPTIONAL - only used for IRC and XMPP protocols at the moment | ||||
| @@ -1427,7 +1641,14 @@ enable=true | ||||
|  | ||||
|         #OPTIONAL - webhookurl only works for discord (it needs a different URL for each cahnnel) | ||||
|         [gateway.inout.options] | ||||
|         webhookurl=""https://discordapp.com/api/webhooks/123456789123456789/C9WPqExYWONPDZabcdef-def1434FGFjstasJX9pYht73y" | ||||
|         webhookurl="https://discordapp.com/api/webhooks/123456789123456789/C9WPqExYWONPDZabcdef-def1434FGFjstasJX9pYht73y" | ||||
|  | ||||
|     [[gateway.inout]] | ||||
|     account="zulip.streamchat" | ||||
|     channel="general" | ||||
|         #OPTIONAL - topic only works for zulip | ||||
|         [gateway.inout.options] | ||||
|         topic="topic1" | ||||
|  | ||||
|     #API example | ||||
|     #[[gateway.inout]] | ||||
|   | ||||
							
								
								
									
										207
									
								
								matterclient/channels.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,207 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| // GetChannels returns all channels we're members off | ||||
| func (m *MMClient) GetChannels() []*model.Channel { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	// our primary team channels first | ||||
| 	channels = append(channels, m.Team.Channels...) | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id != m.Team.Id { | ||||
| 			channels = append(channels, t.Channels...) | ||||
| 		} | ||||
| 	} | ||||
| 	return channels | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelHeader(channelId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Id == channelId { | ||||
| 				return channel.Header | ||||
| 			} | ||||
|  | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelId(name string, teamId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	if teamId != "" { | ||||
| 		return m.getChannelIdTeam(name, teamId) | ||||
| 	} | ||||
|  | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Type == model.CHANNEL_GROUP { | ||||
| 				res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||
| 				res = strings.Replace(res, " ", "_", -1) | ||||
| 				if res == name { | ||||
| 					return channel.Id | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) getChannelIdTeam(name string, teamId string) string { //nolint:golint | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id == teamId { | ||||
| 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 				if channel.Name == name { | ||||
| 					return channel.Id | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelName(channelId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t == nil { | ||||
| 			continue | ||||
| 		} | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Id == channelId { | ||||
| 				if channel.Type == model.CHANNEL_GROUP { | ||||
| 					res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||
| 					res = strings.Replace(res, " ", "_", -1) | ||||
| 					return res | ||||
| 				} | ||||
| 				return channel.Name | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelTeamId(id string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range append(m.OtherTeams, m.Team) { | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Id == id { | ||||
| 				return channel.TeamId | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetLastViewedAt(channelId string) int64 { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return model.GetMillis() | ||||
| 	} | ||||
| 	return res.LastViewedAt | ||||
| } | ||||
|  | ||||
| // GetMoreChannels returns existing channels where we're not a member off. | ||||
| func (m *MMClient) GetMoreChannels() []*model.Channel { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		channels = append(channels, t.MoreChannels...) | ||||
| 	} | ||||
| 	return channels | ||||
| } | ||||
|  | ||||
| // GetTeamFromChannel returns teamId belonging to channel (DM channels have no teamId). | ||||
| func (m *MMClient) GetTeamFromChannel(channelId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		channels = append(channels, t.Channels...) | ||||
| 		if t.MoreChannels != nil { | ||||
| 			channels = append(channels, t.MoreChannels...) | ||||
| 		} | ||||
| 		for _, c := range channels { | ||||
| 			if c.Id == channelId { | ||||
| 				if c.Type == model.CHANNEL_GROUP { | ||||
| 					return "G" | ||||
| 				} | ||||
| 				return t.Id | ||||
| 			} | ||||
| 		} | ||||
| 		channels = nil | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) JoinChannel(channelId string) error { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, c := range m.Team.Channels { | ||||
| 		if c.Id == channelId { | ||||
| 			m.logger.Debug("Not joining ", channelId, " already joined.") | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
| 	m.logger.Debug("Joining ", channelId) | ||||
| 	_, resp := m.Client.AddChannelMember(channelId, m.User.Id) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateChannels() error { | ||||
| 	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
| 	m.Lock() | ||||
| 	m.Team.Channels = mmchannels | ||||
| 	m.Unlock() | ||||
|  | ||||
| 	mmchannels, resp = m.Client.GetPublicChannelsForTeam(m.Team.Id, 0, 5000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
|  | ||||
| 	m.Lock() | ||||
| 	m.Team.MoreChannels = mmchannels | ||||
| 	m.Unlock() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateChannelHeader(channelId string, header string) { //nolint:golint | ||||
| 	channel := &model.Channel{Id: channelId, Header: header} | ||||
| 	m.logger.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||
| 	_, resp := m.Client.UpdateChannel(channel) | ||||
| 	if resp.Error != nil { | ||||
| 		m.logger.Error(resp.Error) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateLastViewed(channelId string) error { //nolint:golint | ||||
| 	m.logger.Debugf("posting lastview %#v", channelId) | ||||
| 	view := &model.ChannelView{ChannelId: channelId} | ||||
| 	_, resp := m.Client.ViewChannel(m.User.Id, view) | ||||
| 	if resp.Error != nil { | ||||
| 		m.logger.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error) | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										282
									
								
								matterclient/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,282 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"crypto/md5" //nolint:gosec | ||||
| 	"crypto/tls" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"net/http" | ||||
| 	"net/http/cookiejar" | ||||
| 	"net/url" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/gorilla/websocket" | ||||
| 	"github.com/jpillora/backoff" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| func (m *MMClient) doLogin(firstConnection bool, b *backoff.Backoff) error { | ||||
| 	var resp *model.Response | ||||
| 	var appErr *model.AppError | ||||
| 	var logmsg = "trying login" | ||||
| 	var err error | ||||
| 	for { | ||||
| 		m.logger.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | ||||
| 		if m.Credentials.Token != "" { | ||||
| 			resp, err = m.doLoginToken() | ||||
| 			if err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} else { | ||||
| 			m.User, resp = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | ||||
| 		} | ||||
| 		appErr = resp.Error | ||||
| 		if appErr != nil { | ||||
| 			d := b.Duration() | ||||
| 			m.logger.Debug(appErr.DetailedError) | ||||
| 			if firstConnection { | ||||
| 				if appErr.Message == "" { | ||||
| 					return errors.New(appErr.DetailedError) | ||||
| 				} | ||||
| 				return errors.New(appErr.Message) | ||||
| 			} | ||||
| 			m.logger.Debugf("LOGIN: %s, reconnecting in %s", appErr, d) | ||||
| 			time.Sleep(d) | ||||
| 			logmsg = "retrying login" | ||||
| 			continue | ||||
| 		} | ||||
| 		break | ||||
| 	} | ||||
| 	// reset timer | ||||
| 	b.Reset() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) doLoginToken() (*model.Response, error) { | ||||
| 	var resp *model.Response | ||||
| 	var logmsg = "trying login" | ||||
| 	m.Client.AuthType = model.HEADER_BEARER | ||||
| 	m.Client.AuthToken = m.Credentials.Token | ||||
| 	if m.Credentials.CookieToken { | ||||
| 		m.logger.Debugf(logmsg + " with cookie (MMAUTH) token") | ||||
| 		m.Client.HttpClient.Jar = m.createCookieJar(m.Credentials.Token) | ||||
| 	} else { | ||||
| 		m.logger.Debugf(logmsg + " with personal token") | ||||
| 	} | ||||
| 	m.User, resp = m.Client.GetMe("") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp, resp.Error | ||||
| 	} | ||||
| 	if m.User == nil { | ||||
| 		m.logger.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | ||||
| 		return resp, errors.New("invalid token") | ||||
| 	} | ||||
| 	return resp, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) handleLoginToken() error { | ||||
| 	switch { | ||||
| 	case strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN): | ||||
| 		token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||
| 		if len(token) != 2 { | ||||
| 			return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||
| 		} | ||||
| 		m.Credentials.Token = token[1] | ||||
| 		m.Credentials.CookieToken = true | ||||
| 	case strings.Contains(m.Credentials.Pass, "token="): | ||||
| 		token := strings.Split(m.Credentials.Pass, "token=") | ||||
| 		if len(token) != 2 { | ||||
| 			return errors.New("incorrect personal token. valid input is token=yourtoken") | ||||
| 		} | ||||
| 		m.Credentials.Token = token[1] | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) initClient(firstConnection bool, b *backoff.Backoff) error { | ||||
| 	uriScheme := "https://" | ||||
| 	if m.NoTLS { | ||||
| 		uriScheme = "http://" | ||||
| 	} | ||||
| 	// login to mattermost | ||||
| 	m.Client = model.NewAPIv4Client(uriScheme + m.Credentials.Server) | ||||
| 	m.Client.HttpClient.Transport = &http.Transport{ | ||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, //nolint:gosec | ||||
| 		Proxy:           http.ProxyFromEnvironment, | ||||
| 	} | ||||
| 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||
|  | ||||
| 	// handle MMAUTHTOKEN and personal token | ||||
| 	if err := m.handleLoginToken(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	// check if server alive, retry until | ||||
| 	if err := m.serverAlive(firstConnection, b); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // initialize user and teams | ||||
| func (m *MMClient) initUser() error { | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	// we only load all team data on initial login. | ||||
| 	// all other updates are for channels from our (primary) team only. | ||||
| 	//m.logger.Debug("initUser(): loading all team data") | ||||
| 	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	for _, team := range teams { | ||||
| 		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return errors.New(resp.Error.DetailedError) | ||||
| 		} | ||||
| 		usermap := make(map[string]*model.User) | ||||
| 		for _, user := range mmusers { | ||||
| 			usermap[user.Id] = user | ||||
| 		} | ||||
|  | ||||
| 		t := &Team{Team: team, Users: usermap, Id: team.Id} | ||||
|  | ||||
| 		mmchannels, resp := m.Client.GetChannelsForTeamForUser(team.Id, m.User.Id, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return resp.Error | ||||
| 		} | ||||
| 		t.Channels = mmchannels | ||||
| 		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return resp.Error | ||||
| 		} | ||||
| 		t.MoreChannels = mmchannels | ||||
| 		m.OtherTeams = append(m.OtherTeams, t) | ||||
| 		if team.Name == m.Credentials.Team { | ||||
| 			m.Team = t | ||||
| 			m.logger.Debugf("initUser(): found our team %s (id: %s)", team.Name, team.Id) | ||||
| 		} | ||||
| 		// add all users | ||||
| 		for k, v := range t.Users { | ||||
| 			m.Users[k] = v | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) serverAlive(firstConnection bool, b *backoff.Backoff) error { | ||||
| 	defer b.Reset() | ||||
| 	for { | ||||
| 		d := b.Duration() | ||||
| 		// bogus call to get the serverversion | ||||
| 		_, resp := m.Client.Logout() | ||||
| 		if resp.Error != nil { | ||||
| 			return fmt.Errorf("%#v", resp.Error.Error()) | ||||
| 		} | ||||
| 		if firstConnection && !supportedVersion(resp.ServerVersion) { | ||||
| 			return fmt.Errorf("unsupported mattermost version: %s", resp.ServerVersion) | ||||
| 		} | ||||
| 		m.ServerVersion = resp.ServerVersion | ||||
| 		if m.ServerVersion == "" { | ||||
| 			m.logger.Debugf("Server not up yet, reconnecting in %s", d) | ||||
| 			time.Sleep(d) | ||||
| 		} else { | ||||
| 			m.logger.Infof("Found version %s", m.ServerVersion) | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) wsConnect() { | ||||
| 	b := &backoff.Backoff{ | ||||
| 		Min:    time.Second, | ||||
| 		Max:    5 * time.Minute, | ||||
| 		Jitter: true, | ||||
| 	} | ||||
|  | ||||
| 	m.WsConnected = false | ||||
| 	wsScheme := "wss://" | ||||
| 	if m.NoTLS { | ||||
| 		wsScheme = "ws://" | ||||
| 	} | ||||
|  | ||||
| 	// setup websocket connection | ||||
| 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V4 + "/websocket" | ||||
| 	header := http.Header{} | ||||
| 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||
|  | ||||
| 	m.logger.Debugf("WsClient: making connection: %s", wsurl) | ||||
| 	for { | ||||
| 		wsDialer := &websocket.Dialer{ | ||||
| 			TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, //nolint:gosec | ||||
| 			Proxy:           http.ProxyFromEnvironment, | ||||
| 		} | ||||
| 		var err error | ||||
| 		m.WsClient, _, err = wsDialer.Dial(wsurl, header) | ||||
| 		if err != nil { | ||||
| 			d := b.Duration() | ||||
| 			m.logger.Debugf("WSS: %s, reconnecting in %s", err, d) | ||||
| 			time.Sleep(d) | ||||
| 			continue | ||||
| 		} | ||||
| 		break | ||||
| 	} | ||||
|  | ||||
| 	m.logger.Debug("WsClient: connected") | ||||
| 	m.WsSequence = 1 | ||||
| 	m.WsPingChan = make(chan *model.WebSocketResponse) | ||||
| 	// only start to parse WS messages when login is completely done | ||||
| 	m.WsConnected = true | ||||
| } | ||||
|  | ||||
| func (m *MMClient) createCookieJar(token string) *cookiejar.Jar { | ||||
| 	var cookies []*http.Cookie | ||||
| 	jar, _ := cookiejar.New(nil) | ||||
| 	firstCookie := &http.Cookie{ | ||||
| 		Name:   "MMAUTHTOKEN", | ||||
| 		Value:  token, | ||||
| 		Path:   "/", | ||||
| 		Domain: m.Credentials.Server, | ||||
| 	} | ||||
| 	cookies = append(cookies, firstCookie) | ||||
| 	cookieURL, _ := url.Parse("https://" + m.Credentials.Server) | ||||
| 	jar.SetCookies(cookieURL, cookies) | ||||
| 	return jar | ||||
| } | ||||
|  | ||||
| func (m *MMClient) checkAlive() error { | ||||
| 	// check if session still is valid | ||||
| 	_, resp := m.Client.GetMe("") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	m.logger.Debug("WS PING") | ||||
| 	return m.sendWSRequest("ping", nil) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error { | ||||
| 	req := &model.WebSocketRequest{} | ||||
| 	req.Seq = m.WsSequence | ||||
| 	req.Action = action | ||||
| 	req.Data = data | ||||
| 	m.WsSequence++ | ||||
| 	m.logger.Debugf("sendWsRequest %#v", req) | ||||
| 	return m.WsClient.WriteJSON(req) | ||||
| } | ||||
|  | ||||
| func supportedVersion(version string) bool { | ||||
| 	if strings.HasPrefix(version, "3.8.0") || | ||||
| 		strings.HasPrefix(version, "3.9.0") || | ||||
| 		strings.HasPrefix(version, "3.10.0") || | ||||
| 		strings.HasPrefix(version, "4.") || | ||||
| 		strings.HasPrefix(version, "5.") { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func digestString(s string) string { | ||||
| 	return fmt.Sprintf("%x", md5.Sum([]byte(s))) //nolint:gosec | ||||
| } | ||||
| @@ -1,31 +1,26 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"crypto/md5" | ||||
| 	"crypto/tls" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"net/http" | ||||
| 	"net/http/cookiejar" | ||||
| 	"net/url" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
|  | ||||
| 	"github.com/gorilla/websocket" | ||||
| 	"github.com/hashicorp/golang-lru" | ||||
| 	lru "github.com/hashicorp/golang-lru" | ||||
| 	"github.com/jpillora/backoff" | ||||
| 	"github.com/mattermost/platform/model" | ||||
| 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Credentials struct { | ||||
| 	Login         string | ||||
| 	Team          string | ||||
| 	Pass          string | ||||
| 	Token         string | ||||
| 	CookieToken   bool | ||||
| 	Server        string | ||||
| 	NoTLS         bool | ||||
| 	SkipTLSVerify bool | ||||
| @@ -42,6 +37,7 @@ type Message struct { | ||||
| 	UserID   string | ||||
| } | ||||
|  | ||||
| //nolint:golint | ||||
| type Team struct { | ||||
| 	Team         *model.Team | ||||
| 	Id           string | ||||
| @@ -53,13 +49,13 @@ type Team struct { | ||||
| type MMClient struct { | ||||
| 	sync.RWMutex | ||||
| 	*Credentials | ||||
|  | ||||
| 	Team          *Team | ||||
| 	OtherTeams    []*Team | ||||
| 	Client        *model.Client4 | ||||
| 	User          *model.User | ||||
| 	Users         map[string]*model.User | ||||
| 	MessageChan   chan *Message | ||||
| 	log           *log.Entry | ||||
| 	WsClient      *websocket.Conn | ||||
| 	WsQuit        bool | ||||
| 	WsAway        bool | ||||
| @@ -68,31 +64,61 @@ type MMClient struct { | ||||
| 	WsPingChan    chan *model.WebSocketResponse | ||||
| 	ServerVersion string | ||||
| 	OnWsConnect   func() | ||||
| 	lruCache      *lru.Cache | ||||
|  | ||||
| 	logger     *logrus.Entry | ||||
| 	rootLogger *logrus.Logger | ||||
| 	lruCache   *lru.Cache | ||||
| } | ||||
|  | ||||
| func New(login, pass, team, server string) *MMClient { | ||||
| 	cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server} | ||||
| 	mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)} | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true}) | ||||
| 	mmclient.log = log.WithFields(log.Fields{"prefix": "matterclient"}) | ||||
| 	mmclient.lruCache, _ = lru.New(500) | ||||
| 	return mmclient | ||||
| } | ||||
| // New will instantiate a new Matterclient with the specified login details without connecting. | ||||
| func New(login string, pass string, team string, server string) *MMClient { | ||||
| 	rootLogger := logrus.New() | ||||
| 	rootLogger.SetFormatter(&prefixed.TextFormatter{ | ||||
| 		PrefixPadding: 13, | ||||
| 		DisableColors: true, | ||||
| 	}) | ||||
|  | ||||
| func (m *MMClient) SetDebugLog() { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SetLogLevel(level string) { | ||||
| 	l, err := log.ParseLevel(level) | ||||
| 	if err != nil { | ||||
| 		log.SetLevel(log.InfoLevel) | ||||
| 		return | ||||
| 	cred := &Credentials{ | ||||
| 		Login:  login, | ||||
| 		Pass:   pass, | ||||
| 		Team:   team, | ||||
| 		Server: server, | ||||
| 	} | ||||
|  | ||||
| 	cache, _ := lru.New(500) | ||||
| 	return &MMClient{ | ||||
| 		Credentials: cred, | ||||
| 		MessageChan: make(chan *Message, 100), | ||||
| 		Users:       make(map[string]*model.User), | ||||
| 		rootLogger:  rootLogger, | ||||
| 		lruCache:    cache, | ||||
| 		logger:      rootLogger.WithFields(logrus.Fields{"prefix": "matterclient"}), | ||||
| 	} | ||||
| 	log.SetLevel(l) | ||||
| } | ||||
|  | ||||
| // SetDebugLog activates debugging logging on all Matterclient log output. | ||||
| func (m *MMClient) SetDebugLog() { | ||||
| 	m.rootLogger.SetFormatter(&prefixed.TextFormatter{ | ||||
| 		PrefixPadding:   13, | ||||
| 		DisableColors:   true, | ||||
| 		FullTimestamp:   false, | ||||
| 		ForceFormatting: true, | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| // SetLogLevel tries to parse the specified level and if successful sets | ||||
| // the log level accordingly. Accepted levels are: 'debug', 'info', 'warn', | ||||
| // 'error', 'fatal' and 'panic'. | ||||
| func (m *MMClient) SetLogLevel(level string) { | ||||
| 	l, err := logrus.ParseLevel(level) | ||||
| 	if err != nil { | ||||
| 		m.logger.Warnf("Failed to parse specified log-level '%s': %#v", level, err) | ||||
| 	} else { | ||||
| 		m.rootLogger.SetLevel(l) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Login tries to connect the client with the loging details with which it was initialized. | ||||
| func (m *MMClient) Login() error { | ||||
| 	// check if this is a first connect or a reconnection | ||||
| 	firstConnection := true | ||||
| @@ -108,84 +134,17 @@ func (m *MMClient) Login() error { | ||||
| 		Max:    5 * time.Minute, | ||||
| 		Jitter: true, | ||||
| 	} | ||||
| 	uriScheme := "https://" | ||||
| 	if m.NoTLS { | ||||
| 		uriScheme = "http://" | ||||
| 	} | ||||
| 	// login to mattermost | ||||
| 	m.Client = model.NewAPIv4Client(uriScheme + m.Credentials.Server) | ||||
| 	m.Client.HttpClient.Transport = &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}, Proxy: http.ProxyFromEnvironment} | ||||
| 	m.Client.HttpClient.Timeout = time.Second * 10 | ||||
|  | ||||
| 	for { | ||||
| 		d := b.Duration() | ||||
| 		// bogus call to get the serverversion | ||||
| 		_, resp := m.Client.Logout() | ||||
| 		if resp.Error != nil { | ||||
| 			return fmt.Errorf("%#v", resp.Error.Error()) | ||||
| 		} | ||||
| 		if firstConnection && !supportedVersion(resp.ServerVersion) { | ||||
| 			return fmt.Errorf("unsupported mattermost version: %s", resp.ServerVersion) | ||||
| 		} | ||||
| 		m.ServerVersion = resp.ServerVersion | ||||
| 		if m.ServerVersion == "" { | ||||
| 			m.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||
| 			time.Sleep(d) | ||||
| 		} else { | ||||
| 			m.log.Infof("Found version %s", m.ServerVersion) | ||||
| 			break | ||||
| 		} | ||||
| 	// do initialization setup | ||||
| 	if err := m.initClient(firstConnection, b); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	b.Reset() | ||||
|  | ||||
| 	var resp *model.Response | ||||
| 	//var myinfo *model.Result | ||||
| 	var appErr *model.AppError | ||||
| 	var logmsg = "trying login" | ||||
| 	for { | ||||
| 		m.log.Debugf("%s %s %s %s", logmsg, m.Credentials.Team, m.Credentials.Login, m.Credentials.Server) | ||||
| 		if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||
| 			m.log.Debugf(logmsg + " with token") | ||||
| 			token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||
| 			if len(token) != 2 { | ||||
| 				return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||
| 			} | ||||
| 			m.Client.HttpClient.Jar = m.createCookieJar(token[1]) | ||||
| 			m.Client.AuthToken = token[1] | ||||
| 			m.Client.AuthType = model.HEADER_BEARER | ||||
| 			m.User, resp = m.Client.GetMe("") | ||||
| 			if resp.Error != nil { | ||||
| 				return resp.Error | ||||
| 			} | ||||
| 			if m.User == nil { | ||||
| 				m.log.Errorf("LOGIN TOKEN: %s is invalid", m.Credentials.Pass) | ||||
| 				return errors.New("invalid " + model.SESSION_COOKIE_TOKEN) | ||||
| 			} | ||||
| 		} else { | ||||
| 			m.User, resp = m.Client.Login(m.Credentials.Login, m.Credentials.Pass) | ||||
| 		} | ||||
| 		appErr = resp.Error | ||||
| 		if appErr != nil { | ||||
| 			d := b.Duration() | ||||
| 			m.log.Debug(appErr.DetailedError) | ||||
| 			if firstConnection { | ||||
| 				if appErr.Message == "" { | ||||
| 					return errors.New(appErr.DetailedError) | ||||
| 				} | ||||
| 				return errors.New(appErr.Message) | ||||
| 			} | ||||
| 			m.log.Debugf("LOGIN: %s, reconnecting in %s", appErr, d) | ||||
| 			time.Sleep(d) | ||||
| 			logmsg = "retrying login" | ||||
| 			continue | ||||
| 		} | ||||
| 		break | ||||
| 	if err := m.doLogin(firstConnection, b); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	// reset timer | ||||
| 	b.Reset() | ||||
|  | ||||
| 	err := m.initUser() | ||||
| 	if err != nil { | ||||
| 	if err := m.initUser(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| @@ -202,52 +161,14 @@ func (m *MMClient) Login() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) wsConnect() { | ||||
| 	b := &backoff.Backoff{ | ||||
| 		Min:    time.Second, | ||||
| 		Max:    5 * time.Minute, | ||||
| 		Jitter: true, | ||||
| 	} | ||||
|  | ||||
| 	m.WsConnected = false | ||||
| 	wsScheme := "wss://" | ||||
| 	if m.NoTLS { | ||||
| 		wsScheme = "ws://" | ||||
| 	} | ||||
|  | ||||
| 	// setup websocket connection | ||||
| 	wsurl := wsScheme + m.Credentials.Server + model.API_URL_SUFFIX_V4 + "/websocket" | ||||
| 	header := http.Header{} | ||||
| 	header.Set(model.HEADER_AUTH, "BEARER "+m.Client.AuthToken) | ||||
|  | ||||
| 	m.log.Debugf("WsClient: making connection: %s", wsurl) | ||||
| 	for { | ||||
| 		wsDialer := &websocket.Dialer{Proxy: http.ProxyFromEnvironment, TLSClientConfig: &tls.Config{InsecureSkipVerify: m.SkipTLSVerify}} | ||||
| 		var err error | ||||
| 		m.WsClient, _, err = wsDialer.Dial(wsurl, header) | ||||
| 		if err != nil { | ||||
| 			d := b.Duration() | ||||
| 			m.log.Debugf("WSS: %s, reconnecting in %s", err, d) | ||||
| 			time.Sleep(d) | ||||
| 			continue | ||||
| 		} | ||||
| 		break | ||||
| 	} | ||||
|  | ||||
| 	m.log.Debug("WsClient: connected") | ||||
| 	m.WsSequence = 1 | ||||
| 	m.WsPingChan = make(chan *model.WebSocketResponse) | ||||
| 	// only start to parse WS messages when login is completely done | ||||
| 	m.WsConnected = true | ||||
| } | ||||
|  | ||||
| // Logout disconnects the client from the chat server. | ||||
| func (m *MMClient) Logout() error { | ||||
| 	m.log.Debugf("logout as %s (team: %s) on %s", m.Credentials.Login, m.Credentials.Team, m.Credentials.Server) | ||||
| 	m.logger.Debugf("logout as %s (team: %s) on %s", m.Credentials.Login, m.Credentials.Team, m.Credentials.Server) | ||||
| 	m.WsQuit = true | ||||
| 	m.WsClient.Close() | ||||
| 	m.WsClient.UnderlyingConn().Close() | ||||
| 	if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||
| 		m.log.Debug("Not invalidating session in logout, credential is a token") | ||||
| 		m.logger.Debug("Not invalidating session in logout, credential is a token") | ||||
| 		return nil | ||||
| 	} | ||||
| 	_, resp := m.Client.Logout() | ||||
| @@ -257,13 +178,16 @@ func (m *MMClient) Logout() error { | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // WsReceiver implements the core loop that manages the connection to the chat server. In | ||||
| // case of a disconnect it will try to reconnect. A call to this method is blocking until | ||||
| // the 'WsQuite' field of the MMClient object is set to 'true'. | ||||
| func (m *MMClient) WsReceiver() { | ||||
| 	for { | ||||
| 		var rawMsg json.RawMessage | ||||
| 		var err error | ||||
|  | ||||
| 		if m.WsQuit { | ||||
| 			m.log.Debug("exiting WsReceiver") | ||||
| 			m.logger.Debug("exiting WsReceiver") | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| @@ -273,14 +197,14 @@ func (m *MMClient) WsReceiver() { | ||||
| 		} | ||||
|  | ||||
| 		if _, rawMsg, err = m.WsClient.ReadMessage(); err != nil { | ||||
| 			m.log.Error("error:", err) | ||||
| 			m.logger.Error("error:", err) | ||||
| 			// reconnect | ||||
| 			m.wsConnect() | ||||
| 		} | ||||
|  | ||||
| 		var event model.WebSocketEvent | ||||
| 		if err := json.Unmarshal(rawMsg, &event); err == nil && event.IsValid() { | ||||
| 			m.log.Debugf("WsReceiver event: %#v", event) | ||||
| 			m.logger.Debugf("WsReceiver event: %#v", event) | ||||
| 			msg := &Message{Raw: &event, Team: m.Credentials.Team} | ||||
| 			m.parseMessage(msg) | ||||
| 			// check if we didn't empty the message | ||||
| @@ -299,570 +223,42 @@ func (m *MMClient) WsReceiver() { | ||||
|  | ||||
| 		var response model.WebSocketResponse | ||||
| 		if err := json.Unmarshal(rawMsg, &response); err == nil && response.IsValid() { | ||||
| 			m.log.Debugf("WsReceiver response: %#v", response) | ||||
| 			m.logger.Debugf("WsReceiver response: %#v", response) | ||||
| 			m.parseResponse(response) | ||||
| 			continue | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) parseMessage(rmsg *Message) { | ||||
| 	switch rmsg.Raw.Event { | ||||
| 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED, model.WEBSOCKET_EVENT_POST_DELETED: | ||||
| 		m.parseActionPost(rmsg) | ||||
| 	case "user_updated": | ||||
| 		user := rmsg.Raw.Data["user"].(map[string]interface{}) | ||||
| 		if _, ok := user["id"].(string); ok { | ||||
| 			m.UpdateUser(user["id"].(string)) | ||||
| 		} | ||||
| 		/* | ||||
| 			case model.ACTION_USER_REMOVED: | ||||
| 				m.handleWsActionUserRemoved(&rmsg) | ||||
| 			case model.ACTION_USER_ADDED: | ||||
| 				m.handleWsActionUserAdded(&rmsg) | ||||
| 		*/ | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) { | ||||
| 	if rmsg.Data != nil { | ||||
| 		// ping reply | ||||
| 		if rmsg.Data["text"].(string) == "pong" { | ||||
| 			m.WsPingChan <- &rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) parseActionPost(rmsg *Message) { | ||||
| 	// add post to cache, if it already exists don't relay this again. | ||||
| 	// this should fix reposts | ||||
| 	if ok, _ := m.lruCache.ContainsOrAdd(digestString(rmsg.Raw.Data["post"].(string)), true); ok { | ||||
| 		m.log.Debugf("message %#v in cache, not processing again", rmsg.Raw.Data["post"].(string)) | ||||
| 		rmsg.Text = "" | ||||
| 		return | ||||
| 	} | ||||
| 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | ||||
| 	// we don't have the user, refresh the userlist | ||||
| 	if m.GetUser(data.UserId) == nil { | ||||
| 		m.log.Infof("User %s is not known, ignoring message %s", data) | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg.Username = m.GetUserName(data.UserId) | ||||
| 	rmsg.Channel = m.GetChannelName(data.ChannelId) | ||||
| 	rmsg.UserID = data.UserId | ||||
| 	rmsg.Type = data.Type | ||||
| 	teamid, _ := rmsg.Raw.Data["team_id"].(string) | ||||
| 	// edit messsages have no team_id for some reason | ||||
| 	if teamid == "" { | ||||
| 		// we can find the team_id from the channelid | ||||
| 		teamid = m.GetChannelTeamId(data.ChannelId) | ||||
| 		rmsg.Raw.Data["team_id"] = teamid | ||||
| 	} | ||||
| 	if teamid != "" { | ||||
| 		rmsg.Team = m.GetTeamName(teamid) | ||||
| 	} | ||||
| 	// direct message | ||||
| 	if rmsg.Raw.Data["channel_type"] == "D" { | ||||
| 		rmsg.Channel = m.GetUser(data.UserId).Username | ||||
| 	} | ||||
| 	rmsg.Text = data.Message | ||||
| 	rmsg.Post = data | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUsers() error { | ||||
| 	mmusers, resp := m.Client.GetUsers(0, 50000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
| 	m.Lock() | ||||
| 	for _, user := range mmusers { | ||||
| 		m.Users[user.Id] = user | ||||
| 	} | ||||
| 	m.Unlock() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateChannels() error { | ||||
| 	mmchannels, resp := m.Client.GetChannelsForTeamForUser(m.Team.Id, m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
| 	m.Lock() | ||||
| 	m.Team.Channels = mmchannels | ||||
| 	m.Unlock() | ||||
|  | ||||
| 	mmchannels, resp = m.Client.GetPublicChannelsForTeam(m.Team.Id, 0, 5000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
|  | ||||
| 	m.Lock() | ||||
| 	m.Team.MoreChannels = mmchannels | ||||
| 	m.Unlock() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelName(channelId string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t == nil { | ||||
| 			continue | ||||
| 		} | ||||
| 		if t.Channels != nil { | ||||
| 			for _, channel := range t.Channels { | ||||
| 				if channel.Id == channelId { | ||||
| 					return channel.Name | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		if t.MoreChannels != nil { | ||||
| 			for _, channel := range t.MoreChannels { | ||||
| 				if channel.Id == channelId { | ||||
| 					return channel.Name | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelId(name string, teamId string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	if teamId == "" { | ||||
| 		teamId = m.Team.Id | ||||
| 	} | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id == teamId { | ||||
| 			for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 				if channel.Name == name { | ||||
| 					return channel.Id | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelTeamId(id string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range append(m.OtherTeams, m.Team) { | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Id == id { | ||||
| 				return channel.TeamId | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetChannelHeader(channelId string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		for _, channel := range append(t.Channels, t.MoreChannels...) { | ||||
| 			if channel.Id == channelId { | ||||
| 				return channel.Header | ||||
| 			} | ||||
|  | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) PostMessage(channelId string, text string) (string, error) { | ||||
| 	post := &model.Post{ChannelId: channelId, Message: text} | ||||
| 	res, resp := m.Client.CreatePost(post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) PostMessageWithFiles(channelId string, text string, fileIds []string) (string, error) { | ||||
| 	post := &model.Post{ChannelId: channelId, Message: text, FileIds: fileIds} | ||||
| 	res, resp := m.Client.CreatePost(post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) EditMessage(postId string, text string) (string, error) { | ||||
| 	post := &model.Post{Message: text} | ||||
| 	res, resp := m.Client.UpdatePost(postId, post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) DeleteMessage(postId string) error { | ||||
| 	_, resp := m.Client.DeletePost(postId) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) JoinChannel(channelId string) error { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, c := range m.Team.Channels { | ||||
| 		if c.Id == channelId { | ||||
| 			m.log.Debug("Not joining ", channelId, " already joined.") | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
| 	m.log.Debug("Joining ", channelId) | ||||
| 	_, resp := m.Client.AddChannelMember(channelId, m.User.Id) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { | ||||
| 	res, resp := m.Client.GetPostsSince(channelId, time) | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SearchPosts(query string) *model.PostList { | ||||
| 	res, resp := m.Client.SearchPosts(m.Team.Id, query, false) | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { | ||||
| 	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPublicLink(filename string) string { | ||||
| 	res, resp := m.Client.GetFileLink(filename) | ||||
| 	if resp.Error != nil { | ||||
| 		return "" | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPublicLinks(filenames []string) []string { | ||||
| 	var output []string | ||||
| 	for _, f := range filenames { | ||||
| 		res, resp := m.Client.GetFileLink(f) | ||||
| 		if resp.Error != nil { | ||||
| 			continue | ||||
| 		} | ||||
| 		output = append(output, res) | ||||
| 	} | ||||
| 	return output | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetFileLinks(filenames []string) []string { | ||||
| 	uriScheme := "https://" | ||||
| 	if m.NoTLS { | ||||
| 		uriScheme = "http://" | ||||
| 	} | ||||
|  | ||||
| 	var output []string | ||||
| 	for _, f := range filenames { | ||||
| 		res, resp := m.Client.GetFileLink(f) | ||||
| 		if resp.Error != nil { | ||||
| 			// public links is probably disabled, create the link ourselves | ||||
| 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V4+"/files/"+f) | ||||
| 			continue | ||||
| 		} | ||||
| 		output = append(output, res) | ||||
| 	} | ||||
| 	return output | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateChannelHeader(channelId string, header string) { | ||||
| 	channel := &model.Channel{Id: channelId, Header: header} | ||||
| 	m.log.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||
| 	_, resp := m.Client.UpdateChannel(channel) | ||||
| 	if resp.Error != nil { | ||||
| 		log.Error(resp.Error) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateLastViewed(channelId string) { | ||||
| 	m.log.Debugf("posting lastview %#v", channelId) | ||||
| 	view := &model.ChannelView{ChannelId: channelId} | ||||
| 	_, resp := m.Client.ViewChannel(m.User.Id, view) | ||||
| 	if resp.Error != nil { | ||||
| 		m.log.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUserNick(nick string) error { | ||||
| 	user := m.User | ||||
| 	user.Nickname = nick | ||||
| 	_, resp := m.Client.UpdateUser(user) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UsernamesInChannel(channelId string) []string { | ||||
| 	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		m.log.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error) | ||||
| 		return []string{} | ||||
| 	} | ||||
| 	allusers := m.GetUsers() | ||||
| 	result := []string{} | ||||
| 	for _, member := range *res { | ||||
| 		result = append(result, allusers[member.UserId].Nickname) | ||||
| 	} | ||||
| 	return result | ||||
| } | ||||
|  | ||||
| func (m *MMClient) createCookieJar(token string) *cookiejar.Jar { | ||||
| 	var cookies []*http.Cookie | ||||
| 	jar, _ := cookiejar.New(nil) | ||||
| 	firstCookie := &http.Cookie{ | ||||
| 		Name:   "MMAUTHTOKEN", | ||||
| 		Value:  token, | ||||
| 		Path:   "/", | ||||
| 		Domain: m.Credentials.Server, | ||||
| 	} | ||||
| 	cookies = append(cookies, firstCookie) | ||||
| 	cookieURL, _ := url.Parse("https://" + m.Credentials.Server) | ||||
| 	jar.SetCookies(cookieURL, cookies) | ||||
| 	return jar | ||||
| } | ||||
|  | ||||
| // SendDirectMessage sends a direct message to specified user | ||||
| func (m *MMClient) SendDirectMessage(toUserId string, msg string) { | ||||
| 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||
| 	// create DM channel (only happens on first message) | ||||
| 	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId) | ||||
| 	if resp.Error != nil { | ||||
| 		m.log.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error) | ||||
| 		return | ||||
| 	} | ||||
| 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||
|  | ||||
| 	// update our channels | ||||
| 	m.UpdateChannels() | ||||
|  | ||||
| 	// build & send the message | ||||
| 	msg = strings.Replace(msg, "\r", "", -1) | ||||
| 	post := &model.Post{ChannelId: m.GetChannelId(channelName, ""), Message: msg} | ||||
| 	m.Client.CreatePost(post) | ||||
| } | ||||
|  | ||||
| // GetTeamName returns the name of the specified teamId | ||||
| func (m *MMClient) GetTeamName(teamId string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id == teamId { | ||||
| 			return t.Team.Name | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| // GetChannels returns all channels we're members off | ||||
| func (m *MMClient) GetChannels() []*model.Channel { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	// our primary team channels first | ||||
| 	channels = append(channels, m.Team.Channels...) | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id != m.Team.Id { | ||||
| 			channels = append(channels, t.Channels...) | ||||
| 		} | ||||
| 	} | ||||
| 	return channels | ||||
| } | ||||
|  | ||||
| // GetMoreChannels returns existing channels where we're not a member off. | ||||
| func (m *MMClient) GetMoreChannels() []*model.Channel { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		channels = append(channels, t.MoreChannels...) | ||||
| 	} | ||||
| 	return channels | ||||
| } | ||||
|  | ||||
| // GetTeamFromChannel returns teamId belonging to channel (DM channels have no teamId). | ||||
| func (m *MMClient) GetTeamFromChannel(channelId string) string { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	var channels []*model.Channel | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		channels = append(channels, t.Channels...) | ||||
| 		if t.MoreChannels != nil { | ||||
| 			channels = append(channels, t.MoreChannels...) | ||||
| 		} | ||||
| 		for _, c := range channels { | ||||
| 			if c.Id == channelId { | ||||
| 				return t.Id | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetLastViewedAt(channelId string) int64 { | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	res, resp := m.Client.GetChannelMember(channelId, m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return model.GetMillis() | ||||
| 	} | ||||
| 	return res.LastViewedAt | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUsers() map[string]*model.User { | ||||
| 	users := make(map[string]*model.User) | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for k, v := range m.Users { | ||||
| 		users[k] = v | ||||
| 	} | ||||
| 	return users | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUser(userId string) *model.User { | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	_, ok := m.Users[userId] | ||||
| 	if !ok { | ||||
| 		res, resp := m.Client.GetUser(userId, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		m.Users[userId] = res | ||||
| 	} | ||||
| 	return m.Users[userId] | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUser(userId string) { | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	res, resp := m.Client.GetUser(userId, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return | ||||
| 	} | ||||
| 	m.Users[userId] = res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUserName(userId string) string { | ||||
| 	user := m.GetUser(userId) | ||||
| 	if user != nil { | ||||
| 		return user.Username | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetNickName(userId string) string { | ||||
| 	user := m.GetUser(userId) | ||||
| 	if user != nil { | ||||
| 		return user.Nickname | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetStatus(userId string) string { | ||||
| 	res, resp := m.Client.GetUserStatus(userId, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return "" | ||||
| 	} | ||||
| 	if res.Status == model.STATUS_AWAY { | ||||
| 		return "away" | ||||
| 	} | ||||
| 	if res.Status == model.STATUS_ONLINE { | ||||
| 		return "online" | ||||
| 	} | ||||
| 	return "offline" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateStatus(userId string, status string) error { | ||||
| 	_, resp := m.Client.UpdateUserStatus(userId, &model.Status{Status: status}) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetStatuses() map[string]string { | ||||
| 	var ids []string | ||||
| 	statuses := make(map[string]string) | ||||
| 	for id := range m.Users { | ||||
| 		ids = append(ids, id) | ||||
| 	} | ||||
| 	res, resp := m.Client.GetUsersStatusesByIds(ids) | ||||
| 	if resp.Error != nil { | ||||
| 		return statuses | ||||
| 	} | ||||
| 	for _, status := range res { | ||||
| 		statuses[status.UserId] = "offline" | ||||
| 		if status.Status == model.STATUS_AWAY { | ||||
| 			statuses[status.UserId] = "away" | ||||
| 		} | ||||
| 		if status.Status == model.STATUS_ONLINE { | ||||
| 			statuses[status.UserId] = "online" | ||||
| 		} | ||||
| 	} | ||||
| 	return statuses | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetTeamId() string { | ||||
| 	return m.Team.Id | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { | ||||
| 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return f.FileInfos[0].Id, nil | ||||
| } | ||||
|  | ||||
| // StatusLoop implements a ping-cycle that ensures that the connection to the chat servers | ||||
| // remains alive. In case of a disconnect it will try to reconnect. A call to this method | ||||
| // is blocking until the 'WsQuite' field of the MMClient object is set to 'true'. | ||||
| func (m *MMClient) StatusLoop() { | ||||
| 	retries := 0 | ||||
| 	backoff := time.Second * 60 | ||||
| 	if m.OnWsConnect != nil { | ||||
| 		m.OnWsConnect() | ||||
| 	} | ||||
| 	m.log.Debug("StatusLoop:", m.OnWsConnect) | ||||
| 	m.logger.Debug("StatusLoop:", m.OnWsConnect != nil) | ||||
| 	for { | ||||
| 		if m.WsQuit { | ||||
| 			return | ||||
| 		} | ||||
| 		if m.WsConnected { | ||||
| 			m.log.Debug("WS PING") | ||||
| 			m.sendWSRequest("ping", nil) | ||||
| 			if err := m.checkAlive(); err != nil { | ||||
| 				m.logger.Errorf("Connection is not alive: %#v", err) | ||||
| 			} | ||||
| 			select { | ||||
| 			case <-m.WsPingChan: | ||||
| 				m.log.Debug("WS PONG received") | ||||
| 				m.logger.Debug("WS PONG received") | ||||
| 				backoff = time.Second * 60 | ||||
| 			case <-time.After(time.Second * 5): | ||||
| 				if retries > 3 { | ||||
| 					m.log.Debug("StatusLoop() timeout") | ||||
| 					m.logger.Debug("StatusLoop() timeout") | ||||
| 					m.Logout() | ||||
| 					m.WsQuit = false | ||||
| 					err := m.Login() | ||||
| 					if err != nil { | ||||
| 						log.Errorf("Login failed: %#v", err) | ||||
| 						m.logger.Errorf("Login failed: %#v", err) | ||||
| 						break | ||||
| 					} | ||||
| 					if m.OnWsConnect != nil { | ||||
| @@ -878,75 +274,3 @@ func (m *MMClient) StatusLoop() { | ||||
| 		time.Sleep(backoff) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // initialize user and teams | ||||
| func (m *MMClient) initUser() error { | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	// we only load all team data on initial login. | ||||
| 	// all other updates are for channels from our (primary) team only. | ||||
| 	//m.log.Debug("initUser(): loading all team data") | ||||
| 	teams, resp := m.Client.GetTeamsForUser(m.User.Id, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	for _, team := range teams { | ||||
| 		mmusers, resp := m.Client.GetUsersInTeam(team.Id, 0, 50000, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return errors.New(resp.Error.DetailedError) | ||||
| 		} | ||||
| 		usermap := make(map[string]*model.User) | ||||
| 		for _, user := range mmusers { | ||||
| 			usermap[user.Id] = user | ||||
| 		} | ||||
|  | ||||
| 		t := &Team{Team: team, Users: usermap, Id: team.Id} | ||||
|  | ||||
| 		mmchannels, resp := m.Client.GetChannelsForTeamForUser(team.Id, m.User.Id, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return resp.Error | ||||
| 		} | ||||
| 		t.Channels = mmchannels | ||||
| 		mmchannels, resp = m.Client.GetPublicChannelsForTeam(team.Id, 0, 5000, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return resp.Error | ||||
| 		} | ||||
| 		t.MoreChannels = mmchannels | ||||
| 		m.OtherTeams = append(m.OtherTeams, t) | ||||
| 		if team.Name == m.Credentials.Team { | ||||
| 			m.Team = t | ||||
| 			m.log.Debugf("initUser(): found our team %s (id: %s)", team.Name, team.Id) | ||||
| 		} | ||||
| 		// add all users | ||||
| 		for k, v := range t.Users { | ||||
| 			m.Users[k] = v | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error { | ||||
| 	req := &model.WebSocketRequest{} | ||||
| 	req.Seq = m.WsSequence | ||||
| 	req.Action = action | ||||
| 	req.Data = data | ||||
| 	m.WsSequence++ | ||||
| 	m.log.Debugf("sendWsRequest %#v", req) | ||||
| 	m.WsClient.WriteJSON(req) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func supportedVersion(version string) bool { | ||||
| 	if strings.HasPrefix(version, "3.8.0") || | ||||
| 		strings.HasPrefix(version, "3.9.0") || | ||||
| 		strings.HasPrefix(version, "3.10.0") || | ||||
| 		strings.HasPrefix(version, "4.") || | ||||
| 		strings.HasPrefix(version, "5.") { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func digestString(s string) string { | ||||
| 	return fmt.Sprintf("%x", md5.Sum([]byte(s))) | ||||
| } | ||||
|   | ||||
							
								
								
									
										207
									
								
								matterclient/messages.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,207 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| func (m *MMClient) parseActionPost(rmsg *Message) { | ||||
| 	// add post to cache, if it already exists don't relay this again. | ||||
| 	// this should fix reposts | ||||
| 	if ok, _ := m.lruCache.ContainsOrAdd(digestString(rmsg.Raw.Data["post"].(string)), true); ok { | ||||
| 		m.logger.Debugf("message %#v in cache, not processing again", rmsg.Raw.Data["post"].(string)) | ||||
| 		rmsg.Text = "" | ||||
| 		return | ||||
| 	} | ||||
| 	data := model.PostFromJson(strings.NewReader(rmsg.Raw.Data["post"].(string))) | ||||
| 	// we don't have the user, refresh the userlist | ||||
| 	if m.GetUser(data.UserId) == nil { | ||||
| 		m.logger.Infof("User '%v' is not known, ignoring message '%#v'", | ||||
| 			data.UserId, data) | ||||
| 		return | ||||
| 	} | ||||
| 	rmsg.Username = m.GetUserName(data.UserId) | ||||
| 	rmsg.Channel = m.GetChannelName(data.ChannelId) | ||||
| 	rmsg.UserID = data.UserId | ||||
| 	rmsg.Type = data.Type | ||||
| 	teamid, _ := rmsg.Raw.Data["team_id"].(string) | ||||
| 	// edit messsages have no team_id for some reason | ||||
| 	if teamid == "" { | ||||
| 		// we can find the team_id from the channelid | ||||
| 		teamid = m.GetChannelTeamId(data.ChannelId) | ||||
| 		rmsg.Raw.Data["team_id"] = teamid | ||||
| 	} | ||||
| 	if teamid != "" { | ||||
| 		rmsg.Team = m.GetTeamName(teamid) | ||||
| 	} | ||||
| 	// direct message | ||||
| 	if rmsg.Raw.Data["channel_type"] == "D" { | ||||
| 		rmsg.Channel = m.GetUser(data.UserId).Username | ||||
| 	} | ||||
| 	rmsg.Text = data.Message | ||||
| 	rmsg.Post = data | ||||
| } | ||||
|  | ||||
| func (m *MMClient) parseMessage(rmsg *Message) { | ||||
| 	switch rmsg.Raw.Event { | ||||
| 	case model.WEBSOCKET_EVENT_POSTED, model.WEBSOCKET_EVENT_POST_EDITED, model.WEBSOCKET_EVENT_POST_DELETED: | ||||
| 		m.parseActionPost(rmsg) | ||||
| 	case "user_updated": | ||||
| 		user := rmsg.Raw.Data["user"].(map[string]interface{}) | ||||
| 		if _, ok := user["id"].(string); ok { | ||||
| 			m.UpdateUser(user["id"].(string)) | ||||
| 		} | ||||
| 	case "group_added": | ||||
| 		if err := m.UpdateChannels(); err != nil { | ||||
| 			m.logger.Errorf("failed to update channels: %#v", err) | ||||
| 		} | ||||
| 		/* | ||||
| 			case model.ACTION_USER_REMOVED: | ||||
| 				m.handleWsActionUserRemoved(&rmsg) | ||||
| 			case model.ACTION_USER_ADDED: | ||||
| 				m.handleWsActionUserAdded(&rmsg) | ||||
| 		*/ | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) parseResponse(rmsg model.WebSocketResponse) { | ||||
| 	if rmsg.Data != nil { | ||||
| 		// ping reply | ||||
| 		if rmsg.Data["text"].(string) == "pong" { | ||||
| 			m.WsPingChan <- &rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) DeleteMessage(postId string) error { //nolint:golint | ||||
| 	_, resp := m.Client.DeletePost(postId) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) EditMessage(postId string, text string) (string, error) { //nolint:golint | ||||
| 	post := &model.Post{Message: text} | ||||
| 	res, resp := m.Client.UpdatePost(postId, post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetFileLinks(filenames []string) []string { | ||||
| 	uriScheme := "https://" | ||||
| 	if m.NoTLS { | ||||
| 		uriScheme = "http://" | ||||
| 	} | ||||
|  | ||||
| 	var output []string | ||||
| 	for _, f := range filenames { | ||||
| 		res, resp := m.Client.GetFileLink(f) | ||||
| 		if resp.Error != nil { | ||||
| 			// public links is probably disabled, create the link ourselves | ||||
| 			output = append(output, uriScheme+m.Credentials.Server+model.API_URL_SUFFIX_V4+"/files/"+f) | ||||
| 			continue | ||||
| 		} | ||||
| 		output = append(output, res) | ||||
| 	} | ||||
| 	return output | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPosts(channelId string, limit int) *model.PostList { //nolint:golint | ||||
| 	res, resp := m.Client.GetPostsForChannel(channelId, 0, limit, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPostsSince(channelId string, time int64) *model.PostList { //nolint:golint | ||||
| 	res, resp := m.Client.GetPostsSince(channelId, time) | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPublicLink(filename string) string { | ||||
| 	res, resp := m.Client.GetFileLink(filename) | ||||
| 	if resp.Error != nil { | ||||
| 		return "" | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetPublicLinks(filenames []string) []string { | ||||
| 	var output []string | ||||
| 	for _, f := range filenames { | ||||
| 		res, resp := m.Client.GetFileLink(f) | ||||
| 		if resp.Error != nil { | ||||
| 			continue | ||||
| 		} | ||||
| 		output = append(output, res) | ||||
| 	} | ||||
| 	return output | ||||
| } | ||||
|  | ||||
| func (m *MMClient) PostMessage(channelId string, text string, rootId string) (string, error) { //nolint:golint | ||||
| 	post := &model.Post{ChannelId: channelId, Message: text, RootId: rootId} | ||||
| 	res, resp := m.Client.CreatePost(post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) PostMessageWithFiles(channelId string, text string, rootId string, fileIds []string) (string, error) { //nolint:golint | ||||
| 	post := &model.Post{ChannelId: channelId, Message: text, RootId: rootId, FileIds: fileIds} | ||||
| 	res, resp := m.Client.CreatePost(post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SearchPosts(query string) *model.PostList { | ||||
| 	res, resp := m.Client.SearchPosts(m.Team.Id, query, false) | ||||
| 	if resp.Error != nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return res | ||||
| } | ||||
|  | ||||
| // SendDirectMessage sends a direct message to specified user | ||||
| func (m *MMClient) SendDirectMessage(toUserId string, msg string, rootId string) { //nolint:golint | ||||
| 	m.SendDirectMessageProps(toUserId, msg, rootId, nil) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SendDirectMessageProps(toUserId string, msg string, rootId string, props map[string]interface{}) { //nolint:golint | ||||
| 	m.logger.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||
| 	// create DM channel (only happens on first message) | ||||
| 	_, resp := m.Client.CreateDirectChannel(m.User.Id, toUserId) | ||||
| 	if resp.Error != nil { | ||||
| 		m.logger.Debugf("SendDirectMessage to %#v failed: %s", toUserId, resp.Error) | ||||
| 		return | ||||
| 	} | ||||
| 	channelName := model.GetDMNameFromIds(toUserId, m.User.Id) | ||||
|  | ||||
| 	// update our channels | ||||
| 	if err := m.UpdateChannels(); err != nil { | ||||
| 		m.logger.Errorf("failed to update channels: %#v", err) | ||||
| 	} | ||||
|  | ||||
| 	// build & send the message | ||||
| 	msg = strings.Replace(msg, "\r", "", -1) | ||||
| 	post := &model.Post{ChannelId: m.GetChannelId(channelName, m.Team.Id), Message: msg, RootId: rootId, Props: props} | ||||
| 	m.Client.CreatePost(post) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { //nolint:golint | ||||
| 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return f.FileInfos[0].Id, nil | ||||
| } | ||||
							
								
								
									
										154
									
								
								matterclient/users.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,154 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
|  | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| ) | ||||
|  | ||||
| func (m *MMClient) GetNickName(userId string) string { //nolint:golint | ||||
| 	user := m.GetUser(userId) | ||||
| 	if user != nil { | ||||
| 		return user.Nickname | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetStatus(userId string) string { //nolint:golint | ||||
| 	res, resp := m.Client.GetUserStatus(userId, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return "" | ||||
| 	} | ||||
| 	if res.Status == model.STATUS_AWAY { | ||||
| 		return "away" | ||||
| 	} | ||||
| 	if res.Status == model.STATUS_ONLINE { | ||||
| 		return "online" | ||||
| 	} | ||||
| 	return "offline" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetStatuses() map[string]string { | ||||
| 	var ids []string | ||||
| 	statuses := make(map[string]string) | ||||
| 	for id := range m.Users { | ||||
| 		ids = append(ids, id) | ||||
| 	} | ||||
| 	res, resp := m.Client.GetUsersStatusesByIds(ids) | ||||
| 	if resp.Error != nil { | ||||
| 		return statuses | ||||
| 	} | ||||
| 	for _, status := range res { | ||||
| 		statuses[status.UserId] = "offline" | ||||
| 		if status.Status == model.STATUS_AWAY { | ||||
| 			statuses[status.UserId] = "away" | ||||
| 		} | ||||
| 		if status.Status == model.STATUS_ONLINE { | ||||
| 			statuses[status.UserId] = "online" | ||||
| 		} | ||||
| 	} | ||||
| 	return statuses | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetTeamId() string { //nolint:golint | ||||
| 	return m.Team.Id | ||||
| } | ||||
|  | ||||
| // GetTeamName returns the name of the specified teamId | ||||
| func (m *MMClient) GetTeamName(teamId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for _, t := range m.OtherTeams { | ||||
| 		if t.Id == teamId { | ||||
| 			return t.Team.Name | ||||
| 		} | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUser(userId string) *model.User { //nolint:golint | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	_, ok := m.Users[userId] | ||||
| 	if !ok { | ||||
| 		res, resp := m.Client.GetUser(userId, "") | ||||
| 		if resp.Error != nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		m.Users[userId] = res | ||||
| 	} | ||||
| 	return m.Users[userId] | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUserName(userId string) string { //nolint:golint | ||||
| 	user := m.GetUser(userId) | ||||
| 	if user != nil { | ||||
| 		return user.Username | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUsers() map[string]*model.User { | ||||
| 	users := make(map[string]*model.User) | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	for k, v := range m.Users { | ||||
| 		users[k] = v | ||||
| 	} | ||||
| 	return users | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUsers() error { | ||||
| 	mmusers, resp := m.Client.GetUsers(0, 50000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return errors.New(resp.Error.DetailedError) | ||||
| 	} | ||||
| 	m.Lock() | ||||
| 	for _, user := range mmusers { | ||||
| 		m.Users[user.Id] = user | ||||
| 	} | ||||
| 	m.Unlock() | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUserNick(nick string) error { | ||||
| 	user := m.User | ||||
| 	user.Nickname = nick | ||||
| 	_, resp := m.Client.UpdateUser(user) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UsernamesInChannel(channelId string) []string { //nolint:golint | ||||
| 	res, resp := m.Client.GetChannelMembers(channelId, 0, 50000, "") | ||||
| 	if resp.Error != nil { | ||||
| 		m.logger.Errorf("UsernamesInChannel(%s) failed: %s", channelId, resp.Error) | ||||
| 		return []string{} | ||||
| 	} | ||||
| 	allusers := m.GetUsers() | ||||
| 	result := []string{} | ||||
| 	for _, member := range *res { | ||||
| 		result = append(result, allusers[member.UserId].Nickname) | ||||
| 	} | ||||
| 	return result | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateStatus(userId string, status string) error { //nolint:golint | ||||
| 	_, resp := m.Client.UpdateUserStatus(userId, &model.Status{Status: status}) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateUser(userId string) { //nolint:golint | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	res, resp := m.Client.GetUser(userId, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return | ||||
| 	} | ||||
| 	m.Users[userId] = res | ||||
| } | ||||
| @@ -41,9 +41,9 @@ type IMessage struct { | ||||
| 	Timestamp   string `schema:"timestamp"` | ||||
| 	UserID      string `schema:"user_id"` | ||||
| 	UserName    string `schema:"user_name"` | ||||
| 	PostId      string `schema:"post_id"` | ||||
| 	PostId      string `schema:"post_id"` //nolint:golint | ||||
| 	RawText     string `schema:"raw_text"` | ||||
| 	ServiceId   string `schema:"service_id"` | ||||
| 	ServiceId   string `schema:"service_id"` //nolint:golint | ||||
| 	Text        string `schema:"text"` | ||||
| 	TriggerWord string `schema:"trigger_word"` | ||||
| 	FileIDs     string `schema:"file_ids"` | ||||
| @@ -51,7 +51,8 @@ type IMessage struct { | ||||
|  | ||||
| // Client for Mattermost. | ||||
| type Client struct { | ||||
| 	Url        string // URL for incoming webhooks on mattermost. | ||||
| 	// URL for incoming webhooks on mattermost. | ||||
| 	Url        string // nolint:golint | ||||
| 	In         chan IMessage | ||||
| 	Out        chan OMessage | ||||
| 	httpclient *http.Client | ||||
| @@ -70,7 +71,7 @@ type Config struct { | ||||
| func New(url string, config Config) *Client { | ||||
| 	c := &Client{Url: url, In: make(chan IMessage), Out: make(chan OMessage), Config: config} | ||||
| 	tr := &http.Transport{ | ||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, | ||||
| 		TLSClientConfig: &tls.Config{InsecureSkipVerify: config.InsecureSkipVerify}, //nolint:gosec | ||||
| 	} | ||||
| 	c.httpclient = &http.Client{Transport: tr} | ||||
| 	if !c.DisableServer { | ||||
|   | ||||
							
								
								
									
										24
									
								
								vendor/github.com/Baozisoftware/qrcode-terminal-go/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,24 @@ | ||||
| # Compiled Object files, Static and Dynamic libs (Shared Objects) | ||||
| *.o | ||||
| *.a | ||||
| *.so | ||||
|  | ||||
| # Folders | ||||
| _obj | ||||
| _test | ||||
|  | ||||
| # Architecture specific extensions/prefixes | ||||
| *.[568vq] | ||||
| [568vq].out | ||||
|  | ||||
| *.cgo1.go | ||||
| *.cgo2.c | ||||
| _cgo_defun.c | ||||
| _cgo_gotypes.go | ||||
| _cgo_export.* | ||||
|  | ||||
| _testmain.go | ||||
|  | ||||
| *.exe | ||||
| *.test | ||||
| *.prof | ||||
							
								
								
									
										29
									
								
								vendor/github.com/Baozisoftware/qrcode-terminal-go/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,29 @@ | ||||
| BSD 3-Clause License | ||||
|  | ||||
| Copyright (c) 2017, Baozisoftware | ||||
| All rights reserved. | ||||
|  | ||||
| Redistribution and use in source and binary forms, with or without | ||||
| modification, are permitted provided that the following conditions are met: | ||||
|  | ||||
| * Redistributions of source code must retain the above copyright notice, this | ||||
|   list of conditions and the following disclaimer. | ||||
|  | ||||
| * Redistributions in binary form must reproduce the above copyright notice, | ||||
|   this list of conditions and the following disclaimer in the documentation | ||||
|   and/or other materials provided with the distribution. | ||||
|  | ||||
| * Neither the name of the copyright holder nor the names of its | ||||
|   contributors may be used to endorse or promote products derived from | ||||
|   this software without specific prior written permission. | ||||
|  | ||||
| THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" | ||||
| AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | ||||
| IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | ||||
| DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE | ||||
| FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | ||||
| DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | ||||
| SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | ||||
| CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, | ||||
| OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | ||||
| OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | ||||
							
								
								
									
										39
									
								
								vendor/github.com/Baozisoftware/qrcode-terminal-go/README.md
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,39 @@ | ||||
|  | ||||
| # qrcode-terminal-go | ||||
| QRCode terminal for golang. | ||||
|  | ||||
| # Example | ||||
| ```go | ||||
| package main | ||||
|  | ||||
| import "github.com/Baozisoftware/qrcode-terminal-go" | ||||
|  | ||||
| func main() { | ||||
| 	Test1() | ||||
| 	Test2() | ||||
| } | ||||
|  | ||||
| func Test1(){ | ||||
| 	content := "Hello, 世界" | ||||
| 	obj := qrcodeTerminal.New() | ||||
| 	obj.Get(content).Print() | ||||
| } | ||||
|  | ||||
| func Test2(){ | ||||
| 	content := "https://github.com/Baozisoftware/qrcode-terminal-go" | ||||
| 	obj := qrcodeTerminal.New2(qrcodeTerminal.ConsoleColors.BrightBlue,qrcodeTerminal.ConsoleColors.BrightGreen,qrcodeTerminal.QRCodeRecoveryLevels.Low) | ||||
| 	obj.Get([]byte(content)).Print() | ||||
| } | ||||
| ``` | ||||
|  | ||||
| ## Screenshots | ||||
| ### Windows XP | ||||
|  | ||||
| ### Windows 7 | ||||
|  | ||||
| ### Windows 10 | ||||
|  | ||||
| ### Ubuntu | ||||
|  | ||||
| ### macOS | ||||
|  | ||||
							
								
								
									
										155
									
								
								vendor/github.com/Baozisoftware/qrcode-terminal-go/qrcodeTerminal.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,155 @@ | ||||
| package qrcodeTerminal | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
|  | ||||
| 	"github.com/skip2/go-qrcode" | ||||
| 	"github.com/mattn/go-colorable" | ||||
| 	"image/png" | ||||
| 	nbytes "bytes" | ||||
| ) | ||||
|  | ||||
| type consoleColor string | ||||
| type consoleColors struct { | ||||
| 	NormalBlack   consoleColor | ||||
| 	NormalRed     consoleColor | ||||
| 	NormalGreen   consoleColor | ||||
| 	NormalYellow  consoleColor | ||||
| 	NormalBlue    consoleColor | ||||
| 	NormalMagenta consoleColor | ||||
| 	NormalCyan    consoleColor | ||||
| 	NormalWhite   consoleColor | ||||
| 	BrightBlack   consoleColor | ||||
| 	BrightRed     consoleColor | ||||
| 	BrightGreen   consoleColor | ||||
| 	BrightYellow  consoleColor | ||||
| 	BrightBlue    consoleColor | ||||
| 	BrightMagenta consoleColor | ||||
| 	BrightCyan    consoleColor | ||||
| 	BrightWhite   consoleColor | ||||
| } | ||||
| type qrcodeRecoveryLevel qrcode.RecoveryLevel | ||||
| type qrcodeRecoveryLevels struct { | ||||
| 	Low     qrcodeRecoveryLevel | ||||
| 	Medium  qrcodeRecoveryLevel | ||||
| 	High    qrcodeRecoveryLevel | ||||
| 	Highest qrcodeRecoveryLevel | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	ConsoleColors consoleColors = consoleColors{ | ||||
| 		NormalBlack:   "\033[38;5;0m  \033[0m", | ||||
| 		NormalRed:     "\033[38;5;1m  \033[0m", | ||||
| 		NormalGreen:   "\033[38;5;2m  \033[0m", | ||||
| 		NormalYellow:  "\033[38;5;3m  \033[0m", | ||||
| 		NormalBlue:    "\033[38;5;4m  \033[0m", | ||||
| 		NormalMagenta: "\033[38;5;5m  \033[0m", | ||||
| 		NormalCyan:    "\033[38;5;6m  \033[0m", | ||||
| 		NormalWhite:   "\033[38;5;7m  \033[0m", | ||||
| 		BrightBlack:   "\033[48;5;0m  \033[0m", | ||||
| 		BrightRed:     "\033[48;5;1m  \033[0m", | ||||
| 		BrightGreen:   "\033[48;5;2m  \033[0m", | ||||
| 		BrightYellow:  "\033[48;5;3m  \033[0m", | ||||
| 		BrightBlue:    "\033[48;5;4m  \033[0m", | ||||
| 		BrightMagenta: "\033[48;5;5m  \033[0m", | ||||
| 		BrightCyan:    "\033[48;5;6m  \033[0m", | ||||
| 		BrightWhite:   "\033[48;5;7m  \033[0m"} | ||||
| 	QRCodeRecoveryLevels = qrcodeRecoveryLevels{ | ||||
| 		Low:     qrcodeRecoveryLevel(qrcode.Low), | ||||
| 		Medium:  qrcodeRecoveryLevel(qrcode.Medium), | ||||
| 		High:    qrcodeRecoveryLevel(qrcode.High), | ||||
| 		Highest: qrcodeRecoveryLevel(qrcode.Highest)} | ||||
| ) | ||||
|  | ||||
| type QRCodeString string | ||||
|  | ||||
| func (v *QRCodeString) Print() { | ||||
| 	fmt.Fprint(outer, *v) | ||||
| } | ||||
|  | ||||
| type qrcodeTerminal struct { | ||||
| 	front consoleColor | ||||
| 	back  consoleColor | ||||
| 	level qrcodeRecoveryLevel | ||||
| } | ||||
|  | ||||
| func (v *qrcodeTerminal) Get(content interface{}) (result *QRCodeString) { | ||||
| 	var qr *qrcode.QRCode | ||||
| 	var err error | ||||
| 	if t, ok := content.(string); ok { | ||||
| 		qr, err = qrcode.New(t, qrcode.RecoveryLevel(v.level)) | ||||
| 	} else if t, ok := content.([]byte); ok { | ||||
| 		qr, err = qrcode.New(string(t), qrcode.RecoveryLevel(v.level)) | ||||
| 	} | ||||
| 	if qr != nil && err == nil { | ||||
| 		data := qr.Bitmap() | ||||
| 		result = v.getQRCodeString(data) | ||||
| 	} | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func (v *qrcodeTerminal) Get2(bytes []byte) (result *QRCodeString) { | ||||
| 	data, err := parseQR(bytes) | ||||
| 	if err == nil { | ||||
| 		result = v.getQRCodeString(data) | ||||
| 	} | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func New2(front, back consoleColor, level qrcodeRecoveryLevel) *qrcodeTerminal { | ||||
| 	obj := qrcodeTerminal{front: front, back: back, level: level} | ||||
| 	return &obj | ||||
| } | ||||
|  | ||||
| func New() *qrcodeTerminal { | ||||
| 	front, back, level := ConsoleColors.BrightBlack, ConsoleColors.BrightWhite, QRCodeRecoveryLevels.Medium | ||||
| 	return New2(front, back, level) | ||||
| } | ||||
|  | ||||
| func (v *qrcodeTerminal) getQRCodeString(data [][]bool) (result *QRCodeString) { | ||||
| 	str := "" | ||||
| 	for ir, row := range data { | ||||
| 		lr := len(row) | ||||
| 		if ir == 0 || ir == 1 || ir == 2 || | ||||
| 			ir == lr-1 || ir == lr-2 || ir == lr-3 { | ||||
| 			continue | ||||
| 		} | ||||
| 		for ic, col := range row { | ||||
| 			lc := len(data) | ||||
| 			if ic == 0 || ic == 1 || ic == 2 || | ||||
| 				ic == lc-1 || ic == lc-2 || ic == lc-3 { | ||||
| 				continue | ||||
| 			} | ||||
| 			if col { | ||||
| 				str += fmt.Sprint(v.front) | ||||
| 			} else { | ||||
| 				str += fmt.Sprint(v.back) | ||||
| 			} | ||||
| 		} | ||||
| 		str += fmt.Sprintln() | ||||
| 	} | ||||
| 	obj := QRCodeString(str) | ||||
| 	result = &obj | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func parseQR(bytes []byte) (data [][]bool, err error) { | ||||
| 	r := nbytes.NewReader(bytes) | ||||
| 	img, err := png.Decode(r) | ||||
| 	if err == nil { | ||||
| 		rect := img.Bounds() | ||||
| 		mx, my := rect.Max.X, rect.Max.Y | ||||
| 		data = make([][]bool, mx) | ||||
| 		for x := 0; x < mx; x++ { | ||||
| 			data[x] = make([]bool, my) | ||||
| 			for y := 0; y < my; y++ { | ||||
| 				c := img.At(x, y) | ||||
| 				r, _, _, _ := c.RGBA() | ||||
| 				data[x][y] = r == 0 | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return | ||||
| } | ||||
|  | ||||
| var outer = colorable.NewColorableStdout() | ||||
							
								
								
									
										5
									
								
								vendor/github.com/BurntSushi/toml/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,5 +0,0 @@ | ||||
| TAGS | ||||
| tags | ||||
| .*.swp | ||||
| tomlcheck/tomlcheck | ||||
| toml.test | ||||
							
								
								
									
										15
									
								
								vendor/github.com/BurntSushi/toml/.travis.yml
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,15 +0,0 @@ | ||||
| language: go | ||||
| go: | ||||
|   - 1.1 | ||||
|   - 1.2 | ||||
|   - 1.3 | ||||
|   - 1.4 | ||||
|   - 1.5 | ||||
|   - 1.6 | ||||
|   - tip | ||||
| install: | ||||
|   - go install ./... | ||||
|   - go get github.com/BurntSushi/toml-test | ||||
| script: | ||||
|   - export PATH="$PATH:$HOME/gopath/bin" | ||||
|   - make test | ||||
							
								
								
									
										3
									
								
								vendor/github.com/BurntSushi/toml/COMPATIBLE
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,3 +0,0 @@ | ||||
| Compatible with TOML version | ||||
| [v0.4.0](https://github.com/toml-lang/toml/blob/v0.4.0/versions/en/toml-v0.4.0.md) | ||||
|  | ||||
							
								
								
									
										14
									
								
								vendor/github.com/BurntSushi/toml/COPYING
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,14 +0,0 @@ | ||||
|             DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE | ||||
|                     Version 2, December 2004 | ||||
|  | ||||
|  Copyright (C) 2004 Sam Hocevar <sam@hocevar.net> | ||||
|  | ||||
|  Everyone is permitted to copy and distribute verbatim or modified | ||||
|  copies of this license document, and changing it is allowed as long | ||||
|  as the name is changed. | ||||
|  | ||||
|             DO WHAT THE FUCK YOU WANT TO PUBLIC LICENSE | ||||
|    TERMS AND CONDITIONS FOR COPYING, DISTRIBUTION AND MODIFICATION | ||||
|  | ||||
|   0. You just DO WHAT THE FUCK YOU WANT TO. | ||||
|  | ||||
							
								
								
									
										19
									
								
								vendor/github.com/BurntSushi/toml/Makefile
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,19 +0,0 @@ | ||||
| install: | ||||
| 	go install ./... | ||||
|  | ||||
| test: install | ||||
| 	go test -v | ||||
| 	toml-test toml-test-decoder | ||||
| 	toml-test -encoder toml-test-encoder | ||||
|  | ||||
| fmt: | ||||
| 	gofmt -w *.go */*.go | ||||
| 	colcheck *.go */*.go | ||||
|  | ||||
| tags: | ||||
| 	find ./ -name '*.go' -print0 | xargs -0 gotags > TAGS | ||||
|  | ||||
| push: | ||||
| 	git push origin master | ||||
| 	git push github master | ||||
|  | ||||
							
								
								
									
										218
									
								
								vendor/github.com/BurntSushi/toml/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,218 +0,0 @@ | ||||
| ## TOML parser and encoder for Go with reflection | ||||
|  | ||||
| TOML stands for Tom's Obvious, Minimal Language. This Go package provides a | ||||
| reflection interface similar to Go's standard library `json` and `xml` | ||||
| packages. This package also supports the `encoding.TextUnmarshaler` and | ||||
| `encoding.TextMarshaler` interfaces so that you can define custom data | ||||
| representations. (There is an example of this below.) | ||||
|  | ||||
| Spec: https://github.com/toml-lang/toml | ||||
|  | ||||
| Compatible with TOML version | ||||
| [v0.4.0](https://github.com/toml-lang/toml/blob/master/versions/en/toml-v0.4.0.md) | ||||
|  | ||||
| Documentation: https://godoc.org/github.com/BurntSushi/toml | ||||
|  | ||||
| Installation: | ||||
|  | ||||
| ```bash | ||||
| go get github.com/BurntSushi/toml | ||||
| ``` | ||||
|  | ||||
| Try the toml validator: | ||||
|  | ||||
| ```bash | ||||
| go get github.com/BurntSushi/toml/cmd/tomlv | ||||
| tomlv some-toml-file.toml | ||||
| ``` | ||||
|  | ||||
| [](https://travis-ci.org/BurntSushi/toml) [](https://godoc.org/github.com/BurntSushi/toml) | ||||
|  | ||||
| ### Testing | ||||
|  | ||||
| This package passes all tests in | ||||
| [toml-test](https://github.com/BurntSushi/toml-test) for both the decoder | ||||
| and the encoder. | ||||
|  | ||||
| ### Examples | ||||
|  | ||||
| This package works similarly to how the Go standard library handles `XML` | ||||
| and `JSON`. Namely, data is loaded into Go values via reflection. | ||||
|  | ||||
| For the simplest example, consider some TOML file as just a list of keys | ||||
| and values: | ||||
|  | ||||
| ```toml | ||||
| Age = 25 | ||||
| Cats = [ "Cauchy", "Plato" ] | ||||
| Pi = 3.14 | ||||
| Perfection = [ 6, 28, 496, 8128 ] | ||||
| DOB = 1987-07-05T05:45:00Z | ||||
| ``` | ||||
|  | ||||
| Which could be defined in Go as: | ||||
|  | ||||
| ```go | ||||
| type Config struct { | ||||
|   Age int | ||||
|   Cats []string | ||||
|   Pi float64 | ||||
|   Perfection []int | ||||
|   DOB time.Time // requires `import time` | ||||
| } | ||||
| ``` | ||||
|  | ||||
| And then decoded with: | ||||
|  | ||||
| ```go | ||||
| var conf Config | ||||
| if _, err := toml.Decode(tomlData, &conf); err != nil { | ||||
|   // handle error | ||||
| } | ||||
| ``` | ||||
|  | ||||
| You can also use struct tags if your struct field name doesn't map to a TOML | ||||
| key value directly: | ||||
|  | ||||
| ```toml | ||||
| some_key_NAME = "wat" | ||||
| ``` | ||||
|  | ||||
| ```go | ||||
| type TOML struct { | ||||
|   ObscureKey string `toml:"some_key_NAME"` | ||||
| } | ||||
| ``` | ||||
|  | ||||
| ### Using the `encoding.TextUnmarshaler` interface | ||||
|  | ||||
| Here's an example that automatically parses duration strings into | ||||
| `time.Duration` values: | ||||
|  | ||||
| ```toml | ||||
| [[song]] | ||||
| name = "Thunder Road" | ||||
| duration = "4m49s" | ||||
|  | ||||
| [[song]] | ||||
| name = "Stairway to Heaven" | ||||
| duration = "8m03s" | ||||
| ``` | ||||
|  | ||||
| Which can be decoded with: | ||||
|  | ||||
| ```go | ||||
| type song struct { | ||||
|   Name     string | ||||
|   Duration duration | ||||
| } | ||||
| type songs struct { | ||||
|   Song []song | ||||
| } | ||||
| var favorites songs | ||||
| if _, err := toml.Decode(blob, &favorites); err != nil { | ||||
|   log.Fatal(err) | ||||
| } | ||||
|  | ||||
| for _, s := range favorites.Song { | ||||
|   fmt.Printf("%s (%s)\n", s.Name, s.Duration) | ||||
| } | ||||
| ``` | ||||
|  | ||||
| And you'll also need a `duration` type that satisfies the | ||||
| `encoding.TextUnmarshaler` interface: | ||||
|  | ||||
| ```go | ||||
| type duration struct { | ||||
| 	time.Duration | ||||
| } | ||||
|  | ||||
| func (d *duration) UnmarshalText(text []byte) error { | ||||
| 	var err error | ||||
| 	d.Duration, err = time.ParseDuration(string(text)) | ||||
| 	return err | ||||
| } | ||||
| ``` | ||||
|  | ||||
| ### More complex usage | ||||
|  | ||||
| Here's an example of how to load the example from the official spec page: | ||||
|  | ||||
| ```toml | ||||
| # This is a TOML document. Boom. | ||||
|  | ||||
| title = "TOML Example" | ||||
|  | ||||
| [owner] | ||||
| name = "Tom Preston-Werner" | ||||
| organization = "GitHub" | ||||
| bio = "GitHub Cofounder & CEO\nLikes tater tots and beer." | ||||
| dob = 1979-05-27T07:32:00Z # First class dates? Why not? | ||||
|  | ||||
| [database] | ||||
| server = "192.168.1.1" | ||||
| ports = [ 8001, 8001, 8002 ] | ||||
| connection_max = 5000 | ||||
| enabled = true | ||||
|  | ||||
| [servers] | ||||
|  | ||||
|   # You can indent as you please. Tabs or spaces. TOML don't care. | ||||
|   [servers.alpha] | ||||
|   ip = "10.0.0.1" | ||||
|   dc = "eqdc10" | ||||
|  | ||||
|   [servers.beta] | ||||
|   ip = "10.0.0.2" | ||||
|   dc = "eqdc10" | ||||
|  | ||||
| [clients] | ||||
| data = [ ["gamma", "delta"], [1, 2] ] # just an update to make sure parsers support it | ||||
|  | ||||
| # Line breaks are OK when inside arrays | ||||
| hosts = [ | ||||
|   "alpha", | ||||
|   "omega" | ||||
| ] | ||||
| ``` | ||||
|  | ||||
| And the corresponding Go types are: | ||||
|  | ||||
| ```go | ||||
| type tomlConfig struct { | ||||
| 	Title string | ||||
| 	Owner ownerInfo | ||||
| 	DB database `toml:"database"` | ||||
| 	Servers map[string]server | ||||
| 	Clients clients | ||||
| } | ||||
|  | ||||
| type ownerInfo struct { | ||||
| 	Name string | ||||
| 	Org string `toml:"organization"` | ||||
| 	Bio string | ||||
| 	DOB time.Time | ||||
| } | ||||
|  | ||||
| type database struct { | ||||
| 	Server string | ||||
| 	Ports []int | ||||
| 	ConnMax int `toml:"connection_max"` | ||||
| 	Enabled bool | ||||
| } | ||||
|  | ||||
| type server struct { | ||||
| 	IP string | ||||
| 	DC string | ||||
| } | ||||
|  | ||||
| type clients struct { | ||||
| 	Data [][]interface{} | ||||
| 	Hosts []string | ||||
| } | ||||
| ``` | ||||
|  | ||||
| Note that a case insensitive match will be tried if an exact match can't be | ||||
| found. | ||||
|  | ||||
| A working example of the above can be found in `_examples/example.{go,toml}`. | ||||
							
								
								
									
										509
									
								
								vendor/github.com/BurntSushi/toml/decode.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,509 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"math" | ||||
| 	"reflect" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| func e(format string, args ...interface{}) error { | ||||
| 	return fmt.Errorf("toml: "+format, args...) | ||||
| } | ||||
|  | ||||
| // Unmarshaler is the interface implemented by objects that can unmarshal a | ||||
| // TOML description of themselves. | ||||
| type Unmarshaler interface { | ||||
| 	UnmarshalTOML(interface{}) error | ||||
| } | ||||
|  | ||||
| // Unmarshal decodes the contents of `p` in TOML format into a pointer `v`. | ||||
| func Unmarshal(p []byte, v interface{}) error { | ||||
| 	_, err := Decode(string(p), v) | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // Primitive is a TOML value that hasn't been decoded into a Go value. | ||||
| // When using the various `Decode*` functions, the type `Primitive` may | ||||
| // be given to any value, and its decoding will be delayed. | ||||
| // | ||||
| // A `Primitive` value can be decoded using the `PrimitiveDecode` function. | ||||
| // | ||||
| // The underlying representation of a `Primitive` value is subject to change. | ||||
| // Do not rely on it. | ||||
| // | ||||
| // N.B. Primitive values are still parsed, so using them will only avoid | ||||
| // the overhead of reflection. They can be useful when you don't know the | ||||
| // exact type of TOML data until run time. | ||||
| type Primitive struct { | ||||
| 	undecoded interface{} | ||||
| 	context   Key | ||||
| } | ||||
|  | ||||
| // DEPRECATED! | ||||
| // | ||||
| // Use MetaData.PrimitiveDecode instead. | ||||
| func PrimitiveDecode(primValue Primitive, v interface{}) error { | ||||
| 	md := MetaData{decoded: make(map[string]bool)} | ||||
| 	return md.unify(primValue.undecoded, rvalue(v)) | ||||
| } | ||||
|  | ||||
| // PrimitiveDecode is just like the other `Decode*` functions, except it | ||||
| // decodes a TOML value that has already been parsed. Valid primitive values | ||||
| // can *only* be obtained from values filled by the decoder functions, | ||||
| // including this method. (i.e., `v` may contain more `Primitive` | ||||
| // values.) | ||||
| // | ||||
| // Meta data for primitive values is included in the meta data returned by | ||||
| // the `Decode*` functions with one exception: keys returned by the Undecoded | ||||
| // method will only reflect keys that were decoded. Namely, any keys hidden | ||||
| // behind a Primitive will be considered undecoded. Executing this method will | ||||
| // update the undecoded keys in the meta data. (See the example.) | ||||
| func (md *MetaData) PrimitiveDecode(primValue Primitive, v interface{}) error { | ||||
| 	md.context = primValue.context | ||||
| 	defer func() { md.context = nil }() | ||||
| 	return md.unify(primValue.undecoded, rvalue(v)) | ||||
| } | ||||
|  | ||||
| // Decode will decode the contents of `data` in TOML format into a pointer | ||||
| // `v`. | ||||
| // | ||||
| // TOML hashes correspond to Go structs or maps. (Dealer's choice. They can be | ||||
| // used interchangeably.) | ||||
| // | ||||
| // TOML arrays of tables correspond to either a slice of structs or a slice | ||||
| // of maps. | ||||
| // | ||||
| // TOML datetimes correspond to Go `time.Time` values. | ||||
| // | ||||
| // All other TOML types (float, string, int, bool and array) correspond | ||||
| // to the obvious Go types. | ||||
| // | ||||
| // An exception to the above rules is if a type implements the | ||||
| // encoding.TextUnmarshaler interface. In this case, any primitive TOML value | ||||
| // (floats, strings, integers, booleans and datetimes) will be converted to | ||||
| // a byte string and given to the value's UnmarshalText method. See the | ||||
| // Unmarshaler example for a demonstration with time duration strings. | ||||
| // | ||||
| // Key mapping | ||||
| // | ||||
| // TOML keys can map to either keys in a Go map or field names in a Go | ||||
| // struct. The special `toml` struct tag may be used to map TOML keys to | ||||
| // struct fields that don't match the key name exactly. (See the example.) | ||||
| // A case insensitive match to struct names will be tried if an exact match | ||||
| // can't be found. | ||||
| // | ||||
| // The mapping between TOML values and Go values is loose. That is, there | ||||
| // may exist TOML values that cannot be placed into your representation, and | ||||
| // there may be parts of your representation that do not correspond to | ||||
| // TOML values. This loose mapping can be made stricter by using the IsDefined | ||||
| // and/or Undecoded methods on the MetaData returned. | ||||
| // | ||||
| // This decoder will not handle cyclic types. If a cyclic type is passed, | ||||
| // `Decode` will not terminate. | ||||
| func Decode(data string, v interface{}) (MetaData, error) { | ||||
| 	rv := reflect.ValueOf(v) | ||||
| 	if rv.Kind() != reflect.Ptr { | ||||
| 		return MetaData{}, e("Decode of non-pointer %s", reflect.TypeOf(v)) | ||||
| 	} | ||||
| 	if rv.IsNil() { | ||||
| 		return MetaData{}, e("Decode of nil %s", reflect.TypeOf(v)) | ||||
| 	} | ||||
| 	p, err := parse(data) | ||||
| 	if err != nil { | ||||
| 		return MetaData{}, err | ||||
| 	} | ||||
| 	md := MetaData{ | ||||
| 		p.mapping, p.types, p.ordered, | ||||
| 		make(map[string]bool, len(p.ordered)), nil, | ||||
| 	} | ||||
| 	return md, md.unify(p.mapping, indirect(rv)) | ||||
| } | ||||
|  | ||||
| // DecodeFile is just like Decode, except it will automatically read the | ||||
| // contents of the file at `fpath` and decode it for you. | ||||
| func DecodeFile(fpath string, v interface{}) (MetaData, error) { | ||||
| 	bs, err := ioutil.ReadFile(fpath) | ||||
| 	if err != nil { | ||||
| 		return MetaData{}, err | ||||
| 	} | ||||
| 	return Decode(string(bs), v) | ||||
| } | ||||
|  | ||||
| // DecodeReader is just like Decode, except it will consume all bytes | ||||
| // from the reader and decode it for you. | ||||
| func DecodeReader(r io.Reader, v interface{}) (MetaData, error) { | ||||
| 	bs, err := ioutil.ReadAll(r) | ||||
| 	if err != nil { | ||||
| 		return MetaData{}, err | ||||
| 	} | ||||
| 	return Decode(string(bs), v) | ||||
| } | ||||
|  | ||||
| // unify performs a sort of type unification based on the structure of `rv`, | ||||
| // which is the client representation. | ||||
| // | ||||
| // Any type mismatch produces an error. Finding a type that we don't know | ||||
| // how to handle produces an unsupported type error. | ||||
| func (md *MetaData) unify(data interface{}, rv reflect.Value) error { | ||||
|  | ||||
| 	// Special case. Look for a `Primitive` value. | ||||
| 	if rv.Type() == reflect.TypeOf((*Primitive)(nil)).Elem() { | ||||
| 		// Save the undecoded data and the key context into the primitive | ||||
| 		// value. | ||||
| 		context := make(Key, len(md.context)) | ||||
| 		copy(context, md.context) | ||||
| 		rv.Set(reflect.ValueOf(Primitive{ | ||||
| 			undecoded: data, | ||||
| 			context:   context, | ||||
| 		})) | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// Special case. Unmarshaler Interface support. | ||||
| 	if rv.CanAddr() { | ||||
| 		if v, ok := rv.Addr().Interface().(Unmarshaler); ok { | ||||
| 			return v.UnmarshalTOML(data) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Special case. Handle time.Time values specifically. | ||||
| 	// TODO: Remove this code when we decide to drop support for Go 1.1. | ||||
| 	// This isn't necessary in Go 1.2 because time.Time satisfies the encoding | ||||
| 	// interfaces. | ||||
| 	if rv.Type().AssignableTo(rvalue(time.Time{}).Type()) { | ||||
| 		return md.unifyDatetime(data, rv) | ||||
| 	} | ||||
|  | ||||
| 	// Special case. Look for a value satisfying the TextUnmarshaler interface. | ||||
| 	if v, ok := rv.Interface().(TextUnmarshaler); ok { | ||||
| 		return md.unifyText(data, v) | ||||
| 	} | ||||
| 	// BUG(burntsushi) | ||||
| 	// The behavior here is incorrect whenever a Go type satisfies the | ||||
| 	// encoding.TextUnmarshaler interface but also corresponds to a TOML | ||||
| 	// hash or array. In particular, the unmarshaler should only be applied | ||||
| 	// to primitive TOML values. But at this point, it will be applied to | ||||
| 	// all kinds of values and produce an incorrect error whenever those values | ||||
| 	// are hashes or arrays (including arrays of tables). | ||||
|  | ||||
| 	k := rv.Kind() | ||||
|  | ||||
| 	// laziness | ||||
| 	if k >= reflect.Int && k <= reflect.Uint64 { | ||||
| 		return md.unifyInt(data, rv) | ||||
| 	} | ||||
| 	switch k { | ||||
| 	case reflect.Ptr: | ||||
| 		elem := reflect.New(rv.Type().Elem()) | ||||
| 		err := md.unify(data, reflect.Indirect(elem)) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		rv.Set(elem) | ||||
| 		return nil | ||||
| 	case reflect.Struct: | ||||
| 		return md.unifyStruct(data, rv) | ||||
| 	case reflect.Map: | ||||
| 		return md.unifyMap(data, rv) | ||||
| 	case reflect.Array: | ||||
| 		return md.unifyArray(data, rv) | ||||
| 	case reflect.Slice: | ||||
| 		return md.unifySlice(data, rv) | ||||
| 	case reflect.String: | ||||
| 		return md.unifyString(data, rv) | ||||
| 	case reflect.Bool: | ||||
| 		return md.unifyBool(data, rv) | ||||
| 	case reflect.Interface: | ||||
| 		// we only support empty interfaces. | ||||
| 		if rv.NumMethod() > 0 { | ||||
| 			return e("unsupported type %s", rv.Type()) | ||||
| 		} | ||||
| 		return md.unifyAnything(data, rv) | ||||
| 	case reflect.Float32: | ||||
| 		fallthrough | ||||
| 	case reflect.Float64: | ||||
| 		return md.unifyFloat64(data, rv) | ||||
| 	} | ||||
| 	return e("unsupported type %s", rv.Kind()) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyStruct(mapping interface{}, rv reflect.Value) error { | ||||
| 	tmap, ok := mapping.(map[string]interface{}) | ||||
| 	if !ok { | ||||
| 		if mapping == nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return e("type mismatch for %s: expected table but found %T", | ||||
| 			rv.Type().String(), mapping) | ||||
| 	} | ||||
|  | ||||
| 	for key, datum := range tmap { | ||||
| 		var f *field | ||||
| 		fields := cachedTypeFields(rv.Type()) | ||||
| 		for i := range fields { | ||||
| 			ff := &fields[i] | ||||
| 			if ff.name == key { | ||||
| 				f = ff | ||||
| 				break | ||||
| 			} | ||||
| 			if f == nil && strings.EqualFold(ff.name, key) { | ||||
| 				f = ff | ||||
| 			} | ||||
| 		} | ||||
| 		if f != nil { | ||||
| 			subv := rv | ||||
| 			for _, i := range f.index { | ||||
| 				subv = indirect(subv.Field(i)) | ||||
| 			} | ||||
| 			if isUnifiable(subv) { | ||||
| 				md.decoded[md.context.add(key).String()] = true | ||||
| 				md.context = append(md.context, key) | ||||
| 				if err := md.unify(datum, subv); err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 				md.context = md.context[0 : len(md.context)-1] | ||||
| 			} else if f.name != "" { | ||||
| 				// Bad user! No soup for you! | ||||
| 				return e("cannot write unexported field %s.%s", | ||||
| 					rv.Type().String(), f.name) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyMap(mapping interface{}, rv reflect.Value) error { | ||||
| 	tmap, ok := mapping.(map[string]interface{}) | ||||
| 	if !ok { | ||||
| 		if tmap == nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return badtype("map", mapping) | ||||
| 	} | ||||
| 	if rv.IsNil() { | ||||
| 		rv.Set(reflect.MakeMap(rv.Type())) | ||||
| 	} | ||||
| 	for k, v := range tmap { | ||||
| 		md.decoded[md.context.add(k).String()] = true | ||||
| 		md.context = append(md.context, k) | ||||
|  | ||||
| 		rvkey := indirect(reflect.New(rv.Type().Key())) | ||||
| 		rvval := reflect.Indirect(reflect.New(rv.Type().Elem())) | ||||
| 		if err := md.unify(v, rvval); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		md.context = md.context[0 : len(md.context)-1] | ||||
|  | ||||
| 		rvkey.SetString(k) | ||||
| 		rv.SetMapIndex(rvkey, rvval) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyArray(data interface{}, rv reflect.Value) error { | ||||
| 	datav := reflect.ValueOf(data) | ||||
| 	if datav.Kind() != reflect.Slice { | ||||
| 		if !datav.IsValid() { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return badtype("slice", data) | ||||
| 	} | ||||
| 	sliceLen := datav.Len() | ||||
| 	if sliceLen != rv.Len() { | ||||
| 		return e("expected array length %d; got TOML array of length %d", | ||||
| 			rv.Len(), sliceLen) | ||||
| 	} | ||||
| 	return md.unifySliceArray(datav, rv) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifySlice(data interface{}, rv reflect.Value) error { | ||||
| 	datav := reflect.ValueOf(data) | ||||
| 	if datav.Kind() != reflect.Slice { | ||||
| 		if !datav.IsValid() { | ||||
| 			return nil | ||||
| 		} | ||||
| 		return badtype("slice", data) | ||||
| 	} | ||||
| 	n := datav.Len() | ||||
| 	if rv.IsNil() || rv.Cap() < n { | ||||
| 		rv.Set(reflect.MakeSlice(rv.Type(), n, n)) | ||||
| 	} | ||||
| 	rv.SetLen(n) | ||||
| 	return md.unifySliceArray(datav, rv) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifySliceArray(data, rv reflect.Value) error { | ||||
| 	sliceLen := data.Len() | ||||
| 	for i := 0; i < sliceLen; i++ { | ||||
| 		v := data.Index(i).Interface() | ||||
| 		sliceval := indirect(rv.Index(i)) | ||||
| 		if err := md.unify(v, sliceval); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyDatetime(data interface{}, rv reflect.Value) error { | ||||
| 	if _, ok := data.(time.Time); ok { | ||||
| 		rv.Set(reflect.ValueOf(data)) | ||||
| 		return nil | ||||
| 	} | ||||
| 	return badtype("time.Time", data) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyString(data interface{}, rv reflect.Value) error { | ||||
| 	if s, ok := data.(string); ok { | ||||
| 		rv.SetString(s) | ||||
| 		return nil | ||||
| 	} | ||||
| 	return badtype("string", data) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyFloat64(data interface{}, rv reflect.Value) error { | ||||
| 	if num, ok := data.(float64); ok { | ||||
| 		switch rv.Kind() { | ||||
| 		case reflect.Float32: | ||||
| 			fallthrough | ||||
| 		case reflect.Float64: | ||||
| 			rv.SetFloat(num) | ||||
| 		default: | ||||
| 			panic("bug") | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	return badtype("float", data) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyInt(data interface{}, rv reflect.Value) error { | ||||
| 	if num, ok := data.(int64); ok { | ||||
| 		if rv.Kind() >= reflect.Int && rv.Kind() <= reflect.Int64 { | ||||
| 			switch rv.Kind() { | ||||
| 			case reflect.Int, reflect.Int64: | ||||
| 				// No bounds checking necessary. | ||||
| 			case reflect.Int8: | ||||
| 				if num < math.MinInt8 || num > math.MaxInt8 { | ||||
| 					return e("value %d is out of range for int8", num) | ||||
| 				} | ||||
| 			case reflect.Int16: | ||||
| 				if num < math.MinInt16 || num > math.MaxInt16 { | ||||
| 					return e("value %d is out of range for int16", num) | ||||
| 				} | ||||
| 			case reflect.Int32: | ||||
| 				if num < math.MinInt32 || num > math.MaxInt32 { | ||||
| 					return e("value %d is out of range for int32", num) | ||||
| 				} | ||||
| 			} | ||||
| 			rv.SetInt(num) | ||||
| 		} else if rv.Kind() >= reflect.Uint && rv.Kind() <= reflect.Uint64 { | ||||
| 			unum := uint64(num) | ||||
| 			switch rv.Kind() { | ||||
| 			case reflect.Uint, reflect.Uint64: | ||||
| 				// No bounds checking necessary. | ||||
| 			case reflect.Uint8: | ||||
| 				if num < 0 || unum > math.MaxUint8 { | ||||
| 					return e("value %d is out of range for uint8", num) | ||||
| 				} | ||||
| 			case reflect.Uint16: | ||||
| 				if num < 0 || unum > math.MaxUint16 { | ||||
| 					return e("value %d is out of range for uint16", num) | ||||
| 				} | ||||
| 			case reflect.Uint32: | ||||
| 				if num < 0 || unum > math.MaxUint32 { | ||||
| 					return e("value %d is out of range for uint32", num) | ||||
| 				} | ||||
| 			} | ||||
| 			rv.SetUint(unum) | ||||
| 		} else { | ||||
| 			panic("unreachable") | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	return badtype("integer", data) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyBool(data interface{}, rv reflect.Value) error { | ||||
| 	if b, ok := data.(bool); ok { | ||||
| 		rv.SetBool(b) | ||||
| 		return nil | ||||
| 	} | ||||
| 	return badtype("boolean", data) | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyAnything(data interface{}, rv reflect.Value) error { | ||||
| 	rv.Set(reflect.ValueOf(data)) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (md *MetaData) unifyText(data interface{}, v TextUnmarshaler) error { | ||||
| 	var s string | ||||
| 	switch sdata := data.(type) { | ||||
| 	case TextMarshaler: | ||||
| 		text, err := sdata.MarshalText() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		s = string(text) | ||||
| 	case fmt.Stringer: | ||||
| 		s = sdata.String() | ||||
| 	case string: | ||||
| 		s = sdata | ||||
| 	case bool: | ||||
| 		s = fmt.Sprintf("%v", sdata) | ||||
| 	case int64: | ||||
| 		s = fmt.Sprintf("%d", sdata) | ||||
| 	case float64: | ||||
| 		s = fmt.Sprintf("%f", sdata) | ||||
| 	default: | ||||
| 		return badtype("primitive (string-like)", data) | ||||
| 	} | ||||
| 	if err := v.UnmarshalText([]byte(s)); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // rvalue returns a reflect.Value of `v`. All pointers are resolved. | ||||
| func rvalue(v interface{}) reflect.Value { | ||||
| 	return indirect(reflect.ValueOf(v)) | ||||
| } | ||||
|  | ||||
| // indirect returns the value pointed to by a pointer. | ||||
| // Pointers are followed until the value is not a pointer. | ||||
| // New values are allocated for each nil pointer. | ||||
| // | ||||
| // An exception to this rule is if the value satisfies an interface of | ||||
| // interest to us (like encoding.TextUnmarshaler). | ||||
| func indirect(v reflect.Value) reflect.Value { | ||||
| 	if v.Kind() != reflect.Ptr { | ||||
| 		if v.CanSet() { | ||||
| 			pv := v.Addr() | ||||
| 			if _, ok := pv.Interface().(TextUnmarshaler); ok { | ||||
| 				return pv | ||||
| 			} | ||||
| 		} | ||||
| 		return v | ||||
| 	} | ||||
| 	if v.IsNil() { | ||||
| 		v.Set(reflect.New(v.Type().Elem())) | ||||
| 	} | ||||
| 	return indirect(reflect.Indirect(v)) | ||||
| } | ||||
|  | ||||
| func isUnifiable(rv reflect.Value) bool { | ||||
| 	if rv.CanSet() { | ||||
| 		return true | ||||
| 	} | ||||
| 	if _, ok := rv.Interface().(TextUnmarshaler); ok { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func badtype(expected string, data interface{}) error { | ||||
| 	return e("cannot load TOML value of type %T into a Go %s", data, expected) | ||||
| } | ||||
							
								
								
									
										121
									
								
								vendor/github.com/BurntSushi/toml/decode_meta.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,121 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| import "strings" | ||||
|  | ||||
| // MetaData allows access to meta information about TOML data that may not | ||||
| // be inferrable via reflection. In particular, whether a key has been defined | ||||
| // and the TOML type of a key. | ||||
| type MetaData struct { | ||||
| 	mapping map[string]interface{} | ||||
| 	types   map[string]tomlType | ||||
| 	keys    []Key | ||||
| 	decoded map[string]bool | ||||
| 	context Key // Used only during decoding. | ||||
| } | ||||
|  | ||||
| // IsDefined returns true if the key given exists in the TOML data. The key | ||||
| // should be specified hierarchially. e.g., | ||||
| // | ||||
| //	// access the TOML key 'a.b.c' | ||||
| //	IsDefined("a", "b", "c") | ||||
| // | ||||
| // IsDefined will return false if an empty key given. Keys are case sensitive. | ||||
| func (md *MetaData) IsDefined(key ...string) bool { | ||||
| 	if len(key) == 0 { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	var hash map[string]interface{} | ||||
| 	var ok bool | ||||
| 	var hashOrVal interface{} = md.mapping | ||||
| 	for _, k := range key { | ||||
| 		if hash, ok = hashOrVal.(map[string]interface{}); !ok { | ||||
| 			return false | ||||
| 		} | ||||
| 		if hashOrVal, ok = hash[k]; !ok { | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| // Type returns a string representation of the type of the key specified. | ||||
| // | ||||
| // Type will return the empty string if given an empty key or a key that | ||||
| // does not exist. Keys are case sensitive. | ||||
| func (md *MetaData) Type(key ...string) string { | ||||
| 	fullkey := strings.Join(key, ".") | ||||
| 	if typ, ok := md.types[fullkey]; ok { | ||||
| 		return typ.typeString() | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| // Key is the type of any TOML key, including key groups. Use (MetaData).Keys | ||||
| // to get values of this type. | ||||
| type Key []string | ||||
|  | ||||
| func (k Key) String() string { | ||||
| 	return strings.Join(k, ".") | ||||
| } | ||||
|  | ||||
| func (k Key) maybeQuotedAll() string { | ||||
| 	var ss []string | ||||
| 	for i := range k { | ||||
| 		ss = append(ss, k.maybeQuoted(i)) | ||||
| 	} | ||||
| 	return strings.Join(ss, ".") | ||||
| } | ||||
|  | ||||
| func (k Key) maybeQuoted(i int) string { | ||||
| 	quote := false | ||||
| 	for _, c := range k[i] { | ||||
| 		if !isBareKeyChar(c) { | ||||
| 			quote = true | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	if quote { | ||||
| 		return "\"" + strings.Replace(k[i], "\"", "\\\"", -1) + "\"" | ||||
| 	} | ||||
| 	return k[i] | ||||
| } | ||||
|  | ||||
| func (k Key) add(piece string) Key { | ||||
| 	newKey := make(Key, len(k)+1) | ||||
| 	copy(newKey, k) | ||||
| 	newKey[len(k)] = piece | ||||
| 	return newKey | ||||
| } | ||||
|  | ||||
| // Keys returns a slice of every key in the TOML data, including key groups. | ||||
| // Each key is itself a slice, where the first element is the top of the | ||||
| // hierarchy and the last is the most specific. | ||||
| // | ||||
| // The list will have the same order as the keys appeared in the TOML data. | ||||
| // | ||||
| // All keys returned are non-empty. | ||||
| func (md *MetaData) Keys() []Key { | ||||
| 	return md.keys | ||||
| } | ||||
|  | ||||
| // Undecoded returns all keys that have not been decoded in the order in which | ||||
| // they appear in the original TOML document. | ||||
| // | ||||
| // This includes keys that haven't been decoded because of a Primitive value. | ||||
| // Once the Primitive value is decoded, the keys will be considered decoded. | ||||
| // | ||||
| // Also note that decoding into an empty interface will result in no decoding, | ||||
| // and so no keys will be considered decoded. | ||||
| // | ||||
| // In this sense, the Undecoded keys correspond to keys in the TOML document | ||||
| // that do not have a concrete type in your representation. | ||||
| func (md *MetaData) Undecoded() []Key { | ||||
| 	undecoded := make([]Key, 0, len(md.keys)) | ||||
| 	for _, key := range md.keys { | ||||
| 		if !md.decoded[key.String()] { | ||||
| 			undecoded = append(undecoded, key) | ||||
| 		} | ||||
| 	} | ||||
| 	return undecoded | ||||
| } | ||||
							
								
								
									
										27
									
								
								vendor/github.com/BurntSushi/toml/doc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,27 +0,0 @@ | ||||
| /* | ||||
| Package toml provides facilities for decoding and encoding TOML configuration | ||||
| files via reflection. There is also support for delaying decoding with | ||||
| the Primitive type, and querying the set of keys in a TOML document with the | ||||
| MetaData type. | ||||
|  | ||||
| The specification implemented: https://github.com/toml-lang/toml | ||||
|  | ||||
| The sub-command github.com/BurntSushi/toml/cmd/tomlv can be used to verify | ||||
| whether a file is a valid TOML document. It can also be used to print the | ||||
| type of each key in a TOML document. | ||||
|  | ||||
| Testing | ||||
|  | ||||
| There are two important types of tests used for this package. The first is | ||||
| contained inside '*_test.go' files and uses the standard Go unit testing | ||||
| framework. These tests are primarily devoted to holistically testing the | ||||
| decoder and encoder. | ||||
|  | ||||
| The second type of testing is used to verify the implementation's adherence | ||||
| to the TOML specification. These tests have been factored into their own | ||||
| project: https://github.com/BurntSushi/toml-test | ||||
|  | ||||
| The reason the tests are in a separate project is so that they can be used by | ||||
| any implementation of TOML. Namely, it is language agnostic. | ||||
| */ | ||||
| package toml | ||||
							
								
								
									
										568
									
								
								vendor/github.com/BurntSushi/toml/encode.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,568 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"reflect" | ||||
| 	"sort" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| type tomlEncodeError struct{ error } | ||||
|  | ||||
| var ( | ||||
| 	errArrayMixedElementTypes = errors.New( | ||||
| 		"toml: cannot encode array with mixed element types") | ||||
| 	errArrayNilElement = errors.New( | ||||
| 		"toml: cannot encode array with nil element") | ||||
| 	errNonString = errors.New( | ||||
| 		"toml: cannot encode a map with non-string key type") | ||||
| 	errAnonNonStruct = errors.New( | ||||
| 		"toml: cannot encode an anonymous field that is not a struct") | ||||
| 	errArrayNoTable = errors.New( | ||||
| 		"toml: TOML array element cannot contain a table") | ||||
| 	errNoKey = errors.New( | ||||
| 		"toml: top-level values must be Go maps or structs") | ||||
| 	errAnything = errors.New("") // used in testing | ||||
| ) | ||||
|  | ||||
| var quotedReplacer = strings.NewReplacer( | ||||
| 	"\t", "\\t", | ||||
| 	"\n", "\\n", | ||||
| 	"\r", "\\r", | ||||
| 	"\"", "\\\"", | ||||
| 	"\\", "\\\\", | ||||
| ) | ||||
|  | ||||
| // Encoder controls the encoding of Go values to a TOML document to some | ||||
| // io.Writer. | ||||
| // | ||||
| // The indentation level can be controlled with the Indent field. | ||||
| type Encoder struct { | ||||
| 	// A single indentation level. By default it is two spaces. | ||||
| 	Indent string | ||||
|  | ||||
| 	// hasWritten is whether we have written any output to w yet. | ||||
| 	hasWritten bool | ||||
| 	w          *bufio.Writer | ||||
| } | ||||
|  | ||||
| // NewEncoder returns a TOML encoder that encodes Go values to the io.Writer | ||||
| // given. By default, a single indentation level is 2 spaces. | ||||
| func NewEncoder(w io.Writer) *Encoder { | ||||
| 	return &Encoder{ | ||||
| 		w:      bufio.NewWriter(w), | ||||
| 		Indent: "  ", | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Encode writes a TOML representation of the Go value to the underlying | ||||
| // io.Writer. If the value given cannot be encoded to a valid TOML document, | ||||
| // then an error is returned. | ||||
| // | ||||
| // The mapping between Go values and TOML values should be precisely the same | ||||
| // as for the Decode* functions. Similarly, the TextMarshaler interface is | ||||
| // supported by encoding the resulting bytes as strings. (If you want to write | ||||
| // arbitrary binary data then you will need to use something like base64 since | ||||
| // TOML does not have any binary types.) | ||||
| // | ||||
| // When encoding TOML hashes (i.e., Go maps or structs), keys without any | ||||
| // sub-hashes are encoded first. | ||||
| // | ||||
| // If a Go map is encoded, then its keys are sorted alphabetically for | ||||
| // deterministic output. More control over this behavior may be provided if | ||||
| // there is demand for it. | ||||
| // | ||||
| // Encoding Go values without a corresponding TOML representation---like map | ||||
| // types with non-string keys---will cause an error to be returned. Similarly | ||||
| // for mixed arrays/slices, arrays/slices with nil elements, embedded | ||||
| // non-struct types and nested slices containing maps or structs. | ||||
| // (e.g., [][]map[string]string is not allowed but []map[string]string is OK | ||||
| // and so is []map[string][]string.) | ||||
| func (enc *Encoder) Encode(v interface{}) error { | ||||
| 	rv := eindirect(reflect.ValueOf(v)) | ||||
| 	if err := enc.safeEncode(Key([]string{}), rv); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return enc.w.Flush() | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) safeEncode(key Key, rv reflect.Value) (err error) { | ||||
| 	defer func() { | ||||
| 		if r := recover(); r != nil { | ||||
| 			if terr, ok := r.(tomlEncodeError); ok { | ||||
| 				err = terr.error | ||||
| 				return | ||||
| 			} | ||||
| 			panic(r) | ||||
| 		} | ||||
| 	}() | ||||
| 	enc.encode(key, rv) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) encode(key Key, rv reflect.Value) { | ||||
| 	// Special case. Time needs to be in ISO8601 format. | ||||
| 	// Special case. If we can marshal the type to text, then we used that. | ||||
| 	// Basically, this prevents the encoder for handling these types as | ||||
| 	// generic structs (or whatever the underlying type of a TextMarshaler is). | ||||
| 	switch rv.Interface().(type) { | ||||
| 	case time.Time, TextMarshaler: | ||||
| 		enc.keyEqElement(key, rv) | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	k := rv.Kind() | ||||
| 	switch k { | ||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, | ||||
| 		reflect.Int64, | ||||
| 		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, | ||||
| 		reflect.Uint64, | ||||
| 		reflect.Float32, reflect.Float64, reflect.String, reflect.Bool: | ||||
| 		enc.keyEqElement(key, rv) | ||||
| 	case reflect.Array, reflect.Slice: | ||||
| 		if typeEqual(tomlArrayHash, tomlTypeOfGo(rv)) { | ||||
| 			enc.eArrayOfTables(key, rv) | ||||
| 		} else { | ||||
| 			enc.keyEqElement(key, rv) | ||||
| 		} | ||||
| 	case reflect.Interface: | ||||
| 		if rv.IsNil() { | ||||
| 			return | ||||
| 		} | ||||
| 		enc.encode(key, rv.Elem()) | ||||
| 	case reflect.Map: | ||||
| 		if rv.IsNil() { | ||||
| 			return | ||||
| 		} | ||||
| 		enc.eTable(key, rv) | ||||
| 	case reflect.Ptr: | ||||
| 		if rv.IsNil() { | ||||
| 			return | ||||
| 		} | ||||
| 		enc.encode(key, rv.Elem()) | ||||
| 	case reflect.Struct: | ||||
| 		enc.eTable(key, rv) | ||||
| 	default: | ||||
| 		panic(e("unsupported type for key '%s': %s", key, k)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // eElement encodes any value that can be an array element (primitives and | ||||
| // arrays). | ||||
| func (enc *Encoder) eElement(rv reflect.Value) { | ||||
| 	switch v := rv.Interface().(type) { | ||||
| 	case time.Time: | ||||
| 		// Special case time.Time as a primitive. Has to come before | ||||
| 		// TextMarshaler below because time.Time implements | ||||
| 		// encoding.TextMarshaler, but we need to always use UTC. | ||||
| 		enc.wf(v.UTC().Format("2006-01-02T15:04:05Z")) | ||||
| 		return | ||||
| 	case TextMarshaler: | ||||
| 		// Special case. Use text marshaler if it's available for this value. | ||||
| 		if s, err := v.MarshalText(); err != nil { | ||||
| 			encPanic(err) | ||||
| 		} else { | ||||
| 			enc.writeQuoted(string(s)) | ||||
| 		} | ||||
| 		return | ||||
| 	} | ||||
| 	switch rv.Kind() { | ||||
| 	case reflect.Bool: | ||||
| 		enc.wf(strconv.FormatBool(rv.Bool())) | ||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, | ||||
| 		reflect.Int64: | ||||
| 		enc.wf(strconv.FormatInt(rv.Int(), 10)) | ||||
| 	case reflect.Uint, reflect.Uint8, reflect.Uint16, | ||||
| 		reflect.Uint32, reflect.Uint64: | ||||
| 		enc.wf(strconv.FormatUint(rv.Uint(), 10)) | ||||
| 	case reflect.Float32: | ||||
| 		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 32))) | ||||
| 	case reflect.Float64: | ||||
| 		enc.wf(floatAddDecimal(strconv.FormatFloat(rv.Float(), 'f', -1, 64))) | ||||
| 	case reflect.Array, reflect.Slice: | ||||
| 		enc.eArrayOrSliceElement(rv) | ||||
| 	case reflect.Interface: | ||||
| 		enc.eElement(rv.Elem()) | ||||
| 	case reflect.String: | ||||
| 		enc.writeQuoted(rv.String()) | ||||
| 	default: | ||||
| 		panic(e("unexpected primitive type: %s", rv.Kind())) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // By the TOML spec, all floats must have a decimal with at least one | ||||
| // number on either side. | ||||
| func floatAddDecimal(fstr string) string { | ||||
| 	if !strings.Contains(fstr, ".") { | ||||
| 		return fstr + ".0" | ||||
| 	} | ||||
| 	return fstr | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) writeQuoted(s string) { | ||||
| 	enc.wf("\"%s\"", quotedReplacer.Replace(s)) | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eArrayOrSliceElement(rv reflect.Value) { | ||||
| 	length := rv.Len() | ||||
| 	enc.wf("[") | ||||
| 	for i := 0; i < length; i++ { | ||||
| 		elem := rv.Index(i) | ||||
| 		enc.eElement(elem) | ||||
| 		if i != length-1 { | ||||
| 			enc.wf(", ") | ||||
| 		} | ||||
| 	} | ||||
| 	enc.wf("]") | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eArrayOfTables(key Key, rv reflect.Value) { | ||||
| 	if len(key) == 0 { | ||||
| 		encPanic(errNoKey) | ||||
| 	} | ||||
| 	for i := 0; i < rv.Len(); i++ { | ||||
| 		trv := rv.Index(i) | ||||
| 		if isNil(trv) { | ||||
| 			continue | ||||
| 		} | ||||
| 		panicIfInvalidKey(key) | ||||
| 		enc.newline() | ||||
| 		enc.wf("%s[[%s]]", enc.indentStr(key), key.maybeQuotedAll()) | ||||
| 		enc.newline() | ||||
| 		enc.eMapOrStruct(key, trv) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eTable(key Key, rv reflect.Value) { | ||||
| 	panicIfInvalidKey(key) | ||||
| 	if len(key) == 1 { | ||||
| 		// Output an extra newline between top-level tables. | ||||
| 		// (The newline isn't written if nothing else has been written though.) | ||||
| 		enc.newline() | ||||
| 	} | ||||
| 	if len(key) > 0 { | ||||
| 		enc.wf("%s[%s]", enc.indentStr(key), key.maybeQuotedAll()) | ||||
| 		enc.newline() | ||||
| 	} | ||||
| 	enc.eMapOrStruct(key, rv) | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eMapOrStruct(key Key, rv reflect.Value) { | ||||
| 	switch rv := eindirect(rv); rv.Kind() { | ||||
| 	case reflect.Map: | ||||
| 		enc.eMap(key, rv) | ||||
| 	case reflect.Struct: | ||||
| 		enc.eStruct(key, rv) | ||||
| 	default: | ||||
| 		panic("eTable: unhandled reflect.Value Kind: " + rv.Kind().String()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eMap(key Key, rv reflect.Value) { | ||||
| 	rt := rv.Type() | ||||
| 	if rt.Key().Kind() != reflect.String { | ||||
| 		encPanic(errNonString) | ||||
| 	} | ||||
|  | ||||
| 	// Sort keys so that we have deterministic output. And write keys directly | ||||
| 	// underneath this key first, before writing sub-structs or sub-maps. | ||||
| 	var mapKeysDirect, mapKeysSub []string | ||||
| 	for _, mapKey := range rv.MapKeys() { | ||||
| 		k := mapKey.String() | ||||
| 		if typeIsHash(tomlTypeOfGo(rv.MapIndex(mapKey))) { | ||||
| 			mapKeysSub = append(mapKeysSub, k) | ||||
| 		} else { | ||||
| 			mapKeysDirect = append(mapKeysDirect, k) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	var writeMapKeys = func(mapKeys []string) { | ||||
| 		sort.Strings(mapKeys) | ||||
| 		for _, mapKey := range mapKeys { | ||||
| 			mrv := rv.MapIndex(reflect.ValueOf(mapKey)) | ||||
| 			if isNil(mrv) { | ||||
| 				// Don't write anything for nil fields. | ||||
| 				continue | ||||
| 			} | ||||
| 			enc.encode(key.add(mapKey), mrv) | ||||
| 		} | ||||
| 	} | ||||
| 	writeMapKeys(mapKeysDirect) | ||||
| 	writeMapKeys(mapKeysSub) | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) eStruct(key Key, rv reflect.Value) { | ||||
| 	// Write keys for fields directly under this key first, because if we write | ||||
| 	// a field that creates a new table, then all keys under it will be in that | ||||
| 	// table (not the one we're writing here). | ||||
| 	rt := rv.Type() | ||||
| 	var fieldsDirect, fieldsSub [][]int | ||||
| 	var addFields func(rt reflect.Type, rv reflect.Value, start []int) | ||||
| 	addFields = func(rt reflect.Type, rv reflect.Value, start []int) { | ||||
| 		for i := 0; i < rt.NumField(); i++ { | ||||
| 			f := rt.Field(i) | ||||
| 			// skip unexported fields | ||||
| 			if f.PkgPath != "" && !f.Anonymous { | ||||
| 				continue | ||||
| 			} | ||||
| 			frv := rv.Field(i) | ||||
| 			if f.Anonymous { | ||||
| 				t := f.Type | ||||
| 				switch t.Kind() { | ||||
| 				case reflect.Struct: | ||||
| 					// Treat anonymous struct fields with | ||||
| 					// tag names as though they are not | ||||
| 					// anonymous, like encoding/json does. | ||||
| 					if getOptions(f.Tag).name == "" { | ||||
| 						addFields(t, frv, f.Index) | ||||
| 						continue | ||||
| 					} | ||||
| 				case reflect.Ptr: | ||||
| 					if t.Elem().Kind() == reflect.Struct && | ||||
| 						getOptions(f.Tag).name == "" { | ||||
| 						if !frv.IsNil() { | ||||
| 							addFields(t.Elem(), frv.Elem(), f.Index) | ||||
| 						} | ||||
| 						continue | ||||
| 					} | ||||
| 					// Fall through to the normal field encoding logic below | ||||
| 					// for non-struct anonymous fields. | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			if typeIsHash(tomlTypeOfGo(frv)) { | ||||
| 				fieldsSub = append(fieldsSub, append(start, f.Index...)) | ||||
| 			} else { | ||||
| 				fieldsDirect = append(fieldsDirect, append(start, f.Index...)) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| 	addFields(rt, rv, nil) | ||||
|  | ||||
| 	var writeFields = func(fields [][]int) { | ||||
| 		for _, fieldIndex := range fields { | ||||
| 			sft := rt.FieldByIndex(fieldIndex) | ||||
| 			sf := rv.FieldByIndex(fieldIndex) | ||||
| 			if isNil(sf) { | ||||
| 				// Don't write anything for nil fields. | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			opts := getOptions(sft.Tag) | ||||
| 			if opts.skip { | ||||
| 				continue | ||||
| 			} | ||||
| 			keyName := sft.Name | ||||
| 			if opts.name != "" { | ||||
| 				keyName = opts.name | ||||
| 			} | ||||
| 			if opts.omitempty && isEmpty(sf) { | ||||
| 				continue | ||||
| 			} | ||||
| 			if opts.omitzero && isZero(sf) { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			enc.encode(key.add(keyName), sf) | ||||
| 		} | ||||
| 	} | ||||
| 	writeFields(fieldsDirect) | ||||
| 	writeFields(fieldsSub) | ||||
| } | ||||
|  | ||||
| // tomlTypeName returns the TOML type name of the Go value's type. It is | ||||
| // used to determine whether the types of array elements are mixed (which is | ||||
| // forbidden). If the Go value is nil, then it is illegal for it to be an array | ||||
| // element, and valueIsNil is returned as true. | ||||
|  | ||||
| // Returns the TOML type of a Go value. The type may be `nil`, which means | ||||
| // no concrete TOML type could be found. | ||||
| func tomlTypeOfGo(rv reflect.Value) tomlType { | ||||
| 	if isNil(rv) || !rv.IsValid() { | ||||
| 		return nil | ||||
| 	} | ||||
| 	switch rv.Kind() { | ||||
| 	case reflect.Bool: | ||||
| 		return tomlBool | ||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, | ||||
| 		reflect.Int64, | ||||
| 		reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, | ||||
| 		reflect.Uint64: | ||||
| 		return tomlInteger | ||||
| 	case reflect.Float32, reflect.Float64: | ||||
| 		return tomlFloat | ||||
| 	case reflect.Array, reflect.Slice: | ||||
| 		if typeEqual(tomlHash, tomlArrayType(rv)) { | ||||
| 			return tomlArrayHash | ||||
| 		} | ||||
| 		return tomlArray | ||||
| 	case reflect.Ptr, reflect.Interface: | ||||
| 		return tomlTypeOfGo(rv.Elem()) | ||||
| 	case reflect.String: | ||||
| 		return tomlString | ||||
| 	case reflect.Map: | ||||
| 		return tomlHash | ||||
| 	case reflect.Struct: | ||||
| 		switch rv.Interface().(type) { | ||||
| 		case time.Time: | ||||
| 			return tomlDatetime | ||||
| 		case TextMarshaler: | ||||
| 			return tomlString | ||||
| 		default: | ||||
| 			return tomlHash | ||||
| 		} | ||||
| 	default: | ||||
| 		panic("unexpected reflect.Kind: " + rv.Kind().String()) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // tomlArrayType returns the element type of a TOML array. The type returned | ||||
| // may be nil if it cannot be determined (e.g., a nil slice or a zero length | ||||
| // slize). This function may also panic if it finds a type that cannot be | ||||
| // expressed in TOML (such as nil elements, heterogeneous arrays or directly | ||||
| // nested arrays of tables). | ||||
| func tomlArrayType(rv reflect.Value) tomlType { | ||||
| 	if isNil(rv) || !rv.IsValid() || rv.Len() == 0 { | ||||
| 		return nil | ||||
| 	} | ||||
| 	firstType := tomlTypeOfGo(rv.Index(0)) | ||||
| 	if firstType == nil { | ||||
| 		encPanic(errArrayNilElement) | ||||
| 	} | ||||
|  | ||||
| 	rvlen := rv.Len() | ||||
| 	for i := 1; i < rvlen; i++ { | ||||
| 		elem := rv.Index(i) | ||||
| 		switch elemType := tomlTypeOfGo(elem); { | ||||
| 		case elemType == nil: | ||||
| 			encPanic(errArrayNilElement) | ||||
| 		case !typeEqual(firstType, elemType): | ||||
| 			encPanic(errArrayMixedElementTypes) | ||||
| 		} | ||||
| 	} | ||||
| 	// If we have a nested array, then we must make sure that the nested | ||||
| 	// array contains ONLY primitives. | ||||
| 	// This checks arbitrarily nested arrays. | ||||
| 	if typeEqual(firstType, tomlArray) || typeEqual(firstType, tomlArrayHash) { | ||||
| 		nest := tomlArrayType(eindirect(rv.Index(0))) | ||||
| 		if typeEqual(nest, tomlHash) || typeEqual(nest, tomlArrayHash) { | ||||
| 			encPanic(errArrayNoTable) | ||||
| 		} | ||||
| 	} | ||||
| 	return firstType | ||||
| } | ||||
|  | ||||
| type tagOptions struct { | ||||
| 	skip      bool // "-" | ||||
| 	name      string | ||||
| 	omitempty bool | ||||
| 	omitzero  bool | ||||
| } | ||||
|  | ||||
| func getOptions(tag reflect.StructTag) tagOptions { | ||||
| 	t := tag.Get("toml") | ||||
| 	if t == "-" { | ||||
| 		return tagOptions{skip: true} | ||||
| 	} | ||||
| 	var opts tagOptions | ||||
| 	parts := strings.Split(t, ",") | ||||
| 	opts.name = parts[0] | ||||
| 	for _, s := range parts[1:] { | ||||
| 		switch s { | ||||
| 		case "omitempty": | ||||
| 			opts.omitempty = true | ||||
| 		case "omitzero": | ||||
| 			opts.omitzero = true | ||||
| 		} | ||||
| 	} | ||||
| 	return opts | ||||
| } | ||||
|  | ||||
| func isZero(rv reflect.Value) bool { | ||||
| 	switch rv.Kind() { | ||||
| 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: | ||||
| 		return rv.Int() == 0 | ||||
| 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: | ||||
| 		return rv.Uint() == 0 | ||||
| 	case reflect.Float32, reflect.Float64: | ||||
| 		return rv.Float() == 0.0 | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func isEmpty(rv reflect.Value) bool { | ||||
| 	switch rv.Kind() { | ||||
| 	case reflect.Array, reflect.Slice, reflect.Map, reflect.String: | ||||
| 		return rv.Len() == 0 | ||||
| 	case reflect.Bool: | ||||
| 		return !rv.Bool() | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) newline() { | ||||
| 	if enc.hasWritten { | ||||
| 		enc.wf("\n") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) keyEqElement(key Key, val reflect.Value) { | ||||
| 	if len(key) == 0 { | ||||
| 		encPanic(errNoKey) | ||||
| 	} | ||||
| 	panicIfInvalidKey(key) | ||||
| 	enc.wf("%s%s = ", enc.indentStr(key), key.maybeQuoted(len(key)-1)) | ||||
| 	enc.eElement(val) | ||||
| 	enc.newline() | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) wf(format string, v ...interface{}) { | ||||
| 	if _, err := fmt.Fprintf(enc.w, format, v...); err != nil { | ||||
| 		encPanic(err) | ||||
| 	} | ||||
| 	enc.hasWritten = true | ||||
| } | ||||
|  | ||||
| func (enc *Encoder) indentStr(key Key) string { | ||||
| 	return strings.Repeat(enc.Indent, len(key)-1) | ||||
| } | ||||
|  | ||||
| func encPanic(err error) { | ||||
| 	panic(tomlEncodeError{err}) | ||||
| } | ||||
|  | ||||
| func eindirect(v reflect.Value) reflect.Value { | ||||
| 	switch v.Kind() { | ||||
| 	case reflect.Ptr, reflect.Interface: | ||||
| 		return eindirect(v.Elem()) | ||||
| 	default: | ||||
| 		return v | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func isNil(rv reflect.Value) bool { | ||||
| 	switch rv.Kind() { | ||||
| 	case reflect.Interface, reflect.Map, reflect.Ptr, reflect.Slice: | ||||
| 		return rv.IsNil() | ||||
| 	default: | ||||
| 		return false | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func panicIfInvalidKey(key Key) { | ||||
| 	for _, k := range key { | ||||
| 		if len(k) == 0 { | ||||
| 			encPanic(e("Key '%s' is not a valid table name. Key names "+ | ||||
| 				"cannot be empty.", key.maybeQuotedAll())) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func isValidKeyName(s string) bool { | ||||
| 	return len(s) != 0 | ||||
| } | ||||
							
								
								
									
										19
									
								
								vendor/github.com/BurntSushi/toml/encoding_types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,19 +0,0 @@ | ||||
| // +build go1.2 | ||||
|  | ||||
| package toml | ||||
|  | ||||
| // In order to support Go 1.1, we define our own TextMarshaler and | ||||
| // TextUnmarshaler types. For Go 1.2+, we just alias them with the | ||||
| // standard library interfaces. | ||||
|  | ||||
| import ( | ||||
| 	"encoding" | ||||
| ) | ||||
|  | ||||
| // TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here | ||||
| // so that Go 1.1 can be supported. | ||||
| type TextMarshaler encoding.TextMarshaler | ||||
|  | ||||
| // TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined | ||||
| // here so that Go 1.1 can be supported. | ||||
| type TextUnmarshaler encoding.TextUnmarshaler | ||||
							
								
								
									
										18
									
								
								vendor/github.com/BurntSushi/toml/encoding_types_1.1.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,18 +0,0 @@ | ||||
| // +build !go1.2 | ||||
|  | ||||
| package toml | ||||
|  | ||||
| // These interfaces were introduced in Go 1.2, so we add them manually when | ||||
| // compiling for Go 1.1. | ||||
|  | ||||
| // TextMarshaler is a synonym for encoding.TextMarshaler. It is defined here | ||||
| // so that Go 1.1 can be supported. | ||||
| type TextMarshaler interface { | ||||
| 	MarshalText() (text []byte, err error) | ||||
| } | ||||
|  | ||||
| // TextUnmarshaler is a synonym for encoding.TextUnmarshaler. It is defined | ||||
| // here so that Go 1.1 can be supported. | ||||
| type TextUnmarshaler interface { | ||||
| 	UnmarshalText(text []byte) error | ||||
| } | ||||
							
								
								
									
										953
									
								
								vendor/github.com/BurntSushi/toml/lex.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,953 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 	"unicode" | ||||
| 	"unicode/utf8" | ||||
| ) | ||||
|  | ||||
| type itemType int | ||||
|  | ||||
| const ( | ||||
| 	itemError itemType = iota | ||||
| 	itemNIL            // used in the parser to indicate no type | ||||
| 	itemEOF | ||||
| 	itemText | ||||
| 	itemString | ||||
| 	itemRawString | ||||
| 	itemMultilineString | ||||
| 	itemRawMultilineString | ||||
| 	itemBool | ||||
| 	itemInteger | ||||
| 	itemFloat | ||||
| 	itemDatetime | ||||
| 	itemArray // the start of an array | ||||
| 	itemArrayEnd | ||||
| 	itemTableStart | ||||
| 	itemTableEnd | ||||
| 	itemArrayTableStart | ||||
| 	itemArrayTableEnd | ||||
| 	itemKeyStart | ||||
| 	itemCommentStart | ||||
| 	itemInlineTableStart | ||||
| 	itemInlineTableEnd | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	eof              = 0 | ||||
| 	comma            = ',' | ||||
| 	tableStart       = '[' | ||||
| 	tableEnd         = ']' | ||||
| 	arrayTableStart  = '[' | ||||
| 	arrayTableEnd    = ']' | ||||
| 	tableSep         = '.' | ||||
| 	keySep           = '=' | ||||
| 	arrayStart       = '[' | ||||
| 	arrayEnd         = ']' | ||||
| 	commentStart     = '#' | ||||
| 	stringStart      = '"' | ||||
| 	stringEnd        = '"' | ||||
| 	rawStringStart   = '\'' | ||||
| 	rawStringEnd     = '\'' | ||||
| 	inlineTableStart = '{' | ||||
| 	inlineTableEnd   = '}' | ||||
| ) | ||||
|  | ||||
| type stateFn func(lx *lexer) stateFn | ||||
|  | ||||
| type lexer struct { | ||||
| 	input string | ||||
| 	start int | ||||
| 	pos   int | ||||
| 	line  int | ||||
| 	state stateFn | ||||
| 	items chan item | ||||
|  | ||||
| 	// Allow for backing up up to three runes. | ||||
| 	// This is necessary because TOML contains 3-rune tokens (""" and '''). | ||||
| 	prevWidths [3]int | ||||
| 	nprev      int // how many of prevWidths are in use | ||||
| 	// If we emit an eof, we can still back up, but it is not OK to call | ||||
| 	// next again. | ||||
| 	atEOF bool | ||||
|  | ||||
| 	// A stack of state functions used to maintain context. | ||||
| 	// The idea is to reuse parts of the state machine in various places. | ||||
| 	// For example, values can appear at the top level or within arbitrarily | ||||
| 	// nested arrays. The last state on the stack is used after a value has | ||||
| 	// been lexed. Similarly for comments. | ||||
| 	stack []stateFn | ||||
| } | ||||
|  | ||||
| type item struct { | ||||
| 	typ  itemType | ||||
| 	val  string | ||||
| 	line int | ||||
| } | ||||
|  | ||||
| func (lx *lexer) nextItem() item { | ||||
| 	for { | ||||
| 		select { | ||||
| 		case item := <-lx.items: | ||||
| 			return item | ||||
| 		default: | ||||
| 			lx.state = lx.state(lx) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func lex(input string) *lexer { | ||||
| 	lx := &lexer{ | ||||
| 		input: input, | ||||
| 		state: lexTop, | ||||
| 		line:  1, | ||||
| 		items: make(chan item, 10), | ||||
| 		stack: make([]stateFn, 0, 10), | ||||
| 	} | ||||
| 	return lx | ||||
| } | ||||
|  | ||||
| func (lx *lexer) push(state stateFn) { | ||||
| 	lx.stack = append(lx.stack, state) | ||||
| } | ||||
|  | ||||
| func (lx *lexer) pop() stateFn { | ||||
| 	if len(lx.stack) == 0 { | ||||
| 		return lx.errorf("BUG in lexer: no states to pop") | ||||
| 	} | ||||
| 	last := lx.stack[len(lx.stack)-1] | ||||
| 	lx.stack = lx.stack[0 : len(lx.stack)-1] | ||||
| 	return last | ||||
| } | ||||
|  | ||||
| func (lx *lexer) current() string { | ||||
| 	return lx.input[lx.start:lx.pos] | ||||
| } | ||||
|  | ||||
| func (lx *lexer) emit(typ itemType) { | ||||
| 	lx.items <- item{typ, lx.current(), lx.line} | ||||
| 	lx.start = lx.pos | ||||
| } | ||||
|  | ||||
| func (lx *lexer) emitTrim(typ itemType) { | ||||
| 	lx.items <- item{typ, strings.TrimSpace(lx.current()), lx.line} | ||||
| 	lx.start = lx.pos | ||||
| } | ||||
|  | ||||
| func (lx *lexer) next() (r rune) { | ||||
| 	if lx.atEOF { | ||||
| 		panic("next called after EOF") | ||||
| 	} | ||||
| 	if lx.pos >= len(lx.input) { | ||||
| 		lx.atEOF = true | ||||
| 		return eof | ||||
| 	} | ||||
|  | ||||
| 	if lx.input[lx.pos] == '\n' { | ||||
| 		lx.line++ | ||||
| 	} | ||||
| 	lx.prevWidths[2] = lx.prevWidths[1] | ||||
| 	lx.prevWidths[1] = lx.prevWidths[0] | ||||
| 	if lx.nprev < 3 { | ||||
| 		lx.nprev++ | ||||
| 	} | ||||
| 	r, w := utf8.DecodeRuneInString(lx.input[lx.pos:]) | ||||
| 	lx.prevWidths[0] = w | ||||
| 	lx.pos += w | ||||
| 	return r | ||||
| } | ||||
|  | ||||
| // ignore skips over the pending input before this point. | ||||
| func (lx *lexer) ignore() { | ||||
| 	lx.start = lx.pos | ||||
| } | ||||
|  | ||||
| // backup steps back one rune. Can be called only twice between calls to next. | ||||
| func (lx *lexer) backup() { | ||||
| 	if lx.atEOF { | ||||
| 		lx.atEOF = false | ||||
| 		return | ||||
| 	} | ||||
| 	if lx.nprev < 1 { | ||||
| 		panic("backed up too far") | ||||
| 	} | ||||
| 	w := lx.prevWidths[0] | ||||
| 	lx.prevWidths[0] = lx.prevWidths[1] | ||||
| 	lx.prevWidths[1] = lx.prevWidths[2] | ||||
| 	lx.nprev-- | ||||
| 	lx.pos -= w | ||||
| 	if lx.pos < len(lx.input) && lx.input[lx.pos] == '\n' { | ||||
| 		lx.line-- | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // accept consumes the next rune if it's equal to `valid`. | ||||
| func (lx *lexer) accept(valid rune) bool { | ||||
| 	if lx.next() == valid { | ||||
| 		return true | ||||
| 	} | ||||
| 	lx.backup() | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| // peek returns but does not consume the next rune in the input. | ||||
| func (lx *lexer) peek() rune { | ||||
| 	r := lx.next() | ||||
| 	lx.backup() | ||||
| 	return r | ||||
| } | ||||
|  | ||||
| // skip ignores all input that matches the given predicate. | ||||
| func (lx *lexer) skip(pred func(rune) bool) { | ||||
| 	for { | ||||
| 		r := lx.next() | ||||
| 		if pred(r) { | ||||
| 			continue | ||||
| 		} | ||||
| 		lx.backup() | ||||
| 		lx.ignore() | ||||
| 		return | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // errorf stops all lexing by emitting an error and returning `nil`. | ||||
| // Note that any value that is a character is escaped if it's a special | ||||
| // character (newlines, tabs, etc.). | ||||
| func (lx *lexer) errorf(format string, values ...interface{}) stateFn { | ||||
| 	lx.items <- item{ | ||||
| 		itemError, | ||||
| 		fmt.Sprintf(format, values...), | ||||
| 		lx.line, | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // lexTop consumes elements at the top level of TOML data. | ||||
| func lexTop(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isWhitespace(r) || isNL(r) { | ||||
| 		return lexSkip(lx, lexTop) | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case commentStart: | ||||
| 		lx.push(lexTop) | ||||
| 		return lexCommentStart | ||||
| 	case tableStart: | ||||
| 		return lexTableStart | ||||
| 	case eof: | ||||
| 		if lx.pos > lx.start { | ||||
| 			return lx.errorf("unexpected EOF") | ||||
| 		} | ||||
| 		lx.emit(itemEOF) | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	// At this point, the only valid item can be a key, so we back up | ||||
| 	// and let the key lexer do the rest. | ||||
| 	lx.backup() | ||||
| 	lx.push(lexTopEnd) | ||||
| 	return lexKeyStart | ||||
| } | ||||
|  | ||||
| // lexTopEnd is entered whenever a top-level item has been consumed. (A value | ||||
| // or a table.) It must see only whitespace, and will turn back to lexTop | ||||
| // upon a newline. If it sees EOF, it will quit the lexer successfully. | ||||
| func lexTopEnd(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case r == commentStart: | ||||
| 		// a comment will read to a newline for us. | ||||
| 		lx.push(lexTop) | ||||
| 		return lexCommentStart | ||||
| 	case isWhitespace(r): | ||||
| 		return lexTopEnd | ||||
| 	case isNL(r): | ||||
| 		lx.ignore() | ||||
| 		return lexTop | ||||
| 	case r == eof: | ||||
| 		lx.emit(itemEOF) | ||||
| 		return nil | ||||
| 	} | ||||
| 	return lx.errorf("expected a top-level item to end with a newline, "+ | ||||
| 		"comment, or EOF, but got %q instead", r) | ||||
| } | ||||
|  | ||||
| // lexTable lexes the beginning of a table. Namely, it makes sure that | ||||
| // it starts with a character other than '.' and ']'. | ||||
| // It assumes that '[' has already been consumed. | ||||
| // It also handles the case that this is an item in an array of tables. | ||||
| // e.g., '[[name]]'. | ||||
| func lexTableStart(lx *lexer) stateFn { | ||||
| 	if lx.peek() == arrayTableStart { | ||||
| 		lx.next() | ||||
| 		lx.emit(itemArrayTableStart) | ||||
| 		lx.push(lexArrayTableEnd) | ||||
| 	} else { | ||||
| 		lx.emit(itemTableStart) | ||||
| 		lx.push(lexTableEnd) | ||||
| 	} | ||||
| 	return lexTableNameStart | ||||
| } | ||||
|  | ||||
| func lexTableEnd(lx *lexer) stateFn { | ||||
| 	lx.emit(itemTableEnd) | ||||
| 	return lexTopEnd | ||||
| } | ||||
|  | ||||
| func lexArrayTableEnd(lx *lexer) stateFn { | ||||
| 	if r := lx.next(); r != arrayTableEnd { | ||||
| 		return lx.errorf("expected end of table array name delimiter %q, "+ | ||||
| 			"but got %q instead", arrayTableEnd, r) | ||||
| 	} | ||||
| 	lx.emit(itemArrayTableEnd) | ||||
| 	return lexTopEnd | ||||
| } | ||||
|  | ||||
| func lexTableNameStart(lx *lexer) stateFn { | ||||
| 	lx.skip(isWhitespace) | ||||
| 	switch r := lx.peek(); { | ||||
| 	case r == tableEnd || r == eof: | ||||
| 		return lx.errorf("unexpected end of table name " + | ||||
| 			"(table names cannot be empty)") | ||||
| 	case r == tableSep: | ||||
| 		return lx.errorf("unexpected table separator " + | ||||
| 			"(table names cannot be empty)") | ||||
| 	case r == stringStart || r == rawStringStart: | ||||
| 		lx.ignore() | ||||
| 		lx.push(lexTableNameEnd) | ||||
| 		return lexValue // reuse string lexing | ||||
| 	default: | ||||
| 		return lexBareTableName | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // lexBareTableName lexes the name of a table. It assumes that at least one | ||||
| // valid character for the table has already been read. | ||||
| func lexBareTableName(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isBareKeyChar(r) { | ||||
| 		return lexBareTableName | ||||
| 	} | ||||
| 	lx.backup() | ||||
| 	lx.emit(itemText) | ||||
| 	return lexTableNameEnd | ||||
| } | ||||
|  | ||||
| // lexTableNameEnd reads the end of a piece of a table name, optionally | ||||
| // consuming whitespace. | ||||
| func lexTableNameEnd(lx *lexer) stateFn { | ||||
| 	lx.skip(isWhitespace) | ||||
| 	switch r := lx.next(); { | ||||
| 	case isWhitespace(r): | ||||
| 		return lexTableNameEnd | ||||
| 	case r == tableSep: | ||||
| 		lx.ignore() | ||||
| 		return lexTableNameStart | ||||
| 	case r == tableEnd: | ||||
| 		return lx.pop() | ||||
| 	default: | ||||
| 		return lx.errorf("expected '.' or ']' to end table name, "+ | ||||
| 			"but got %q instead", r) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // lexKeyStart consumes a key name up until the first non-whitespace character. | ||||
| // lexKeyStart will ignore whitespace. | ||||
| func lexKeyStart(lx *lexer) stateFn { | ||||
| 	r := lx.peek() | ||||
| 	switch { | ||||
| 	case r == keySep: | ||||
| 		return lx.errorf("unexpected key separator %q", keySep) | ||||
| 	case isWhitespace(r) || isNL(r): | ||||
| 		lx.next() | ||||
| 		return lexSkip(lx, lexKeyStart) | ||||
| 	case r == stringStart || r == rawStringStart: | ||||
| 		lx.ignore() | ||||
| 		lx.emit(itemKeyStart) | ||||
| 		lx.push(lexKeyEnd) | ||||
| 		return lexValue // reuse string lexing | ||||
| 	default: | ||||
| 		lx.ignore() | ||||
| 		lx.emit(itemKeyStart) | ||||
| 		return lexBareKey | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // lexBareKey consumes the text of a bare key. Assumes that the first character | ||||
| // (which is not whitespace) has not yet been consumed. | ||||
| func lexBareKey(lx *lexer) stateFn { | ||||
| 	switch r := lx.next(); { | ||||
| 	case isBareKeyChar(r): | ||||
| 		return lexBareKey | ||||
| 	case isWhitespace(r): | ||||
| 		lx.backup() | ||||
| 		lx.emit(itemText) | ||||
| 		return lexKeyEnd | ||||
| 	case r == keySep: | ||||
| 		lx.backup() | ||||
| 		lx.emit(itemText) | ||||
| 		return lexKeyEnd | ||||
| 	default: | ||||
| 		return lx.errorf("bare keys cannot contain %q", r) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // lexKeyEnd consumes the end of a key and trims whitespace (up to the key | ||||
| // separator). | ||||
| func lexKeyEnd(lx *lexer) stateFn { | ||||
| 	switch r := lx.next(); { | ||||
| 	case r == keySep: | ||||
| 		return lexSkip(lx, lexValue) | ||||
| 	case isWhitespace(r): | ||||
| 		return lexSkip(lx, lexKeyEnd) | ||||
| 	default: | ||||
| 		return lx.errorf("expected key separator %q, but got %q instead", | ||||
| 			keySep, r) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // lexValue starts the consumption of a value anywhere a value is expected. | ||||
| // lexValue will ignore whitespace. | ||||
| // After a value is lexed, the last state on the next is popped and returned. | ||||
| func lexValue(lx *lexer) stateFn { | ||||
| 	// We allow whitespace to precede a value, but NOT newlines. | ||||
| 	// In array syntax, the array states are responsible for ignoring newlines. | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case isWhitespace(r): | ||||
| 		return lexSkip(lx, lexValue) | ||||
| 	case isDigit(r): | ||||
| 		lx.backup() // avoid an extra state and use the same as above | ||||
| 		return lexNumberOrDateStart | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case arrayStart: | ||||
| 		lx.ignore() | ||||
| 		lx.emit(itemArray) | ||||
| 		return lexArrayValue | ||||
| 	case inlineTableStart: | ||||
| 		lx.ignore() | ||||
| 		lx.emit(itemInlineTableStart) | ||||
| 		return lexInlineTableValue | ||||
| 	case stringStart: | ||||
| 		if lx.accept(stringStart) { | ||||
| 			if lx.accept(stringStart) { | ||||
| 				lx.ignore() // Ignore """ | ||||
| 				return lexMultilineString | ||||
| 			} | ||||
| 			lx.backup() | ||||
| 		} | ||||
| 		lx.ignore() // ignore the '"' | ||||
| 		return lexString | ||||
| 	case rawStringStart: | ||||
| 		if lx.accept(rawStringStart) { | ||||
| 			if lx.accept(rawStringStart) { | ||||
| 				lx.ignore() // Ignore """ | ||||
| 				return lexMultilineRawString | ||||
| 			} | ||||
| 			lx.backup() | ||||
| 		} | ||||
| 		lx.ignore() // ignore the "'" | ||||
| 		return lexRawString | ||||
| 	case '+', '-': | ||||
| 		return lexNumberStart | ||||
| 	case '.': // special error case, be kind to users | ||||
| 		return lx.errorf("floats must start with a digit, not '.'") | ||||
| 	} | ||||
| 	if unicode.IsLetter(r) { | ||||
| 		// Be permissive here; lexBool will give a nice error if the | ||||
| 		// user wrote something like | ||||
| 		//   x = foo | ||||
| 		// (i.e. not 'true' or 'false' but is something else word-like.) | ||||
| 		lx.backup() | ||||
| 		return lexBool | ||||
| 	} | ||||
| 	return lx.errorf("expected value but found %q instead", r) | ||||
| } | ||||
|  | ||||
| // lexArrayValue consumes one value in an array. It assumes that '[' or ',' | ||||
| // have already been consumed. All whitespace and newlines are ignored. | ||||
| func lexArrayValue(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case isWhitespace(r) || isNL(r): | ||||
| 		return lexSkip(lx, lexArrayValue) | ||||
| 	case r == commentStart: | ||||
| 		lx.push(lexArrayValue) | ||||
| 		return lexCommentStart | ||||
| 	case r == comma: | ||||
| 		return lx.errorf("unexpected comma") | ||||
| 	case r == arrayEnd: | ||||
| 		// NOTE(caleb): The spec isn't clear about whether you can have | ||||
| 		// a trailing comma or not, so we'll allow it. | ||||
| 		return lexArrayEnd | ||||
| 	} | ||||
|  | ||||
| 	lx.backup() | ||||
| 	lx.push(lexArrayValueEnd) | ||||
| 	return lexValue | ||||
| } | ||||
|  | ||||
| // lexArrayValueEnd consumes everything between the end of an array value and | ||||
| // the next value (or the end of the array): it ignores whitespace and newlines | ||||
| // and expects either a ',' or a ']'. | ||||
| func lexArrayValueEnd(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case isWhitespace(r) || isNL(r): | ||||
| 		return lexSkip(lx, lexArrayValueEnd) | ||||
| 	case r == commentStart: | ||||
| 		lx.push(lexArrayValueEnd) | ||||
| 		return lexCommentStart | ||||
| 	case r == comma: | ||||
| 		lx.ignore() | ||||
| 		return lexArrayValue // move on to the next value | ||||
| 	case r == arrayEnd: | ||||
| 		return lexArrayEnd | ||||
| 	} | ||||
| 	return lx.errorf( | ||||
| 		"expected a comma or array terminator %q, but got %q instead", | ||||
| 		arrayEnd, r, | ||||
| 	) | ||||
| } | ||||
|  | ||||
| // lexArrayEnd finishes the lexing of an array. | ||||
| // It assumes that a ']' has just been consumed. | ||||
| func lexArrayEnd(lx *lexer) stateFn { | ||||
| 	lx.ignore() | ||||
| 	lx.emit(itemArrayEnd) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexInlineTableValue consumes one key/value pair in an inline table. | ||||
| // It assumes that '{' or ',' have already been consumed. Whitespace is ignored. | ||||
| func lexInlineTableValue(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case isWhitespace(r): | ||||
| 		return lexSkip(lx, lexInlineTableValue) | ||||
| 	case isNL(r): | ||||
| 		return lx.errorf("newlines not allowed within inline tables") | ||||
| 	case r == commentStart: | ||||
| 		lx.push(lexInlineTableValue) | ||||
| 		return lexCommentStart | ||||
| 	case r == comma: | ||||
| 		return lx.errorf("unexpected comma") | ||||
| 	case r == inlineTableEnd: | ||||
| 		return lexInlineTableEnd | ||||
| 	} | ||||
| 	lx.backup() | ||||
| 	lx.push(lexInlineTableValueEnd) | ||||
| 	return lexKeyStart | ||||
| } | ||||
|  | ||||
| // lexInlineTableValueEnd consumes everything between the end of an inline table | ||||
| // key/value pair and the next pair (or the end of the table): | ||||
| // it ignores whitespace and expects either a ',' or a '}'. | ||||
| func lexInlineTableValueEnd(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case isWhitespace(r): | ||||
| 		return lexSkip(lx, lexInlineTableValueEnd) | ||||
| 	case isNL(r): | ||||
| 		return lx.errorf("newlines not allowed within inline tables") | ||||
| 	case r == commentStart: | ||||
| 		lx.push(lexInlineTableValueEnd) | ||||
| 		return lexCommentStart | ||||
| 	case r == comma: | ||||
| 		lx.ignore() | ||||
| 		return lexInlineTableValue | ||||
| 	case r == inlineTableEnd: | ||||
| 		return lexInlineTableEnd | ||||
| 	} | ||||
| 	return lx.errorf("expected a comma or an inline table terminator %q, "+ | ||||
| 		"but got %q instead", inlineTableEnd, r) | ||||
| } | ||||
|  | ||||
| // lexInlineTableEnd finishes the lexing of an inline table. | ||||
| // It assumes that a '}' has just been consumed. | ||||
| func lexInlineTableEnd(lx *lexer) stateFn { | ||||
| 	lx.ignore() | ||||
| 	lx.emit(itemInlineTableEnd) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexString consumes the inner contents of a string. It assumes that the | ||||
| // beginning '"' has already been consumed and ignored. | ||||
| func lexString(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case r == eof: | ||||
| 		return lx.errorf("unexpected EOF") | ||||
| 	case isNL(r): | ||||
| 		return lx.errorf("strings cannot contain newlines") | ||||
| 	case r == '\\': | ||||
| 		lx.push(lexString) | ||||
| 		return lexStringEscape | ||||
| 	case r == stringEnd: | ||||
| 		lx.backup() | ||||
| 		lx.emit(itemString) | ||||
| 		lx.next() | ||||
| 		lx.ignore() | ||||
| 		return lx.pop() | ||||
| 	} | ||||
| 	return lexString | ||||
| } | ||||
|  | ||||
| // lexMultilineString consumes the inner contents of a string. It assumes that | ||||
| // the beginning '"""' has already been consumed and ignored. | ||||
| func lexMultilineString(lx *lexer) stateFn { | ||||
| 	switch lx.next() { | ||||
| 	case eof: | ||||
| 		return lx.errorf("unexpected EOF") | ||||
| 	case '\\': | ||||
| 		return lexMultilineStringEscape | ||||
| 	case stringEnd: | ||||
| 		if lx.accept(stringEnd) { | ||||
| 			if lx.accept(stringEnd) { | ||||
| 				lx.backup() | ||||
| 				lx.backup() | ||||
| 				lx.backup() | ||||
| 				lx.emit(itemMultilineString) | ||||
| 				lx.next() | ||||
| 				lx.next() | ||||
| 				lx.next() | ||||
| 				lx.ignore() | ||||
| 				return lx.pop() | ||||
| 			} | ||||
| 			lx.backup() | ||||
| 		} | ||||
| 	} | ||||
| 	return lexMultilineString | ||||
| } | ||||
|  | ||||
| // lexRawString consumes a raw string. Nothing can be escaped in such a string. | ||||
| // It assumes that the beginning "'" has already been consumed and ignored. | ||||
| func lexRawString(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch { | ||||
| 	case r == eof: | ||||
| 		return lx.errorf("unexpected EOF") | ||||
| 	case isNL(r): | ||||
| 		return lx.errorf("strings cannot contain newlines") | ||||
| 	case r == rawStringEnd: | ||||
| 		lx.backup() | ||||
| 		lx.emit(itemRawString) | ||||
| 		lx.next() | ||||
| 		lx.ignore() | ||||
| 		return lx.pop() | ||||
| 	} | ||||
| 	return lexRawString | ||||
| } | ||||
|  | ||||
| // lexMultilineRawString consumes a raw string. Nothing can be escaped in such | ||||
| // a string. It assumes that the beginning "'''" has already been consumed and | ||||
| // ignored. | ||||
| func lexMultilineRawString(lx *lexer) stateFn { | ||||
| 	switch lx.next() { | ||||
| 	case eof: | ||||
| 		return lx.errorf("unexpected EOF") | ||||
| 	case rawStringEnd: | ||||
| 		if lx.accept(rawStringEnd) { | ||||
| 			if lx.accept(rawStringEnd) { | ||||
| 				lx.backup() | ||||
| 				lx.backup() | ||||
| 				lx.backup() | ||||
| 				lx.emit(itemRawMultilineString) | ||||
| 				lx.next() | ||||
| 				lx.next() | ||||
| 				lx.next() | ||||
| 				lx.ignore() | ||||
| 				return lx.pop() | ||||
| 			} | ||||
| 			lx.backup() | ||||
| 		} | ||||
| 	} | ||||
| 	return lexMultilineRawString | ||||
| } | ||||
|  | ||||
| // lexMultilineStringEscape consumes an escaped character. It assumes that the | ||||
| // preceding '\\' has already been consumed. | ||||
| func lexMultilineStringEscape(lx *lexer) stateFn { | ||||
| 	// Handle the special case first: | ||||
| 	if isNL(lx.next()) { | ||||
| 		return lexMultilineString | ||||
| 	} | ||||
| 	lx.backup() | ||||
| 	lx.push(lexMultilineString) | ||||
| 	return lexStringEscape(lx) | ||||
| } | ||||
|  | ||||
| func lexStringEscape(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	switch r { | ||||
| 	case 'b': | ||||
| 		fallthrough | ||||
| 	case 't': | ||||
| 		fallthrough | ||||
| 	case 'n': | ||||
| 		fallthrough | ||||
| 	case 'f': | ||||
| 		fallthrough | ||||
| 	case 'r': | ||||
| 		fallthrough | ||||
| 	case '"': | ||||
| 		fallthrough | ||||
| 	case '\\': | ||||
| 		return lx.pop() | ||||
| 	case 'u': | ||||
| 		return lexShortUnicodeEscape | ||||
| 	case 'U': | ||||
| 		return lexLongUnicodeEscape | ||||
| 	} | ||||
| 	return lx.errorf("invalid escape character %q; only the following "+ | ||||
| 		"escape characters are allowed: "+ | ||||
| 		`\b, \t, \n, \f, \r, \", \\, \uXXXX, and \UXXXXXXXX`, r) | ||||
| } | ||||
|  | ||||
| func lexShortUnicodeEscape(lx *lexer) stateFn { | ||||
| 	var r rune | ||||
| 	for i := 0; i < 4; i++ { | ||||
| 		r = lx.next() | ||||
| 		if !isHexadecimal(r) { | ||||
| 			return lx.errorf(`expected four hexadecimal digits after '\u', `+ | ||||
| 				"but got %q instead", lx.current()) | ||||
| 		} | ||||
| 	} | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| func lexLongUnicodeEscape(lx *lexer) stateFn { | ||||
| 	var r rune | ||||
| 	for i := 0; i < 8; i++ { | ||||
| 		r = lx.next() | ||||
| 		if !isHexadecimal(r) { | ||||
| 			return lx.errorf(`expected eight hexadecimal digits after '\U', `+ | ||||
| 				"but got %q instead", lx.current()) | ||||
| 		} | ||||
| 	} | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexNumberOrDateStart consumes either an integer, a float, or datetime. | ||||
| func lexNumberOrDateStart(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isDigit(r) { | ||||
| 		return lexNumberOrDate | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case '_': | ||||
| 		return lexNumber | ||||
| 	case 'e', 'E': | ||||
| 		return lexFloat | ||||
| 	case '.': | ||||
| 		return lx.errorf("floats must start with a digit, not '.'") | ||||
| 	} | ||||
| 	return lx.errorf("expected a digit but got %q", r) | ||||
| } | ||||
|  | ||||
| // lexNumberOrDate consumes either an integer, float or datetime. | ||||
| func lexNumberOrDate(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isDigit(r) { | ||||
| 		return lexNumberOrDate | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case '-': | ||||
| 		return lexDatetime | ||||
| 	case '_': | ||||
| 		return lexNumber | ||||
| 	case '.', 'e', 'E': | ||||
| 		return lexFloat | ||||
| 	} | ||||
|  | ||||
| 	lx.backup() | ||||
| 	lx.emit(itemInteger) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexDatetime consumes a Datetime, to a first approximation. | ||||
| // The parser validates that it matches one of the accepted formats. | ||||
| func lexDatetime(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isDigit(r) { | ||||
| 		return lexDatetime | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case '-', 'T', ':', '.', 'Z': | ||||
| 		return lexDatetime | ||||
| 	} | ||||
|  | ||||
| 	lx.backup() | ||||
| 	lx.emit(itemDatetime) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexNumberStart consumes either an integer or a float. It assumes that a sign | ||||
| // has already been read, but that *no* digits have been consumed. | ||||
| // lexNumberStart will move to the appropriate integer or float states. | ||||
| func lexNumberStart(lx *lexer) stateFn { | ||||
| 	// We MUST see a digit. Even floats have to start with a digit. | ||||
| 	r := lx.next() | ||||
| 	if !isDigit(r) { | ||||
| 		if r == '.' { | ||||
| 			return lx.errorf("floats must start with a digit, not '.'") | ||||
| 		} | ||||
| 		return lx.errorf("expected a digit but got %q", r) | ||||
| 	} | ||||
| 	return lexNumber | ||||
| } | ||||
|  | ||||
| // lexNumber consumes an integer or a float after seeing the first digit. | ||||
| func lexNumber(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isDigit(r) { | ||||
| 		return lexNumber | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case '_': | ||||
| 		return lexNumber | ||||
| 	case '.', 'e', 'E': | ||||
| 		return lexFloat | ||||
| 	} | ||||
|  | ||||
| 	lx.backup() | ||||
| 	lx.emit(itemInteger) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexFloat consumes the elements of a float. It allows any sequence of | ||||
| // float-like characters, so floats emitted by the lexer are only a first | ||||
| // approximation and must be validated by the parser. | ||||
| func lexFloat(lx *lexer) stateFn { | ||||
| 	r := lx.next() | ||||
| 	if isDigit(r) { | ||||
| 		return lexFloat | ||||
| 	} | ||||
| 	switch r { | ||||
| 	case '_', '.', '-', '+', 'e', 'E': | ||||
| 		return lexFloat | ||||
| 	} | ||||
|  | ||||
| 	lx.backup() | ||||
| 	lx.emit(itemFloat) | ||||
| 	return lx.pop() | ||||
| } | ||||
|  | ||||
| // lexBool consumes a bool string: 'true' or 'false. | ||||
| func lexBool(lx *lexer) stateFn { | ||||
| 	var rs []rune | ||||
| 	for { | ||||
| 		r := lx.next() | ||||
| 		if r == eof || isWhitespace(r) || isNL(r) { | ||||
| 			lx.backup() | ||||
| 			break | ||||
| 		} | ||||
| 		rs = append(rs, r) | ||||
| 	} | ||||
| 	s := string(rs) | ||||
| 	switch s { | ||||
| 	case "true", "false": | ||||
| 		lx.emit(itemBool) | ||||
| 		return lx.pop() | ||||
| 	} | ||||
| 	return lx.errorf("expected value but found %q instead", s) | ||||
| } | ||||
|  | ||||
| // lexCommentStart begins the lexing of a comment. It will emit | ||||
| // itemCommentStart and consume no characters, passing control to lexComment. | ||||
| func lexCommentStart(lx *lexer) stateFn { | ||||
| 	lx.ignore() | ||||
| 	lx.emit(itemCommentStart) | ||||
| 	return lexComment | ||||
| } | ||||
|  | ||||
| // lexComment lexes an entire comment. It assumes that '#' has been consumed. | ||||
| // It will consume *up to* the first newline character, and pass control | ||||
| // back to the last state on the stack. | ||||
| func lexComment(lx *lexer) stateFn { | ||||
| 	r := lx.peek() | ||||
| 	if isNL(r) || r == eof { | ||||
| 		lx.emit(itemText) | ||||
| 		return lx.pop() | ||||
| 	} | ||||
| 	lx.next() | ||||
| 	return lexComment | ||||
| } | ||||
|  | ||||
| // lexSkip ignores all slurped input and moves on to the next state. | ||||
| func lexSkip(lx *lexer, nextState stateFn) stateFn { | ||||
| 	return func(lx *lexer) stateFn { | ||||
| 		lx.ignore() | ||||
| 		return nextState | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // isWhitespace returns true if `r` is a whitespace character according | ||||
| // to the spec. | ||||
| func isWhitespace(r rune) bool { | ||||
| 	return r == '\t' || r == ' ' | ||||
| } | ||||
|  | ||||
| func isNL(r rune) bool { | ||||
| 	return r == '\n' || r == '\r' | ||||
| } | ||||
|  | ||||
| func isDigit(r rune) bool { | ||||
| 	return r >= '0' && r <= '9' | ||||
| } | ||||
|  | ||||
| func isHexadecimal(r rune) bool { | ||||
| 	return (r >= '0' && r <= '9') || | ||||
| 		(r >= 'a' && r <= 'f') || | ||||
| 		(r >= 'A' && r <= 'F') | ||||
| } | ||||
|  | ||||
| func isBareKeyChar(r rune) bool { | ||||
| 	return (r >= 'A' && r <= 'Z') || | ||||
| 		(r >= 'a' && r <= 'z') || | ||||
| 		(r >= '0' && r <= '9') || | ||||
| 		r == '_' || | ||||
| 		r == '-' | ||||
| } | ||||
|  | ||||
| func (itype itemType) String() string { | ||||
| 	switch itype { | ||||
| 	case itemError: | ||||
| 		return "Error" | ||||
| 	case itemNIL: | ||||
| 		return "NIL" | ||||
| 	case itemEOF: | ||||
| 		return "EOF" | ||||
| 	case itemText: | ||||
| 		return "Text" | ||||
| 	case itemString, itemRawString, itemMultilineString, itemRawMultilineString: | ||||
| 		return "String" | ||||
| 	case itemBool: | ||||
| 		return "Bool" | ||||
| 	case itemInteger: | ||||
| 		return "Integer" | ||||
| 	case itemFloat: | ||||
| 		return "Float" | ||||
| 	case itemDatetime: | ||||
| 		return "DateTime" | ||||
| 	case itemTableStart: | ||||
| 		return "TableStart" | ||||
| 	case itemTableEnd: | ||||
| 		return "TableEnd" | ||||
| 	case itemKeyStart: | ||||
| 		return "KeyStart" | ||||
| 	case itemArray: | ||||
| 		return "Array" | ||||
| 	case itemArrayEnd: | ||||
| 		return "ArrayEnd" | ||||
| 	case itemCommentStart: | ||||
| 		return "CommentStart" | ||||
| 	} | ||||
| 	panic(fmt.Sprintf("BUG: Unknown type '%d'.", int(itype))) | ||||
| } | ||||
|  | ||||
| func (item item) String() string { | ||||
| 	return fmt.Sprintf("(%s, %s)", item.typ.String(), item.val) | ||||
| } | ||||
							
								
								
									
										592
									
								
								vendor/github.com/BurntSushi/toml/parse.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,592 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| 	"unicode" | ||||
| 	"unicode/utf8" | ||||
| ) | ||||
|  | ||||
| type parser struct { | ||||
| 	mapping map[string]interface{} | ||||
| 	types   map[string]tomlType | ||||
| 	lx      *lexer | ||||
|  | ||||
| 	// A list of keys in the order that they appear in the TOML data. | ||||
| 	ordered []Key | ||||
|  | ||||
| 	// the full key for the current hash in scope | ||||
| 	context Key | ||||
|  | ||||
| 	// the base key name for everything except hashes | ||||
| 	currentKey string | ||||
|  | ||||
| 	// rough approximation of line number | ||||
| 	approxLine int | ||||
|  | ||||
| 	// A map of 'key.group.names' to whether they were created implicitly. | ||||
| 	implicits map[string]bool | ||||
| } | ||||
|  | ||||
| type parseError string | ||||
|  | ||||
| func (pe parseError) Error() string { | ||||
| 	return string(pe) | ||||
| } | ||||
|  | ||||
| func parse(data string) (p *parser, err error) { | ||||
| 	defer func() { | ||||
| 		if r := recover(); r != nil { | ||||
| 			var ok bool | ||||
| 			if err, ok = r.(parseError); ok { | ||||
| 				return | ||||
| 			} | ||||
| 			panic(r) | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	p = &parser{ | ||||
| 		mapping:   make(map[string]interface{}), | ||||
| 		types:     make(map[string]tomlType), | ||||
| 		lx:        lex(data), | ||||
| 		ordered:   make([]Key, 0), | ||||
| 		implicits: make(map[string]bool), | ||||
| 	} | ||||
| 	for { | ||||
| 		item := p.next() | ||||
| 		if item.typ == itemEOF { | ||||
| 			break | ||||
| 		} | ||||
| 		p.topLevel(item) | ||||
| 	} | ||||
|  | ||||
| 	return p, nil | ||||
| } | ||||
|  | ||||
| func (p *parser) panicf(format string, v ...interface{}) { | ||||
| 	msg := fmt.Sprintf("Near line %d (last key parsed '%s'): %s", | ||||
| 		p.approxLine, p.current(), fmt.Sprintf(format, v...)) | ||||
| 	panic(parseError(msg)) | ||||
| } | ||||
|  | ||||
| func (p *parser) next() item { | ||||
| 	it := p.lx.nextItem() | ||||
| 	if it.typ == itemError { | ||||
| 		p.panicf("%s", it.val) | ||||
| 	} | ||||
| 	return it | ||||
| } | ||||
|  | ||||
| func (p *parser) bug(format string, v ...interface{}) { | ||||
| 	panic(fmt.Sprintf("BUG: "+format+"\n\n", v...)) | ||||
| } | ||||
|  | ||||
| func (p *parser) expect(typ itemType) item { | ||||
| 	it := p.next() | ||||
| 	p.assertEqual(typ, it.typ) | ||||
| 	return it | ||||
| } | ||||
|  | ||||
| func (p *parser) assertEqual(expected, got itemType) { | ||||
| 	if expected != got { | ||||
| 		p.bug("Expected '%s' but got '%s'.", expected, got) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (p *parser) topLevel(item item) { | ||||
| 	switch item.typ { | ||||
| 	case itemCommentStart: | ||||
| 		p.approxLine = item.line | ||||
| 		p.expect(itemText) | ||||
| 	case itemTableStart: | ||||
| 		kg := p.next() | ||||
| 		p.approxLine = kg.line | ||||
|  | ||||
| 		var key Key | ||||
| 		for ; kg.typ != itemTableEnd && kg.typ != itemEOF; kg = p.next() { | ||||
| 			key = append(key, p.keyString(kg)) | ||||
| 		} | ||||
| 		p.assertEqual(itemTableEnd, kg.typ) | ||||
|  | ||||
| 		p.establishContext(key, false) | ||||
| 		p.setType("", tomlHash) | ||||
| 		p.ordered = append(p.ordered, key) | ||||
| 	case itemArrayTableStart: | ||||
| 		kg := p.next() | ||||
| 		p.approxLine = kg.line | ||||
|  | ||||
| 		var key Key | ||||
| 		for ; kg.typ != itemArrayTableEnd && kg.typ != itemEOF; kg = p.next() { | ||||
| 			key = append(key, p.keyString(kg)) | ||||
| 		} | ||||
| 		p.assertEqual(itemArrayTableEnd, kg.typ) | ||||
|  | ||||
| 		p.establishContext(key, true) | ||||
| 		p.setType("", tomlArrayHash) | ||||
| 		p.ordered = append(p.ordered, key) | ||||
| 	case itemKeyStart: | ||||
| 		kname := p.next() | ||||
| 		p.approxLine = kname.line | ||||
| 		p.currentKey = p.keyString(kname) | ||||
|  | ||||
| 		val, typ := p.value(p.next()) | ||||
| 		p.setValue(p.currentKey, val) | ||||
| 		p.setType(p.currentKey, typ) | ||||
| 		p.ordered = append(p.ordered, p.context.add(p.currentKey)) | ||||
| 		p.currentKey = "" | ||||
| 	default: | ||||
| 		p.bug("Unexpected type at top level: %s", item.typ) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Gets a string for a key (or part of a key in a table name). | ||||
| func (p *parser) keyString(it item) string { | ||||
| 	switch it.typ { | ||||
| 	case itemText: | ||||
| 		return it.val | ||||
| 	case itemString, itemMultilineString, | ||||
| 		itemRawString, itemRawMultilineString: | ||||
| 		s, _ := p.value(it) | ||||
| 		return s.(string) | ||||
| 	default: | ||||
| 		p.bug("Unexpected key type: %s", it.typ) | ||||
| 		panic("unreachable") | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // value translates an expected value from the lexer into a Go value wrapped | ||||
| // as an empty interface. | ||||
| func (p *parser) value(it item) (interface{}, tomlType) { | ||||
| 	switch it.typ { | ||||
| 	case itemString: | ||||
| 		return p.replaceEscapes(it.val), p.typeOfPrimitive(it) | ||||
| 	case itemMultilineString: | ||||
| 		trimmed := stripFirstNewline(stripEscapedWhitespace(it.val)) | ||||
| 		return p.replaceEscapes(trimmed), p.typeOfPrimitive(it) | ||||
| 	case itemRawString: | ||||
| 		return it.val, p.typeOfPrimitive(it) | ||||
| 	case itemRawMultilineString: | ||||
| 		return stripFirstNewline(it.val), p.typeOfPrimitive(it) | ||||
| 	case itemBool: | ||||
| 		switch it.val { | ||||
| 		case "true": | ||||
| 			return true, p.typeOfPrimitive(it) | ||||
| 		case "false": | ||||
| 			return false, p.typeOfPrimitive(it) | ||||
| 		} | ||||
| 		p.bug("Expected boolean value, but got '%s'.", it.val) | ||||
| 	case itemInteger: | ||||
| 		if !numUnderscoresOK(it.val) { | ||||
| 			p.panicf("Invalid integer %q: underscores must be surrounded by digits", | ||||
| 				it.val) | ||||
| 		} | ||||
| 		val := strings.Replace(it.val, "_", "", -1) | ||||
| 		num, err := strconv.ParseInt(val, 10, 64) | ||||
| 		if err != nil { | ||||
| 			// Distinguish integer values. Normally, it'd be a bug if the lexer | ||||
| 			// provides an invalid integer, but it's possible that the number is | ||||
| 			// out of range of valid values (which the lexer cannot determine). | ||||
| 			// So mark the former as a bug but the latter as a legitimate user | ||||
| 			// error. | ||||
| 			if e, ok := err.(*strconv.NumError); ok && | ||||
| 				e.Err == strconv.ErrRange { | ||||
|  | ||||
| 				p.panicf("Integer '%s' is out of the range of 64-bit "+ | ||||
| 					"signed integers.", it.val) | ||||
| 			} else { | ||||
| 				p.bug("Expected integer value, but got '%s'.", it.val) | ||||
| 			} | ||||
| 		} | ||||
| 		return num, p.typeOfPrimitive(it) | ||||
| 	case itemFloat: | ||||
| 		parts := strings.FieldsFunc(it.val, func(r rune) bool { | ||||
| 			switch r { | ||||
| 			case '.', 'e', 'E': | ||||
| 				return true | ||||
| 			} | ||||
| 			return false | ||||
| 		}) | ||||
| 		for _, part := range parts { | ||||
| 			if !numUnderscoresOK(part) { | ||||
| 				p.panicf("Invalid float %q: underscores must be "+ | ||||
| 					"surrounded by digits", it.val) | ||||
| 			} | ||||
| 		} | ||||
| 		if !numPeriodsOK(it.val) { | ||||
| 			// As a special case, numbers like '123.' or '1.e2', | ||||
| 			// which are valid as far as Go/strconv are concerned, | ||||
| 			// must be rejected because TOML says that a fractional | ||||
| 			// part consists of '.' followed by 1+ digits. | ||||
| 			p.panicf("Invalid float %q: '.' must be followed "+ | ||||
| 				"by one or more digits", it.val) | ||||
| 		} | ||||
| 		val := strings.Replace(it.val, "_", "", -1) | ||||
| 		num, err := strconv.ParseFloat(val, 64) | ||||
| 		if err != nil { | ||||
| 			if e, ok := err.(*strconv.NumError); ok && | ||||
| 				e.Err == strconv.ErrRange { | ||||
|  | ||||
| 				p.panicf("Float '%s' is out of the range of 64-bit "+ | ||||
| 					"IEEE-754 floating-point numbers.", it.val) | ||||
| 			} else { | ||||
| 				p.panicf("Invalid float value: %q", it.val) | ||||
| 			} | ||||
| 		} | ||||
| 		return num, p.typeOfPrimitive(it) | ||||
| 	case itemDatetime: | ||||
| 		var t time.Time | ||||
| 		var ok bool | ||||
| 		var err error | ||||
| 		for _, format := range []string{ | ||||
| 			"2006-01-02T15:04:05Z07:00", | ||||
| 			"2006-01-02T15:04:05", | ||||
| 			"2006-01-02", | ||||
| 		} { | ||||
| 			t, err = time.ParseInLocation(format, it.val, time.Local) | ||||
| 			if err == nil { | ||||
| 				ok = true | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		if !ok { | ||||
| 			p.panicf("Invalid TOML Datetime: %q.", it.val) | ||||
| 		} | ||||
| 		return t, p.typeOfPrimitive(it) | ||||
| 	case itemArray: | ||||
| 		array := make([]interface{}, 0) | ||||
| 		types := make([]tomlType, 0) | ||||
|  | ||||
| 		for it = p.next(); it.typ != itemArrayEnd; it = p.next() { | ||||
| 			if it.typ == itemCommentStart { | ||||
| 				p.expect(itemText) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			val, typ := p.value(it) | ||||
| 			array = append(array, val) | ||||
| 			types = append(types, typ) | ||||
| 		} | ||||
| 		return array, p.typeOfArray(types) | ||||
| 	case itemInlineTableStart: | ||||
| 		var ( | ||||
| 			hash         = make(map[string]interface{}) | ||||
| 			outerContext = p.context | ||||
| 			outerKey     = p.currentKey | ||||
| 		) | ||||
|  | ||||
| 		p.context = append(p.context, p.currentKey) | ||||
| 		p.currentKey = "" | ||||
| 		for it := p.next(); it.typ != itemInlineTableEnd; it = p.next() { | ||||
| 			if it.typ != itemKeyStart { | ||||
| 				p.bug("Expected key start but instead found %q, around line %d", | ||||
| 					it.val, p.approxLine) | ||||
| 			} | ||||
| 			if it.typ == itemCommentStart { | ||||
| 				p.expect(itemText) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			// retrieve key | ||||
| 			k := p.next() | ||||
| 			p.approxLine = k.line | ||||
| 			kname := p.keyString(k) | ||||
|  | ||||
| 			// retrieve value | ||||
| 			p.currentKey = kname | ||||
| 			val, typ := p.value(p.next()) | ||||
| 			// make sure we keep metadata up to date | ||||
| 			p.setType(kname, typ) | ||||
| 			p.ordered = append(p.ordered, p.context.add(p.currentKey)) | ||||
| 			hash[kname] = val | ||||
| 		} | ||||
| 		p.context = outerContext | ||||
| 		p.currentKey = outerKey | ||||
| 		return hash, tomlHash | ||||
| 	} | ||||
| 	p.bug("Unexpected value type: %s", it.typ) | ||||
| 	panic("unreachable") | ||||
| } | ||||
|  | ||||
| // numUnderscoresOK checks whether each underscore in s is surrounded by | ||||
| // characters that are not underscores. | ||||
| func numUnderscoresOK(s string) bool { | ||||
| 	accept := false | ||||
| 	for _, r := range s { | ||||
| 		if r == '_' { | ||||
| 			if !accept { | ||||
| 				return false | ||||
| 			} | ||||
| 			accept = false | ||||
| 			continue | ||||
| 		} | ||||
| 		accept = true | ||||
| 	} | ||||
| 	return accept | ||||
| } | ||||
|  | ||||
| // numPeriodsOK checks whether every period in s is followed by a digit. | ||||
| func numPeriodsOK(s string) bool { | ||||
| 	period := false | ||||
| 	for _, r := range s { | ||||
| 		if period && !isDigit(r) { | ||||
| 			return false | ||||
| 		} | ||||
| 		period = r == '.' | ||||
| 	} | ||||
| 	return !period | ||||
| } | ||||
|  | ||||
| // establishContext sets the current context of the parser, | ||||
| // where the context is either a hash or an array of hashes. Which one is | ||||
| // set depends on the value of the `array` parameter. | ||||
| // | ||||
| // Establishing the context also makes sure that the key isn't a duplicate, and | ||||
| // will create implicit hashes automatically. | ||||
| func (p *parser) establishContext(key Key, array bool) { | ||||
| 	var ok bool | ||||
|  | ||||
| 	// Always start at the top level and drill down for our context. | ||||
| 	hashContext := p.mapping | ||||
| 	keyContext := make(Key, 0) | ||||
|  | ||||
| 	// We only need implicit hashes for key[0:-1] | ||||
| 	for _, k := range key[0 : len(key)-1] { | ||||
| 		_, ok = hashContext[k] | ||||
| 		keyContext = append(keyContext, k) | ||||
|  | ||||
| 		// No key? Make an implicit hash and move on. | ||||
| 		if !ok { | ||||
| 			p.addImplicit(keyContext) | ||||
| 			hashContext[k] = make(map[string]interface{}) | ||||
| 		} | ||||
|  | ||||
| 		// If the hash context is actually an array of tables, then set | ||||
| 		// the hash context to the last element in that array. | ||||
| 		// | ||||
| 		// Otherwise, it better be a table, since this MUST be a key group (by | ||||
| 		// virtue of it not being the last element in a key). | ||||
| 		switch t := hashContext[k].(type) { | ||||
| 		case []map[string]interface{}: | ||||
| 			hashContext = t[len(t)-1] | ||||
| 		case map[string]interface{}: | ||||
| 			hashContext = t | ||||
| 		default: | ||||
| 			p.panicf("Key '%s' was already created as a hash.", keyContext) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	p.context = keyContext | ||||
| 	if array { | ||||
| 		// If this is the first element for this array, then allocate a new | ||||
| 		// list of tables for it. | ||||
| 		k := key[len(key)-1] | ||||
| 		if _, ok := hashContext[k]; !ok { | ||||
| 			hashContext[k] = make([]map[string]interface{}, 0, 5) | ||||
| 		} | ||||
|  | ||||
| 		// Add a new table. But make sure the key hasn't already been used | ||||
| 		// for something else. | ||||
| 		if hash, ok := hashContext[k].([]map[string]interface{}); ok { | ||||
| 			hashContext[k] = append(hash, make(map[string]interface{})) | ||||
| 		} else { | ||||
| 			p.panicf("Key '%s' was already created and cannot be used as "+ | ||||
| 				"an array.", keyContext) | ||||
| 		} | ||||
| 	} else { | ||||
| 		p.setValue(key[len(key)-1], make(map[string]interface{})) | ||||
| 	} | ||||
| 	p.context = append(p.context, key[len(key)-1]) | ||||
| } | ||||
|  | ||||
| // setValue sets the given key to the given value in the current context. | ||||
| // It will make sure that the key hasn't already been defined, account for | ||||
| // implicit key groups. | ||||
| func (p *parser) setValue(key string, value interface{}) { | ||||
| 	var tmpHash interface{} | ||||
| 	var ok bool | ||||
|  | ||||
| 	hash := p.mapping | ||||
| 	keyContext := make(Key, 0) | ||||
| 	for _, k := range p.context { | ||||
| 		keyContext = append(keyContext, k) | ||||
| 		if tmpHash, ok = hash[k]; !ok { | ||||
| 			p.bug("Context for key '%s' has not been established.", keyContext) | ||||
| 		} | ||||
| 		switch t := tmpHash.(type) { | ||||
| 		case []map[string]interface{}: | ||||
| 			// The context is a table of hashes. Pick the most recent table | ||||
| 			// defined as the current hash. | ||||
| 			hash = t[len(t)-1] | ||||
| 		case map[string]interface{}: | ||||
| 			hash = t | ||||
| 		default: | ||||
| 			p.bug("Expected hash to have type 'map[string]interface{}', but "+ | ||||
| 				"it has '%T' instead.", tmpHash) | ||||
| 		} | ||||
| 	} | ||||
| 	keyContext = append(keyContext, key) | ||||
|  | ||||
| 	if _, ok := hash[key]; ok { | ||||
| 		// Typically, if the given key has already been set, then we have | ||||
| 		// to raise an error since duplicate keys are disallowed. However, | ||||
| 		// it's possible that a key was previously defined implicitly. In this | ||||
| 		// case, it is allowed to be redefined concretely. (See the | ||||
| 		// `tests/valid/implicit-and-explicit-after.toml` test in `toml-test`.) | ||||
| 		// | ||||
| 		// But we have to make sure to stop marking it as an implicit. (So that | ||||
| 		// another redefinition provokes an error.) | ||||
| 		// | ||||
| 		// Note that since it has already been defined (as a hash), we don't | ||||
| 		// want to overwrite it. So our business is done. | ||||
| 		if p.isImplicit(keyContext) { | ||||
| 			p.removeImplicit(keyContext) | ||||
| 			return | ||||
| 		} | ||||
|  | ||||
| 		// Otherwise, we have a concrete key trying to override a previous | ||||
| 		// key, which is *always* wrong. | ||||
| 		p.panicf("Key '%s' has already been defined.", keyContext) | ||||
| 	} | ||||
| 	hash[key] = value | ||||
| } | ||||
|  | ||||
| // setType sets the type of a particular value at a given key. | ||||
| // It should be called immediately AFTER setValue. | ||||
| // | ||||
| // Note that if `key` is empty, then the type given will be applied to the | ||||
| // current context (which is either a table or an array of tables). | ||||
| func (p *parser) setType(key string, typ tomlType) { | ||||
| 	keyContext := make(Key, 0, len(p.context)+1) | ||||
| 	for _, k := range p.context { | ||||
| 		keyContext = append(keyContext, k) | ||||
| 	} | ||||
| 	if len(key) > 0 { // allow type setting for hashes | ||||
| 		keyContext = append(keyContext, key) | ||||
| 	} | ||||
| 	p.types[keyContext.String()] = typ | ||||
| } | ||||
|  | ||||
| // addImplicit sets the given Key as having been created implicitly. | ||||
| func (p *parser) addImplicit(key Key) { | ||||
| 	p.implicits[key.String()] = true | ||||
| } | ||||
|  | ||||
| // removeImplicit stops tagging the given key as having been implicitly | ||||
| // created. | ||||
| func (p *parser) removeImplicit(key Key) { | ||||
| 	p.implicits[key.String()] = false | ||||
| } | ||||
|  | ||||
| // isImplicit returns true if the key group pointed to by the key was created | ||||
| // implicitly. | ||||
| func (p *parser) isImplicit(key Key) bool { | ||||
| 	return p.implicits[key.String()] | ||||
| } | ||||
|  | ||||
| // current returns the full key name of the current context. | ||||
| func (p *parser) current() string { | ||||
| 	if len(p.currentKey) == 0 { | ||||
| 		return p.context.String() | ||||
| 	} | ||||
| 	if len(p.context) == 0 { | ||||
| 		return p.currentKey | ||||
| 	} | ||||
| 	return fmt.Sprintf("%s.%s", p.context, p.currentKey) | ||||
| } | ||||
|  | ||||
| func stripFirstNewline(s string) string { | ||||
| 	if len(s) == 0 || s[0] != '\n' { | ||||
| 		return s | ||||
| 	} | ||||
| 	return s[1:] | ||||
| } | ||||
|  | ||||
| func stripEscapedWhitespace(s string) string { | ||||
| 	esc := strings.Split(s, "\\\n") | ||||
| 	if len(esc) > 1 { | ||||
| 		for i := 1; i < len(esc); i++ { | ||||
| 			esc[i] = strings.TrimLeftFunc(esc[i], unicode.IsSpace) | ||||
| 		} | ||||
| 	} | ||||
| 	return strings.Join(esc, "") | ||||
| } | ||||
|  | ||||
| func (p *parser) replaceEscapes(str string) string { | ||||
| 	var replaced []rune | ||||
| 	s := []byte(str) | ||||
| 	r := 0 | ||||
| 	for r < len(s) { | ||||
| 		if s[r] != '\\' { | ||||
| 			c, size := utf8.DecodeRune(s[r:]) | ||||
| 			r += size | ||||
| 			replaced = append(replaced, c) | ||||
| 			continue | ||||
| 		} | ||||
| 		r += 1 | ||||
| 		if r >= len(s) { | ||||
| 			p.bug("Escape sequence at end of string.") | ||||
| 			return "" | ||||
| 		} | ||||
| 		switch s[r] { | ||||
| 		default: | ||||
| 			p.bug("Expected valid escape code after \\, but got %q.", s[r]) | ||||
| 			return "" | ||||
| 		case 'b': | ||||
| 			replaced = append(replaced, rune(0x0008)) | ||||
| 			r += 1 | ||||
| 		case 't': | ||||
| 			replaced = append(replaced, rune(0x0009)) | ||||
| 			r += 1 | ||||
| 		case 'n': | ||||
| 			replaced = append(replaced, rune(0x000A)) | ||||
| 			r += 1 | ||||
| 		case 'f': | ||||
| 			replaced = append(replaced, rune(0x000C)) | ||||
| 			r += 1 | ||||
| 		case 'r': | ||||
| 			replaced = append(replaced, rune(0x000D)) | ||||
| 			r += 1 | ||||
| 		case '"': | ||||
| 			replaced = append(replaced, rune(0x0022)) | ||||
| 			r += 1 | ||||
| 		case '\\': | ||||
| 			replaced = append(replaced, rune(0x005C)) | ||||
| 			r += 1 | ||||
| 		case 'u': | ||||
| 			// At this point, we know we have a Unicode escape of the form | ||||
| 			// `uXXXX` at [r, r+5). (Because the lexer guarantees this | ||||
| 			// for us.) | ||||
| 			escaped := p.asciiEscapeToUnicode(s[r+1 : r+5]) | ||||
| 			replaced = append(replaced, escaped) | ||||
| 			r += 5 | ||||
| 		case 'U': | ||||
| 			// At this point, we know we have a Unicode escape of the form | ||||
| 			// `uXXXX` at [r, r+9). (Because the lexer guarantees this | ||||
| 			// for us.) | ||||
| 			escaped := p.asciiEscapeToUnicode(s[r+1 : r+9]) | ||||
| 			replaced = append(replaced, escaped) | ||||
| 			r += 9 | ||||
| 		} | ||||
| 	} | ||||
| 	return string(replaced) | ||||
| } | ||||
|  | ||||
| func (p *parser) asciiEscapeToUnicode(bs []byte) rune { | ||||
| 	s := string(bs) | ||||
| 	hex, err := strconv.ParseUint(strings.ToLower(s), 16, 32) | ||||
| 	if err != nil { | ||||
| 		p.bug("Could not parse '%s' as a hexadecimal number, but the "+ | ||||
| 			"lexer claims it's OK: %s", s, err) | ||||
| 	} | ||||
| 	if !utf8.ValidRune(rune(hex)) { | ||||
| 		p.panicf("Escaped character '\\u%s' is not valid UTF-8.", s) | ||||
| 	} | ||||
| 	return rune(hex) | ||||
| } | ||||
|  | ||||
| func isStringType(ty itemType) bool { | ||||
| 	return ty == itemString || ty == itemMultilineString || | ||||
| 		ty == itemRawString || ty == itemRawMultilineString | ||||
| } | ||||
							
								
								
									
										1
									
								
								vendor/github.com/BurntSushi/toml/session.vim
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1 +0,0 @@ | ||||
| au BufWritePost *.go silent!make tags > /dev/null 2>&1 | ||||
							
								
								
									
										91
									
								
								vendor/github.com/BurntSushi/toml/type_check.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,91 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| // tomlType represents any Go type that corresponds to a TOML type. | ||||
| // While the first draft of the TOML spec has a simplistic type system that | ||||
| // probably doesn't need this level of sophistication, we seem to be militating | ||||
| // toward adding real composite types. | ||||
| type tomlType interface { | ||||
| 	typeString() string | ||||
| } | ||||
|  | ||||
| // typeEqual accepts any two types and returns true if they are equal. | ||||
| func typeEqual(t1, t2 tomlType) bool { | ||||
| 	if t1 == nil || t2 == nil { | ||||
| 		return false | ||||
| 	} | ||||
| 	return t1.typeString() == t2.typeString() | ||||
| } | ||||
|  | ||||
| func typeIsHash(t tomlType) bool { | ||||
| 	return typeEqual(t, tomlHash) || typeEqual(t, tomlArrayHash) | ||||
| } | ||||
|  | ||||
| type tomlBaseType string | ||||
|  | ||||
| func (btype tomlBaseType) typeString() string { | ||||
| 	return string(btype) | ||||
| } | ||||
|  | ||||
| func (btype tomlBaseType) String() string { | ||||
| 	return btype.typeString() | ||||
| } | ||||
|  | ||||
| var ( | ||||
| 	tomlInteger   tomlBaseType = "Integer" | ||||
| 	tomlFloat     tomlBaseType = "Float" | ||||
| 	tomlDatetime  tomlBaseType = "Datetime" | ||||
| 	tomlString    tomlBaseType = "String" | ||||
| 	tomlBool      tomlBaseType = "Bool" | ||||
| 	tomlArray     tomlBaseType = "Array" | ||||
| 	tomlHash      tomlBaseType = "Hash" | ||||
| 	tomlArrayHash tomlBaseType = "ArrayHash" | ||||
| ) | ||||
|  | ||||
| // typeOfPrimitive returns a tomlType of any primitive value in TOML. | ||||
| // Primitive values are: Integer, Float, Datetime, String and Bool. | ||||
| // | ||||
| // Passing a lexer item other than the following will cause a BUG message | ||||
| // to occur: itemString, itemBool, itemInteger, itemFloat, itemDatetime. | ||||
| func (p *parser) typeOfPrimitive(lexItem item) tomlType { | ||||
| 	switch lexItem.typ { | ||||
| 	case itemInteger: | ||||
| 		return tomlInteger | ||||
| 	case itemFloat: | ||||
| 		return tomlFloat | ||||
| 	case itemDatetime: | ||||
| 		return tomlDatetime | ||||
| 	case itemString: | ||||
| 		return tomlString | ||||
| 	case itemMultilineString: | ||||
| 		return tomlString | ||||
| 	case itemRawString: | ||||
| 		return tomlString | ||||
| 	case itemRawMultilineString: | ||||
| 		return tomlString | ||||
| 	case itemBool: | ||||
| 		return tomlBool | ||||
| 	} | ||||
| 	p.bug("Cannot infer primitive type of lex item '%s'.", lexItem) | ||||
| 	panic("unreachable") | ||||
| } | ||||
|  | ||||
| // typeOfArray returns a tomlType for an array given a list of types of its | ||||
| // values. | ||||
| // | ||||
| // In the current spec, if an array is homogeneous, then its type is always | ||||
| // "Array". If the array is not homogeneous, an error is generated. | ||||
| func (p *parser) typeOfArray(types []tomlType) tomlType { | ||||
| 	// Empty arrays are cool. | ||||
| 	if len(types) == 0 { | ||||
| 		return tomlArray | ||||
| 	} | ||||
|  | ||||
| 	theType := types[0] | ||||
| 	for _, t := range types[1:] { | ||||
| 		if !typeEqual(theType, t) { | ||||
| 			p.panicf("Array contains values of type '%s' and '%s', but "+ | ||||
| 				"arrays must be homogeneous.", theType, t) | ||||
| 		} | ||||
| 	} | ||||
| 	return tomlArray | ||||
| } | ||||
							
								
								
									
										242
									
								
								vendor/github.com/BurntSushi/toml/type_fields.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -1,242 +0,0 @@ | ||||
| package toml | ||||
|  | ||||
| // Struct field handling is adapted from code in encoding/json: | ||||
| // | ||||
| // Copyright 2010 The Go Authors.  All rights reserved. | ||||
| // Use of this source code is governed by a BSD-style | ||||
| // license that can be found in the Go distribution. | ||||
|  | ||||
| import ( | ||||
| 	"reflect" | ||||
| 	"sort" | ||||
| 	"sync" | ||||
| ) | ||||
|  | ||||
| // A field represents a single field found in a struct. | ||||
| type field struct { | ||||
| 	name  string       // the name of the field (`toml` tag included) | ||||
| 	tag   bool         // whether field has a `toml` tag | ||||
| 	index []int        // represents the depth of an anonymous field | ||||
| 	typ   reflect.Type // the type of the field | ||||
| } | ||||
|  | ||||
| // byName sorts field by name, breaking ties with depth, | ||||
| // then breaking ties with "name came from toml tag", then | ||||
| // breaking ties with index sequence. | ||||
| type byName []field | ||||
|  | ||||
| func (x byName) Len() int { return len(x) } | ||||
|  | ||||
| func (x byName) Swap(i, j int) { x[i], x[j] = x[j], x[i] } | ||||
|  | ||||
| func (x byName) Less(i, j int) bool { | ||||
| 	if x[i].name != x[j].name { | ||||
| 		return x[i].name < x[j].name | ||||
| 	} | ||||
| 	if len(x[i].index) != len(x[j].index) { | ||||
| 		return len(x[i].index) < len(x[j].index) | ||||
| 	} | ||||
| 	if x[i].tag != x[j].tag { | ||||
| 		return x[i].tag | ||||
| 	} | ||||
| 	return byIndex(x).Less(i, j) | ||||
| } | ||||
|  | ||||
| // byIndex sorts field by index sequence. | ||||
| type byIndex []field | ||||
|  | ||||
| func (x byIndex) Len() int { return len(x) } | ||||
|  | ||||
| func (x byIndex) Swap(i, j int) { x[i], x[j] = x[j], x[i] } | ||||
|  | ||||
| func (x byIndex) Less(i, j int) bool { | ||||
| 	for k, xik := range x[i].index { | ||||
| 		if k >= len(x[j].index) { | ||||
| 			return false | ||||
| 		} | ||||
| 		if xik != x[j].index[k] { | ||||
| 			return xik < x[j].index[k] | ||||
| 		} | ||||
| 	} | ||||
| 	return len(x[i].index) < len(x[j].index) | ||||
| } | ||||
|  | ||||
| // typeFields returns a list of fields that TOML should recognize for the given | ||||
| // type. The algorithm is breadth-first search over the set of structs to | ||||
| // include - the top struct and then any reachable anonymous structs. | ||||
| func typeFields(t reflect.Type) []field { | ||||
| 	// Anonymous fields to explore at the current level and the next. | ||||
| 	current := []field{} | ||||
| 	next := []field{{typ: t}} | ||||
|  | ||||
| 	// Count of queued names for current level and the next. | ||||
| 	count := map[reflect.Type]int{} | ||||
| 	nextCount := map[reflect.Type]int{} | ||||
|  | ||||
| 	// Types already visited at an earlier level. | ||||
| 	visited := map[reflect.Type]bool{} | ||||
|  | ||||
| 	// Fields found. | ||||
| 	var fields []field | ||||
|  | ||||
| 	for len(next) > 0 { | ||||
| 		current, next = next, current[:0] | ||||
| 		count, nextCount = nextCount, map[reflect.Type]int{} | ||||
|  | ||||
| 		for _, f := range current { | ||||
| 			if visited[f.typ] { | ||||
| 				continue | ||||
| 			} | ||||
| 			visited[f.typ] = true | ||||
|  | ||||
| 			// Scan f.typ for fields to include. | ||||
| 			for i := 0; i < f.typ.NumField(); i++ { | ||||
| 				sf := f.typ.Field(i) | ||||
| 				if sf.PkgPath != "" && !sf.Anonymous { // unexported | ||||
| 					continue | ||||
| 				} | ||||
| 				opts := getOptions(sf.Tag) | ||||
| 				if opts.skip { | ||||
| 					continue | ||||
| 				} | ||||
| 				index := make([]int, len(f.index)+1) | ||||
| 				copy(index, f.index) | ||||
| 				index[len(f.index)] = i | ||||
|  | ||||
| 				ft := sf.Type | ||||
| 				if ft.Name() == "" && ft.Kind() == reflect.Ptr { | ||||
| 					// Follow pointer. | ||||
| 					ft = ft.Elem() | ||||
| 				} | ||||
|  | ||||
| 				// Record found field and index sequence. | ||||
| 				if opts.name != "" || !sf.Anonymous || ft.Kind() != reflect.Struct { | ||||
| 					tagged := opts.name != "" | ||||
| 					name := opts.name | ||||
| 					if name == "" { | ||||
| 						name = sf.Name | ||||
| 					} | ||||
| 					fields = append(fields, field{name, tagged, index, ft}) | ||||
| 					if count[f.typ] > 1 { | ||||
| 						// If there were multiple instances, add a second, | ||||
| 						// so that the annihilation code will see a duplicate. | ||||
| 						// It only cares about the distinction between 1 or 2, | ||||
| 						// so don't bother generating any more copies. | ||||
| 						fields = append(fields, fields[len(fields)-1]) | ||||
| 					} | ||||
| 					continue | ||||
| 				} | ||||
|  | ||||
| 				// Record new anonymous struct to explore in next round. | ||||
| 				nextCount[ft]++ | ||||
| 				if nextCount[ft] == 1 { | ||||
| 					f := field{name: ft.Name(), index: index, typ: ft} | ||||
| 					next = append(next, f) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	sort.Sort(byName(fields)) | ||||
|  | ||||
| 	// Delete all fields that are hidden by the Go rules for embedded fields, | ||||
| 	// except that fields with TOML tags are promoted. | ||||
|  | ||||
| 	// The fields are sorted in primary order of name, secondary order | ||||
| 	// of field index length. Loop over names; for each name, delete | ||||
| 	// hidden fields by choosing the one dominant field that survives. | ||||
| 	out := fields[:0] | ||||
| 	for advance, i := 0, 0; i < len(fields); i += advance { | ||||
| 		// One iteration per name. | ||||
| 		// Find the sequence of fields with the name of this first field. | ||||
| 		fi := fields[i] | ||||
| 		name := fi.name | ||||
| 		for advance = 1; i+advance < len(fields); advance++ { | ||||
| 			fj := fields[i+advance] | ||||
| 			if fj.name != name { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		if advance == 1 { // Only one field with this name | ||||
| 			out = append(out, fi) | ||||
| 			continue | ||||
| 		} | ||||
| 		dominant, ok := dominantField(fields[i : i+advance]) | ||||
| 		if ok { | ||||
| 			out = append(out, dominant) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	fields = out | ||||
| 	sort.Sort(byIndex(fields)) | ||||
|  | ||||
| 	return fields | ||||
| } | ||||
|  | ||||
| // dominantField looks through the fields, all of which are known to | ||||
| // have the same name, to find the single field that dominates the | ||||
| // others using Go's embedding rules, modified by the presence of | ||||
| // TOML tags. If there are multiple top-level fields, the boolean | ||||
| // will be false: This condition is an error in Go and we skip all | ||||
| // the fields. | ||||
| func dominantField(fields []field) (field, bool) { | ||||
| 	// The fields are sorted in increasing index-length order. The winner | ||||
| 	// must therefore be one with the shortest index length. Drop all | ||||
| 	// longer entries, which is easy: just truncate the slice. | ||||
| 	length := len(fields[0].index) | ||||
| 	tagged := -1 // Index of first tagged field. | ||||
| 	for i, f := range fields { | ||||
| 		if len(f.index) > length { | ||||
| 			fields = fields[:i] | ||||
| 			break | ||||
| 		} | ||||
| 		if f.tag { | ||||
| 			if tagged >= 0 { | ||||
| 				// Multiple tagged fields at the same level: conflict. | ||||
| 				// Return no field. | ||||
| 				return field{}, false | ||||
| 			} | ||||
| 			tagged = i | ||||
| 		} | ||||
| 	} | ||||
| 	if tagged >= 0 { | ||||
| 		return fields[tagged], true | ||||
| 	} | ||||
| 	// All remaining fields have the same length. If there's more than one, | ||||
| 	// we have a conflict (two fields named "X" at the same level) and we | ||||
| 	// return no field. | ||||
| 	if len(fields) > 1 { | ||||
| 		return field{}, false | ||||
| 	} | ||||
| 	return fields[0], true | ||||
| } | ||||
|  | ||||
| var fieldCache struct { | ||||
| 	sync.RWMutex | ||||
| 	m map[reflect.Type][]field | ||||
| } | ||||
|  | ||||
| // cachedTypeFields is like typeFields but uses a cache to avoid repeated work. | ||||
| func cachedTypeFields(t reflect.Type) []field { | ||||
| 	fieldCache.RLock() | ||||
| 	f := fieldCache.m[t] | ||||
| 	fieldCache.RUnlock() | ||||
| 	if f != nil { | ||||
| 		return f | ||||
| 	} | ||||
|  | ||||
| 	// Compute fields without lock. | ||||
| 	// Might duplicate effort but won't hold other computations back. | ||||
| 	f = typeFields(t) | ||||
| 	if f == nil { | ||||
| 		f = []field{} | ||||
| 	} | ||||
|  | ||||
| 	fieldCache.Lock() | ||||
| 	if fieldCache.m == nil { | ||||
| 		fieldCache.m = map[reflect.Type][]field{} | ||||
| 	} | ||||
| 	fieldCache.m[t] = f | ||||
| 	fieldCache.Unlock() | ||||
| 	return f | ||||
| } | ||||
							
								
								
									
										19
									
								
								vendor/github.com/Jeffail/gabs/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,19 @@ | ||||
| Copyright (c) 2014 Ashley Jeffs | ||||
|  | ||||
| Permission is hereby granted, free of charge, to any person obtaining a copy | ||||
| of this software and associated documentation files (the "Software"), to deal | ||||
| in the Software without restriction, including without limitation the rights | ||||
| to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||||
| copies of the Software, and to permit persons to whom the Software is | ||||
| furnished to do so, subject to the following conditions: | ||||
|  | ||||
| The above copyright notice and this permission notice shall be included in | ||||
| all copies or substantial portions of the Software. | ||||
|  | ||||
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||||
| IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||||
| FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||||
| AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||||
| LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||||
| OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||||
| THE SOFTWARE. | ||||
							
								
								
									
										315
									
								
								vendor/github.com/Jeffail/gabs/README.md
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,315 @@ | ||||
|  | ||||
|  | ||||
| Gabs is a small utility for dealing with dynamic or unknown JSON structures in | ||||
| golang. It's pretty much just a helpful wrapper around the golang | ||||
| `json.Marshal/json.Unmarshal` behaviour and `map[string]interface{}` objects. | ||||
| It does nothing spectacular except for being fabulous. | ||||
|  | ||||
| https://godoc.org/github.com/Jeffail/gabs | ||||
|  | ||||
| ## How to install: | ||||
|  | ||||
| ``` bash | ||||
| go get github.com/Jeffail/gabs | ||||
| ``` | ||||
|  | ||||
| ## How to use | ||||
|  | ||||
| ### Parsing and searching JSON | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| import "github.com/Jeffail/gabs" | ||||
|  | ||||
| jsonParsed, err := gabs.ParseJSON([]byte(`{ | ||||
| 	"outter":{ | ||||
| 		"inner":{ | ||||
| 			"value1":10, | ||||
| 			"value2":22 | ||||
| 		}, | ||||
| 		"alsoInner":{ | ||||
| 			"value1":20 | ||||
| 		} | ||||
| 	} | ||||
| }`)) | ||||
|  | ||||
| var value float64 | ||||
| var ok bool | ||||
|  | ||||
| value, ok = jsonParsed.Path("outter.inner.value1").Data().(float64) | ||||
| // value == 10.0, ok == true | ||||
|  | ||||
| value, ok = jsonParsed.Search("outter", "inner", "value1").Data().(float64) | ||||
| // value == 10.0, ok == true | ||||
|  | ||||
| value, ok = jsonParsed.Path("does.not.exist").Data().(float64) | ||||
| // value == 0.0, ok == false | ||||
|  | ||||
| exists := jsonParsed.Exists("outter", "inner", "value1") | ||||
| // exists == true | ||||
|  | ||||
| exists := jsonParsed.Exists("does", "not", "exist") | ||||
| // exists == false | ||||
|  | ||||
| exists := jsonParsed.ExistsP("does.not.exist") | ||||
| // exists == false | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| ### Iterating objects | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonParsed, _ := gabs.ParseJSON([]byte(`{"object":{ "first": 1, "second": 2, "third": 3 }}`)) | ||||
|  | ||||
| // S is shorthand for Search | ||||
| children, _ := jsonParsed.S("object").ChildrenMap() | ||||
| for key, child := range children { | ||||
| 	fmt.Printf("key: %v, value: %v\n", key, child.Data().(string)) | ||||
| } | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| ### Iterating arrays | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonParsed, _ := gabs.ParseJSON([]byte(`{"array":[ "first", "second", "third" ]}`)) | ||||
|  | ||||
| // S is shorthand for Search | ||||
| children, _ := jsonParsed.S("array").Children() | ||||
| for _, child := range children { | ||||
| 	fmt.Println(child.Data().(string)) | ||||
| } | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| first | ||||
| second | ||||
| third | ||||
| ``` | ||||
|  | ||||
| Children() will return all children of an array in order. This also works on | ||||
| objects, however, the children will be returned in a random order. | ||||
|  | ||||
| ### Searching through arrays | ||||
|  | ||||
| If your JSON structure contains arrays you can still search the fields of the | ||||
| objects within the array, this returns a JSON array containing the results for | ||||
| each element. | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonParsed, _ := gabs.ParseJSON([]byte(`{"array":[ {"value":1}, {"value":2}, {"value":3} ]}`)) | ||||
| fmt.Println(jsonParsed.Path("array.value").String()) | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| [1,2,3] | ||||
| ``` | ||||
|  | ||||
| ### Generating JSON | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonObj := gabs.New() | ||||
| // or gabs.Consume(jsonObject) to work on an existing map[string]interface{} | ||||
|  | ||||
| jsonObj.Set(10, "outter", "inner", "value") | ||||
| jsonObj.SetP(20, "outter.inner.value2") | ||||
| jsonObj.Set(30, "outter", "inner2", "value3") | ||||
|  | ||||
| fmt.Println(jsonObj.String()) | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| {"outter":{"inner":{"value":10,"value2":20},"inner2":{"value3":30}}} | ||||
| ``` | ||||
|  | ||||
| To pretty-print: | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| fmt.Println(jsonObj.StringIndent("", "  ")) | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| { | ||||
|   "outter": { | ||||
|     "inner": { | ||||
|       "value": 10, | ||||
|       "value2": 20 | ||||
|     }, | ||||
|     "inner2": { | ||||
|       "value3": 30 | ||||
|     } | ||||
|   } | ||||
| } | ||||
| ``` | ||||
|  | ||||
| ### Generating Arrays | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonObj := gabs.New() | ||||
|  | ||||
| jsonObj.Array("foo", "array") | ||||
| // Or .ArrayP("foo.array") | ||||
|  | ||||
| jsonObj.ArrayAppend(10, "foo", "array") | ||||
| jsonObj.ArrayAppend(20, "foo", "array") | ||||
| jsonObj.ArrayAppend(30, "foo", "array") | ||||
|  | ||||
| fmt.Println(jsonObj.String()) | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| {"foo":{"array":[10,20,30]}} | ||||
| ``` | ||||
|  | ||||
| Working with arrays by index: | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonObj := gabs.New() | ||||
|  | ||||
| // Create an array with the length of 3 | ||||
| jsonObj.ArrayOfSize(3, "foo") | ||||
|  | ||||
| jsonObj.S("foo").SetIndex("test1", 0) | ||||
| jsonObj.S("foo").SetIndex("test2", 1) | ||||
|  | ||||
| // Create an embedded array with the length of 3 | ||||
| jsonObj.S("foo").ArrayOfSizeI(3, 2) | ||||
|  | ||||
| jsonObj.S("foo").Index(2).SetIndex(1, 0) | ||||
| jsonObj.S("foo").Index(2).SetIndex(2, 1) | ||||
| jsonObj.S("foo").Index(2).SetIndex(3, 2) | ||||
|  | ||||
| fmt.Println(jsonObj.String()) | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| Will print: | ||||
|  | ||||
| ``` | ||||
| {"foo":["test1","test2",[1,2,3]]} | ||||
| ``` | ||||
|  | ||||
| ### Converting back to JSON | ||||
|  | ||||
| This is the easiest part: | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonParsedObj, _ := gabs.ParseJSON([]byte(`{ | ||||
| 	"outter":{ | ||||
| 		"values":{ | ||||
| 			"first":10, | ||||
| 			"second":11 | ||||
| 		} | ||||
| 	}, | ||||
| 	"outter2":"hello world" | ||||
| }`)) | ||||
|  | ||||
| jsonOutput := jsonParsedObj.String() | ||||
| // Becomes `{"outter":{"values":{"first":10,"second":11}},"outter2":"hello world"}` | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| And to serialize a specific segment is as simple as: | ||||
|  | ||||
| ``` go | ||||
| ... | ||||
|  | ||||
| jsonParsedObj := gabs.ParseJSON([]byte(`{ | ||||
| 	"outter":{ | ||||
| 		"values":{ | ||||
| 			"first":10, | ||||
| 			"second":11 | ||||
| 		} | ||||
| 	}, | ||||
| 	"outter2":"hello world" | ||||
| }`)) | ||||
|  | ||||
| jsonOutput := jsonParsedObj.Search("outter").String() | ||||
| // Becomes `{"values":{"first":10,"second":11}}` | ||||
|  | ||||
| ... | ||||
| ``` | ||||
|  | ||||
| ### Merge two containers | ||||
|  | ||||
| You can merge a JSON structure into an existing one, where collisions will be | ||||
| converted into a JSON array. | ||||
|  | ||||
| ``` go | ||||
| jsonParsed1, _ := ParseJSON([]byte(`{"outter": {"value1": "one"}}`)) | ||||
| jsonParsed2, _ := ParseJSON([]byte(`{"outter": {"inner": {"value3": "three"}}, "outter2": {"value2": "two"}}`)) | ||||
|  | ||||
| jsonParsed1.Merge(jsonParsed2) | ||||
| // Becomes `{"outter":{"inner":{"value3":"three"},"value1":"one"},"outter2":{"value2":"two"}}` | ||||
| ``` | ||||
|  | ||||
| Arrays are merged: | ||||
|  | ||||
| ``` go | ||||
| jsonParsed1, _ := ParseJSON([]byte(`{"array": ["one"]}`)) | ||||
| jsonParsed2, _ := ParseJSON([]byte(`{"array": ["two"]}`)) | ||||
|  | ||||
| jsonParsed1.Merge(jsonParsed2) | ||||
| // Becomes `{"array":["one", "two"]}` | ||||
| ``` | ||||
|  | ||||
| ### Parsing Numbers | ||||
|  | ||||
| Gabs uses the `json` package under the bonnet, which by default will parse all | ||||
| number values into `float64`. If you need to parse `Int` values then you should | ||||
| use a `json.Decoder` (https://golang.org/pkg/encoding/json/#Decoder): | ||||
|  | ||||
| ``` go | ||||
| sample := []byte(`{"test":{"int":10, "float":6.66}}`) | ||||
| dec := json.NewDecoder(bytes.NewReader(sample)) | ||||
| dec.UseNumber() | ||||
|  | ||||
| val, err := gabs.ParseJSONDecoder(dec) | ||||
| if err != nil { | ||||
|     t.Errorf("Failed to parse: %v", err) | ||||
|     return | ||||
| } | ||||
|  | ||||
| intValue, err := val.Path("test.int").Data().(json.Number).Int64() | ||||
| ``` | ||||
							
								
								
									
										581
									
								
								vendor/github.com/Jeffail/gabs/gabs.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,581 @@ | ||||
| /* | ||||
| Copyright (c) 2014 Ashley Jeffs | ||||
|  | ||||
| Permission is hereby granted, free of charge, to any person obtaining a copy | ||||
| of this software and associated documentation files (the "Software"), to deal | ||||
| in the Software without restriction, including without limitation the rights | ||||
| to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||||
| copies of the Software, and to permit persons to whom the Software is | ||||
| furnished to do so, subject to the following conditions: | ||||
|  | ||||
| The above copyright notice and this permission notice shall be included in | ||||
| all copies or substantial portions of the Software. | ||||
|  | ||||
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||||
| IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||||
| FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||||
| AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||||
| LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||||
| OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN | ||||
| THE SOFTWARE. | ||||
| */ | ||||
|  | ||||
| // Package gabs implements a simplified wrapper around creating and parsing JSON. | ||||
| package gabs | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| var ( | ||||
| 	// ErrOutOfBounds - Index out of bounds. | ||||
| 	ErrOutOfBounds = errors.New("out of bounds") | ||||
|  | ||||
| 	// ErrNotObjOrArray - The target is not an object or array type. | ||||
| 	ErrNotObjOrArray = errors.New("not an object or array") | ||||
|  | ||||
| 	// ErrNotObj - The target is not an object type. | ||||
| 	ErrNotObj = errors.New("not an object") | ||||
|  | ||||
| 	// ErrNotArray - The target is not an array type. | ||||
| 	ErrNotArray = errors.New("not an array") | ||||
|  | ||||
| 	// ErrPathCollision - Creating a path failed because an element collided with an existing value. | ||||
| 	ErrPathCollision = errors.New("encountered value collision whilst building path") | ||||
|  | ||||
| 	// ErrInvalidInputObj - The input value was not a map[string]interface{}. | ||||
| 	ErrInvalidInputObj = errors.New("invalid input object") | ||||
|  | ||||
| 	// ErrInvalidInputText - The input data could not be parsed. | ||||
| 	ErrInvalidInputText = errors.New("input text could not be parsed") | ||||
|  | ||||
| 	// ErrInvalidPath - The filepath was not valid. | ||||
| 	ErrInvalidPath = errors.New("invalid file path") | ||||
|  | ||||
| 	// ErrInvalidBuffer - The input buffer contained an invalid JSON string | ||||
| 	ErrInvalidBuffer = errors.New("input buffer contained invalid JSON") | ||||
| ) | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| // Container - an internal structure that holds a reference to the core interface map of the parsed | ||||
| // json. Use this container to move context. | ||||
| type Container struct { | ||||
| 	object interface{} | ||||
| } | ||||
|  | ||||
| // Data - Return the contained data as an interface{}. | ||||
| func (g *Container) Data() interface{} { | ||||
| 	if g == nil { | ||||
| 		return nil | ||||
| 	} | ||||
| 	return g.object | ||||
| } | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| // Path - Search for a value using dot notation. | ||||
| func (g *Container) Path(path string) *Container { | ||||
| 	return g.Search(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // Search - Attempt to find and return an object within the JSON structure by specifying the | ||||
| // hierarchy of field names to locate the target. If the search encounters an array and has not | ||||
| // reached the end target then it will iterate each object of the array for the target and return | ||||
| // all of the results in a JSON array. | ||||
| func (g *Container) Search(hierarchy ...string) *Container { | ||||
| 	var object interface{} | ||||
|  | ||||
| 	object = g.Data() | ||||
| 	for target := 0; target < len(hierarchy); target++ { | ||||
| 		if mmap, ok := object.(map[string]interface{}); ok { | ||||
| 			object, ok = mmap[hierarchy[target]] | ||||
| 			if !ok { | ||||
| 				return nil | ||||
| 			} | ||||
| 		} else if marray, ok := object.([]interface{}); ok { | ||||
| 			tmpArray := []interface{}{} | ||||
| 			for _, val := range marray { | ||||
| 				tmpGabs := &Container{val} | ||||
| 				res := tmpGabs.Search(hierarchy[target:]...) | ||||
| 				if res != nil { | ||||
| 					tmpArray = append(tmpArray, res.Data()) | ||||
| 				} | ||||
| 			} | ||||
| 			if len(tmpArray) == 0 { | ||||
| 				return nil | ||||
| 			} | ||||
| 			return &Container{tmpArray} | ||||
| 		} else { | ||||
| 			return nil | ||||
| 		} | ||||
| 	} | ||||
| 	return &Container{object} | ||||
| } | ||||
|  | ||||
| // S - Shorthand method, does the same thing as Search. | ||||
| func (g *Container) S(hierarchy ...string) *Container { | ||||
| 	return g.Search(hierarchy...) | ||||
| } | ||||
|  | ||||
| // Exists - Checks whether a path exists. | ||||
| func (g *Container) Exists(hierarchy ...string) bool { | ||||
| 	return g.Search(hierarchy...) != nil | ||||
| } | ||||
|  | ||||
| // ExistsP - Checks whether a dot notation path exists. | ||||
| func (g *Container) ExistsP(path string) bool { | ||||
| 	return g.Exists(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // Index - Attempt to find and return an object within a JSON array by index. | ||||
| func (g *Container) Index(index int) *Container { | ||||
| 	if array, ok := g.Data().([]interface{}); ok { | ||||
| 		if index >= len(array) { | ||||
| 			return &Container{nil} | ||||
| 		} | ||||
| 		return &Container{array[index]} | ||||
| 	} | ||||
| 	return &Container{nil} | ||||
| } | ||||
|  | ||||
| // Children - Return a slice of all the children of the array. This also works for objects, however, | ||||
| // the children returned for an object will NOT be in order and you lose the names of the returned | ||||
| // objects this way. | ||||
| func (g *Container) Children() ([]*Container, error) { | ||||
| 	if array, ok := g.Data().([]interface{}); ok { | ||||
| 		children := make([]*Container, len(array)) | ||||
| 		for i := 0; i < len(array); i++ { | ||||
| 			children[i] = &Container{array[i]} | ||||
| 		} | ||||
| 		return children, nil | ||||
| 	} | ||||
| 	if mmap, ok := g.Data().(map[string]interface{}); ok { | ||||
| 		children := []*Container{} | ||||
| 		for _, obj := range mmap { | ||||
| 			children = append(children, &Container{obj}) | ||||
| 		} | ||||
| 		return children, nil | ||||
| 	} | ||||
| 	return nil, ErrNotObjOrArray | ||||
| } | ||||
|  | ||||
| // ChildrenMap - Return a map of all the children of an object. | ||||
| func (g *Container) ChildrenMap() (map[string]*Container, error) { | ||||
| 	if mmap, ok := g.Data().(map[string]interface{}); ok { | ||||
| 		children := map[string]*Container{} | ||||
| 		for name, obj := range mmap { | ||||
| 			children[name] = &Container{obj} | ||||
| 		} | ||||
| 		return children, nil | ||||
| 	} | ||||
| 	return nil, ErrNotObj | ||||
| } | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| // Set - Set the value of a field at a JSON path, any parts of the path that do not exist will be | ||||
| // constructed, and if a collision occurs with a non object type whilst iterating the path an error | ||||
| // is returned. | ||||
| func (g *Container) Set(value interface{}, path ...string) (*Container, error) { | ||||
| 	if len(path) == 0 { | ||||
| 		g.object = value | ||||
| 		return g, nil | ||||
| 	} | ||||
| 	var object interface{} | ||||
| 	if g.object == nil { | ||||
| 		g.object = map[string]interface{}{} | ||||
| 	} | ||||
| 	object = g.object | ||||
| 	for target := 0; target < len(path); target++ { | ||||
| 		if mmap, ok := object.(map[string]interface{}); ok { | ||||
| 			if target == len(path)-1 { | ||||
| 				mmap[path[target]] = value | ||||
| 			} else if mmap[path[target]] == nil { | ||||
| 				mmap[path[target]] = map[string]interface{}{} | ||||
| 			} | ||||
| 			object = mmap[path[target]] | ||||
| 		} else { | ||||
| 			return &Container{nil}, ErrPathCollision | ||||
| 		} | ||||
| 	} | ||||
| 	return &Container{object}, nil | ||||
| } | ||||
|  | ||||
| // SetP - Does the same as Set, but using a dot notation JSON path. | ||||
| func (g *Container) SetP(value interface{}, path string) (*Container, error) { | ||||
| 	return g.Set(value, strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // SetIndex - Set a value of an array element based on the index. | ||||
| func (g *Container) SetIndex(value interface{}, index int) (*Container, error) { | ||||
| 	if array, ok := g.Data().([]interface{}); ok { | ||||
| 		if index >= len(array) { | ||||
| 			return &Container{nil}, ErrOutOfBounds | ||||
| 		} | ||||
| 		array[index] = value | ||||
| 		return &Container{array[index]}, nil | ||||
| 	} | ||||
| 	return &Container{nil}, ErrNotArray | ||||
| } | ||||
|  | ||||
| // Object - Create a new JSON object at a path. Returns an error if the path contains a collision | ||||
| // with a non object type. | ||||
| func (g *Container) Object(path ...string) (*Container, error) { | ||||
| 	return g.Set(map[string]interface{}{}, path...) | ||||
| } | ||||
|  | ||||
| // ObjectP - Does the same as Object, but using a dot notation JSON path. | ||||
| func (g *Container) ObjectP(path string) (*Container, error) { | ||||
| 	return g.Object(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ObjectI - Create a new JSON object at an array index. Returns an error if the object is not an | ||||
| // array or the index is out of bounds. | ||||
| func (g *Container) ObjectI(index int) (*Container, error) { | ||||
| 	return g.SetIndex(map[string]interface{}{}, index) | ||||
| } | ||||
|  | ||||
| // Array - Create a new JSON array at a path. Returns an error if the path contains a collision with | ||||
| // a non object type. | ||||
| func (g *Container) Array(path ...string) (*Container, error) { | ||||
| 	return g.Set([]interface{}{}, path...) | ||||
| } | ||||
|  | ||||
| // ArrayP - Does the same as Array, but using a dot notation JSON path. | ||||
| func (g *Container) ArrayP(path string) (*Container, error) { | ||||
| 	return g.Array(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ArrayI - Create a new JSON array at an array index. Returns an error if the object is not an | ||||
| // array or the index is out of bounds. | ||||
| func (g *Container) ArrayI(index int) (*Container, error) { | ||||
| 	return g.SetIndex([]interface{}{}, index) | ||||
| } | ||||
|  | ||||
| // ArrayOfSize - Create a new JSON array of a particular size at a path. Returns an error if the | ||||
| // path contains a collision with a non object type. | ||||
| func (g *Container) ArrayOfSize(size int, path ...string) (*Container, error) { | ||||
| 	a := make([]interface{}, size) | ||||
| 	return g.Set(a, path...) | ||||
| } | ||||
|  | ||||
| // ArrayOfSizeP - Does the same as ArrayOfSize, but using a dot notation JSON path. | ||||
| func (g *Container) ArrayOfSizeP(size int, path string) (*Container, error) { | ||||
| 	return g.ArrayOfSize(size, strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ArrayOfSizeI - Create a new JSON array of a particular size at an array index. Returns an error | ||||
| // if the object is not an array or the index is out of bounds. | ||||
| func (g *Container) ArrayOfSizeI(size, index int) (*Container, error) { | ||||
| 	a := make([]interface{}, size) | ||||
| 	return g.SetIndex(a, index) | ||||
| } | ||||
|  | ||||
| // Delete - Delete an element at a JSON path, an error is returned if the element does not exist. | ||||
| func (g *Container) Delete(path ...string) error { | ||||
| 	var object interface{} | ||||
|  | ||||
| 	if g.object == nil { | ||||
| 		return ErrNotObj | ||||
| 	} | ||||
| 	object = g.object | ||||
| 	for target := 0; target < len(path); target++ { | ||||
| 		if mmap, ok := object.(map[string]interface{}); ok { | ||||
| 			if target == len(path)-1 { | ||||
| 				if _, ok := mmap[path[target]]; ok { | ||||
| 					delete(mmap, path[target]) | ||||
| 				} else { | ||||
| 					return ErrNotObj | ||||
| 				} | ||||
| 			} | ||||
| 			object = mmap[path[target]] | ||||
| 		} else { | ||||
| 			return ErrNotObj | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // DeleteP - Does the same as Delete, but using a dot notation JSON path. | ||||
| func (g *Container) DeleteP(path string) error { | ||||
| 	return g.Delete(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // Merge - Merges two gabs-containers | ||||
| func (g *Container) Merge(toMerge *Container) error { | ||||
| 	var recursiveFnc func(map[string]interface{}, []string) error | ||||
| 	recursiveFnc = func(mmap map[string]interface{}, path []string) error { | ||||
| 		for key, value := range mmap { | ||||
| 			newPath := append(path, key) | ||||
| 			if g.Exists(newPath...) { | ||||
| 				target := g.Search(newPath...) | ||||
| 				switch t := value.(type) { | ||||
| 				case map[string]interface{}: | ||||
| 					switch targetV := target.Data().(type) { | ||||
| 					case map[string]interface{}: | ||||
| 						if err := recursiveFnc(t, newPath); err != nil { | ||||
| 							return err | ||||
| 						} | ||||
| 					case []interface{}: | ||||
| 						g.Set(append(targetV, t), newPath...) | ||||
| 					default: | ||||
| 						newSlice := append([]interface{}{}, targetV) | ||||
| 						g.Set(append(newSlice, t), newPath...) | ||||
| 					} | ||||
| 				case []interface{}: | ||||
| 					for _, valueOfSlice := range t { | ||||
| 						if err := g.ArrayAppend(valueOfSlice, newPath...); err != nil { | ||||
| 							return err | ||||
| 						} | ||||
| 					} | ||||
| 				default: | ||||
| 					switch targetV := target.Data().(type) { | ||||
| 					case []interface{}: | ||||
| 						g.Set(append(targetV, t), newPath...) | ||||
| 					default: | ||||
| 						newSlice := append([]interface{}{}, targetV) | ||||
| 						g.Set(append(newSlice, t), newPath...) | ||||
| 					} | ||||
| 				} | ||||
| 			} else { | ||||
| 				// path doesn't exist. So set the value | ||||
| 				if _, err := g.Set(value, newPath...); err != nil { | ||||
| 					return err | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		return nil | ||||
| 	} | ||||
| 	if mmap, ok := toMerge.Data().(map[string]interface{}); ok { | ||||
| 		return recursiveFnc(mmap, []string{}) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| /* | ||||
| Array modification/search - Keeping these options simple right now, no need for anything more | ||||
| complicated since you can just cast to []interface{}, modify and then reassign with Set. | ||||
| */ | ||||
|  | ||||
| // ArrayAppend - Append a value onto a JSON array. If the target is not a JSON array then it will be | ||||
| // converted into one, with its contents as the first element of the array. | ||||
| func (g *Container) ArrayAppend(value interface{}, path ...string) error { | ||||
| 	if array, ok := g.Search(path...).Data().([]interface{}); ok { | ||||
| 		array = append(array, value) | ||||
| 		_, err := g.Set(array, path...) | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	newArray := []interface{}{} | ||||
| 	if d := g.Search(path...).Data(); d != nil { | ||||
| 		newArray = append(newArray, d) | ||||
| 	} | ||||
| 	newArray = append(newArray, value) | ||||
|  | ||||
| 	_, err := g.Set(newArray, path...) | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // ArrayAppendP - Append a value onto a JSON array using a dot notation JSON path. | ||||
| func (g *Container) ArrayAppendP(value interface{}, path string) error { | ||||
| 	return g.ArrayAppend(value, strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ArrayRemove - Remove an element from a JSON array. | ||||
| func (g *Container) ArrayRemove(index int, path ...string) error { | ||||
| 	if index < 0 { | ||||
| 		return ErrOutOfBounds | ||||
| 	} | ||||
| 	array, ok := g.Search(path...).Data().([]interface{}) | ||||
| 	if !ok { | ||||
| 		return ErrNotArray | ||||
| 	} | ||||
| 	if index < len(array) { | ||||
| 		array = append(array[:index], array[index+1:]...) | ||||
| 	} else { | ||||
| 		return ErrOutOfBounds | ||||
| 	} | ||||
| 	_, err := g.Set(array, path...) | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // ArrayRemoveP - Remove an element from a JSON array using a dot notation JSON path. | ||||
| func (g *Container) ArrayRemoveP(index int, path string) error { | ||||
| 	return g.ArrayRemove(index, strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ArrayElement - Access an element from a JSON array. | ||||
| func (g *Container) ArrayElement(index int, path ...string) (*Container, error) { | ||||
| 	if index < 0 { | ||||
| 		return &Container{nil}, ErrOutOfBounds | ||||
| 	} | ||||
| 	array, ok := g.Search(path...).Data().([]interface{}) | ||||
| 	if !ok { | ||||
| 		return &Container{nil}, ErrNotArray | ||||
| 	} | ||||
| 	if index < len(array) { | ||||
| 		return &Container{array[index]}, nil | ||||
| 	} | ||||
| 	return &Container{nil}, ErrOutOfBounds | ||||
| } | ||||
|  | ||||
| // ArrayElementP - Access an element from a JSON array using a dot notation JSON path. | ||||
| func (g *Container) ArrayElementP(index int, path string) (*Container, error) { | ||||
| 	return g.ArrayElement(index, strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| // ArrayCount - Count the number of elements in a JSON array. | ||||
| func (g *Container) ArrayCount(path ...string) (int, error) { | ||||
| 	if array, ok := g.Search(path...).Data().([]interface{}); ok { | ||||
| 		return len(array), nil | ||||
| 	} | ||||
| 	return 0, ErrNotArray | ||||
| } | ||||
|  | ||||
| // ArrayCountP - Count the number of elements in a JSON array using a dot notation JSON path. | ||||
| func (g *Container) ArrayCountP(path string) (int, error) { | ||||
| 	return g.ArrayCount(strings.Split(path, ".")...) | ||||
| } | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
|  | ||||
| // Bytes - Converts the contained object back to a JSON []byte blob. | ||||
| func (g *Container) Bytes() []byte { | ||||
| 	if g.Data() != nil { | ||||
| 		if bytes, err := json.Marshal(g.object); err == nil { | ||||
| 			return bytes | ||||
| 		} | ||||
| 	} | ||||
| 	return []byte("{}") | ||||
| } | ||||
|  | ||||
| // BytesIndent - Converts the contained object to a JSON []byte blob formatted with prefix, indent. | ||||
| func (g *Container) BytesIndent(prefix string, indent string) []byte { | ||||
| 	if g.object != nil { | ||||
| 		if bytes, err := json.MarshalIndent(g.object, prefix, indent); err == nil { | ||||
| 			return bytes | ||||
| 		} | ||||
| 	} | ||||
| 	return []byte("{}") | ||||
| } | ||||
|  | ||||
| // String - Converts the contained object to a JSON formatted string. | ||||
| func (g *Container) String() string { | ||||
| 	return string(g.Bytes()) | ||||
| } | ||||
|  | ||||
| // StringIndent - Converts the contained object back to a JSON formatted string with prefix, indent. | ||||
| func (g *Container) StringIndent(prefix string, indent string) string { | ||||
| 	return string(g.BytesIndent(prefix, indent)) | ||||
| } | ||||
|  | ||||
| // EncodeOpt is a functional option for the EncodeJSON method. | ||||
| type EncodeOpt func(e *json.Encoder) | ||||
|  | ||||
| // EncodeOptHTMLEscape sets the encoder to escape the JSON for html. | ||||
| func EncodeOptHTMLEscape(doEscape bool) EncodeOpt { | ||||
| 	return func(e *json.Encoder) { | ||||
| 		e.SetEscapeHTML(doEscape) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // EncodeOptIndent sets the encoder to indent the JSON output. | ||||
| func EncodeOptIndent(prefix string, indent string) EncodeOpt { | ||||
| 	return func(e *json.Encoder) { | ||||
| 		e.SetIndent(prefix, indent) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // EncodeJSON - Encodes the contained object back to a JSON formatted []byte | ||||
| // using a variant list of modifier functions for the encoder being used. | ||||
| // Functions for modifying the output are prefixed with EncodeOpt, e.g. | ||||
| // EncodeOptHTMLEscape. | ||||
| func (g *Container) EncodeJSON(encodeOpts ...EncodeOpt) []byte { | ||||
| 	var b bytes.Buffer | ||||
| 	encoder := json.NewEncoder(&b) | ||||
| 	encoder.SetEscapeHTML(false) // Do not escape by default. | ||||
| 	for _, opt := range encodeOpts { | ||||
| 		opt(encoder) | ||||
| 	} | ||||
| 	if err := encoder.Encode(g.object); err != nil { | ||||
| 		return []byte("{}") | ||||
| 	} | ||||
| 	result := b.Bytes() | ||||
| 	if len(result) > 0 { | ||||
| 		result = result[:len(result)-1] | ||||
| 	} | ||||
| 	return result | ||||
| } | ||||
|  | ||||
| // New - Create a new gabs JSON object. | ||||
| func New() *Container { | ||||
| 	return &Container{map[string]interface{}{}} | ||||
| } | ||||
|  | ||||
| // Consume - Gobble up an already converted JSON object, or a fresh map[string]interface{} object. | ||||
| func Consume(root interface{}) (*Container, error) { | ||||
| 	return &Container{root}, nil | ||||
| } | ||||
|  | ||||
| // ParseJSON - Convert a string into a representation of the parsed JSON. | ||||
| func ParseJSON(sample []byte) (*Container, error) { | ||||
| 	var gabs Container | ||||
|  | ||||
| 	if err := json.Unmarshal(sample, &gabs.object); err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return &gabs, nil | ||||
| } | ||||
|  | ||||
| // ParseJSONDecoder - Convert a json.Decoder into a representation of the parsed JSON. | ||||
| func ParseJSONDecoder(decoder *json.Decoder) (*Container, error) { | ||||
| 	var gabs Container | ||||
|  | ||||
| 	if err := decoder.Decode(&gabs.object); err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return &gabs, nil | ||||
| } | ||||
|  | ||||
| // ParseJSONFile - Read a file and convert into a representation of the parsed JSON. | ||||
| func ParseJSONFile(path string) (*Container, error) { | ||||
| 	if len(path) > 0 { | ||||
| 		cBytes, err := ioutil.ReadFile(path) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
|  | ||||
| 		container, err := ParseJSON(cBytes) | ||||
| 		if err != nil { | ||||
| 			return nil, err | ||||
| 		} | ||||
|  | ||||
| 		return container, nil | ||||
| 	} | ||||
| 	return nil, ErrInvalidPath | ||||
| } | ||||
|  | ||||
| // ParseJSONBuffer - Read the contents of a buffer into a representation of the parsed JSON. | ||||
| func ParseJSONBuffer(buffer io.Reader) (*Container, error) { | ||||
| 	var gabs Container | ||||
| 	jsonDecoder := json.NewDecoder(buffer) | ||||
| 	if err := jsonDecoder.Decode(&gabs.object); err != nil { | ||||
| 		return nil, err | ||||
| 	} | ||||
|  | ||||
| 	return &gabs, nil | ||||
| } | ||||
|  | ||||
| //-------------------------------------------------------------------------------------------------- | ||||
							
								
								
									
										
											BIN
										
									
								
								vendor/github.com/Jeffail/gabs/gabs_logo.png
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| After Width: | Height: | Size: 164 KiB | 
							
								
								
									
										41
									
								
								vendor/github.com/Philipp15b/go-steam/auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						| @@ -2,13 +2,14 @@ package steam | ||||
|  | ||||
| import ( | ||||
| 	"crypto/sha1" | ||||
| 	"sync/atomic" | ||||
| 	"time" | ||||
|  | ||||
| 	. "github.com/Philipp15b/go-steam/protocol" | ||||
| 	. "github.com/Philipp15b/go-steam/protocol/protobuf" | ||||
| 	. "github.com/Philipp15b/go-steam/protocol/steamlang" | ||||
| 	. "github.com/Philipp15b/go-steam/steamid" | ||||
| 	"github.com/golang/protobuf/proto" | ||||
| 	"sync/atomic" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| type Auth struct { | ||||
| @@ -19,23 +20,41 @@ type Auth struct { | ||||
| type SentryHash []byte | ||||
|  | ||||
| type LogOnDetails struct { | ||||
| 	Username       string | ||||
| 	Password       string | ||||
| 	AuthCode       string | ||||
| 	Username string | ||||
|  | ||||
| 	// If logging into an account without a login key, the account's password. | ||||
| 	Password string | ||||
|  | ||||
| 	// If you have a Steam Guard email code, you can provide it here. | ||||
| 	AuthCode string | ||||
|  | ||||
| 	// If you have a Steam Guard mobile two-factor authentication code, you can provide it here. | ||||
| 	TwoFactorCode  string | ||||
| 	SentryFileHash SentryHash | ||||
| 	LoginKey       string | ||||
|  | ||||
| 	// true if you want to get a login key which can be used in lieu of | ||||
| 	// a password for subsequent logins. false or omitted otherwise. | ||||
| 	ShouldRememberPassword bool | ||||
| } | ||||
|  | ||||
| // Log on with the given details. You must always specify username and | ||||
| // password. For the first login, don't set an authcode or a hash and you'll receive an error | ||||
| // password OR username and loginkey. For the first login, don't set an authcode or a hash and you'll | ||||
| //  receive an error (EResult_AccountLogonDenied) | ||||
| // and Steam will send you an authcode. Then you have to login again, this time with the authcode. | ||||
| // Shortly after logging in, you'll receive a MachineAuthUpdateEvent with a hash which allows | ||||
| // you to login without using an authcode in the future. | ||||
| // | ||||
| // If you don't use Steam Guard, username and password are enough. | ||||
| // | ||||
| // After the event EMsg_ClientNewLoginKey is received you can use the LoginKey | ||||
| // to login instead of using the password. | ||||
| func (a *Auth) LogOn(details *LogOnDetails) { | ||||
| 	if len(details.Username) == 0 || len(details.Password) == 0 { | ||||
| 		panic("Username and password must be set!") | ||||
| 	if details.Username == "" { | ||||
| 		panic("Username must be set!") | ||||
| 	} | ||||
| 	if details.Password == "" && details.LoginKey == "" { | ||||
| 		panic("Password or LoginKey must be set!") | ||||
| 	} | ||||
|  | ||||
| 	logon := new(CMsgClientLogon) | ||||
| @@ -50,6 +69,12 @@ func (a *Auth) LogOn(details *LogOnDetails) { | ||||
| 	logon.ClientLanguage = proto.String("english") | ||||
| 	logon.ProtocolVersion = proto.Uint32(MsgClientLogon_CurrentProtocol) | ||||
| 	logon.ShaSentryfile = details.SentryFileHash | ||||
| 	if details.LoginKey != "" { | ||||
| 		logon.LoginKey = proto.String(details.LoginKey) | ||||
| 	} | ||||
| 	if details.ShouldRememberPassword { | ||||
| 		logon.ShouldRememberPassword = proto.Bool(details.ShouldRememberPassword) | ||||
| 	} | ||||
|  | ||||
| 	atomic.StoreUint64(&a.client.steamId, uint64(NewIdAdv(0, 1, int32(EUniverse_Public), int32(EAccountType_Individual)))) | ||||
|  | ||||
|   | ||||
							
								
								
									
										2
									
								
								vendor/github.com/Rhymen/go-whatsapp/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,2 @@ | ||||
| .idea/ | ||||
| docs/ | ||||
							
								
								
									
										21
									
								
								vendor/github.com/Rhymen/go-whatsapp/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						| @@ -0,0 +1,21 @@ | ||||
| MIT License | ||||
|  | ||||
| Copyright (c) 2018 | ||||
|  | ||||
| Permission is hereby granted, free of charge, to any person obtaining a copy | ||||
| of this software and associated documentation files (the "Software"), to deal | ||||
| in the Software without restriction, including without limitation the rights | ||||
| to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | ||||
| copies of the Software, and to permit persons to whom the Software is | ||||
| furnished to do so, subject to the following conditions: | ||||
|  | ||||
| The above copyright notice and this permission notice shall be included in all | ||||
| copies or substantial portions of the Software. | ||||
|  | ||||
| THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | ||||
| IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | ||||
| FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | ||||
| AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | ||||
| LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | ||||
| OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | ||||
| SOFTWARE. | ||||