forked from lug/matterbridge
		
	Compare commits
	
		
			84 Commits
		
	
	
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|   | 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 | 
							
								
								
									
										213
									
								
								.golangci.yaml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										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 | ||||
|     enabled-checks: | ||||
|       - appendAssign | ||||
|       - assignOp | ||||
|       - boolExprSimplify | ||||
|       - builtinShadow | ||||
|       - captLocal | ||||
|       - caseOrder | ||||
|       - commentedOutImport | ||||
|       - defaultCaseOrder | ||||
|       - dupArg | ||||
|       - dupBranchBody | ||||
|       - dupCase | ||||
|       - dupSubExpr | ||||
|       - elseif | ||||
|       - emptyFallthrough | ||||
|       - hugeParam | ||||
|       - 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" | ||||
							
								
								
									
										40
									
								
								.travis.yml
									
									
									
									
									
								
							
							
						
						
									
										40
									
								
								.travis.yml
									
									
									
									
									
								
							| @@ -1,6 +1,7 @@ | ||||
| language: go | ||||
| go: | ||||
|     - 1.11.x | ||||
| go_import_path: github.com/42wim/matterbridge | ||||
|  | ||||
| # we have everything vendored | ||||
| install: true | ||||
| @@ -9,9 +10,9 @@ git: | ||||
|   depth: 200 | ||||
|  | ||||
| env: | ||||
|   global: | ||||
|     - GOOS=linux GOARCH=amd64 | ||||
|       #    - GOOS=windows GOARCH=amd64 | ||||
|       #- GOOS=linux GOARCH=arm | ||||
|     - GOLANGCI_VERSION="v1.12.3" | ||||
|  | ||||
| matrix: | ||||
|   # It's ok if our code fails on unstable development versions of Go. | ||||
| @@ -25,25 +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 | ||||
|   - curl -sfL https://install.goreleaser.com/github.com/golangci/golangci-lint.sh | bash -s -- -b $GOPATH/bin v1.12.2 | ||||
|   # 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 "$(go fmt ./...)"         # Fail if a .go file hasn't been formatted with gofmt | ||||
|   - go test -v -race $PKGS            # Run all the tests with the race detector enabled | ||||
|   #- go vet $PKGS                      # go vet is the official Go static analyzer | ||||
|   - golangci-lint run --enable-all -D lll -D errcheck -D gosec -D maligned -D prealloc -D gocyclo -D gochecknoglobals | ||||
|   #- megacheck $PKGS                   # "go vet on steroids" + linter | ||||
|   # 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 | ||||
| @@ -52,4 +54,4 @@ deploy: | ||||
|   file: ci/deploy.json | ||||
|   user: 42wim | ||||
|   key: | ||||
|      secure: "CeXXe6JOmt7HYR81MdWLua0ltQHhDdkIeRGBFbgd7hkb1wi8eF9DgpAcQrTso8NIlHNZmSAP46uhFgsRvkuezzX0ygalZ7DCJyAyn3sAMEh+UQSHV1WGThRehTtidqRGjetzsIGSwdrJOWil+XTfbO1Z8DGzfakhSuAZka8CM4BAoe3YeP9rYK8h+84x0GHfczvsLtXZ3mWLvQuwe4pK6+ItBCUg0ae7O7ZUpWHy0xQQkkWztY/6RAzXfaG7DuGjIw+20fhx3WOXRNpHCtZ6Bc3qERCpk0s1HhlQWlrN9wDaFTBWYwlvSnNgvxxMbNXJ6RrRJ0l0bA7FUswYwyroxhzrGLdzWDg8dHaQkypocngdalfhpsnoO9j3ApJhomUFJ3UoEq5nOGRUrKn8MPi+dP0zE4kNQ3e4VNa1ufNrvfpWolMg3xh8OXuhQdD5wIM5zFAbRJLqWSCVAjPq4DDPecmvXBOlIial7oa312lN5qnBnUjvAcxszZ+FUyDHT1Grxzna4tMwxY9obPzZUzm7359AOCCwIQFVB8GLqD2nwIstcXS0zGRz+fhviPipHuBa02q5bGUZwmkvrSNab0s8Jo7pCrel2Rz3nWPKaiCfq2WjbW1CLheSMkOQrjsdUd1hhbqNWFPUjJPInTc77NAKCfm5runv5uyowRLh4NNd0sI=" | ||||
|      secure: "CeXXe6JOmt7HYR81MdWLua0ltQHhDdkIeRGBFbgd7hkb1wi8eF9DgpAcQrTso8NIlHNZmSAP46uhFgsRvkuezzX0ygalZ7DCJyAyn3sAMEh+UQSHV1WGThRehTtidqRGjetzsIGSwdrJOWil+XTfbO1Z8DGzfakhSuAZka8CM4BAoe3YeP9rYK8h+84x0GHfczvsLtXZ3mWLvQuwe4pK6+ItBCUg0ae7O7ZUpWHy0xQQkkWztY/6RAzXfaG7DuGjIw+20fhx3WOXRNpHCtZ6Bc3qERCpk0s1HhlQWlrN9wDaFTBWYwlvSnNgvxxMbNXJ6RrRJ0l0bA7FUswYwyroxhzrGLdzWDg8dHaQkypocngdalfhpsnoO9j3ApJhomUFJ3UoEq5nOGRUrKn8MPi+dP0zE4kNQ3e4VNa1ufNrvfpWolMg3xh8OXuhQdD5wIM5zFAbRJLqWSCVAjPq4DDPecmvXBOlIial7oa312lN5qnBnUjvAcxszZ+FUyDHT1Grxzna4tMwxY9obPzZUzm7359AOCCwIQFVB8GLqD2nwIstcXS0zGRz+fhviPipHuBa02q5bGUZwmkvrSNab0s8Jo7pCrel2Rz3nWPKaiCfq2WjbW1CLheSMkOQrjsdUd1hhbqNWFPUjJPInTc77NAKCfm5runv5uyowRLh4NNd0sI=" | ||||
							
								
								
									
										24
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										24
									
								
								README.md
									
									
									
									
									
								
							| @@ -62,7 +62,8 @@ | ||||
| * [API](https://github.com/42wim/matterbridge/wiki/Features#api) | ||||
|  | ||||
| ### API | ||||
| The API is very basic at the moment and rather undocumented. | ||||
| The API is very basic at the moment.    | ||||
| More info and examples on the [wiki](https://github.com/42wim/matterbridge/wiki/Api). | ||||
|  | ||||
| Used by at least 3 projects. Feel free to make a PR to add your project to this list. | ||||
|  | ||||
| @@ -72,9 +73,9 @@ Used by at least 3 projects. Feel free to make a PR to add your project to this | ||||
|  | ||||
| ## 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) | ||||
| @@ -92,9 +93,12 @@ See https://github.com/42wim/matterbridge/wiki | ||||
|  | ||||
| ## Installing | ||||
| ### Binaries | ||||
| * Latest stable release [v1.12.0](https://github.com/42wim/matterbridge/releases/latest) | ||||
| * Latest stable release [v1.12.2](https://github.com/42wim/matterbridge/releases/latest) | ||||
| * Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/) | ||||
|  | ||||
| ### Packages | ||||
| * [Overview](https://repology.org/metapackage/matterbridge/versions) | ||||
|  | ||||
| ### 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). | ||||
|  | ||||
| @@ -210,15 +214,17 @@ Want to tip ? | ||||
| * btc: 1N7cKHj5SfqBHBzDJ6kad4BzeqUBBS2zhs | ||||
|  | ||||
| ## Related projects | ||||
| * [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||
| * [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||
| * [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) | ||||
| * [matterlink](https://github.com/elytra/MatterLink) | ||||
| * [matterbridge config viewer](https://github.com/patcon/matterbridge-heroku-viewer) | ||||
| * [matterbridge-heroku](https://github.com/cadecairos/matterbridge-heroku) | ||||
| * [mattereddit](https://github.com/bonehurtingjuice/mattereddit) | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
| * [matterlink](https://github.com/elytra/MatterLink) | ||||
| * [mattermost-plugin](https://github.com/matterbridge/mattermost-plugin) - Run matterbridge as a plugin in mattermost | ||||
| * [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot) | ||||
|  | ||||
| ## Articles | ||||
| * [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/ | ||||
| @@ -237,7 +243,7 @@ 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 | ||||
| * slack - https://github.com/nlopes/slack | ||||
| * steam - https://github.com/Philipp15b/go-steam | ||||
|   | ||||
| @@ -1,10 +1,11 @@ | ||||
| package bridge | ||||
|  | ||||
| import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
|  | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"sync" | ||||
| ) | ||||
|  | ||||
| type Bridger interface { | ||||
| @@ -16,20 +17,22 @@ 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 | ||||
| 	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 | ||||
| 	*sync.RWMutex | ||||
| } | ||||
|  | ||||
| type Config struct { | ||||
| 	//	General *config.Protocol | ||||
| 	Remote chan config.Message | ||||
| 	Log    *log.Entry | ||||
| 	Log    *logrus.Entry | ||||
| 	*Bridge | ||||
| } | ||||
|  | ||||
| @@ -37,15 +40,17 @@ type Config struct { | ||||
| type Factory func(*Config) Bridger | ||||
|  | ||||
| func New(bridge *config.Bridge) *Bridge { | ||||
| 	b := new(Bridge) | ||||
| 	b.Channels = make(map[string]config.ChannelInfo) | ||||
| 	b := &Bridge{ | ||||
| 		Channels: make(map[string]config.ChannelInfo), | ||||
| 		RWMutex:  new(sync.RWMutex), | ||||
| 		Joined:   make(map[string]bool), | ||||
| 	} | ||||
| 	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 | ||||
| } | ||||
|  | ||||
| @@ -54,6 +59,13 @@ func (b *Bridge) JoinChannels() error { | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| // 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 { | ||||
| 	for ID, channel := range channels { | ||||
| 		if !exists[ID] { | ||||
|   | ||||
| @@ -9,21 +9,22 @@ import ( | ||||
|  | ||||
| 	"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 ( | ||||
| 	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" | ||||
| 	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 { | ||||
| @@ -61,6 +62,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 | ||||
| @@ -72,6 +83,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 | ||||
| @@ -108,6 +120,7 @@ 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 | ||||
| @@ -115,6 +128,7 @@ type Protocol struct { | ||||
| 	ShowEmbeds             bool       // discord | ||||
| 	SkipTLSVerify          bool       // IRC, mattermost | ||||
| 	StripNick              bool       // all protocols | ||||
| 	SyncTopic              bool       // slack | ||||
| 	Team                   string     // mattermost | ||||
| 	Token                  string     // gitter, slack, discord, api | ||||
| 	Topic                  string     // zulip | ||||
| @@ -127,7 +141,6 @@ type Protocol struct { | ||||
| 	UseInsecureURL         bool       // telegram | ||||
| 	WebhookBindAddress     string     // mattermost, slack | ||||
| 	WebhookURL             string     // mattermost, slack | ||||
| 	WebhookUse             string     // mattermost, slack, discord | ||||
| } | ||||
|  | ||||
| type ChannelOptions struct { | ||||
| @@ -194,12 +207,12 @@ type config struct { | ||||
| } | ||||
|  | ||||
| func NewConfig(cfgfile string) Config { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||
| 	flog := log.WithFields(log.Fields{"prefix": "config"}) | ||||
| 	logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false}) | ||||
| 	flog := logrus.WithFields(logrus.Fields{"prefix": "config"}) | ||||
| 	viper.SetConfigFile(cfgfile) | ||||
| 	input, err := getFileContents(cfgfile) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 		logrus.Fatal(err) | ||||
| 	} | ||||
| 	mycfg := newConfigFromString(input) | ||||
| 	if mycfg.cv.General.MediaDownloadSize == 0 { | ||||
| @@ -215,7 +228,7 @@ func NewConfig(cfgfile string) Config { | ||||
| func getFileContents(filename string) ([]byte, error) { | ||||
| 	input, err := ioutil.ReadFile(filename) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 		logrus.Fatal(err) | ||||
| 		return []byte(nil), err | ||||
| 	} | ||||
| 	return input, nil | ||||
| @@ -233,13 +246,13 @@ func newConfigFromString(input []byte) *config { | ||||
| 	viper.AutomaticEnv() | ||||
| 	err := viper.ReadConfig(bytes.NewBuffer(input)) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 		logrus.Fatal(err) | ||||
| 	} | ||||
|  | ||||
| 	cfg := &BridgeValues{} | ||||
| 	err = viper.Unmarshal(cfg) | ||||
| 	if err != nil { | ||||
| 		log.Fatal(err) | ||||
| 		logrus.Fatal(err) | ||||
| 	} | ||||
| 	return &config{ | ||||
| 		v:  viper.GetViper(), | ||||
|   | ||||
| @@ -4,7 +4,6 @@ import ( | ||||
| 	"bytes" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
|  | ||||
| @@ -17,18 +16,24 @@ import ( | ||||
| const MessageLength = 1950 | ||||
|  | ||||
| type Bdiscord struct { | ||||
| 	c              *discordgo.Session | ||||
| 	Channels       []*discordgo.Channel | ||||
| 	Nick           string | ||||
| 	UseChannelID   bool | ||||
| 	userMemberMap  map[string]*discordgo.Member | ||||
| 	nickMemberMap  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 { | ||||
| @@ -45,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") | ||||
| @@ -55,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 | ||||
| @@ -76,28 +87,71 @@ 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) | ||||
| 		} | ||||
| 	} | ||||
| 	// obtaining guild members and initializing nickname mapping | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
|  | ||||
| 	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 guild members", err) | ||||
| 		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 != "" { | ||||
| @@ -112,10 +166,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 | ||||
| } | ||||
| @@ -133,16 +190,20 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 		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 != "" { | ||||
| @@ -153,8 +214,14 @@ func (b *Bdiscord) Send(msg config.Message) (string, error) { | ||||
| 		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 | ||||
| @@ -164,6 +231,23 @@ 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, | ||||
| @@ -191,7 +275,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 { | ||||
| @@ -216,246 +302,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.EventMsgDelete, Text: config.EventMsgDelete} | ||||
| 	rmsg.Channel = b.getChannelName(m.ChannelID) | ||||
| 	if b.UseChannelID { | ||||
| 		rmsg.Channel = "ID:" + m.ChannelID | ||||
| 	} | ||||
| 	b.Log.Debugf("<= Sending message from %s to gateway", b.Account) | ||||
| 	b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 	b.Remote <- rmsg | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdate) { | ||||
| 	if b.GetBool("EditDisable") { | ||||
| 		return | ||||
| 	} | ||||
| 	// 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) { | ||||
| 	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.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) { | ||||
| 	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.nickMemberMap[m.Member.User.Username] = m.Member | ||||
| 	if m.Member.Nick != "" { | ||||
| 		b.nickMemberMap[m.Member.Nick] = 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) getGuildMemberByNick(nick string) (*discordgo.Member, error) { | ||||
| 	b.Lock() | ||||
| 	defer b.Unlock() | ||||
| 	if _, ok := b.nickMemberMap[nick]; ok { | ||||
| 		if b.nickMemberMap[nick] != nil { | ||||
| 			return b.nickMemberMap[nick], nil | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return nil, errors.New("Couldn't find guild member with nick " + nick) // This will most likely get ignored by the caller | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) getChannelID(name string) string { | ||||
| 	idcheck := strings.Split(name, "ID:") | ||||
| 	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) replaceUserMentions(text string) string { | ||||
| 	re := regexp.MustCompile("@[^@]{1,32}") | ||||
| 	text = re.ReplaceAllStringFunc(text, func(m string) string { | ||||
| 		mention := strings.TrimSpace(m[1:]) | ||||
| 		var member *discordgo.Member | ||||
| 		var err error | ||||
| 		for { | ||||
| 			b.Log.Debugf("Testing mention: '%s'", mention) | ||||
| 			member, err = b.getGuildMemberByNick(mention) | ||||
| 			if err != nil { | ||||
| 				lastSpace := strings.LastIndex(mention, " ") | ||||
| 				if lastSpace == -1 { | ||||
| 					break | ||||
| 				} | ||||
| 				mention = strings.TrimSpace(mention[0:lastSpace]) | ||||
| 			} else { | ||||
| 				break | ||||
| 			} | ||||
| 		} | ||||
| 		if err != nil { | ||||
| 			return m | ||||
| 		} | ||||
| 		return strings.Replace(m, "@"+mention, member.User.Mention(), -1) | ||||
| 	}) | ||||
| 	b.Log.Debugf("Message with mention replaced: %s", text) | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func (b *Bdiscord) replaceAction(text string) (string, bool) { | ||||
| 	if strings.HasPrefix(text, "_") && strings.HasSuffix(text, "_") { | ||||
| 		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 | ||||
| @@ -472,6 +327,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) | ||||
|   | ||||
							
								
								
									
										125
									
								
								bridge/discord/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										125
									
								
								bridge/discord/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,125 @@ | ||||
| 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 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 | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										189
									
								
								bridge/discord/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										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
									
								
							
							
						
						
									
										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) | ||||
| 	} | ||||
| } | ||||
| @@ -11,7 +11,8 @@ import ( | ||||
| 	"unicode/utf8" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"gitlab.com/golang-commonmark/markdown" | ||||
| ) | ||||
|  | ||||
| func DownloadFile(url string) (*[]byte, error) { | ||||
| @@ -97,7 +98,7 @@ 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 { | ||||
| func HandleDownloadSize(flog *logrus.Entry, msg *config.Message, name string, size int64, general *config.Protocol) error { | ||||
| 	// check blacklist here | ||||
| 	for _, entry := range general.MediaDownloadBlackList { | ||||
| 		if entry != "" { | ||||
| @@ -120,7 +121,7 @@ func HandleDownloadSize(flog *log.Entry, msg *config.Message, name string, size | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func HandleDownloadData(flog *log.Entry, msg *config.Message, name, comment, url string, data *[]byte, general *config.Protocol) { | ||||
| func HandleDownloadData(flog *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.EventAvatarDownload { | ||||
| @@ -151,3 +152,8 @@ func ClipMessage(text string, length int) string { | ||||
| 	} | ||||
| 	return text | ||||
| } | ||||
|  | ||||
| func ParseMarkdown(input string) string { | ||||
| 	md := markdown.New(markdown.XHTMLOutput(true), markdown.Breaks(true)) | ||||
| 	return (md.RenderToString([]byte(input))) | ||||
| } | ||||
|   | ||||
							
								
								
									
										235
									
								
								bridge/irc/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										235
									
								
								bridge/irc/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,235 @@ | ||||
| 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.Trailing, "Ping timeout") { | ||||
| 			b.Log.Infof("%s reconnecting ..", b.Account) | ||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EventFailure} | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	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.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.Trailing, 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,14 +1,10 @@ | ||||
| package birc | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/tls" | ||||
| 	"fmt" | ||||
| 	"hash/crc32" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"net" | ||||
| 	"regexp" | ||||
| 	"sort" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| @@ -17,10 +13,7 @@ import ( | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/dfordsoft/golib/ic" | ||||
| 	"github.com/lrstanley/girc" | ||||
| 	"github.com/paulrosania/go-charset/charset" | ||||
| 	"github.com/saintfish/chardet" | ||||
|  | ||||
| 	// We need to import the 'data' package as an implicit dependency. | ||||
| 	// See: https://godoc.org/github.com/paulrosania/go-charset/charset | ||||
| @@ -31,9 +24,9 @@ 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 | ||||
| @@ -44,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 { | ||||
| @@ -68,7 +61,7 @@ func (b *Birc) Command(msg *config.Message) string { | ||||
| 	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 "" | ||||
| } | ||||
| @@ -76,35 +69,11 @@ 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")) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	port, err := strconv.Atoi(portstr) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	// fix strict user handling of girc | ||||
| 	user := b.GetString("Nick") | ||||
| 	for !girc.IsValidUser(user) { | ||||
| 		if len(user) == 1 { | ||||
| 			user = "matterbridge" | ||||
| 			break | ||||
| 		} | ||||
| 		user = user[1:] | ||||
| 	} | ||||
|  | ||||
| 	i := girc.New(girc.Config{ | ||||
| 		Server:     server, | ||||
| 		ServerPass: b.GetString("Password"), | ||||
| 		Port:       port, | ||||
| 		Nick:       b.GetString("Nick"), | ||||
| 		User:       user, | ||||
| 		Name:       b.GetString("Nick"), | ||||
| 		SSL:        b.GetBool("UseTLS"), | ||||
| 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, | ||||
| 		PingDelay:  time.Minute, | ||||
| 	}) | ||||
| 	i, err := b.getClient() | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("UseSASL") { | ||||
| 		i.Config.SASL = &girc.SASLPlain{ | ||||
| @@ -115,33 +84,18 @@ func (b *Birc) Connect() error { | ||||
|  | ||||
| 	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) | ||||
| 	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.EventRejoinChannels} | ||||
| 				// 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") | ||||
|  | ||||
| 	go b.doConnect() | ||||
|  | ||||
| 	err = <-b.connected | ||||
| 	if err != nil { | ||||
| 		return fmt.Errorf("connection failed %s", err) | ||||
| 	} | ||||
| 	//i.Debug = false | ||||
| 	b.Log.Info("Connection succeeded") | ||||
| 	b.FirstConnection = false | ||||
| 	if b.GetInt("DebugLevel") == 0 { | ||||
| 		i.Handlers.Clear(girc.ALL_EVENTS) | ||||
| 	} | ||||
| @@ -156,6 +110,13 @@ func (b *Birc) Disconnect() error { | ||||
| } | ||||
|  | ||||
| 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) | ||||
| @@ -184,44 +145,13 @@ func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// 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() | ||||
| 		} | ||||
| 	if err := b.handleCharset(&msg); err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// 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 | ||||
| 		} | ||||
| 	// handle files, return if we're done here | ||||
| 	if ok := b.handleFiles(&msg); ok { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	var msgLines []string | ||||
| @@ -246,6 +176,28 @@ func (b *Birc) Send(msg config.Message) (string, error) { | ||||
| 	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) | ||||
| @@ -266,6 +218,40 @@ func (b *Birc) doSend() { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // 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 nil, err | ||||
| 	} | ||||
| 	// fix strict user handling of girc | ||||
| 	user := b.GetString("Nick") | ||||
| 	for !girc.IsValidUser(user) { | ||||
| 		if len(user) == 1 { | ||||
| 			user = "matterbridge" | ||||
| 			break | ||||
| 		} | ||||
| 		user = user[1:] | ||||
| 	} | ||||
|  | ||||
| 	i := girc.New(girc.Config{ | ||||
| 		Server:     server, | ||||
| 		ServerPass: b.GetString("Password"), | ||||
| 		Port:       port, | ||||
| 		Nick:       b.GetString("Nick"), | ||||
| 		User:       user, | ||||
| 		Name:       b.GetString("Nick"), | ||||
| 		SSL:        b.GetBool("UseTLS"), | ||||
| 		TLSConfig:  &tls.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"), ServerName: server}, //nolint:gosec | ||||
| 		PingDelay:  time.Minute, | ||||
| 	}) | ||||
| 	return i, nil | ||||
| } | ||||
|  | ||||
| func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||
| 	channel := event.Params[1] | ||||
| 	sort.Strings(b.names[channel]) | ||||
| @@ -282,87 +268,6 @@ func (b *Birc) endNames(client *girc.Client, event girc.Event) { | ||||
| 	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.EventRejoinChannels} | ||||
| 		return | ||||
| 	} | ||||
| 	if event.Command == "QUIT" { | ||||
| 		if event.Source.Name == b.Nick && strings.Contains(event.Trailing, "Ping timeout") { | ||||
| 			b.Log.Infof("%s reconnecting ..", b.Account) | ||||
| 			b.Remote <- config.Message{Username: "system", Text: "reconnect", Channel: channel, Account: b.Account, Event: config.EventFailure} | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
| 	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.EventJoinLeave} | ||||
| 		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.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 | ||||
| 	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 | ||||
| } | ||||
|   | ||||
| @@ -3,6 +3,7 @@ package bmatrix | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"html" | ||||
| 	"mime" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| @@ -99,19 +100,21 @@ 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) | ||||
| 	// Post normal message with HTML support (eg riot.im) | ||||
| 	resp, err := b.mc.SendHTML(channel, msg.Username+msg.Text, html.EscapeString(msg.Username)+helper.ParseMarkdown(msg.Text)) | ||||
| 	if err != nil { | ||||
| 		return "", err | ||||
| 	} | ||||
| @@ -257,47 +260,54 @@ 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) | ||||
| 		} | ||||
| 	} | ||||
| 	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
									
								
							
							
						
						
									
										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
									
								
							
							
						
						
									
										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,14 +3,12 @@ package bmattermost | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/42wim/matterbridge/matterclient" | ||||
| 	"github.com/42wim/matterbridge/matterhook" | ||||
| 	"github.com/mattermost/platform/model" | ||||
| 	"github.com/rs/xid" | ||||
| ) | ||||
|  | ||||
| @@ -40,54 +38,18 @@ func (b *Bmattermost) Connect() error { | ||||
| 		return nil | ||||
| 	} | ||||
| 	if b.GetString("WebhookBindAddress") != "" { | ||||
| 		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")}) | ||||
| 		if err := b.doConnectWebhookBind(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 		return nil | ||||
| 	} | ||||
| 	switch { | ||||
| 	case 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() | ||||
| 		if err := b.doConnectWebhookURL(); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		go b.handleMatter() | ||||
| 		return nil | ||||
| 	case b.GetString("Token") != "": | ||||
| 		b.Log.Info("Connecting using token (sending and receiving)") | ||||
| @@ -104,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 | ||||
| @@ -161,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, b.TeamID), 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) | ||||
| @@ -179,304 +144,5 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Post normal message | ||||
| 	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text) | ||||
| } | ||||
|  | ||||
| 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, 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 == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") { | ||||
| 			rmsg.Text = message.Text + b.GetString("EditSuffix") | ||||
| 		} | ||||
|  | ||||
| 		if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED { | ||||
| 			rmsg.Event = config.EventMsgDelete | ||||
| 		} | ||||
|  | ||||
| 		if len(message.Post.FileIds) > 0 { | ||||
| 			for _, id := range message.Post.FileIds { | ||||
| 				err := b.handleDownloadFile(rmsg, id) | ||||
| 				if err != nil { | ||||
| 					b.Log.Errorf("download failed: %s", err) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		// Use nickname instead of username if defined | ||||
| 		if nick := b.mc.GetNickName(rmsg.UserID); nick != "" { | ||||
| 			rmsg.Username = nick | ||||
| 		} | ||||
|  | ||||
| 		messages <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bmattermost) handleMatterHook(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 = "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 | ||||
| } | ||||
|  | ||||
| // handleDownloadAvatar downloads the avatar of userid from channel | ||||
| // sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful. | ||||
| // logs an error message if it fails | ||||
| func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) { | ||||
| 	rmsg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: userid, Event: config.EventAvatarDownload, Extra: make(map[string][]interface{})} | ||||
| 	if _, ok := b.avatarMap[userid]; !ok { | ||||
| 		data, resp := b.mc.Client.GetProfileImage(userid, "") | ||||
| 		if resp.Error != nil { | ||||
| 			b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error) | ||||
| 			return | ||||
| 		} | ||||
| 		err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) | ||||
| 		if err != nil { | ||||
| 			b.Log.Error(err) | ||||
| 			return | ||||
| 		} | ||||
| 		helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General) | ||||
| 		b.Remote <- rmsg | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error { | ||||
| 	url, _ := b.mc.Client.GetFileLink(id) | ||||
| 	finfo, resp := b.mc.Client.GetFileInfo(id) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, resp := b.mc.Client.DownloadFile(id, true) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) { | ||||
| 	var err error | ||||
| 	var res, id string | ||||
| 	channelID := b.mc.GetChannelId(msg.Channel, b.TeamID) | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		msg.Text = fi.Comment | ||||
| 		if b.GetBool("PrefixMessagesWithNick") { | ||||
| 			msg.Text = msg.Username + msg.Text | ||||
| 		} | ||||
| 		res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, []string{id}) | ||||
| 	} | ||||
| 	return res, err | ||||
| } | ||||
|  | ||||
| // sendWebhook uses the configured WebhookURL to send the message | ||||
| func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) { | ||||
| 	// skip events | ||||
| 	if msg.Event != "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("PrefixMessagesWithNick") { | ||||
| 		msg.Text = msg.Username + msg.Text | ||||
| 	} | ||||
| 	if msg.Extra != nil { | ||||
| 		// this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			rmsg := rmsg // scopelint | ||||
| 			iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl")) | ||||
| 			matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text, Props: make(map[string]interface{})} | ||||
| 			matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 			b.mh.Send(matterMessage) | ||||
| 		} | ||||
|  | ||||
| 		// webhook doesn't support file uploads, so we add the url manually | ||||
| 		if len(msg.Extra["file"]) > 0 { | ||||
| 			for _, f := range msg.Extra["file"] { | ||||
| 				fi := f.(config.FileInfo) | ||||
| 				if fi.URL != "" { | ||||
| 					msg.Text += fi.URL | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	iconURL := config.GetIconURL(&msg, b.GetString("iconurl")) | ||||
| 	matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: msg.Channel, UserName: msg.Username, Text: msg.Text, Props: make(map[string]interface{})} | ||||
| 	if msg.Avatar != "" { | ||||
| 		matterMessage.IconURL = msg.Avatar | ||||
| 	} | ||||
| 	matterMessage.Props["matterbridge_"+b.uuid] = true | ||||
| 	err := b.mh.Send(matterMessage) | ||||
| 	if err != nil { | ||||
| 		b.Log.Info(err) | ||||
| 		return "", err | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| // skipMessages returns true if this message should not be handled | ||||
| func (b *Bmattermost) skipMessage(message *matterclient.Message) bool { | ||||
| 	// Handle join/leave | ||||
| 	if message.Type == "system_join_leave" || | ||||
| 		message.Type == "system_join_channel" || | ||||
| 		message.Type == "system_leave_channel" { | ||||
| 		if b.GetBool("nosendjoinpart") { | ||||
| 			return true | ||||
| 		} | ||||
| 		b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) | ||||
| 		b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EventJoinLeave} | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Handle edited messages | ||||
| 	if (message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED) && b.GetBool("EditDisable") { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from matterbridge | ||||
| 	if message.Post.Props != nil { | ||||
| 		if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok { | ||||
| 			b.Log.Debugf("sent by matterbridge, ignoring") | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Ignore messages sent from a user logged in as the bot | ||||
| 	if b.mc.User.Username == message.Username { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// if the message has reactions don't repost it (for now, until we can correlate reaction with message) | ||||
| 	if message.Post.HasReactions { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// ignore messages from other teams than ours | ||||
| 	if message.Raw.Data["team_id"].(string) != b.TeamID { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// only handle posted, edited or deleted events | ||||
| 	if !(message.Raw.Event == "posted" || message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED || message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED) { | ||||
| 		return true | ||||
| 	} | ||||
| 	return false | ||||
| 	return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text, msg.ParentID) | ||||
| } | ||||
|   | ||||
| @@ -75,21 +75,24 @@ func (b *Bslack) handleSlackClient(messages chan *config.Message) { | ||||
| 			// When we join a channel we update the full list of users as | ||||
| 			// well as the information for the channel that we joined as this | ||||
| 			// should now tell that we are a member of it. | ||||
| 			b.populateUsers() | ||||
|  | ||||
| 			b.channelsMutex.Lock() | ||||
| 			b.channelsByID[ev.Channel.ID] = &ev.Channel | ||||
| 			b.channelsByName[ev.Channel.Name] = &ev.Channel | ||||
| 			b.channelsMutex.Unlock() | ||||
| 		case *slack.ConnectedEvent: | ||||
| 			b.si = ev.Info | ||||
| 			b.populateChannels() | ||||
| 			b.populateUsers() | ||||
| 			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) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -116,6 +119,11 @@ func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool { | ||||
| 		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). | ||||
| @@ -136,7 +144,6 @@ func (b *Bslack) skipMessageEvent(ev *slack.MessageEvent) bool { | ||||
| 	if len(ev.Files) > 0 { | ||||
| 		return b.filesCached(ev.Files) | ||||
| 	} | ||||
|  | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| @@ -193,7 +200,6 @@ func (b *Bslack) handleMessageEvent(ev *slack.MessageEvent) (*config.Message, er | ||||
| func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) bool { | ||||
| 	switch ev.SubType { | ||||
| 	case sChannelJoined, sMemberJoined: | ||||
| 		b.populateUsers() | ||||
| 		// There's no further processing needed on channel events | ||||
| 		// so we return 'true'. | ||||
| 		return true | ||||
| @@ -201,7 +207,15 @@ func (b *Bslack) handleStatusEvent(ev *slack.MessageEvent, rmsg *config.Message) | ||||
| 		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 | ||||
| @@ -242,7 +256,7 @@ func (b *Bslack) handleAttachments(ev *slack.MessageEvent, rmsg *config.Message) | ||||
|  | ||||
| 	// If we have files attached, download them (in memory) and put a pointer to it in msg.Extra. | ||||
| 	for i := range ev.Files { | ||||
| 		if err := b.handleDownloadFile(rmsg, &ev.Files[i]); err != nil { | ||||
| 		if err := b.handleDownloadFile(rmsg, &ev.Files[i], false); err != nil { | ||||
| 			b.Log.Errorf("Could not download incoming file: %#v", err) | ||||
| 		} | ||||
| 	} | ||||
| @@ -261,7 +275,7 @@ func (b *Bslack) handleTypingEvent(ev *slack.UserTypingEvent) (*config.Message, | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) error { | ||||
| func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File, retry bool) error { | ||||
| 	if b.fileCached(file) { | ||||
| 		return nil | ||||
| 	} | ||||
| @@ -277,6 +291,12 @@ func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) erro | ||||
| 		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. | ||||
| @@ -286,6 +306,58 @@ func (b *Bslack) handleDownloadFile(rmsg *config.Message, file *slack.File) erro | ||||
| 	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. | ||||
| // | ||||
|   | ||||
| @@ -12,6 +12,13 @@ import ( | ||||
| ) | ||||
|  | ||||
| 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() | ||||
|  | ||||
| @@ -44,44 +51,69 @@ func (b *Bslack) getChannel(channel string) (*slack.Channel, error) { | ||||
| } | ||||
|  | ||||
| func (b *Bslack) getChannelByName(name string) (*slack.Channel, error) { | ||||
| 	b.channelsMutex.RLock() | ||||
| 	defer b.channelsMutex.RUnlock() | ||||
|  | ||||
| 	if channel, ok := b.channelsByName[name]; ok { | ||||
| 		return channel, nil | ||||
| 	} | ||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, name) | ||||
| 	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 := b.channelsByID[ID]; ok { | ||||
| 	if channel, ok := lookupMap[lookupKey]; ok { | ||||
| 		return channel, nil | ||||
| 	} | ||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, ID) | ||||
| 	return nil, fmt.Errorf("%s: channel %s not found", b.Account, lookupKey) | ||||
| } | ||||
|  | ||||
| const minimumRefreshInterval = 10 * time.Second | ||||
|  | ||||
| func (b *Bslack) populateUsers() { | ||||
| 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 time.Now().Before(b.earliestUserRefresh) || b.refreshInProgress { | ||||
| 	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 | ||||
| @@ -97,6 +129,13 @@ func (b *Bslack) populateUsers() { | ||||
| 		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() | ||||
| @@ -109,19 +148,25 @@ func (b *Bslack) populateUsers() { | ||||
| 	b.refreshInProgress = false | ||||
| } | ||||
|  | ||||
| func (b *Bslack) populateChannels() { | ||||
| func (b *Bslack) populateChannels(wait bool) { | ||||
| 	b.refreshMutex.Lock() | ||||
| 	if time.Now().Before(b.earliestChannelRefresh) || b.refreshInProgress { | ||||
| 	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. | ||||
| @@ -142,7 +187,21 @@ func (b *Bslack) populateChannels() { | ||||
| 		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 | ||||
| 		} | ||||
| @@ -154,6 +213,10 @@ func (b *Bslack) populateChannels() { | ||||
| 	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) | ||||
| @@ -250,8 +313,9 @@ func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| 	b.Log.Debugf("Found bot %#v", bot) | ||||
|  | ||||
| 	if bot.Name != "" && bot.Name != "Slack API Tester" { | ||||
| 	if bot.Name != "" { | ||||
| 		rmsg.Username = bot.Name | ||||
| 		if ev.Username != "" { | ||||
| 			rmsg.Username = ev.Username | ||||
| @@ -262,12 +326,29 @@ func (b *Bslack) populateMessageWithBotInfo(ev *slack.MessageEvent, rmsg *config | ||||
| } | ||||
|  | ||||
| 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(`<(.*?)(\|.*?)?>`) | ||||
| 	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 { | ||||
| @@ -312,6 +393,10 @@ func (b *Bslack) replaceURL(text string) string { | ||||
| 	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 { | ||||
| @@ -321,3 +406,29 @@ func (b *Bslack) handleRateLimit(err error) error { | ||||
| 	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
									
								
							
							
						
						
									
										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{Log: logger.WithFields(nil)} | ||||
| 	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) | ||||
| 	} | ||||
| } | ||||
| @@ -37,6 +37,9 @@ type Bslack struct { | ||||
| 	channelsByName map[string]*slack.Channel | ||||
| 	channelsMutex  sync.RWMutex | ||||
|  | ||||
| 	channelMembers      map[string][]string | ||||
| 	channelMembersMutex sync.RWMutex | ||||
|  | ||||
| 	refreshInProgress      bool | ||||
| 	earliestChannelRefresh time.Time | ||||
| 	earliestUserRefresh    time.Time | ||||
| @@ -48,6 +51,7 @@ const ( | ||||
| 	sChannelLeave    = "channel_leave" | ||||
| 	sChannelJoined   = "channel_joined" | ||||
| 	sMemberJoined    = "member_joined_channel" | ||||
| 	sMessageChanged  = "message_changed" | ||||
| 	sMessageDeleted  = "message_deleted" | ||||
| 	sSlackAttachment = "slack_attachment" | ||||
| 	sPinnedItem      = "pinned_item" | ||||
| @@ -76,14 +80,9 @@ func New(cfg *bridge.Config) bridge.Bridger { | ||||
| 	// Print a deprecation warning for legacy non-bot tokens (#527). | ||||
| 	token := cfg.GetString(tokenConfig) | ||||
| 	if token != "" && !strings.HasPrefix(token, "xoxb") { | ||||
| 		cfg.Log.Error("Non-bot token detected. It is STRONGLY recommended to use a proper bot-token instead.") | ||||
| 		cfg.Log.Error("Legacy tokens may be deprecated by Slack at short notice. See the Matterbridge GitHub wiki for a migration guide.") | ||||
| 		cfg.Log.Error("See https://github.com/42wim/matterbridge/wiki/Slack-bot-setup") | ||||
| 		cfg.Log.Error("") | ||||
| 		cfg.Log.Error("To continue using a legacy token please move your configuration to a \"slack-legacy\" bridge instead.") | ||||
| 		cfg.Log.Error("See https://github.com/42wim/matterbridge/wiki/Section-Slack-(basic)#legacy-configuration)") | ||||
| 		cfg.Log.Error("Delaying start of bridge by 30 seconds. Future Matterbridge release will fail here unless you use a \"slack-legacy\" bridge.") | ||||
| 		time.Sleep(30 * time.Second) | ||||
| 		cfg.Log.Warn("Non-bot token detected. It is STRONGLY recommended to use a proper bot-token instead.") | ||||
| 		cfg.Log.Warn("Legacy tokens may be deprecated by Slack at short notice. See the Matterbridge GitHub wiki for a migration guide.") | ||||
| 		cfg.Log.Warn("See https://github.com/42wim/matterbridge/wiki/Slack-bot-setup") | ||||
| 		return NewLegacy(cfg) | ||||
| 	} | ||||
| 	return newBridge(cfg) | ||||
| @@ -164,7 +163,7 @@ func (b *Bslack) JoinChannel(channel config.ChannelInfo) error { | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	b.populateChannels() | ||||
| 	b.populateChannels(false) | ||||
|  | ||||
| 	channelInfo, err := b.getChannel(channel.Name) | ||||
| 	if err != nil { | ||||
| @@ -192,6 +191,8 @@ func (b *Bslack) Send(msg config.Message) (string, error) { | ||||
| 		b.Log.Debugf("=> Receiving %#v", msg) | ||||
| 	} | ||||
|  | ||||
| 	msg.Text = b.replaceCodeFence(msg.Text) | ||||
|  | ||||
| 	// Make a action /me of the message | ||||
| 	if msg.Event == config.EventUserAction { | ||||
| 		msg.Text = "_" + msg.Text + "_" | ||||
| @@ -199,16 +200,16 @@ func (b *Bslack) Send(msg config.Message) (string, error) { | ||||
|  | ||||
| 	// Use webhook to send the message | ||||
| 	if b.GetString(outgoingWebhookConfig) != "" { | ||||
| 		return b.sendWebhook(msg) | ||||
| 		return "", b.sendWebhook(msg) | ||||
| 	} | ||||
| 	return b.sendRTM(msg) | ||||
| } | ||||
|  | ||||
| // sendWebhook uses the configured WebhookURL to send the message | ||||
| func (b *Bslack) sendWebhook(msg config.Message) (string, error) { | ||||
| func (b *Bslack) sendWebhook(msg config.Message) error { | ||||
| 	// Skip events. | ||||
| 	if msg.Event != "" { | ||||
| 		return "", nil | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool(useNickPrefixConfig) { | ||||
| @@ -263,12 +264,17 @@ func (b *Bslack) sendWebhook(msg config.Message) (string, error) { | ||||
| 	} | ||||
| 	if err := b.mh.Send(matterMessage); err != nil { | ||||
| 		b.Log.Errorf("Failed to send message via webhook: %#v", err) | ||||
| 		return "", err | ||||
| 		return err | ||||
| 	} | ||||
| 	return "", nil | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 	// Handle channelmember messages. | ||||
| 	if handled := b.handleGetChannelMembers(&msg); handled { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	channelInfo, err := b.getChannel(msg.Channel) | ||||
| 	if err != nil { | ||||
| 		return "", fmt.Errorf("could not send message: %v", err) | ||||
| @@ -280,8 +286,14 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 		return "", nil | ||||
| 	} | ||||
|  | ||||
| 	// Handle message deletions. | ||||
| 	var handled bool | ||||
|  | ||||
| 	// Handle topic/purpose updates. | ||||
| 	if handled, err = b.handleTopicOrPurpose(&msg, channelInfo); handled { | ||||
| 		return "", err | ||||
| 	} | ||||
|  | ||||
| 	// Handle message deletions. | ||||
| 	if handled, err = b.deleteMessage(&msg, channelInfo); handled { | ||||
| 		return msg.ID, err | ||||
| 	} | ||||
| @@ -296,12 +308,13 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 		return msg.ID, err | ||||
| 	} | ||||
|  | ||||
| 	messageParameters := b.prepareMessageParameters(&msg) | ||||
|  | ||||
| 	// Upload a file if it exists. | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, rmsg := range helper.HandleExtra(&msg, b.General) { | ||||
| 			_, _, err = b.rtm.PostMessage(channelInfo.ID, rmsg.Username+rmsg.Text, *messageParameters) | ||||
| 		extraMsgs := helper.HandleExtra(&msg, b.General) | ||||
| 		for i := range extraMsgs { | ||||
| 			rmsg := &extraMsgs[i] | ||||
| 			rmsg.Text = rmsg.Username + rmsg.Text | ||||
| 			_, err = b.postMessage(rmsg, channelInfo) | ||||
| 			if err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 			} | ||||
| @@ -311,7 +324,50 @@ func (b *Bslack) sendRTM(msg config.Message) (string, error) { | ||||
| 	} | ||||
|  | ||||
| 	// Post message. | ||||
| 	return b.postMessage(&msg, messageParameters, channelInfo) | ||||
| 	return b.postMessage(&msg, channelInfo) | ||||
| } | ||||
|  | ||||
| func (b *Bslack) updateTopicOrPurpose(msg *config.Message, channelInfo *slack.Channel) error { | ||||
| 	var updateFunc func(channelID string, value string) (*slack.Channel, error) | ||||
|  | ||||
| 	incomingChangeType, text := b.extractTopicOrPurpose(msg.Text) | ||||
| 	switch incomingChangeType { | ||||
| 	case "topic": | ||||
| 		updateFunc = b.rtm.SetTopicOfConversation | ||||
| 	case "purpose": | ||||
| 		updateFunc = b.rtm.SetPurposeOfConversation | ||||
| 	default: | ||||
| 		b.Log.Errorf("Unhandled type received from extractTopicOrPurpose: %s", incomingChangeType) | ||||
| 		return nil | ||||
| 	} | ||||
| 	for { | ||||
| 		_, err := updateFunc(channelInfo.ID, text) | ||||
| 		if err == nil { | ||||
| 			return nil | ||||
| 		} | ||||
| 		if err = b.handleRateLimit(err); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handles updating topic/purpose and determining whether to further propagate update messages. | ||||
| func (b *Bslack) handleTopicOrPurpose(msg *config.Message, channelInfo *slack.Channel) (bool, error) { | ||||
| 	if msg.Event != config.EventTopicChange { | ||||
| 		return false, nil | ||||
| 	} | ||||
|  | ||||
| 	if b.GetBool("SyncTopic") { | ||||
| 		return true, b.updateTopicOrPurpose(msg, channelInfo) | ||||
| 	} | ||||
|  | ||||
| 	// Pass along to normal message handlers. | ||||
| 	if b.GetBool("ShowTopicChange") { | ||||
| 		return false, nil | ||||
| 	} | ||||
|  | ||||
| 	// Swallow message as handled no-op. | ||||
| 	return true, nil | ||||
| } | ||||
|  | ||||
| func (b *Bslack) deleteMessage(msg *config.Message, channelInfo *slack.Channel) (bool, error) { | ||||
| @@ -341,9 +397,10 @@ func (b *Bslack) editMessage(msg *config.Message, channelInfo *slack.Channel) (b | ||||
| 	if msg.ID == "" { | ||||
| 		return false, nil | ||||
| 	} | ||||
|  | ||||
| 	messageOptions := b.prepareMessageOptions(msg) | ||||
| 	for { | ||||
| 		_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, msg.Text) | ||||
| 		messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false)) | ||||
| 		_, _, _, err := b.rtm.UpdateMessage(channelInfo.ID, msg.ID, messageOptions...) | ||||
| 		if err == nil { | ||||
| 			return true, nil | ||||
| 		} | ||||
| @@ -355,9 +412,15 @@ func (b *Bslack) editMessage(msg *config.Message, channelInfo *slack.Channel) (b | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bslack) postMessage(msg *config.Message, messageParameters *slack.PostMessageParameters, channelInfo *slack.Channel) (string, error) { | ||||
| func (b *Bslack) postMessage(msg *config.Message, channelInfo *slack.Channel) (string, error) { | ||||
| 	// don't post empty messages | ||||
| 	if msg.Text == "" { | ||||
| 		return "", nil | ||||
| 	} | ||||
| 	messageOptions := b.prepareMessageOptions(msg) | ||||
| 	messageOptions = append(messageOptions, slack.MsgOptionText(msg.Text, false)) | ||||
| 	for { | ||||
| 		_, id, err := b.rtm.PostMessage(channelInfo.ID, msg.Text, *messageParameters) | ||||
| 		_, id, err := b.rtm.PostMessage(channelInfo.ID, messageOptions...) | ||||
| 		if err == nil { | ||||
| 			return id, nil | ||||
| 		} | ||||
| @@ -385,11 +448,16 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID string) { | ||||
| 		ts := time.Now() | ||||
| 		b.Log.Debugf("Adding file %s to cache at %s with timestamp", fi.Name, ts.String()) | ||||
| 		b.cache.Add("filename"+fi.Name, ts) | ||||
| 		initialComment := fmt.Sprintf("File from %s", msg.Username) | ||||
| 		if fi.Comment != "" { | ||||
| 			initialComment += fmt.Sprintf("with comment: %s", fi.Comment) | ||||
| 		} | ||||
| 		res, err := b.sc.UploadFile(slack.FileUploadParameters{ | ||||
| 			Reader:         bytes.NewReader(*fi.Data), | ||||
| 			Filename:       fi.Name, | ||||
| 			Channels:       []string{channelID}, | ||||
| 			InitialComment: fi.Comment, | ||||
| 			Reader:          bytes.NewReader(*fi.Data), | ||||
| 			Filename:        fi.Name, | ||||
| 			Channels:        []string{channelID}, | ||||
| 			InitialComment:  initialComment, | ||||
| 			ThreadTimestamp: msg.ParentID, | ||||
| 		}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("uploadfile %#v", err) | ||||
| @@ -402,7 +470,7 @@ func (b *Bslack) uploadFile(msg *config.Message, channelID string) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Bslack) prepareMessageParameters(msg *config.Message) *slack.PostMessageParameters { | ||||
| func (b *Bslack) prepareMessageOptions(msg *config.Message) []slack.MsgOption { | ||||
| 	params := slack.NewPostMessageParameters() | ||||
| 	if b.GetBool(useNickPrefixConfig) { | ||||
| 		params.AsUser = true | ||||
| @@ -414,17 +482,23 @@ func (b *Bslack) prepareMessageParameters(msg *config.Message) *slack.PostMessag | ||||
| 	if msg.Avatar != "" { | ||||
| 		params.IconURL = msg.Avatar | ||||
| 	} | ||||
|  | ||||
| 	var attachments []slack.Attachment | ||||
| 	// add a callback ID so we can see we created it | ||||
| 	params.Attachments = append(params.Attachments, slack.Attachment{CallbackID: "matterbridge_" + b.uuid}) | ||||
| 	attachments = append(attachments, slack.Attachment{CallbackID: "matterbridge_" + b.uuid}) | ||||
| 	// add file attachments | ||||
| 	params.Attachments = append(params.Attachments, b.createAttach(msg.Extra)...) | ||||
| 	attachments = append(attachments, b.createAttach(msg.Extra)...) | ||||
| 	// add slack attachments (from another slack bridge) | ||||
| 	if msg.Extra != nil { | ||||
| 		for _, attach := range msg.Extra[sSlackAttachment] { | ||||
| 			params.Attachments = append(params.Attachments, attach.([]slack.Attachment)...) | ||||
| 			attachments = append(attachments, attach.([]slack.Attachment)...) | ||||
| 		} | ||||
| 	} | ||||
| 	return ¶ms | ||||
|  | ||||
| 	var opts []slack.MsgOption | ||||
| 	opts = append(opts, slack.MsgOptionAttachments(attachments...)) | ||||
| 	opts = append(opts, slack.MsgOptionPostMessageParameters(params)) | ||||
| 	return opts | ||||
| } | ||||
|  | ||||
| func (b *Bslack) createAttach(extra map[string][]interface{}) []slack.Attachment { | ||||
|   | ||||
| @@ -9,7 +9,7 @@ import ( | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/bridge/helper" | ||||
| 	"github.com/shazow/ssh-chat/sshd" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Bsshchat struct { | ||||
| @@ -23,21 +23,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 | ||||
| @@ -59,27 +73,16 @@ func (b *Bsshchat) Send(msg config.Message) (string, error) { | ||||
| 	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 | ||||
| } | ||||
|  | ||||
| /* | ||||
| @@ -125,17 +128,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") | ||||
| 				logrus.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
									
								
							
							
						
						
									
										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,7 +2,6 @@ package bsteam | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| @@ -73,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) | ||||
| @@ -103,78 +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) | ||||
| 		default: | ||||
| 			b.Log.Debugf("unknown event %#v", e) | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										346
									
								
								bridge/telegram/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										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 | ||||
| } | ||||
| @@ -34,7 +34,7 @@ func (options *customHTML) Header(out *bytes.Buffer, text func() bool, level int | ||||
| } | ||||
|  | ||||
| func (options *customHTML) HRule(out io.ByteWriter) { | ||||
| 	out.WriteByte('\n') | ||||
| 	out.WriteByte('\n') //nolint:errcheck | ||||
| } | ||||
|  | ||||
| func (options *customHTML) BlockQuote(out *bytes.Buffer, text []byte) { | ||||
|   | ||||
| @@ -2,7 +2,6 @@ package btelegram | ||||
|  | ||||
| import ( | ||||
| 	"html" | ||||
| 	"regexp" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
|  | ||||
| @@ -76,21 +75,15 @@ func (b *Btelegram) Send(msg config.Message) (string, error) { | ||||
|  | ||||
| 	// Delete message | ||||
| 	if msg.Event == config.EventMsgDelete { | ||||
| 		if msg.ID == "" { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 		msgid, err := strconv.Atoi(msg.ID) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		_, err = b.c.DeleteMessage(tgbotapi.DeleteMessageConfig{ChatID: chatid, MessageID: msgid}) | ||||
| 		return "", err | ||||
| 		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 { | ||||
| @@ -100,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") == HTMLFormat { | ||||
| 			b.Log.Debug("Using mode HTML") | ||||
| 			m.ParseMode = tgbotapi.ModeHTML | ||||
| 		} | ||||
| 		if b.GetString("MessageFormat") == "Markdown" { | ||||
| 			b.Log.Debug("Using mode markdown") | ||||
| 			m.ParseMode = tgbotapi.ModeMarkdown | ||||
| 		} | ||||
| 		if strings.ToLower(b.GetString("MessageFormat")) == HTMLNick { | ||||
| 			b.Log.Debug("Using mode HTML - nick only") | ||||
| 			m.ParseMode = tgbotapi.ModeHTML | ||||
| 		} | ||||
| 		_, err = b.c.Send(m) | ||||
| 		if err != nil { | ||||
| 			return "", err | ||||
| 		} | ||||
| 		return "", nil | ||||
| 		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 = unknownUser | ||||
| 		} | ||||
|  | ||||
| 		// 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 = unknownUser | ||||
| 			} | ||||
| 			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 = unknownUser | ||||
| 			} | ||||
| 			if !b.GetBool("QuoteDisable") { | ||||
| 				rmsg.Text = b.handleQuote(rmsg.Text, usernameReply, message.ReplyToMessage.Text) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		if rmsg.Text != "" || len(rmsg.Extra) > 0 { | ||||
| 			rmsg.Text = helper.RemoveEmptyNewLines(rmsg.Text) | ||||
| 			// channels don't have (always?) user information. see #410 | ||||
| 			if message.From != nil { | ||||
| 				rmsg.Avatar = helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General) | ||||
| 			} | ||||
|  | ||||
| 			b.Log.Debugf("<= Sending message from %s on %s to gateway", rmsg.Username, b.Account) | ||||
| 			b.Log.Debugf("<= Message is %#v", rmsg) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) getFileDirectURL(id string) string { | ||||
| 	res, err := b.c.GetFileDirectURL(id) | ||||
| 	if err != nil { | ||||
| @@ -264,146 +108,6 @@ 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.EventAvatarDownload, Extra: make(map[string][]interface{})} | ||||
| 	if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok { | ||||
| 		photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1}) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("Userprofile download failed for %#v %s", userid, err) | ||||
| 		} | ||||
|  | ||||
| 		if len(photos.Photos) > 0 { | ||||
| 			photo := photos.Photos[0][0] | ||||
| 			url := b.getFileDirectURL(photo.FileID) | ||||
| 			name := strconv.Itoa(userid) + ".png" | ||||
| 			b.Log.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize) | ||||
|  | ||||
| 			err := helper.HandleDownloadSize(b.Log, &rmsg, name, int64(photo.FileSize), b.General) | ||||
| 			if err != nil { | ||||
| 				b.Log.Error(err) | ||||
| 				return | ||||
| 			} | ||||
| 			data, err := helper.DownloadFile(url) | ||||
| 			if err != nil { | ||||
| 				b.Log.Errorf("download %s failed %#v", url, err) | ||||
| 				return | ||||
| 			} | ||||
| 			helper.HandleDownloadData(b.Log, &rmsg, name, rmsg.Text, "", data, b.General) | ||||
| 			b.Remote <- rmsg | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleDownloadFile handles file download | ||||
| func (b *Btelegram) handleDownload(message *tgbotapi.Message, rmsg *config.Message) error { | ||||
| 	size := 0 | ||||
| 	var url, name, text string | ||||
|  | ||||
| 	if message.Sticker != nil { | ||||
| 		v := message.Sticker | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		if !strings.HasSuffix(name, ".webp") { | ||||
| 			name += ".webp" | ||||
| 		} | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Video != nil { | ||||
| 		v := message.Video | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Photo != nil { | ||||
| 		photos := *message.Photo | ||||
| 		size = photos[len(photos)-1].FileSize | ||||
| 		url = b.getFileDirectURL(photos[len(photos)-1].FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	if message.Document != nil { | ||||
| 		v := message.Document | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		name = v.FileName | ||||
| 		text = " " + v.FileName + " : " + url | ||||
| 	} | ||||
| 	if message.Voice != nil { | ||||
| 		v := message.Voice | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 		if !strings.HasSuffix(name, ".ogg") { | ||||
| 			name += ".ogg" | ||||
| 		} | ||||
| 	} | ||||
| 	if message.Audio != nil { | ||||
| 		v := message.Audio | ||||
| 		size = v.FileSize | ||||
| 		url = b.getFileDirectURL(v.FileID) | ||||
| 		urlPart := strings.Split(url, "/") | ||||
| 		name = urlPart[len(urlPart)-1] | ||||
| 		text = " " + url | ||||
| 	} | ||||
| 	// if name is empty we didn't match a thing to download | ||||
| 	if name == "" { | ||||
| 		return nil | ||||
| 	} | ||||
| 	// use the URL instead of native upload | ||||
| 	if b.GetBool("UseInsecureURL") { | ||||
| 		b.Log.Debugf("Setting message text to :%s", text) | ||||
| 		rmsg.Text += text | ||||
| 		return nil | ||||
| 	} | ||||
| 	// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra | ||||
| 	err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	data, err := helper.DownloadFile(url) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	helper.HandleDownloadData(b.Log, rmsg, name, message.Caption, "", data, b.General) | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // handleUploadFile handles native upload of files | ||||
| func (b *Btelegram) handleUploadFile(msg *config.Message, chatid int64) (string, error) { | ||||
| 	var c tgbotapi.Chattable | ||||
| 	for _, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		file := tgbotapi.FileBytes{ | ||||
| 			Name:  fi.Name, | ||||
| 			Bytes: *fi.Data, | ||||
| 		} | ||||
| 		re := regexp.MustCompile(".(jpg|png)$") | ||||
| 		if re.MatchString(fi.Name) { | ||||
| 			c = tgbotapi.NewPhotoUpload(chatid, file) | ||||
| 		} else { | ||||
| 			c = tgbotapi.NewDocumentUpload(chatid, file) | ||||
| 		} | ||||
| 		_, err := b.c.Send(c) | ||||
| 		if err != nil { | ||||
| 			b.Log.Errorf("file upload failed: %#v", err) | ||||
| 		} | ||||
| 		if fi.Comment != "" { | ||||
| 			b.sendMessage(chatid, msg.Username, fi.Comment) | ||||
| 		} | ||||
| 	} | ||||
| 	return "", nil | ||||
| } | ||||
|  | ||||
| func (b *Btelegram) sendMessage(chatid int64, username, text string) (string, error) { | ||||
| 	m := tgbotapi.NewMessage(chatid, "") | ||||
| 	m.Text = username + text | ||||
| @@ -437,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 | ||||
| } | ||||
|   | ||||
							
								
								
									
										51
									
								
								changelog.md
									
									
									
									
									
								
							
							
						
						
									
										51
									
								
								changelog.md
									
									
									
									
									
								
							| @@ -1,3 +1,54 @@ | ||||
| # 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 | ||||
|   | ||||
							
								
								
									
										210
									
								
								contrib/api.yaml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										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: [] | ||||
| @@ -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"] | ||||
|   | ||||
							
								
								
									
										35
									
								
								gateway/bridgemap/bridgemap.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										35
									
								
								gateway/bridgemap/bridgemap.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,35 @@ | ||||
| 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/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, | ||||
| 	"xmpp":         bxmpp.New, | ||||
| 	"zulip":        bzulip.New, | ||||
| } | ||||
| @@ -1,35 +1,16 @@ | ||||
| package gateway | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/sha1" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"regexp" | ||||
| 	"strings" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| 	"github.com/42wim/matterbridge/bridge/api" | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	bdiscord "github.com/42wim/matterbridge/bridge/discord" | ||||
| 	bgitter "github.com/42wim/matterbridge/bridge/gitter" | ||||
| 	birc "github.com/42wim/matterbridge/bridge/irc" | ||||
| 	bmatrix "github.com/42wim/matterbridge/bridge/matrix" | ||||
| 	bmattermost "github.com/42wim/matterbridge/bridge/mattermost" | ||||
| 	brocketchat "github.com/42wim/matterbridge/bridge/rocketchat" | ||||
| 	bslack "github.com/42wim/matterbridge/bridge/slack" | ||||
| 	bsshchat "github.com/42wim/matterbridge/bridge/sshchat" | ||||
| 	bsteam "github.com/42wim/matterbridge/bridge/steam" | ||||
| 	btelegram "github.com/42wim/matterbridge/bridge/telegram" | ||||
| 	bxmpp "github.com/42wim/matterbridge/bridge/xmpp" | ||||
| 	bzulip "github.com/42wim/matterbridge/bridge/zulip" | ||||
| 	"github.com/hashicorp/golang-lru" | ||||
| 	"github.com/peterhellberg/emojilib" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Gateway struct { | ||||
| @@ -51,36 +32,21 @@ type BrMsgID struct { | ||||
| 	ChannelID string | ||||
| } | ||||
|  | ||||
| var flog *log.Entry | ||||
|  | ||||
| var bridgeMap = map[string]bridge.Factory{ | ||||
| 	"api":          api.New, | ||||
| 	"discord":      bdiscord.New, | ||||
| 	"gitter":       bgitter.New, | ||||
| 	"irc":          birc.New, | ||||
| 	"mattermost":   bmattermost.New, | ||||
| 	"matrix":       bmatrix.New, | ||||
| 	"rocketchat":   brocketchat.New, | ||||
| 	"slack-legacy": bslack.NewLegacy, | ||||
| 	"slack":        bslack.New, | ||||
| 	"sshchat":      bsshchat.New, | ||||
| 	"steam":        bsteam.New, | ||||
| 	"telegram":     btelegram.New, | ||||
| 	"xmpp":         bxmpp.New, | ||||
| 	"zulip":        bzulip.New, | ||||
| } | ||||
| var flog *logrus.Entry | ||||
|  | ||||
| const ( | ||||
| 	apiProtocol = "api" | ||||
| ) | ||||
|  | ||||
| func New(cfg config.Gateway, r *Router) *Gateway { | ||||
| 	flog = log.WithFields(log.Fields{"prefix": "gateway"}) | ||||
| 	flog = logrus.WithFields(logrus.Fields{"prefix": "gateway"}) | ||||
| 	gw := &Gateway{Channels: make(map[string]*config.ChannelInfo), Message: r.Message, | ||||
| 		Router: r, Bridges: make(map[string]*bridge.Bridge), Config: r.Config} | ||||
| 	cache, _ := lru.New(5000) | ||||
| 	gw.Messages = cache | ||||
| 	gw.AddConfig(&cfg) | ||||
| 	if err := gw.AddConfig(&cfg); err != nil { | ||||
| 		flog.Errorf("AddConfig failed: %s", err) | ||||
| 	} | ||||
| 	return gw | ||||
| } | ||||
|  | ||||
| @@ -111,10 +77,10 @@ func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||
| 		br.Config = gw.Router.Config | ||||
| 		br.General = &gw.BridgeValues().General | ||||
| 		// set logging | ||||
| 		br.Log = log.WithFields(log.Fields{"prefix": "bridge"}) | ||||
| 		brconfig := &bridge.Config{Remote: gw.Message, Log: log.WithFields(log.Fields{"prefix": br.Protocol}), Bridge: br} | ||||
| 		br.Log = logrus.WithFields(logrus.Fields{"prefix": "bridge"}) | ||||
| 		brconfig := &bridge.Config{Remote: gw.Message, Log: logrus.WithFields(logrus.Fields{"prefix": br.Protocol}), Bridge: br} | ||||
| 		// add the actual bridger for this protocol to this bridge using the bridgeMap | ||||
| 		br.Bridger = bridgeMap[br.Protocol](brconfig) | ||||
| 		br.Bridger = gw.Router.BridgeMap[br.Protocol](brconfig) | ||||
| 	} | ||||
| 	gw.mapChannelsToBridge(br) | ||||
| 	gw.Bridges[cfg.Account] = br | ||||
| @@ -124,7 +90,9 @@ func (gw *Gateway) AddBridge(cfg *config.Bridge) error { | ||||
| func (gw *Gateway) AddConfig(cfg *config.Gateway) error { | ||||
| 	gw.Name = cfg.Name | ||||
| 	gw.MyConfig = cfg | ||||
| 	gw.mapChannels() | ||||
| 	if err := gw.mapChannels(); err != nil { | ||||
| 		flog.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) | ||||
| @@ -144,7 +112,9 @@ func (gw *Gateway) mapChannelsToBridge(br *bridge.Bridge) { | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) reconnectBridge(br *bridge.Bridge) { | ||||
| 	br.Disconnect() | ||||
| 	if err := br.Disconnect(); err != nil { | ||||
| 		flog.Errorf("Disconnect() %s failed: %s", br.Account, err) | ||||
| 	} | ||||
| 	time.Sleep(time.Second * 5) | ||||
| RECONNECT: | ||||
| 	flog.Infof("Reconnecting %s", br.Account) | ||||
| @@ -155,7 +125,9 @@ RECONNECT: | ||||
| 		goto RECONNECT | ||||
| 	} | ||||
| 	br.Joined = make(map[string]bool) | ||||
| 	br.JoinChannels() | ||||
| 	if err := br.JoinChannels(); err != nil { | ||||
| 		flog.Errorf("JoinChannels() %s failed: %s", br.Account, err) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | ||||
| @@ -167,6 +139,10 @@ func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) { | ||||
| 		if strings.HasPrefix(br.Account, "irc.") { | ||||
| 			br.Channel = strings.ToLower(br.Channel) | ||||
| 		} | ||||
| 		if strings.HasPrefix(br.Account, "mattermost.") && strings.HasPrefix(br.Channel, "#") { | ||||
| 			flog.Errorf("Mattermost channels do not start with a #: remove the # in %s", br.Channel) | ||||
| 			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, | ||||
| @@ -242,103 +218,55 @@ func (gw *Gateway) getDestMsgID(msgID string, dest *bridge.Bridge, channel confi | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||
| 	var brMsgIDs []*BrMsgID | ||||
|  | ||||
| 	// if we have an attached file, or other info | ||||
| 	if msg.Extra != nil { | ||||
| 		if len(msg.Extra[config.EventFileFailureSize]) != 0 { | ||||
| 			if msg.Text == "" { | ||||
| 				return brMsgIDs | ||||
| 			} | ||||
| 		} | ||||
| // 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 | ||||
| 	} | ||||
|  | ||||
| 	// Avatar downloads are only relevant for telegram and mattermost for now | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		if dest.Protocol != "mattermost" && | ||||
| 			dest.Protocol != "telegram" { | ||||
| 			return brMsgIDs | ||||
| 		} | ||||
| 	if msg.Event == config.EventUserTyping { | ||||
| 		return false | ||||
| 	} | ||||
|  | ||||
| 	// only relay join/part when configured | ||||
| 	if msg.Event == config.EventJoinLeave && !gw.Bridges[dest.Account].GetBool("ShowJoinPart") { | ||||
| 		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 | ||||
| 	} | ||||
| 	flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| 	// only relay topic change when configured | ||||
| 	if msg.Event == config.EventTopicChange && !gw.Bridges[dest.Account].GetBool("ShowTopicChange") { | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// broadcast to every out channel (irc QUIT) | ||||
| 	if msg.Channel == "" && msg.Event != config.EventJoinLeave { | ||||
| 		flog.Debug("empty channel") | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// Get the ID of the parent message in thread | ||||
| 	var canonicalParentMsgID string | ||||
| 	if msg.ParentID != "" && (gw.BridgeValues().General.PreserveThreading || dest.GetBool("PreserveThreading")) { | ||||
| 		canonicalParentMsgID = gw.FindCanonicalMsgID(msg.Protocol, msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	originchannel := msg.Channel | ||||
| 	origmsg := msg | ||||
| 	channels := gw.getDestChannel(&msg, *dest) | ||||
| 	for _, channel := range channels { | ||||
| 		// Only send the avatar download event to ourselves. | ||||
| 		if msg.Event == config.EventAvatarDownload { | ||||
| 			if channel.ID != getChannelID(origmsg) { | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// do not send to ourself for any other event | ||||
| 			if channel.ID == getChannelID(origmsg) { | ||||
| 				continue | ||||
| 			} | ||||
| // ignoreTexts returns true if msg.Text matches any of the input regexes. | ||||
| func (gw *Gateway) ignoreTexts(msg *config.Message, input []string) bool { | ||||
| 	for _, entry := range input { | ||||
| 		if entry == "" { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// Too noisy to log like other events | ||||
| 		if msg.Event != config.EventUserTyping { | ||||
| 			flog.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name) | ||||
| 		} | ||||
|  | ||||
| 		msg.Channel = channel.Name | ||||
| 		msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||
| 		msg.Username = gw.modifyUsername(origmsg, dest) | ||||
|  | ||||
| 		msg.ID = gw.getDestMsgID(origmsg.Protocol+" "+origmsg.ID, dest, channel) | ||||
|  | ||||
| 		// for api we need originchannel as channel | ||||
| 		if dest.Protocol == apiProtocol { | ||||
| 			msg.Channel = originchannel | ||||
| 		} | ||||
|  | ||||
| 		msg.ParentID = gw.getDestMsgID(origmsg.Protocol+" "+canonicalParentMsgID, dest, channel) | ||||
| 		if msg.ParentID == "" { | ||||
| 			msg.ParentID = canonicalParentMsgID | ||||
| 		} | ||||
|  | ||||
| 		// if we are using mattermost plugin account, send messages to MattermostPlugin channel | ||||
| 		// that can be picked up by the mattermost matterbridge plugin | ||||
| 		if dest.Account == "mattermost.plugin" { | ||||
| 			gw.Router.MattermostPlugin <- msg | ||||
| 		} | ||||
|  | ||||
| 		mID, err := dest.Send(msg) | ||||
| 		// TODO do not compile regexps everytime | ||||
| 		re, err := regexp.Compile(entry) | ||||
| 		if err != nil { | ||||
| 			flog.Error(err) | ||||
| 			flog.Errorf("incorrect regexp %s for %s", entry, msg.Account) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||
| 		if mID != "" { | ||||
| 			flog.Debugf("mID %s: %s", dest.Account, mID) | ||||
| 			brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + mID, channel.ID}) | ||||
| 		if re.MatchString(msg.Text) { | ||||
| 			flog.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account) | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return brMsgIDs | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| // ignoreNicks returns true if msg.Username matches any of the input regexes. | ||||
| func (gw *Gateway) ignoreNicks(msg *config.Message, input []string) bool { | ||||
| 	// is the username in IgnoreNicks field | ||||
| 	for _, entry := range input { | ||||
| 		if msg.Username == entry { | ||||
| 			flog.Debugf("ignoring %s from %s", msg.Username, msg.Account) | ||||
| 			return true | ||||
| 		} | ||||
| 	} | ||||
| 	return false | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||
| @@ -347,59 +275,23 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool { | ||||
| 		return true | ||||
| 	} | ||||
|  | ||||
| 	// check if we need to ignore a empty message | ||||
| 	if msg.Text == "" { | ||||
| 		if msg.Event == config.EventUserTyping { | ||||
| 			return false | ||||
| 		} | ||||
| 		// we have an attachment or actual bytes, do not ignore | ||||
| 		if msg.Extra != nil && | ||||
| 			(msg.Extra["attachments"] != nil || | ||||
| 				len(msg.Extra["file"]) > 0 || | ||||
| 				len(msg.Extra[config.EventFileFailureSize]) > 0) { | ||||
| 			return false | ||||
| 		} | ||||
| 		flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account) | ||||
| 	igNicks := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreNicks")) | ||||
| 	igMessages := strings.Fields(gw.Bridges[msg.Account].GetString("IgnoreMessages")) | ||||
| 	if gw.ignoreTextEmpty(msg) || gw.ignoreNicks(msg, igNicks) || gw.ignoreTexts(msg, 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 { | ||||
| 	br := gw.Bridges[msg.Account] | ||||
| 	msg.Protocol = br.Protocol | ||||
| 	if gw.BridgeValues().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.BridgeValues().General.RemoteNickFormat | ||||
| 	} | ||||
|  | ||||
| 	// loop to replace nicks | ||||
| 	for _, outer := range br.GetStringSlice2D("ReplaceNicks") { | ||||
| @@ -437,10 +329,7 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin | ||||
| } | ||||
|  | ||||
| func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string { | ||||
| 	iconurl := gw.BridgeValues().General.IconURL | ||||
| 	if iconurl == "" { | ||||
| 		iconurl = dest.GetString("IconURL") | ||||
| 	} | ||||
| 	iconurl := dest.GetString("IconURL") | ||||
| 	iconurl = strings.Replace(iconurl, "{NICK}", msg.Username, -1) | ||||
| 	if msg.Avatar == "" { | ||||
| 		msg.Avatar = iconurl | ||||
| @@ -472,86 +361,61 @@ func (gw *Gateway) modifyMessage(msg *config.Message) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // handleFiles uploads or places all files on the given msg to the MediaServer and | ||||
| // adds the new URL of the file on the MediaServer onto the given msg. | ||||
| func (gw *Gateway) handleFiles(msg *config.Message) { | ||||
| 	reg := regexp.MustCompile("[^a-zA-Z0-9]+") | ||||
|  | ||||
| 	// If we don't have a attachfield or we don't have a mediaserver configured return | ||||
| 	if msg.Extra == nil || | ||||
| 		(gw.BridgeValues().General.MediaServerUpload == "" && | ||||
| 			gw.BridgeValues().General.MediaDownloadPath == "") { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// If we don't have files, nothing to upload. | ||||
| 	if len(msg.Extra["file"]) == 0 { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	client := &http.Client{ | ||||
| 		Timeout: time.Second * 5, | ||||
| 	} | ||||
|  | ||||
| 	for i, f := range msg.Extra["file"] { | ||||
| 		fi := f.(config.FileInfo) | ||||
| 		ext := filepath.Ext(fi.Name) | ||||
| 		fi.Name = fi.Name[0 : len(fi.Name)-len(ext)] | ||||
| 		fi.Name = reg.ReplaceAllString(fi.Name, "_") | ||||
| 		fi.Name += ext | ||||
|  | ||||
| 		sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))[:8] | ||||
|  | ||||
| 		if gw.BridgeValues().General.MediaServerUpload != "" { | ||||
| 			// Use MediaServerUpload. Upload using a PUT HTTP request and basicauth. | ||||
|  | ||||
| 			url := gw.BridgeValues().General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 			req, err := http.NewRequest("PUT", url, bytes.NewReader(*fi.Data)) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver upload failed, could not create request: %#v", err) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			flog.Debugf("mediaserver upload url: %s", url) | ||||
|  | ||||
| 			req.Header.Set("Content-Type", "binary/octet-stream") | ||||
| 			_, err = client.Do(req) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver upload failed, could not Do request: %#v", err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// Use MediaServerPath. Place the file on the current filesystem. | ||||
|  | ||||
| 			dir := gw.BridgeValues().General.MediaDownloadPath + "/" + sha1sum | ||||
| 			err := os.Mkdir(dir, os.ModePerm) | ||||
| 			if err != nil && !os.IsExist(err) { | ||||
| 				flog.Errorf("mediaserver path failed, could not mkdir: %s %#v", err, err) | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| 			path := dir + "/" + fi.Name | ||||
| 			flog.Debugf("mediaserver path placing file: %s", path) | ||||
|  | ||||
| 			err = ioutil.WriteFile(path, *fi.Data, os.ModePerm) | ||||
| 			if err != nil { | ||||
| 				flog.Errorf("mediaserver path failed, could not writefile: %s %#v", err, err) | ||||
| 				continue | ||||
| 			} | ||||
| // SendMessage sends a message (with specified parentID) to the channel on the selected destination bridge. | ||||
| // returns a message id and error. | ||||
| func (gw *Gateway) SendMessage(origmsg config.Message, dest *bridge.Bridge, channel config.ChannelInfo, canonicalParentMsgID string) (string, error) { | ||||
| 	msg := origmsg | ||||
| 	// Only send the avatar download event to ourselves. | ||||
| 	if msg.Event == config.EventAvatarDownload { | ||||
| 		if channel.ID != getChannelID(origmsg) { | ||||
| 			return "", nil | ||||
| 		} | ||||
| 	} else { | ||||
| 		// do not send to ourself for any other event | ||||
| 		if channel.ID == getChannelID(origmsg) { | ||||
| 			return "", nil | ||||
| 		} | ||||
|  | ||||
| 		// Download URL. | ||||
| 		durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 		flog.Debugf("mediaserver download URL = %s", durl) | ||||
|  | ||||
| 		// We uploaded/placed the file successfully. Add the SHA and URL. | ||||
| 		extra := msg.Extra["file"][i].(config.FileInfo) | ||||
| 		extra.URL = durl | ||||
| 		extra.SHA = sha1sum | ||||
| 		msg.Extra["file"][i] = extra | ||||
| 	} | ||||
|  | ||||
| 	// Too noisy to log like other events | ||||
| 	if msg.Event != config.EventUserTyping { | ||||
| 		flog.Debugf("=> Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, origmsg.Channel, dest.Account, channel.Name) | ||||
| 	} | ||||
|  | ||||
| 	msg.Channel = channel.Name | ||||
| 	msg.Avatar = gw.modifyAvatar(origmsg, dest) | ||||
| 	msg.Username = gw.modifyUsername(origmsg, dest) | ||||
|  | ||||
| 	msg.ID = gw.getDestMsgID(origmsg.Protocol+" "+origmsg.ID, dest, channel) | ||||
|  | ||||
| 	// for api we need originchannel as channel | ||||
| 	if dest.Protocol == apiProtocol { | ||||
| 		msg.Channel = origmsg.Channel | ||||
| 	} | ||||
|  | ||||
| 	msg.ParentID = gw.getDestMsgID(origmsg.Protocol+" "+canonicalParentMsgID, dest, channel) | ||||
| 	if msg.ParentID == "" { | ||||
| 		msg.ParentID = canonicalParentMsgID | ||||
| 	} | ||||
|  | ||||
| 	// if we are using mattermost plugin account, send messages to MattermostPlugin channel | ||||
| 	// that can be picked up by the mattermost matterbridge plugin | ||||
| 	if dest.Account == "mattermost.plugin" { | ||||
| 		gw.Router.MattermostPlugin <- msg | ||||
| 	} | ||||
|  | ||||
| 	mID, err := dest.Send(msg) | ||||
| 	if err != nil { | ||||
| 		return mID, err | ||||
| 	} | ||||
|  | ||||
| 	// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice | ||||
| 	if mID != "" { | ||||
| 		flog.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) bool { | ||||
|   | ||||
| @@ -3,11 +3,11 @@ package gateway | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"strconv" | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/42wim/matterbridge/gateway/bridgemap" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
|  | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| var testconfig = []byte(` | ||||
| @@ -160,7 +160,7 @@ const ( | ||||
|  | ||||
| func maketestRouter(input []byte) *Router { | ||||
| 	cfg := config.NewConfigFromString(input) | ||||
| 	r, err := NewRouter(cfg) | ||||
| 	r, err := NewRouter(cfg, bridgemap.FullMap) | ||||
| 	if err != nil { | ||||
| 		fmt.Println(err) | ||||
| 	} | ||||
| @@ -386,3 +386,116 @@ func TestGetDestChannelAdvanced(t *testing.T) { | ||||
| 	} | ||||
| 	assert.Equal(t, map[string]int{"bridge3": 4, "bridge": 9, "announcements": 3, "bridge2": 4}, hits) | ||||
| } | ||||
|  | ||||
| func TestIgnoreTextEmpty(t *testing.T) { | ||||
| 	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, | ||||
| 		}, | ||||
| 	} | ||||
| 	gw := &Gateway{} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := gw.ignoreTextEmpty(testcase.input) | ||||
| 		assert.Equalf(t, testcase.output, output, "case '%s' failed", testname) | ||||
| 	} | ||||
|  | ||||
| } | ||||
|  | ||||
| func TestIgnoreTexts(t *testing.T) { | ||||
| 	msgTests := map[string]struct { | ||||
| 		input  *config.Message | ||||
| 		re     []string | ||||
| 		output bool | ||||
| 	}{ | ||||
| 		"no regex": { | ||||
| 			input:  &config.Message{Text: "a text message"}, | ||||
| 			re:     []string{}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"simple regex": { | ||||
| 			input:  &config.Message{Text: "a text message"}, | ||||
| 			re:     []string{"text"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple regex fail": { | ||||
| 			input:  &config.Message{Text: "a text message"}, | ||||
| 			re:     []string{"abc", "123$"}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"multiple regex pass": { | ||||
| 			input:  &config.Message{Text: "a text message"}, | ||||
| 			re:     []string{"lala", "sage$"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 	} | ||||
| 	gw := &Gateway{} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := gw.ignoreTexts(testcase.input, testcase.re) | ||||
| 		assert.Equalf(t, testcase.output, output, "case '%s' failed", testname) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func TestIgnoreNicks(t *testing.T) { | ||||
| 	msgTests := map[string]struct { | ||||
| 		input  *config.Message | ||||
| 		re     []string | ||||
| 		output bool | ||||
| 	}{ | ||||
| 		"no entry": { | ||||
| 			input:  &config.Message{Username: "user", Text: "a text message"}, | ||||
| 			re:     []string{}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 		"one entry": { | ||||
| 			input:  &config.Message{Username: "user", Text: "a text message"}, | ||||
| 			re:     []string{"user"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple entries": { | ||||
| 			input:  &config.Message{Username: "user", Text: "a text message"}, | ||||
| 			re:     []string{"abc", "user"}, | ||||
| 			output: true, | ||||
| 		}, | ||||
| 		"multiple entries fail": { | ||||
| 			input:  &config.Message{Username: "user", Text: "a text message"}, | ||||
| 			re:     []string{"abc", "def"}, | ||||
| 			output: false, | ||||
| 		}, | ||||
| 	} | ||||
| 	gw := &Gateway{} | ||||
| 	for testname, testcase := range msgTests { | ||||
| 		output := gw.ignoreNicks(testcase.input, testcase.re) | ||||
| 		assert.Equalf(t, testcase.output, output, "case '%s' failed", testname) | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										227
									
								
								gateway/handlers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										227
									
								
								gateway/handlers.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,227 @@ | ||||
| package gateway | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"crypto/sha1" //nolint:gosec | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"net/http" | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 	"regexp" | ||||
| 	"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) | ||||
| 				flog.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 { | ||||
| 					flog.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 { | ||||
| 				flog.Error(err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} else { | ||||
| 			// Use MediaServerPath. Place the file on the current filesystem. | ||||
| 			if err := gw.handleFilesLocal(&fi); err != nil { | ||||
| 				flog.Error(err) | ||||
| 				continue | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// Download URL. | ||||
| 		durl := gw.BridgeValues().General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name | ||||
|  | ||||
| 		flog.Debugf("mediaserver download URL = %s", durl) | ||||
|  | ||||
| 		// We uploaded/placed the file successfully. Add the SHA and URL. | ||||
| 		extra := msg.Extra["file"][i].(config.FileInfo) | ||||
| 		extra.URL = durl | ||||
| 		extra.SHA = sha1sum | ||||
| 		msg.Extra["file"][i] = extra | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // 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) | ||||
| 	} | ||||
|  | ||||
| 	flog.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 | ||||
| 	flog.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(msg config.Message, dest *bridge.Bridge) []*BrMsgID { | ||||
| 	var brMsgIDs []*BrMsgID | ||||
|  | ||||
| 	// if we have an attached file, or other info | ||||
| 	if msg.Extra != nil && len(msg.Extra[config.EventFileFailureSize]) != 0 && msg.Text == "" { | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	if gw.ignoreEvent(msg.Event, dest) { | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// broadcast to every out channel (irc QUIT) | ||||
| 	if msg.Channel == "" && msg.Event != config.EventJoinLeave { | ||||
| 		flog.Debug("empty channel") | ||||
| 		return brMsgIDs | ||||
| 	} | ||||
|  | ||||
| 	// Get the ID of the parent message in thread | ||||
| 	var canonicalParentMsgID string | ||||
| 	if msg.ParentID != "" && dest.GetBool("PreserveThreading") { | ||||
| 		canonicalParentMsgID = gw.FindCanonicalMsgID(msg.Protocol, msg.ParentID) | ||||
| 	} | ||||
|  | ||||
| 	origmsg := msg | ||||
| 	channels := gw.getDestChannel(&msg, *dest) | ||||
| 	for _, channel := range channels { | ||||
| 		msgID, err := gw.SendMessage(origmsg, dest, channel, canonicalParentMsgID) | ||||
| 		if err != nil { | ||||
| 			flog.Errorf("SendMessage failed: %s", err) | ||||
| 			continue | ||||
| 		} | ||||
| 		if msgID == "" { | ||||
| 			continue | ||||
| 		} | ||||
| 		brMsgIDs = append(brMsgIDs, &BrMsgID{dest, dest.Protocol + " " + msgID, channel.ID}) | ||||
| 	} | ||||
| 	return brMsgIDs | ||||
| } | ||||
| @@ -2,6 +2,7 @@ package gateway | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge" | ||||
| @@ -12,14 +13,17 @@ import ( | ||||
| type Router struct { | ||||
| 	config.Config | ||||
|  | ||||
| 	BridgeMap        map[string]bridge.Factory | ||||
| 	Gateways         map[string]*Gateway | ||||
| 	Message          chan config.Message | ||||
| 	MattermostPlugin chan config.Message | ||||
| 	sync.RWMutex | ||||
| } | ||||
|  | ||||
| func NewRouter(cfg config.Config) (*Router, error) { | ||||
| func NewRouter(cfg config.Config, bridgeMap map[string]bridge.Factory) (*Router, error) { | ||||
| 	r := &Router{ | ||||
| 		Config:           cfg, | ||||
| 		BridgeMap:        bridgeMap, | ||||
| 		Message:          make(chan config.Message), | ||||
| 		MattermostPlugin: make(chan config.Message), | ||||
| 		Gateways:         make(map[string]*Gateway), | ||||
| @@ -54,17 +58,47 @@ func (r *Router) Start() error { | ||||
| 		flog.Infof("Starting bridge: %s ", br.Account) | ||||
| 		err := br.Connect() | ||||
| 		if err != nil { | ||||
| 			return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err) | ||||
| 			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 { | ||||
| 				flog.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 { | ||||
| 		flog.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 { | ||||
| @@ -77,42 +111,47 @@ func (r *Router) getBridge(account string) *bridge.Bridge { | ||||
| func (r *Router) handleReceive() { | ||||
| 	for msg := range r.Message { | ||||
| 		msg := msg // scopelint | ||||
| 		if msg.Event == config.EventFailure { | ||||
| 		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.EventRejoinChannels { | ||||
| 			for _, gw := range r.Gateways { | ||||
| 				for _, br := range gw.Bridges { | ||||
| 					if msg.Account == br.Account { | ||||
| 						br.Joined = make(map[string]bool) | ||||
| 						br.JoinChannels() | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		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.Protocol + " " + msg.ID); !ok && msg.ID != "" { | ||||
| 					gw.Messages.Add(msg.Protocol+" "+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 | ||||
| 				} | ||||
| 				flog.Debugf("sending %s to %s", config.EventGetChannelMembers, br.Account) | ||||
| 				if _, err := br.Send(config.Message{Event: config.EventGetChannelMembers}); err != nil { | ||||
| 					flog.Errorf("updateChannelMembers: %s", err) | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		time.Sleep(time.Minute) | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,10 +1,10 @@ | ||||
| package samechannelgateway | ||||
|  | ||||
| import ( | ||||
| 	"testing" | ||||
|  | ||||
| 	"github.com/42wim/matterbridge/bridge/config" | ||||
| 	"github.com/stretchr/testify/assert" | ||||
|  | ||||
| 	"testing" | ||||
| ) | ||||
|  | ||||
| const testConfig = ` | ||||
|   | ||||
							
								
								
									
										57
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										57
									
								
								go.mod
									
									
									
									
									
								
							| @@ -3,78 +3,79 @@ module github.com/42wim/matterbridge | ||||
| require ( | ||||
| 	github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 | ||||
| 	github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 // indirect | ||||
| 	github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 | ||||
| 	github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b // indirect | ||||
| 	github.com/Philipp15b/go-steam v1.0.1-0.20180818081528-681bd9573329 | ||||
| 	github.com/bwmarrin/discordgo v0.19.0 | ||||
| 	github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d | ||||
| 	github.com/dfordsoft/golib v0.0.0-20180902042739-76ee6ab99bec | ||||
| 	github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a // indirect | ||||
| 	github.com/fsnotify/fsnotify v1.4.7 | ||||
| 	github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c | ||||
| 	github.com/go-telegram-bot-api/telegram-bot-api v4.6.5-0.20181225215658-ec221ba9ea45+incompatible | ||||
| 	github.com/golang/protobuf v0.0.0-20170613224224-e325f446bebc // indirect | ||||
| 	github.com/google/gops v0.0.0-20170319002943-62f833fc9f6c | ||||
| 	github.com/google/gops v0.3.5 | ||||
| 	github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f // indirect | ||||
| 	github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c | ||||
| 	github.com/gorilla/schema v1.0.2 | ||||
| 	github.com/gorilla/websocket v1.4.0 | ||||
| 	github.com/hashicorp/golang-lru v0.0.0-20160813221303-0a025b7e63ad | ||||
| 	github.com/hashicorp/hcl v0.0.0-20171017181929-23c074d0eceb // indirect | ||||
| 	github.com/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/kr/pretty v0.1.0 // indirect | ||||
| 	github.com/labstack/echo v0.0.0-20180219162101-7eec915044a1 | ||||
| 	github.com/labstack/echo v3.3.5+incompatible | ||||
| 	github.com/labstack/gommon v0.2.1 // indirect | ||||
| 	github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e | ||||
| 	github.com/lrstanley/girc v0.0.0-20190102153329-c1e59a02f488 | ||||
| 	github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 // indirect | ||||
| 	github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 // indirect | ||||
| 	github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 // indirect | ||||
| 	github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 | ||||
| 	github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f | ||||
| 	github.com/matterbridge/gomatrix v0.0.0-20190102230110-6f9631ca6dea | ||||
| 	github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544 | ||||
| 	github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 | ||||
| 	github.com/mattermost/platform v4.6.2+incompatible | ||||
| 	github.com/mattermost/mattermost-server v5.5.0+incompatible | ||||
| 	github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597 // indirect | ||||
| 	github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc // indirect | ||||
| 	github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b // indirect | ||||
| 	github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 // indirect | ||||
| 	github.com/mitchellh/mapstructure v1.1.2 // indirect | ||||
| 	github.com/mreiferson/go-httpclient v0.0.0-20160630210159-31f0106b4474 // indirect | ||||
| 	github.com/mrexodia/wray v0.0.0-20160318003008-78a2c1f284ff // indirect | ||||
| 	github.com/nicksnyder/go-i18n v1.4.0 // indirect | ||||
| 	github.com/nlopes/slack v0.4.0 | ||||
| 	github.com/nlopes/slack v0.4.1-0.20181111125009-5963eafd777b | ||||
| 	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-20180820090156-eeb3823dab9a | ||||
| 	github.com/pkg/errors v0.8.0 // indirect | ||||
| 	github.com/rs/xid v0.0.0-20180525034800-088c5cf1423a | ||||
| 	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/shazow/ssh-chat v0.0.0-20181028152505-f36d7eb9ccc6 | ||||
| 	github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95 // indirect | ||||
| 	github.com/sirupsen/logrus v1.2.0 | ||||
| 	github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 // indirect | ||||
| 	github.com/smartystreets/goconvey v0.0.0-20180222194500-ef6db91d284a // indirect | ||||
| 	github.com/spf13/afero v0.0.0-20180211162714-bbf41cb36dff // indirect | ||||
| 	github.com/spf13/cast v1.2.0 // indirect | ||||
| 	github.com/spf13/jwalterweatherman v0.0.0-20180109140146-7c0cea34c8ec // indirect | ||||
| 	github.com/spf13/pflag v0.0.0-20180220143236-ee5fd03fd6ac // indirect | ||||
| 	github.com/spf13/viper v0.0.0-20171227194143-aafc9e6bc7b7 | ||||
| 	github.com/spf13/cast v1.3.0 // indirect | ||||
| 	github.com/spf13/pflag v1.0.3 // indirect | ||||
| 	github.com/spf13/viper v1.2.1 | ||||
| 	github.com/stretchr/testify v1.2.2 | ||||
| 	github.com/technoweenie/multipartstreamer v1.0.1 // indirect | ||||
| 	github.com/valyala/bytebufferpool v0.0.0-20160817181652-e746df99fe4a // indirect | ||||
| 	github.com/valyala/fasttemplate v0.0.0-20170224212429-dcecefd839c4 // indirect | ||||
| 	github.com/x-cray/logrus-prefixed-formatter v0.5.2 // indirect | ||||
| 	github.com/zfjagann/golang-ring v0.0.0-20141111230621-17637388c9f6 | ||||
| 	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 | ||||
| 	golang.org/x/crypto v0.0.0-20181112202954-3d3f9f413869 // indirect | ||||
| 	golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 // indirect | ||||
| 	golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f // indirect | ||||
| 	golang.org/x/sys v0.0.0-20181116161606-93218def8b18 // indirect | ||||
| 	golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 // indirect | ||||
| 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 // indirect | ||||
| 	gopkg.in/fsnotify.v1 v1.4.7 // indirect | ||||
| 	gopkg.in/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 | ||||
| ) | ||||
|   | ||||
							
								
								
									
										134
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										134
									
								
								go.sum
									
									
									
									
									
								
							| @@ -2,40 +2,42 @@ github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557 h1:IZtuWGfzQnKnCSu | ||||
| github.com/42wim/go-gitter v0.0.0-20170828205020-017310c2d557/go.mod h1:jL0YSXMs/txjtGJ4PWrmETOk6KUHMDPMshgQZlTeB3Y= | ||||
| github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14 h1:v/zr4ns/4sSahF9KBm4Uc933bLsEEv7LuT63CJ019yo= | ||||
| github.com/BurntSushi/toml v0.0.0-20170318202913-d94612f9fc14/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= | ||||
| github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3 h1:V4+1E1SRYUySqwOoI3ZphFADtabbF568zTHa5ix/zU0= | ||||
| github.com/Philipp15b/go-steam v0.0.0-20161020161927-e0f3bb9566e3/go.mod h1:HuVM+sZFzumUdKPWiz+IlCMb4RdsKdT3T+nQBKL+sYg= | ||||
| github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b h1:1OpGXps6UOY5HtQaQcLowsV1qMWCNBzhFvK7q4fgXtc= | ||||
| github.com/alecthomas/log4go v0.0.0-20160307011253-e5dc62318d9b/go.mod h1:iCVmQ9g4TfaRX5m5jq5sXY7RXYWPv9/PynM/GocbG3w= | ||||
| github.com/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/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/bwmarrin/discordgo v0.19.0 h1:kMED/DB0NR1QhRcalb85w0Cu3Ep2OrGAqZH1R5awQiY= | ||||
| github.com/bwmarrin/discordgo v0.19.0/go.mod h1:O9S4p+ofTFwB02em7jkpkV8M3R0/PUVOwN61zSZ0r4Q= | ||||
| github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= | ||||
| github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= | ||||
| github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d h1:rONNnZDE5CYuaSFQk+gP4GEQTXEUcyQ5p6p/dgxIHas= | ||||
| github.com/dfordsoft/golib v0.0.0-20180313113957-2ea3495aee1d/go.mod h1:UGa5M2Sz/Uh13AMse4+RELKCDw7kqgqlTjeGae+7vUY= | ||||
| github.com/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 v0.0.0-20170508165458-6c8dedd55f8a h1:MuHMeSsXbNEeUyxjB7T9P8s1+5k8OLTC/M27qsVwixM= | ||||
| github.com/dgrijalva/jwt-go v0.0.0-20170508165458-6c8dedd55f8a/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ= | ||||
| github.com/fsnotify/fsnotify v1.4.7 h1:IXs+QLmnXW2CcXuY+8Mzv/fWEsPGWxqefPtCP5CnV9I= | ||||
| github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c h1:3gMh737vMGqAkkkSfNbwjO8VRHOSaCjYRG4y9xVMEIQ= | ||||
| github.com/go-telegram-bot-api/telegram-bot-api v0.0.0-20180428185002-212b1541150c/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM= | ||||
| github.com/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 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/google/gops v0.3.5 h1:SIWvPLiYvy5vMwjxB3rVFTE4QBhUFj2KKWr3Xm7CKhw= | ||||
| github.com/google/gops v0.3.5/go.mod h1:pMQgrscwEK/aUSW1IFSaBPbJX82FPHWaSoJw1axQfD0= | ||||
| github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f h1:FDM3EtwZLyhW48YRiyqjivNlNZjAObv4xt4NnJaU+NQ= | ||||
| github.com/gopherjs/gopherjs v0.0.0-20180628210949-0892b62f0d9f/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= | ||||
| github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c h1:mORYpib1aLu3M2Oi50Z1pNTXuDJEHcoLb6oo6VdOutk= | ||||
| github.com/gorilla/schema v0.0.0-20170317173100-f3c80893412c/go.mod h1:kgLaKoK1FELgZqMAVxx/5cbj0kT+57qxUrAlIO2eleU= | ||||
| github.com/gorilla/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.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/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= | ||||
| @@ -47,44 +49,45 @@ github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORN | ||||
| 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/echo v3.3.5+incompatible h1:9PfxPUmasKzeJor9uQTaXLT6WUG/r+vSTmvXxvv3JO4= | ||||
| github.com/labstack/echo v3.3.5+incompatible/go.mod h1:0INS7j/VjnFxD4E2wkz67b8cVwCLbBmJyDaka6Cmk1s= | ||||
| github.com/labstack/gommon v0.2.1 h1:C+I4NYknueQncqKYZQ34kHsLZJVeB5KwPUhnO0nmbpU= | ||||
| github.com/labstack/gommon v0.2.1/go.mod h1:/tj9csK2iPSBvn+3NLM9e52usepMtrd5ilFYA+wQNJ4= | ||||
| github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e h1:RpktB2igr6nS1EN7bCvjldAEfngrM5GyAbmOa4/cafU= | ||||
| github.com/lrstanley/girc v0.0.0-20180913221000-0fb5b684054e/go.mod h1:7cRs1SIBfKQ7e3Tam6GKTILSNHzR862JD0JpINaZoJk= | ||||
| github.com/lrstanley/girc v0.0.0-20190102153329-c1e59a02f488 h1:dDEQN5oaa0WOzEiPDSbOugW/e2I/SWY98HYRdcwmGfY= | ||||
| github.com/lrstanley/girc v0.0.0-20190102153329-c1e59a02f488/go.mod h1:7cRs1SIBfKQ7e3Tam6GKTILSNHzR862JD0JpINaZoJk= | ||||
| github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5 h1:AsEBgzv3DhuYHI/GiQh2HxvTP71HCCE9E/tzGUzGdtU= | ||||
| github.com/lusis/go-slackbot v0.0.0-20180109053408-401027ccfef5/go.mod h1:c2mYKRyMb1BPkO5St0c/ps62L4S0W2NAkaTXj9qEI+0= | ||||
| github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6 h1:iOAVXzZyXtW408TMYejlUPo6BIn92HmOacWtIfNyYns= | ||||
| github.com/lusis/slack-test v0.0.0-20180109053238-3c758769bfa6/go.mod h1:sFlOUpQL1YcjhFVXhg1CG8ZASEs/Mf1oVb6H75JL/zg= | ||||
| github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885 h1:HWxJJvF+QceKcql4r9PC93NtMEgEBfBxlQrZPvbcQvs= | ||||
| github.com/magiconair/properties v0.0.0-20180217134545-2c9e95027885/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= | ||||
| github.com/magiconair/properties v1.8.0 h1:LLgXmsheXeRoUOBOjtwPQCWIYqM/LU1ayDtDePerRcY= | ||||
| github.com/magiconair/properties v1.8.0/go.mod h1:PppfXfuXeibc/6YijjN8zIbojt8czPbwD3XqdrwzmxQ= | ||||
| github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91 h1:KzDEcy8eDbTx881giW8a6llsAck3e2bJvMyKvh1IK+k= | ||||
| github.com/matterbridge/go-xmpp v0.0.0-20180529212104-cd19799fba91/go.mod h1:ECDRehsR9TYTKCAsRS8/wLeOk6UUqDydw47ln7wG41Q= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f h1:2eKh6Qi/sJ8bXvYMoyVfQxHgR8UcCDWjOmhV1oCstMU= | ||||
| github.com/matterbridge/gomatrix v0.0.0-20171224233421-78ac6a1a0f5f/go.mod h1:+jWeaaUtXQbBRdKYWfjW6JDDYiI2XXE+3NnTjW5kg8g= | ||||
| github.com/matterbridge/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-20180507190239-b6bb12d33544 h1:A8lLG3DAu75B5jITHs9z4JBmU6oCq1WiUNnDAmqKCZc= | ||||
| github.com/matterbridge/gozulipbot v0.0.0-20180507190239-b6bb12d33544/go.mod h1:yAjnZ34DuDyPHMPHHjOsTk/FefW4JJjoMMCGt/8uuQA= | ||||
| github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61 h1:R/MgM/eUyRBQx2FiH6JVmXck8PaAuKfe2M1tWIzW7nE= | ||||
| github.com/matterbridge/logrus-prefixed-formatter v0.0.0-20180806162718-01618749af61/go.mod h1:iXGEotOvwI1R1SjLxRc+BF5rUORTMtE0iMZBT2lxqAU= | ||||
| github.com/mattermost/platform v4.6.2+incompatible h1:9WqKNuJFIp6SDYn5wl1RF5urdhEw8d7o5tAOwT1MW0A= | ||||
| github.com/mattermost/platform v4.6.2+incompatible/go.mod h1:HjGKtkQNu3HXTOykPMQckMnH11WHvNvQqDBNnVXVbfM= | ||||
| github.com/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.0-20170210172801-5411d3eea597 h1:hGizH4aMDFFt1iOA4HNKC13lqIBoCyxIjWcAnWIy7aU= | ||||
| github.com/mattn/go-colorable v0.0.0-20170210172801-5411d3eea597/go.mod h1:9vuHe8Xs5qXnSaW/c/ABM9alt+Vo+STaOChaDxuIBZU= | ||||
| github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc h1:pK7tzC30erKOTfEDCYGvPZQCkmM9X5iSmmAR5m9x3Yc= | ||||
| github.com/mattn/go-isatty v0.0.0-20170216235908-dda3de49cbfc/go.mod h1:M+lRXTBqGeGNdLjl/ufCoiOlB5xdOkqRJdNxMWT7Zi4= | ||||
| github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b h1:j7+1HpAFS1zy5+Q4qx1fWh90gTKwiN4QCGoY9TWyyO4= | ||||
| github.com/mgutz/ansi v0.0.0-20170206155736-9520e82c474b/go.mod h1:01TrycV0kFyexm33Z7vhZRXopbI8J3TDReVlkTgMUxE= | ||||
| github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238 h1:+MZW2uvHgN8kYvksEN3f7eFL2wpzk0GxmlFsMybWc7E= | ||||
| github.com/mitchellh/mapstructure v0.0.0-20180220230111-00c29f56e238/go.mod h1:FVVH3fgwuzCH5S8UJGiWEs2h04kUh9fWfEaFds41c1Y= | ||||
| github.com/mitchellh/mapstructure v1.0.0/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/nicksnyder/go-i18n v1.4.0 h1:AgLl+Yq7kg5OYlzCgu9cKTZOyI4tD/NgukKqLqC8E+I= | ||||
| github.com/nicksnyder/go-i18n v1.4.0/go.mod h1:HrK7VCrbOvQoUAQ7Vpy7i87N7JZZZ7R2xBGjv0j365Q= | ||||
| github.com/nlopes/slack v0.4.0 h1:OVnHm7lv5gGT5gkcHsZAyw++oHVFihbjWbL3UceUpiA= | ||||
| github.com/nlopes/slack v0.4.0/go.mod h1:jVI4BBK3lSktibKahxBF74txcK2vyvkza1z/+rRnVAM= | ||||
| github.com/nlopes/slack v0.4.1-0.20181111125009-5963eafd777b h1:8ncrr7Xps0GafXIxBzrq1qSjy1zhiCDp/9C4cOrE+GU= | ||||
| github.com/nlopes/slack v0.4.1-0.20181111125009-5963eafd777b/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,24 +96,24 @@ 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-20180820090156-eeb3823dab9a h1:zAss6STq7oejKWTMEUYDUKYZhqXe0xALo8pJhJ3JJAs= | ||||
| github.com/peterhellberg/emojilib v0.0.0-20180820090156-eeb3823dab9a/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/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/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/shazow/ssh-chat v0.0.0-20181028152505-f36d7eb9ccc6 h1:qNoZx1RWPGKiqfs8ZZAYsYtw3ejo3HIF7iECaeaJhFk= | ||||
| github.com/shazow/ssh-chat v0.0.0-20181028152505-f36d7eb9ccc6/go.mod h1:SA/9+Wy3zV0UvPjttpGgs90FS9ZZ5D/LTffnVqdIBE8= | ||||
| github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95 h1:/vdW8Cb7EXrkqWGufVMES1OH2sU9gKVb2n9/1y5NMBY= | ||||
| github.com/shurcooL/sanitized_anchor_name v0.0.0-20170918181015-86672fcb3f95/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc= | ||||
| github.com/sirupsen/logrus v1.2.0 h1:juTguoYk5qI21pwyTXY3B3Y5cOTH3ZUyZCg1v/mihuo= | ||||
| @@ -119,16 +122,19 @@ github.com/smartystreets/assertions v0.0.0-20180803164922-886ec427f6b9 h1:lXQ+j+ | ||||
| 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/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.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/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.2/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= | ||||
| 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.2.1 h1:bIcUwXqLseLF3BDAZduuNfekWG87ibtFxi59Bq+oI9M= | ||||
| github.com/spf13/viper v1.2.1/go.mod h1:P4AexN0a+C9tGAnUFNwDMYYZv3pjFuvmeiMyKRaNVlI= | ||||
| 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= | ||||
| @@ -143,6 +149,25 @@ github.com/x-cray/logrus-prefixed-formatter v0.5.2 h1:00txxvfBM9muc0jiLIEAkAcIMJ | ||||
| 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= | ||||
| 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-20180119074636-ee41a25c63fb/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= | ||||
| 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= | ||||
| @@ -152,16 +177,21 @@ golang.org/x/net v0.0.0-20180108090419-434ec0c7fe37 h1:BkNcmLtAVeWe9h5k0jt24CQga | ||||
| 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-20180117170059-2c42eef0765b/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20180906133057-8cf3aee42992/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/sys v0.0.0-20181116161606-93218def8b18 h1:Wh+XCfg3kNpjhdq2LXrsiOProjtQZKme5XUx7VcxwAw= | ||||
| golang.org/x/sys v0.0.0-20181116161606-93218def8b18/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= | ||||
| golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978 h1:WNm0tmiuBMW4FJRuXKWOqaQfmKptHs0n8nTCyG0ayjc= | ||||
| golang.org/x/text v0.0.0-20180511172408-5c1cf69b5978/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= | ||||
| 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/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.1 h1:mUhvW9EsL+naU5Q3cakzfE91YhliOondGd6ZrsDBHQE= | ||||
| gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
|   | ||||
| @@ -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) | ||||
|   | ||||
| @@ -8,36 +8,40 @@ 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.12.0" | ||||
| 	version = "1.13.0" | ||||
| 	githash string | ||||
| ) | ||||
|  | ||||
| func main() { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true}) | ||||
| 	flog := log.WithFields(log.Fields{"prefix": "main"}) | ||||
| 	logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true}) | ||||
| 	flog := logrus.WithFields(logrus.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 err := agent.Listen(agent.Options{}); err != nil { | ||||
| 			flog.Errorf("failed to start gops agent: %#v", err) | ||||
| 		} else { | ||||
| 			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}) | ||||
| 		logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||
| 		flog.Info("Enabling debug") | ||||
| 		log.SetLevel(log.DebugLevel) | ||||
| 		logrus.SetLevel(logrus.DebugLevel) | ||||
| 	} | ||||
| 	flog.Printf("Running version %s %s", version, githash) | ||||
| 	if strings.Contains(version, "-dev") { | ||||
| @@ -45,7 +49,7 @@ func main() { | ||||
| 	} | ||||
| 	cfg := config.NewConfig(*flagConfig) | ||||
| 	cfg.BridgeValues().General.Debug = *flagDebug | ||||
| 	r, err := gateway.NewRouter(cfg) | ||||
| 	r, err := gateway.NewRouter(cfg, bridgemap.FullMap) | ||||
| 	if err != nil { | ||||
| 		flog.Fatalf("Starting gateway failed: %s", err) | ||||
| 	} | ||||
|   | ||||
| @@ -96,6 +96,11 @@ 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.  | ||||
| #Messages from those users will not be sent to other bridges. | ||||
| #OPTIONAL | ||||
| @@ -662,6 +667,8 @@ 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 | ||||
|  | ||||
| @@ -760,11 +767,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 | ||||
| ################################################################### | ||||
| @@ -1303,6 +1315,12 @@ 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 | ||||
|  | ||||
| ################################################################### | ||||
| #Gateway configuration | ||||
| ################################################################### | ||||
| @@ -1389,7 +1407,7 @@ 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" | ||||
|  | ||||
|     #API example | ||||
|     #[[gateway.inout]] | ||||
|   | ||||
							
								
								
									
										208
									
								
								matterclient/channels.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										208
									
								
								matterclient/channels.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,208 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"strings" | ||||
|  | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| // 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.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) 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.log.Debugf("updating channelheader %#v, %#v", channelId, header) | ||||
| 	_, resp := m.Client.UpdateChannel(channel) | ||||
| 	if resp.Error != nil { | ||||
| 		logrus.Error(resp.Error) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UpdateLastViewed(channelId string) error { //nolint:golint | ||||
| 	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) | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										282
									
								
								matterclient/helpers.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										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.log.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.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 | ||||
| 	} | ||||
| 	// 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.log.Debugf(logmsg + " with cookie (MMAUTH) token") | ||||
| 		m.Client.HttpClient.Jar = m.createCookieJar(m.Credentials.Token) | ||||
| 	} else { | ||||
| 		m.log.Debugf(logmsg + " with personal token") | ||||
| 	} | ||||
| 	m.User, resp = m.Client.GetMe("") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp, resp.Error | ||||
| 	} | ||||
| 	if m.User == nil { | ||||
| 		m.log.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.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) 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.log.Debugf("Server not up yet, reconnecting in %s", d) | ||||
| 			time.Sleep(d) | ||||
| 		} else { | ||||
| 			m.log.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.log.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.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 | ||||
| } | ||||
|  | ||||
| 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.log.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.log.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,14 +1,8 @@ | ||||
| package matterclient | ||||
|  | ||||
| import ( | ||||
| 	"crypto/md5" | ||||
| 	"crypto/tls" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"net/http" | ||||
| 	"net/http/cookiejar" | ||||
| 	"net/url" | ||||
| 	"strings" | ||||
| 	"sync" | ||||
| 	"time" | ||||
| @@ -17,8 +11,8 @@ import ( | ||||
| 	"github.com/hashicorp/golang-lru" | ||||
| 	"github.com/jpillora/backoff" | ||||
| 	prefixed "github.com/matterbridge/logrus-prefixed-formatter" | ||||
| 	"github.com/mattermost/platform/model" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/mattermost/mattermost-server/model" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| ) | ||||
|  | ||||
| type Credentials struct { | ||||
| @@ -61,7 +55,7 @@ type MMClient struct { | ||||
| 	User          *model.User | ||||
| 	Users         map[string]*model.User | ||||
| 	MessageChan   chan *Message | ||||
| 	log           *log.Entry | ||||
| 	log           *logrus.Entry | ||||
| 	WsClient      *websocket.Conn | ||||
| 	WsQuit        bool | ||||
| 	WsAway        bool | ||||
| @@ -76,23 +70,23 @@ type MMClient struct { | ||||
| 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"}) | ||||
| 	logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true}) | ||||
| 	mmclient.log = logrus.WithFields(logrus.Fields{"prefix": "matterclient"}) | ||||
| 	mmclient.lruCache, _ = lru.New(500) | ||||
| 	return mmclient | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SetDebugLog() { | ||||
| 	log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||
| 	logrus.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false, ForceFormatting: true}) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SetLogLevel(level string) { | ||||
| 	l, err := log.ParseLevel(level) | ||||
| 	l, err := logrus.ParseLevel(level) | ||||
| 	if err != nil { | ||||
| 		log.SetLevel(log.InfoLevel) | ||||
| 		logrus.SetLevel(logrus.InfoLevel) | ||||
| 		return | ||||
| 	} | ||||
| 	log.SetLevel(l) | ||||
| 	logrus.SetLevel(l) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) Login() error { | ||||
| @@ -110,101 +104,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 | ||||
|  | ||||
| 	if strings.Contains(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN) { | ||||
| 		token := strings.Split(m.Credentials.Pass, model.SESSION_COOKIE_TOKEN+"=") | ||||
| 		if len(token) != 2 { | ||||
| 			return errors.New("incorrect MMAUTHTOKEN. valid input is MMAUTHTOKEN=yourtoken") | ||||
| 		} | ||||
| 		m.Credentials.Token = token[1] | ||||
| 		m.Credentials.CookieToken = true | ||||
| 	// do initialization setup | ||||
| 	if err := m.initClient(firstConnection, b); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	if strings.Contains(m.Credentials.Pass, "token=") { | ||||
| 		token := strings.Split(m.Credentials.Pass, "token=") | ||||
| 		if len(token) != 2 { | ||||
| 			return errors.New("incorrect personal token. valid input is token=yourtoken") | ||||
| 		} | ||||
| 		m.Credentials.Token = token[1] | ||||
| 	if err := m.doLogin(firstConnection, b); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| 	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 | ||||
| 		} | ||||
| 	} | ||||
| 	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 m.Credentials.Token != "" { | ||||
| 			m.Client.AuthType = model.HEADER_BEARER | ||||
| 			m.Client.AuthToken = m.Credentials.Token | ||||
| 			if m.Credentials.CookieToken { | ||||
| 				m.log.Debugf(logmsg + " with cookie (MMAUTH) token") | ||||
| 				m.Client.HttpClient.Jar = m.createCookieJar(m.Credentials.Token) | ||||
| 			} else { | ||||
| 				m.log.Debugf(logmsg + " with personal token") | ||||
| 			} | ||||
| 			m.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 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 | ||||
| 	} | ||||
| 	// reset timer | ||||
| 	b.Reset() | ||||
|  | ||||
| 	err := m.initUser() | ||||
| 	if err != nil { | ||||
| 	if err := m.initUser(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
|  | ||||
| @@ -221,45 +131,6 @@ 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 | ||||
| } | ||||
|  | ||||
| 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.WsQuit = true | ||||
| @@ -325,572 +196,6 @@ func (m *MMClient) WsReceiver() { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| 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": | ||||
| 		m.UpdateChannels() | ||||
| 		/* | ||||
| 			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 '%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) 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 { //nolint:golint | ||||
| 	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 { | ||||
| 					if channel.Type == model.CHANNEL_GROUP { | ||||
| 						res := strings.Replace(channel.DisplayName, ", ", "-", -1) | ||||
| 						res = strings.Replace(res, " ", "_", -1) | ||||
| 						return res | ||||
| 					} | ||||
| 					return channel.Name | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		if t.MoreChannels != nil { | ||||
| 			for _, channel := range 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) GetChannelId(name string, teamId string) string { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	defer m.RUnlock() | ||||
| 	if 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 | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	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 { //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) 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) PostMessage(channelId string, text string) (string, error) { //nolint:golint | ||||
| 	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) { //nolint:golint | ||||
| 	post := &model.Post{ChannelId: channelId, Message: text, FileIds: fileIds} | ||||
| 	res, resp := m.Client.CreatePost(post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) EditMessage(postId string, text string) (string, error) { //nolint:golint | ||||
| 	post := &model.Post{Message: text} | ||||
| 	res, resp := m.Client.UpdatePost(postId, post) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return res.Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) DeleteMessage(postId string) error { //nolint:golint | ||||
| 	_, resp := m.Client.DeletePost(postId) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) JoinChannel(channelId string) error { //nolint:golint | ||||
| 	m.RLock() | ||||
| 	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 { //nolint:golint | ||||
| 	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 { //nolint:golint | ||||
| 	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) { //nolint:golint | ||||
| 	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) error { //nolint:golint | ||||
| 	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) | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	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.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) { //nolint:golint | ||||
| 	m.SendDirectMessageProps(toUserId, msg, nil) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) SendDirectMessageProps(toUserId string, msg string, props map[string]interface{}) { //nolint:golint | ||||
| 	m.log.Debugf("SendDirectMessage to %s, msg %s", toUserId, msg) | ||||
| 	// 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, m.Team.Id), Message: msg, Props: props} | ||||
| 	m.Client.CreatePost(post) | ||||
| } | ||||
|  | ||||
| // 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 "" | ||||
| } | ||||
|  | ||||
| // 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 { //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) 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 | ||||
| } | ||||
|  | ||||
| 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 { //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) UpdateUser(userId string) { //nolint:golint | ||||
| 	m.Lock() | ||||
| 	defer m.Unlock() | ||||
| 	res, resp := m.Client.GetUser(userId, "") | ||||
| 	if resp.Error != nil { | ||||
| 		return | ||||
| 	} | ||||
| 	m.Users[userId] = res | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetUserName(userId string) string { //nolint:golint | ||||
| 	user := m.GetUser(userId) | ||||
| 	if user != nil { | ||||
| 		return user.Username | ||||
| 	} | ||||
| 	return "" | ||||
| } | ||||
|  | ||||
| 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) UpdateStatus(userId string, status string) error { //nolint:golint | ||||
| 	_, resp := m.Client.UpdateUserStatus(userId, &model.Status{Status: status}) | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) GetStatuses() map[string]string { | ||||
| 	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 | ||||
| } | ||||
|  | ||||
| func (m *MMClient) UploadFile(data []byte, channelId string, filename string) (string, error) { //nolint:golint | ||||
| 	f, resp := m.Client.UploadFile(data, channelId, filename) | ||||
| 	if resp.Error != nil { | ||||
| 		return "", resp.Error | ||||
| 	} | ||||
| 	return f.FileInfos[0].Id, nil | ||||
| } | ||||
|  | ||||
| func (m *MMClient) StatusLoop() { | ||||
| 	retries := 0 | ||||
| 	backoff := time.Second * 60 | ||||
| @@ -903,7 +208,9 @@ func (m *MMClient) StatusLoop() { | ||||
| 			return | ||||
| 		} | ||||
| 		if m.WsConnected { | ||||
| 			m.checkAlive() | ||||
| 			if err := m.checkAlive(); err != nil { | ||||
| 				logrus.Errorf("Connection is not alive: %#v", err) | ||||
| 			} | ||||
| 			select { | ||||
| 			case <-m.WsPingChan: | ||||
| 				m.log.Debug("WS PONG received") | ||||
| @@ -915,7 +222,7 @@ func (m *MMClient) StatusLoop() { | ||||
| 					m.WsQuit = false | ||||
| 					err := m.Login() | ||||
| 					if err != nil { | ||||
| 						log.Errorf("Login failed: %#v", err) | ||||
| 						logrus.Errorf("Login failed: %#v", err) | ||||
| 						break | ||||
| 					} | ||||
| 					if m.OnWsConnect != nil { | ||||
| @@ -931,85 +238,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) checkAlive() error { | ||||
| 	// check if session still is valid | ||||
| 	_, resp := m.Client.GetMe("") | ||||
| 	if resp.Error != nil { | ||||
| 		return resp.Error | ||||
| 	} | ||||
| 	m.log.Debug("WS PING") | ||||
| 	return m.sendWSRequest("ping", nil) | ||||
| } | ||||
|  | ||||
| func (m *MMClient) sendWSRequest(action string, data map[string]interface{}) error { | ||||
| 	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
									
								
							
							
						
						
									
										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.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 '%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.log.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.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 | ||||
| 	if err := m.UpdateChannels(); err != nil { | ||||
| 		m.log.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
									
								
							
							
						
						
									
										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.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) 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 | ||||
| } | ||||
| @@ -71,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 { | ||||
|   | ||||
							
								
								
									
										41
									
								
								vendor/github.com/Philipp15b/go-steam/auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										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/alecthomas/log4go/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/alecthomas/log4go/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,2 +0,0 @@ | ||||
| *.sw[op] | ||||
| .DS_Store | ||||
							
								
								
									
										13
									
								
								vendor/github.com/alecthomas/log4go/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/alecthomas/log4go/LICENSE
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,13 +0,0 @@ | ||||
| Copyright (c) 2010, Kyle Lemons <kyle@kylelemons.net>. 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. | ||||
|  | ||||
| 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. | ||||
							
								
								
									
										14
									
								
								vendor/github.com/alecthomas/log4go/README
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										14
									
								
								vendor/github.com/alecthomas/log4go/README
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,14 +0,0 @@ | ||||
| # This is an unmaintained fork, left only so it doesn't break imports. | ||||
|  | ||||
| Please see http://log4go.googlecode.com/ | ||||
|  | ||||
| Installation: | ||||
| - Run `goinstall log4go.googlecode.com/hg` | ||||
|  | ||||
| Usage: | ||||
| - Add the following import: | ||||
| import l4g "log4go.googlecode.com/hg" | ||||
|  | ||||
| Acknowledgements: | ||||
| - pomack | ||||
|   For providing awesome patches to bring log4go up to the latest Go spec | ||||
							
								
								
									
										288
									
								
								vendor/github.com/alecthomas/log4go/config.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										288
									
								
								vendor/github.com/alecthomas/log4go/config.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,288 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"encoding/xml" | ||||
| 	"fmt" | ||||
| 	"io/ioutil" | ||||
| 	"os" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| type xmlProperty struct { | ||||
| 	Name  string `xml:"name,attr"` | ||||
| 	Value string `xml:",chardata"` | ||||
| } | ||||
|  | ||||
| type xmlFilter struct { | ||||
| 	Enabled  string        `xml:"enabled,attr"` | ||||
| 	Tag      string        `xml:"tag"` | ||||
| 	Level    string        `xml:"level"` | ||||
| 	Type     string        `xml:"type"` | ||||
| 	Property []xmlProperty `xml:"property"` | ||||
| } | ||||
|  | ||||
| type xmlLoggerConfig struct { | ||||
| 	Filter []xmlFilter `xml:"filter"` | ||||
| } | ||||
|  | ||||
| // Load XML configuration; see examples/example.xml for documentation | ||||
| func (log Logger) LoadConfiguration(filename string) { | ||||
| 	log.Close() | ||||
|  | ||||
| 	// Open the configuration file | ||||
| 	fd, err := os.Open(filename) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not open %q for reading: %s\n", filename, err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
|  | ||||
| 	contents, err := ioutil.ReadAll(fd) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not read %q: %s\n", filename, err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
|  | ||||
| 	xc := new(xmlLoggerConfig) | ||||
| 	if err := xml.Unmarshal(contents, xc); err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not parse XML configuration in %q: %s\n", filename, err) | ||||
| 		os.Exit(1) | ||||
| 	} | ||||
|  | ||||
| 	for _, xmlfilt := range xc.Filter { | ||||
| 		var filt LogWriter | ||||
| 		var lvl Level | ||||
| 		bad, good, enabled := false, true, false | ||||
|  | ||||
| 		// Check required children | ||||
| 		if len(xmlfilt.Enabled) == 0 { | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required attribute %s for filter missing in %s\n", "enabled", filename) | ||||
| 			bad = true | ||||
| 		} else { | ||||
| 			enabled = xmlfilt.Enabled != "false" | ||||
| 		} | ||||
| 		if len(xmlfilt.Tag) == 0 { | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "tag", filename) | ||||
| 			bad = true | ||||
| 		} | ||||
| 		if len(xmlfilt.Type) == 0 { | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "type", filename) | ||||
| 			bad = true | ||||
| 		} | ||||
| 		if len(xmlfilt.Level) == 0 { | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter missing in %s\n", "level", filename) | ||||
| 			bad = true | ||||
| 		} | ||||
|  | ||||
| 		switch xmlfilt.Level { | ||||
| 		case "FINEST": | ||||
| 			lvl = FINEST | ||||
| 		case "FINE": | ||||
| 			lvl = FINE | ||||
| 		case "DEBUG": | ||||
| 			lvl = DEBUG | ||||
| 		case "TRACE": | ||||
| 			lvl = TRACE | ||||
| 		case "INFO": | ||||
| 			lvl = INFO | ||||
| 		case "WARNING": | ||||
| 			lvl = WARNING | ||||
| 		case "ERROR": | ||||
| 			lvl = ERROR | ||||
| 		case "CRITICAL": | ||||
| 			lvl = CRITICAL | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required child <%s> for filter has unknown value in %s: %s\n", "level", filename, xmlfilt.Level) | ||||
| 			bad = true | ||||
| 		} | ||||
|  | ||||
| 		// Just so all of the required attributes are errored at the same time if missing | ||||
| 		if bad { | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
|  | ||||
| 		switch xmlfilt.Type { | ||||
| 		case "console": | ||||
| 			filt, good = xmlToConsoleLogWriter(filename, xmlfilt.Property, enabled) | ||||
| 		case "file": | ||||
| 			filt, good = xmlToFileLogWriter(filename, xmlfilt.Property, enabled) | ||||
| 		case "xml": | ||||
| 			filt, good = xmlToXMLLogWriter(filename, xmlfilt.Property, enabled) | ||||
| 		case "socket": | ||||
| 			filt, good = xmlToSocketLogWriter(filename, xmlfilt.Property, enabled) | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Could not load XML configuration in %s: unknown filter type \"%s\"\n", filename, xmlfilt.Type) | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
|  | ||||
| 		// Just so all of the required params are errored at the same time if wrong | ||||
| 		if !good { | ||||
| 			os.Exit(1) | ||||
| 		} | ||||
|  | ||||
| 		// If we're disabled (syntax and correctness checks only), don't add to logger | ||||
| 		if !enabled { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		log[xmlfilt.Tag] = &Filter{lvl, filt} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func xmlToConsoleLogWriter(filename string, props []xmlProperty, enabled bool) (*ConsoleLogWriter, bool) { | ||||
| 	// Parse properties | ||||
| 	for _, prop := range props { | ||||
| 		switch prop.Name { | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for console filter in %s\n", prop.Name, filename) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// If it's disabled, we're just checking syntax | ||||
| 	if !enabled { | ||||
| 		return nil, true | ||||
| 	} | ||||
|  | ||||
| 	return NewConsoleLogWriter(), true | ||||
| } | ||||
|  | ||||
| // Parse a number with K/M/G suffixes based on thousands (1000) or 2^10 (1024) | ||||
| func strToNumSuffix(str string, mult int) int { | ||||
| 	num := 1 | ||||
| 	if len(str) > 1 { | ||||
| 		switch str[len(str)-1] { | ||||
| 		case 'G', 'g': | ||||
| 			num *= mult | ||||
| 			fallthrough | ||||
| 		case 'M', 'm': | ||||
| 			num *= mult | ||||
| 			fallthrough | ||||
| 		case 'K', 'k': | ||||
| 			num *= mult | ||||
| 			str = str[0 : len(str)-1] | ||||
| 		} | ||||
| 	} | ||||
| 	parsed, _ := strconv.Atoi(str) | ||||
| 	return parsed * num | ||||
| } | ||||
| func xmlToFileLogWriter(filename string, props []xmlProperty, enabled bool) (*FileLogWriter, bool) { | ||||
| 	file := "" | ||||
| 	format := "[%D %T] [%L] (%S) %M" | ||||
| 	maxlines := 0 | ||||
| 	maxsize := 0 | ||||
| 	daily := false | ||||
| 	rotate := false | ||||
|  | ||||
| 	// Parse properties | ||||
| 	for _, prop := range props { | ||||
| 		switch prop.Name { | ||||
| 		case "filename": | ||||
| 			file = strings.Trim(prop.Value, " \r\n") | ||||
| 		case "format": | ||||
| 			format = strings.Trim(prop.Value, " \r\n") | ||||
| 		case "maxlines": | ||||
| 			maxlines = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1000) | ||||
| 		case "maxsize": | ||||
| 			maxsize = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1024) | ||||
| 		case "daily": | ||||
| 			daily = strings.Trim(prop.Value, " \r\n") != "false" | ||||
| 		case "rotate": | ||||
| 			rotate = strings.Trim(prop.Value, " \r\n") != "false" | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for file filter in %s\n", prop.Name, filename) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Check properties | ||||
| 	if len(file) == 0 { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for file filter missing in %s\n", "filename", filename) | ||||
| 		return nil, false | ||||
| 	} | ||||
|  | ||||
| 	// If it's disabled, we're just checking syntax | ||||
| 	if !enabled { | ||||
| 		return nil, true | ||||
| 	} | ||||
|  | ||||
| 	flw := NewFileLogWriter(file, rotate) | ||||
| 	flw.SetFormat(format) | ||||
| 	flw.SetRotateLines(maxlines) | ||||
| 	flw.SetRotateSize(maxsize) | ||||
| 	flw.SetRotateDaily(daily) | ||||
| 	return flw, true | ||||
| } | ||||
|  | ||||
| func xmlToXMLLogWriter(filename string, props []xmlProperty, enabled bool) (*FileLogWriter, bool) { | ||||
| 	file := "" | ||||
| 	maxrecords := 0 | ||||
| 	maxsize := 0 | ||||
| 	daily := false | ||||
| 	rotate := false | ||||
|  | ||||
| 	// Parse properties | ||||
| 	for _, prop := range props { | ||||
| 		switch prop.Name { | ||||
| 		case "filename": | ||||
| 			file = strings.Trim(prop.Value, " \r\n") | ||||
| 		case "maxrecords": | ||||
| 			maxrecords = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1000) | ||||
| 		case "maxsize": | ||||
| 			maxsize = strToNumSuffix(strings.Trim(prop.Value, " \r\n"), 1024) | ||||
| 		case "daily": | ||||
| 			daily = strings.Trim(prop.Value, " \r\n") != "false" | ||||
| 		case "rotate": | ||||
| 			rotate = strings.Trim(prop.Value, " \r\n") != "false" | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for xml filter in %s\n", prop.Name, filename) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Check properties | ||||
| 	if len(file) == 0 { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for xml filter missing in %s\n", "filename", filename) | ||||
| 		return nil, false | ||||
| 	} | ||||
|  | ||||
| 	// If it's disabled, we're just checking syntax | ||||
| 	if !enabled { | ||||
| 		return nil, true | ||||
| 	} | ||||
|  | ||||
| 	xlw := NewXMLLogWriter(file, rotate) | ||||
| 	xlw.SetRotateLines(maxrecords) | ||||
| 	xlw.SetRotateSize(maxsize) | ||||
| 	xlw.SetRotateDaily(daily) | ||||
| 	return xlw, true | ||||
| } | ||||
|  | ||||
| func xmlToSocketLogWriter(filename string, props []xmlProperty, enabled bool) (SocketLogWriter, bool) { | ||||
| 	endpoint := "" | ||||
| 	protocol := "udp" | ||||
|  | ||||
| 	// Parse properties | ||||
| 	for _, prop := range props { | ||||
| 		switch prop.Name { | ||||
| 		case "endpoint": | ||||
| 			endpoint = strings.Trim(prop.Value, " \r\n") | ||||
| 		case "protocol": | ||||
| 			protocol = strings.Trim(prop.Value, " \r\n") | ||||
| 		default: | ||||
| 			fmt.Fprintf(os.Stderr, "LoadConfiguration: Warning: Unknown property \"%s\" for file filter in %s\n", prop.Name, filename) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Check properties | ||||
| 	if len(endpoint) == 0 { | ||||
| 		fmt.Fprintf(os.Stderr, "LoadConfiguration: Error: Required property \"%s\" for file filter missing in %s\n", "endpoint", filename) | ||||
| 		return nil, false | ||||
| 	} | ||||
|  | ||||
| 	// If it's disabled, we're just checking syntax | ||||
| 	if !enabled { | ||||
| 		return nil, true | ||||
| 	} | ||||
|  | ||||
| 	return NewSocketLogWriter(protocol, endpoint), true | ||||
| } | ||||
							
								
								
									
										264
									
								
								vendor/github.com/alecthomas/log4go/filelog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										264
									
								
								vendor/github.com/alecthomas/log4go/filelog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,264 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| // This log writer sends output to a file | ||||
| type FileLogWriter struct { | ||||
| 	rec chan *LogRecord | ||||
| 	rot chan bool | ||||
|  | ||||
| 	// The opened file | ||||
| 	filename string | ||||
| 	file     *os.File | ||||
|  | ||||
| 	// The logging format | ||||
| 	format string | ||||
|  | ||||
| 	// File header/trailer | ||||
| 	header, trailer string | ||||
|  | ||||
| 	// Rotate at linecount | ||||
| 	maxlines          int | ||||
| 	maxlines_curlines int | ||||
|  | ||||
| 	// Rotate at size | ||||
| 	maxsize         int | ||||
| 	maxsize_cursize int | ||||
|  | ||||
| 	// Rotate daily | ||||
| 	daily          bool | ||||
| 	daily_opendate int | ||||
|  | ||||
| 	// Keep old logfiles (.001, .002, etc) | ||||
| 	rotate    bool | ||||
| 	maxbackup int | ||||
| } | ||||
|  | ||||
| // This is the FileLogWriter's output method | ||||
| func (w *FileLogWriter) LogWrite(rec *LogRecord) { | ||||
| 	w.rec <- rec | ||||
| } | ||||
|  | ||||
| func (w *FileLogWriter) Close() { | ||||
| 	close(w.rec) | ||||
| 	w.file.Sync() | ||||
| } | ||||
|  | ||||
| // NewFileLogWriter creates a new LogWriter which writes to the given file and | ||||
| // has rotation enabled if rotate is true. | ||||
| // | ||||
| // If rotate is true, any time a new log file is opened, the old one is renamed | ||||
| // with a .### extension to preserve it.  The various Set* methods can be used | ||||
| // to configure log rotation based on lines, size, and daily. | ||||
| // | ||||
| // The standard log-line format is: | ||||
| //   [%D %T] [%L] (%S) %M | ||||
| func NewFileLogWriter(fname string, rotate bool) *FileLogWriter { | ||||
| 	w := &FileLogWriter{ | ||||
| 		rec:       make(chan *LogRecord, LogBufferLength), | ||||
| 		rot:       make(chan bool), | ||||
| 		filename:  fname, | ||||
| 		format:    "[%D %T] [%L] (%S) %M", | ||||
| 		rotate:    rotate, | ||||
| 		maxbackup: 999, | ||||
| 	} | ||||
|  | ||||
| 	// open the file for the first time | ||||
| 	if err := w.intRotate(); err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err) | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	go func() { | ||||
| 		defer func() { | ||||
| 			if w.file != nil { | ||||
| 				fmt.Fprint(w.file, FormatLogRecord(w.trailer, &LogRecord{Created: time.Now()})) | ||||
| 				w.file.Close() | ||||
| 			} | ||||
| 		}() | ||||
|  | ||||
| 		for { | ||||
| 			select { | ||||
| 			case <-w.rot: | ||||
| 				if err := w.intRotate(); err != nil { | ||||
| 					fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err) | ||||
| 					return | ||||
| 				} | ||||
| 			case rec, ok := <-w.rec: | ||||
| 				if !ok { | ||||
| 					return | ||||
| 				} | ||||
| 				now := time.Now() | ||||
| 				if (w.maxlines > 0 && w.maxlines_curlines >= w.maxlines) || | ||||
| 					(w.maxsize > 0 && w.maxsize_cursize >= w.maxsize) || | ||||
| 					(w.daily && now.Day() != w.daily_opendate) { | ||||
| 					if err := w.intRotate(); err != nil { | ||||
| 						fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err) | ||||
| 						return | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 				// Perform the write | ||||
| 				n, err := fmt.Fprint(w.file, FormatLogRecord(w.format, rec)) | ||||
| 				if err != nil { | ||||
| 					fmt.Fprintf(os.Stderr, "FileLogWriter(%q): %s\n", w.filename, err) | ||||
| 					return | ||||
| 				} | ||||
|  | ||||
| 				// Update the counts | ||||
| 				w.maxlines_curlines++ | ||||
| 				w.maxsize_cursize += n | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Request that the logs rotate | ||||
| func (w *FileLogWriter) Rotate() { | ||||
| 	w.rot <- true | ||||
| } | ||||
|  | ||||
| // If this is called in a threaded context, it MUST be synchronized | ||||
| func (w *FileLogWriter) intRotate() error { | ||||
| 	// Close any log file that may be open | ||||
| 	if w.file != nil { | ||||
| 		fmt.Fprint(w.file, FormatLogRecord(w.trailer, &LogRecord{Created: time.Now()})) | ||||
| 		w.file.Close() | ||||
| 	} | ||||
|  | ||||
| 	// If we are keeping log files, move it to the next available number | ||||
| 	if w.rotate { | ||||
| 		_, err := os.Lstat(w.filename) | ||||
| 		if err == nil { // file exists | ||||
| 			// Find the next available number | ||||
| 			num := 1 | ||||
| 			fname := "" | ||||
| 			if w.daily && time.Now().Day() != w.daily_opendate { | ||||
| 				yesterday := time.Now().AddDate(0, 0, -1).Format("2006-01-02") | ||||
|  | ||||
| 				for ; err == nil && num <= 999; num++ { | ||||
| 					fname = w.filename + fmt.Sprintf(".%s.%03d", yesterday, num) | ||||
| 					_, err = os.Lstat(fname) | ||||
| 				} | ||||
| 				// return error if the last file checked still existed | ||||
| 				if err == nil { | ||||
| 					return fmt.Errorf("Rotate: Cannot find free log number to rename %s\n", w.filename) | ||||
| 				} | ||||
| 			} else { | ||||
| 				num = w.maxbackup - 1 | ||||
| 				for ; num >= 1; num-- { | ||||
| 					fname = w.filename + fmt.Sprintf(".%d", num) | ||||
| 					nfname := w.filename + fmt.Sprintf(".%d", num+1) | ||||
| 					_, err = os.Lstat(fname) | ||||
| 					if err == nil { | ||||
| 						os.Rename(fname, nfname) | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			w.file.Close() | ||||
| 			// Rename the file to its newfound home | ||||
| 			err = os.Rename(w.filename, fname) | ||||
| 			if err != nil { | ||||
| 				return fmt.Errorf("Rotate: %s\n", err) | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Open the log file | ||||
| 	fd, err := os.OpenFile(w.filename, os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0660) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	w.file = fd | ||||
|  | ||||
| 	now := time.Now() | ||||
| 	fmt.Fprint(w.file, FormatLogRecord(w.header, &LogRecord{Created: now})) | ||||
|  | ||||
| 	// Set the daily open date to the current date | ||||
| 	w.daily_opendate = now.Day() | ||||
|  | ||||
| 	// initialize rotation values | ||||
| 	w.maxlines_curlines = 0 | ||||
| 	w.maxsize_cursize = 0 | ||||
|  | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Set the logging format (chainable).  Must be called before the first log | ||||
| // message is written. | ||||
| func (w *FileLogWriter) SetFormat(format string) *FileLogWriter { | ||||
| 	w.format = format | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Set the logfile header and footer (chainable).  Must be called before the first log | ||||
| // message is written.  These are formatted similar to the FormatLogRecord (e.g. | ||||
| // you can use %D and %T in your header/footer for date and time). | ||||
| func (w *FileLogWriter) SetHeadFoot(head, foot string) *FileLogWriter { | ||||
| 	w.header, w.trailer = head, foot | ||||
| 	if w.maxlines_curlines == 0 { | ||||
| 		fmt.Fprint(w.file, FormatLogRecord(w.header, &LogRecord{Created: time.Now()})) | ||||
| 	} | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Set rotate at linecount (chainable). Must be called before the first log | ||||
| // message is written. | ||||
| func (w *FileLogWriter) SetRotateLines(maxlines int) *FileLogWriter { | ||||
| 	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateLines: %v\n", maxlines) | ||||
| 	w.maxlines = maxlines | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Set rotate at size (chainable). Must be called before the first log message | ||||
| // is written. | ||||
| func (w *FileLogWriter) SetRotateSize(maxsize int) *FileLogWriter { | ||||
| 	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateSize: %v\n", maxsize) | ||||
| 	w.maxsize = maxsize | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Set rotate daily (chainable). Must be called before the first log message is | ||||
| // written. | ||||
| func (w *FileLogWriter) SetRotateDaily(daily bool) *FileLogWriter { | ||||
| 	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotateDaily: %v\n", daily) | ||||
| 	w.daily = daily | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // Set max backup files. Must be called before the first log message | ||||
| // is written. | ||||
| func (w *FileLogWriter) SetRotateMaxBackup(maxbackup int) *FileLogWriter { | ||||
| 	w.maxbackup = maxbackup | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // SetRotate changes whether or not the old logs are kept. (chainable) Must be | ||||
| // called before the first log message is written.  If rotate is false, the | ||||
| // files are overwritten; otherwise, they are rotated to another file before the | ||||
| // new log is opened. | ||||
| func (w *FileLogWriter) SetRotate(rotate bool) *FileLogWriter { | ||||
| 	//fmt.Fprintf(os.Stderr, "FileLogWriter.SetRotate: %v\n", rotate) | ||||
| 	w.rotate = rotate | ||||
| 	return w | ||||
| } | ||||
|  | ||||
| // NewXMLLogWriter is a utility method for creating a FileLogWriter set up to | ||||
| // output XML record log messages instead of line-based ones. | ||||
| func NewXMLLogWriter(fname string, rotate bool) *FileLogWriter { | ||||
| 	return NewFileLogWriter(fname, rotate).SetFormat( | ||||
| 		`	<record level="%L"> | ||||
| 		<timestamp>%D %T</timestamp> | ||||
| 		<source>%S</source> | ||||
| 		<message>%M</message> | ||||
| 	</record>`).SetHeadFoot("<log created=\"%D %T\">", "</log>") | ||||
| } | ||||
							
								
								
									
										484
									
								
								vendor/github.com/alecthomas/log4go/log4go.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										484
									
								
								vendor/github.com/alecthomas/log4go/log4go.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,484 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| // Package log4go provides level-based and highly configurable logging. | ||||
| // | ||||
| // Enhanced Logging | ||||
| // | ||||
| // This is inspired by the logging functionality in Java.  Essentially, you create a Logger | ||||
| // object and create output filters for it.  You can send whatever you want to the Logger, | ||||
| // and it will filter that based on your settings and send it to the outputs.  This way, you | ||||
| // can put as much debug code in your program as you want, and when you're done you can filter | ||||
| // out the mundane messages so only the important ones show up. | ||||
| // | ||||
| // Utility functions are provided to make life easier. Here is some example code to get started: | ||||
| // | ||||
| // log := log4go.NewLogger() | ||||
| // log.AddFilter("stdout", log4go.DEBUG, log4go.NewConsoleLogWriter()) | ||||
| // log.AddFilter("log",    log4go.FINE,  log4go.NewFileLogWriter("example.log", true)) | ||||
| // log.Info("The time is now: %s", time.LocalTime().Format("15:04:05 MST 2006/01/02")) | ||||
| // | ||||
| // The first two lines can be combined with the utility NewDefaultLogger: | ||||
| // | ||||
| // log := log4go.NewDefaultLogger(log4go.DEBUG) | ||||
| // log.AddFilter("log",    log4go.FINE,  log4go.NewFileLogWriter("example.log", true)) | ||||
| // log.Info("The time is now: %s", time.LocalTime().Format("15:04:05 MST 2006/01/02")) | ||||
| // | ||||
| // Usage notes: | ||||
| // - The ConsoleLogWriter does not display the source of the message to standard | ||||
| //   output, but the FileLogWriter does. | ||||
| // - The utility functions (Info, Debug, Warn, etc) derive their source from the | ||||
| //   calling function, and this incurs extra overhead. | ||||
| // | ||||
| // Changes from 2.0: | ||||
| // - The external interface has remained mostly stable, but a lot of the | ||||
| //   internals have been changed, so if you depended on any of this or created | ||||
| //   your own LogWriter, then you will probably have to update your code.  In | ||||
| //   particular, Logger is now a map and ConsoleLogWriter is now a channel | ||||
| //   behind-the-scenes, and the LogWrite method no longer has return values. | ||||
| // | ||||
| // Future work: (please let me know if you think I should work on any of these particularly) | ||||
| // - Log file rotation | ||||
| // - Logging configuration files ala log4j | ||||
| // - Have the ability to remove filters? | ||||
| // - Have GetInfoChannel, GetDebugChannel, etc return a chan string that allows | ||||
| //   for another method of logging | ||||
| // - Add an XML filter type | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"runtime" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| // Version information | ||||
| const ( | ||||
| 	L4G_VERSION = "log4go-v3.0.1" | ||||
| 	L4G_MAJOR   = 3 | ||||
| 	L4G_MINOR   = 0 | ||||
| 	L4G_BUILD   = 1 | ||||
| ) | ||||
|  | ||||
| /****** Constants ******/ | ||||
|  | ||||
| // These are the integer logging levels used by the logger | ||||
| type Level int | ||||
|  | ||||
| const ( | ||||
| 	FINEST Level = iota | ||||
| 	FINE | ||||
| 	DEBUG | ||||
| 	TRACE | ||||
| 	INFO | ||||
| 	WARNING | ||||
| 	ERROR | ||||
| 	CRITICAL | ||||
| ) | ||||
|  | ||||
| // Logging level strings | ||||
| var ( | ||||
| 	levelStrings = [...]string{"FNST", "FINE", "DEBG", "TRAC", "INFO", "WARN", "EROR", "CRIT"} | ||||
| ) | ||||
|  | ||||
| func (l Level) String() string { | ||||
| 	if l < 0 || int(l) > len(levelStrings) { | ||||
| 		return "UNKNOWN" | ||||
| 	} | ||||
| 	return levelStrings[int(l)] | ||||
| } | ||||
|  | ||||
| /****** Variables ******/ | ||||
| var ( | ||||
| 	// LogBufferLength specifies how many log messages a particular log4go | ||||
| 	// logger can buffer at a time before writing them. | ||||
| 	LogBufferLength = 32 | ||||
| ) | ||||
|  | ||||
| /****** LogRecord ******/ | ||||
|  | ||||
| // A LogRecord contains all of the pertinent information for each message | ||||
| type LogRecord struct { | ||||
| 	Level   Level     // The log level | ||||
| 	Created time.Time // The time at which the log message was created (nanoseconds) | ||||
| 	Source  string    // The message source | ||||
| 	Message string    // The log message | ||||
| } | ||||
|  | ||||
| /****** LogWriter ******/ | ||||
|  | ||||
| // This is an interface for anything that should be able to write logs | ||||
| type LogWriter interface { | ||||
| 	// This will be called to log a LogRecord message. | ||||
| 	LogWrite(rec *LogRecord) | ||||
|  | ||||
| 	// This should clean up anything lingering about the LogWriter, as it is called before | ||||
| 	// the LogWriter is removed.  LogWrite should not be called after Close. | ||||
| 	Close() | ||||
| } | ||||
|  | ||||
| /****** Logger ******/ | ||||
|  | ||||
| // A Filter represents the log level below which no log records are written to | ||||
| // the associated LogWriter. | ||||
| type Filter struct { | ||||
| 	Level Level | ||||
| 	LogWriter | ||||
| } | ||||
|  | ||||
| // A Logger represents a collection of Filters through which log messages are | ||||
| // written. | ||||
| type Logger map[string]*Filter | ||||
|  | ||||
| // Create a new logger. | ||||
| // | ||||
| // DEPRECATED: Use make(Logger) instead. | ||||
| func NewLogger() Logger { | ||||
| 	os.Stderr.WriteString("warning: use of deprecated NewLogger\n") | ||||
| 	return make(Logger) | ||||
| } | ||||
|  | ||||
| // Create a new logger with a "stdout" filter configured to send log messages at | ||||
| // or above lvl to standard output. | ||||
| // | ||||
| // DEPRECATED: use NewDefaultLogger instead. | ||||
| func NewConsoleLogger(lvl Level) Logger { | ||||
| 	os.Stderr.WriteString("warning: use of deprecated NewConsoleLogger\n") | ||||
| 	return Logger{ | ||||
| 		"stdout": &Filter{lvl, NewConsoleLogWriter()}, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Create a new logger with a "stdout" filter configured to send log messages at | ||||
| // or above lvl to standard output. | ||||
| func NewDefaultLogger(lvl Level) Logger { | ||||
| 	return Logger{ | ||||
| 		"stdout": &Filter{lvl, NewConsoleLogWriter()}, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Closes all log writers in preparation for exiting the program or a | ||||
| // reconfiguration of logging.  Calling this is not really imperative, unless | ||||
| // you want to guarantee that all log messages are written.  Close removes | ||||
| // all filters (and thus all LogWriters) from the logger. | ||||
| func (log Logger) Close() { | ||||
| 	// Close all open loggers | ||||
| 	for name, filt := range log { | ||||
| 		filt.Close() | ||||
| 		delete(log, name) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Add a new LogWriter to the Logger which will only log messages at lvl or | ||||
| // higher.  This function should not be called from multiple goroutines. | ||||
| // Returns the logger for chaining. | ||||
| func (log Logger) AddFilter(name string, lvl Level, writer LogWriter) Logger { | ||||
| 	log[name] = &Filter{lvl, writer} | ||||
| 	return log | ||||
| } | ||||
|  | ||||
| /******* Logging *******/ | ||||
| // Send a formatted log message internally | ||||
| func (log Logger) intLogf(lvl Level, format string, args ...interface{}) { | ||||
| 	skip := true | ||||
|  | ||||
| 	// Determine if any logging will be done | ||||
| 	for _, filt := range log { | ||||
| 		if lvl >= filt.Level { | ||||
| 			skip = false | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	if skip { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Determine caller func | ||||
| 	pc, _, lineno, ok := runtime.Caller(2) | ||||
| 	src := "" | ||||
| 	if ok { | ||||
| 		src = fmt.Sprintf("%s:%d", runtime.FuncForPC(pc).Name(), lineno) | ||||
| 	} | ||||
|  | ||||
| 	msg := format | ||||
| 	if len(args) > 0 { | ||||
| 		msg = fmt.Sprintf(format, args...) | ||||
| 	} | ||||
|  | ||||
| 	// Make the log record | ||||
| 	rec := &LogRecord{ | ||||
| 		Level:   lvl, | ||||
| 		Created: time.Now(), | ||||
| 		Source:  src, | ||||
| 		Message: msg, | ||||
| 	} | ||||
|  | ||||
| 	// Dispatch the logs | ||||
| 	for _, filt := range log { | ||||
| 		if lvl < filt.Level { | ||||
| 			continue | ||||
| 		} | ||||
| 		filt.LogWrite(rec) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Send a closure log message internally | ||||
| func (log Logger) intLogc(lvl Level, closure func() string) { | ||||
| 	skip := true | ||||
|  | ||||
| 	// Determine if any logging will be done | ||||
| 	for _, filt := range log { | ||||
| 		if lvl >= filt.Level { | ||||
| 			skip = false | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	if skip { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Determine caller func | ||||
| 	pc, _, lineno, ok := runtime.Caller(2) | ||||
| 	src := "" | ||||
| 	if ok { | ||||
| 		src = fmt.Sprintf("%s:%d", runtime.FuncForPC(pc).Name(), lineno) | ||||
| 	} | ||||
|  | ||||
| 	// Make the log record | ||||
| 	rec := &LogRecord{ | ||||
| 		Level:   lvl, | ||||
| 		Created: time.Now(), | ||||
| 		Source:  src, | ||||
| 		Message: closure(), | ||||
| 	} | ||||
|  | ||||
| 	// Dispatch the logs | ||||
| 	for _, filt := range log { | ||||
| 		if lvl < filt.Level { | ||||
| 			continue | ||||
| 		} | ||||
| 		filt.LogWrite(rec) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Send a log message with manual level, source, and message. | ||||
| func (log Logger) Log(lvl Level, source, message string) { | ||||
| 	skip := true | ||||
|  | ||||
| 	// Determine if any logging will be done | ||||
| 	for _, filt := range log { | ||||
| 		if lvl >= filt.Level { | ||||
| 			skip = false | ||||
| 			break | ||||
| 		} | ||||
| 	} | ||||
| 	if skip { | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	// Make the log record | ||||
| 	rec := &LogRecord{ | ||||
| 		Level:   lvl, | ||||
| 		Created: time.Now(), | ||||
| 		Source:  source, | ||||
| 		Message: message, | ||||
| 	} | ||||
|  | ||||
| 	// Dispatch the logs | ||||
| 	for _, filt := range log { | ||||
| 		if lvl < filt.Level { | ||||
| 			continue | ||||
| 		} | ||||
| 		filt.LogWrite(rec) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Logf logs a formatted log message at the given log level, using the caller as | ||||
| // its source. | ||||
| func (log Logger) Logf(lvl Level, format string, args ...interface{}) { | ||||
| 	log.intLogf(lvl, format, args...) | ||||
| } | ||||
|  | ||||
| // Logc logs a string returned by the closure at the given log level, using the caller as | ||||
| // its source.  If no log message would be written, the closure is never called. | ||||
| func (log Logger) Logc(lvl Level, closure func() string) { | ||||
| 	log.intLogc(lvl, closure) | ||||
| } | ||||
|  | ||||
| // Finest logs a message at the finest log level. | ||||
| // See Debug for an explanation of the arguments. | ||||
| func (log Logger) Finest(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = FINEST | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		log.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		log.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Fine logs a message at the fine log level. | ||||
| // See Debug for an explanation of the arguments. | ||||
| func (log Logger) Fine(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = FINE | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		log.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		log.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Debug is a utility method for debug log messages. | ||||
| // The behavior of Debug depends on the first argument: | ||||
| // - arg0 is a string | ||||
| //   When given a string as the first argument, this behaves like Logf but with | ||||
| //   the DEBUG log level: the first argument is interpreted as a format for the | ||||
| //   latter arguments. | ||||
| // - arg0 is a func()string | ||||
| //   When given a closure of type func()string, this logs the string returned by | ||||
| //   the closure iff it will be logged.  The closure runs at most one time. | ||||
| // - arg0 is interface{} | ||||
| //   When given anything else, the log message will be each of the arguments | ||||
| //   formatted with %v and separated by spaces (ala Sprint). | ||||
| func (log Logger) Debug(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = DEBUG | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		log.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		log.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Trace logs a message at the trace log level. | ||||
| // See Debug for an explanation of the arguments. | ||||
| func (log Logger) Trace(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = TRACE | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		log.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		log.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Info logs a message at the info log level. | ||||
| // See Debug for an explanation of the arguments. | ||||
| func (log Logger) Info(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = INFO | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		log.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		log.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		log.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Warn logs a message at the warning log level and returns the formatted error. | ||||
| // At the warning level and higher, there is no performance benefit if the | ||||
| // message is not actually logged, because all formats are processed and all | ||||
| // closures are executed to format the error message. | ||||
| // See Debug for further explanation of the arguments. | ||||
| func (log Logger) Warn(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = WARNING | ||||
| 	) | ||||
| 	var msg string | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		msg = fmt.Sprintf(first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		msg = first() | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| 	log.intLogf(lvl, msg) | ||||
| 	return errors.New(msg) | ||||
| } | ||||
|  | ||||
| // Error logs a message at the error log level and returns the formatted error, | ||||
| // See Warn for an explanation of the performance and Debug for an explanation | ||||
| // of the parameters. | ||||
| func (log Logger) Error(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = ERROR | ||||
| 	) | ||||
| 	var msg string | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		msg = fmt.Sprintf(first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		msg = first() | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| 	log.intLogf(lvl, msg) | ||||
| 	return errors.New(msg) | ||||
| } | ||||
|  | ||||
| // Critical logs a message at the critical log level and returns the formatted error, | ||||
| // See Warn for an explanation of the performance and Debug for an explanation | ||||
| // of the parameters. | ||||
| func (log Logger) Critical(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = CRITICAL | ||||
| 	) | ||||
| 	var msg string | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		msg = fmt.Sprintf(first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		msg = first() | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		msg = fmt.Sprintf(fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| 	log.intLogf(lvl, msg) | ||||
| 	return errors.New(msg) | ||||
| } | ||||
							
								
								
									
										126
									
								
								vendor/github.com/alecthomas/log4go/pattlog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										126
									
								
								vendor/github.com/alecthomas/log4go/pattlog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,126 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	FORMAT_DEFAULT = "[%D %T] [%L] (%S) %M" | ||||
| 	FORMAT_SHORT   = "[%t %d] [%L] %M" | ||||
| 	FORMAT_ABBREV  = "[%L] %M" | ||||
| ) | ||||
|  | ||||
| type formatCacheType struct { | ||||
| 	LastUpdateSeconds    int64 | ||||
| 	shortTime, shortDate string | ||||
| 	longTime, longDate   string | ||||
| } | ||||
|  | ||||
| var formatCache = &formatCacheType{} | ||||
|  | ||||
| // Known format codes: | ||||
| // %T - Time (15:04:05 MST) | ||||
| // %t - Time (15:04) | ||||
| // %D - Date (2006/01/02) | ||||
| // %d - Date (01/02/06) | ||||
| // %L - Level (FNST, FINE, DEBG, TRAC, WARN, EROR, CRIT) | ||||
| // %S - Source | ||||
| // %M - Message | ||||
| // Ignores unknown formats | ||||
| // Recommended: "[%D %T] [%L] (%S) %M" | ||||
| func FormatLogRecord(format string, rec *LogRecord) string { | ||||
| 	if rec == nil { | ||||
| 		return "<nil>" | ||||
| 	} | ||||
| 	if len(format) == 0 { | ||||
| 		return "" | ||||
| 	} | ||||
|  | ||||
| 	out := bytes.NewBuffer(make([]byte, 0, 64)) | ||||
| 	secs := rec.Created.UnixNano() / 1e9 | ||||
|  | ||||
| 	cache := *formatCache | ||||
| 	if cache.LastUpdateSeconds != secs { | ||||
| 		month, day, year := rec.Created.Month(), rec.Created.Day(), rec.Created.Year() | ||||
| 		hour, minute, second := rec.Created.Hour(), rec.Created.Minute(), rec.Created.Second() | ||||
| 		zone, _ := rec.Created.Zone() | ||||
| 		updated := &formatCacheType{ | ||||
| 			LastUpdateSeconds: secs, | ||||
| 			shortTime:         fmt.Sprintf("%02d:%02d", hour, minute), | ||||
| 			shortDate:         fmt.Sprintf("%02d/%02d/%02d", day, month, year%100), | ||||
| 			longTime:          fmt.Sprintf("%02d:%02d:%02d %s", hour, minute, second, zone), | ||||
| 			longDate:          fmt.Sprintf("%04d/%02d/%02d", year, month, day), | ||||
| 		} | ||||
| 		cache = *updated | ||||
| 		formatCache = updated | ||||
| 	} | ||||
|  | ||||
| 	// Split the string into pieces by % signs | ||||
| 	pieces := bytes.Split([]byte(format), []byte{'%'}) | ||||
|  | ||||
| 	// Iterate over the pieces, replacing known formats | ||||
| 	for i, piece := range pieces { | ||||
| 		if i > 0 && len(piece) > 0 { | ||||
| 			switch piece[0] { | ||||
| 			case 'T': | ||||
| 				out.WriteString(cache.longTime) | ||||
| 			case 't': | ||||
| 				out.WriteString(cache.shortTime) | ||||
| 			case 'D': | ||||
| 				out.WriteString(cache.longDate) | ||||
| 			case 'd': | ||||
| 				out.WriteString(cache.shortDate) | ||||
| 			case 'L': | ||||
| 				out.WriteString(levelStrings[rec.Level]) | ||||
| 			case 'S': | ||||
| 				out.WriteString(rec.Source) | ||||
| 			case 's': | ||||
| 				slice := strings.Split(rec.Source, "/") | ||||
| 				out.WriteString(slice[len(slice)-1]) | ||||
| 			case 'M': | ||||
| 				out.WriteString(rec.Message) | ||||
| 			} | ||||
| 			if len(piece) > 1 { | ||||
| 				out.Write(piece[1:]) | ||||
| 			} | ||||
| 		} else if len(piece) > 0 { | ||||
| 			out.Write(piece) | ||||
| 		} | ||||
| 	} | ||||
| 	out.WriteByte('\n') | ||||
|  | ||||
| 	return out.String() | ||||
| } | ||||
|  | ||||
| // This is the standard writer that prints to standard output. | ||||
| type FormatLogWriter chan *LogRecord | ||||
|  | ||||
| // This creates a new FormatLogWriter | ||||
| func NewFormatLogWriter(out io.Writer, format string) FormatLogWriter { | ||||
| 	records := make(FormatLogWriter, LogBufferLength) | ||||
| 	go records.run(out, format) | ||||
| 	return records | ||||
| } | ||||
|  | ||||
| func (w FormatLogWriter) run(out io.Writer, format string) { | ||||
| 	for rec := range w { | ||||
| 		fmt.Fprint(out, FormatLogRecord(format, rec)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // This is the FormatLogWriter's output method.  This will block if the output | ||||
| // buffer is full. | ||||
| func (w FormatLogWriter) LogWrite(rec *LogRecord) { | ||||
| 	w <- rec | ||||
| } | ||||
|  | ||||
| // Close stops the logger from sending messages to standard output.  Attempts to | ||||
| // send log messages to this logger after a Close have undefined behavior. | ||||
| func (w FormatLogWriter) Close() { | ||||
| 	close(w) | ||||
| } | ||||
							
								
								
									
										57
									
								
								vendor/github.com/alecthomas/log4go/socklog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										57
									
								
								vendor/github.com/alecthomas/log4go/socklog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,57 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"encoding/json" | ||||
| 	"fmt" | ||||
| 	"net" | ||||
| 	"os" | ||||
| ) | ||||
|  | ||||
| // This log writer sends output to a socket | ||||
| type SocketLogWriter chan *LogRecord | ||||
|  | ||||
| // This is the SocketLogWriter's output method | ||||
| func (w SocketLogWriter) LogWrite(rec *LogRecord) { | ||||
| 	w <- rec | ||||
| } | ||||
|  | ||||
| func (w SocketLogWriter) Close() { | ||||
| 	close(w) | ||||
| } | ||||
|  | ||||
| func NewSocketLogWriter(proto, hostport string) SocketLogWriter { | ||||
| 	sock, err := net.Dial(proto, hostport) | ||||
| 	if err != nil { | ||||
| 		fmt.Fprintf(os.Stderr, "NewSocketLogWriter(%q): %s\n", hostport, err) | ||||
| 		return nil | ||||
| 	} | ||||
|  | ||||
| 	w := SocketLogWriter(make(chan *LogRecord, LogBufferLength)) | ||||
|  | ||||
| 	go func() { | ||||
| 		defer func() { | ||||
| 			if sock != nil && proto == "tcp" { | ||||
| 				sock.Close() | ||||
| 			} | ||||
| 		}() | ||||
|  | ||||
| 		for rec := range w { | ||||
| 			// Marshall into JSON | ||||
| 			js, err := json.Marshal(rec) | ||||
| 			if err != nil { | ||||
| 				fmt.Fprint(os.Stderr, "SocketLogWriter(%q): %s", hostport, err) | ||||
| 				return | ||||
| 			} | ||||
|  | ||||
| 			_, err = sock.Write(js) | ||||
| 			if err != nil { | ||||
| 				fmt.Fprint(os.Stderr, "SocketLogWriter(%q): %s", hostport, err) | ||||
| 				return | ||||
| 			} | ||||
| 		} | ||||
| 	}() | ||||
|  | ||||
| 	return w | ||||
| } | ||||
							
								
								
									
										49
									
								
								vendor/github.com/alecthomas/log4go/termlog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										49
									
								
								vendor/github.com/alecthomas/log4go/termlog.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,49 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"os" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| var stdout io.Writer = os.Stdout | ||||
|  | ||||
| // This is the standard writer that prints to standard output. | ||||
| type ConsoleLogWriter struct { | ||||
| 	format string | ||||
| 	w      chan *LogRecord | ||||
| } | ||||
|  | ||||
| // This creates a new ConsoleLogWriter | ||||
| func NewConsoleLogWriter() *ConsoleLogWriter { | ||||
| 	consoleWriter := &ConsoleLogWriter{ | ||||
| 		format: "[%T %D] [%L] (%S) %M", | ||||
| 		w:      make(chan *LogRecord, LogBufferLength), | ||||
| 	} | ||||
| 	go consoleWriter.run(stdout) | ||||
| 	return consoleWriter | ||||
| } | ||||
| func (c *ConsoleLogWriter) SetFormat(format string) { | ||||
| 	c.format = format | ||||
| } | ||||
| func (c *ConsoleLogWriter) run(out io.Writer) { | ||||
| 	for rec := range c.w { | ||||
| 		fmt.Fprint(out, FormatLogRecord(c.format, rec)) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // This is the ConsoleLogWriter's output method.  This will block if the output | ||||
| // buffer is full. | ||||
| func (c *ConsoleLogWriter) LogWrite(rec *LogRecord) { | ||||
| 	c.w <- rec | ||||
| } | ||||
|  | ||||
| // Close stops the logger from sending messages to standard output.  Attempts to | ||||
| // send log messages to this logger after a Close have undefined behavior. | ||||
| func (c *ConsoleLogWriter) Close() { | ||||
| 	close(c.w) | ||||
| 	time.Sleep(50 * time.Millisecond) // Try to give console I/O time to complete | ||||
| } | ||||
							
								
								
									
										278
									
								
								vendor/github.com/alecthomas/log4go/wrapper.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										278
									
								
								vendor/github.com/alecthomas/log4go/wrapper.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,278 +0,0 @@ | ||||
| // Copyright (C) 2010, Kyle Lemons <kyle@kylelemons.net>.  All rights reserved. | ||||
|  | ||||
| package log4go | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| var ( | ||||
| 	Global Logger | ||||
| ) | ||||
|  | ||||
| func init() { | ||||
| 	Global = NewDefaultLogger(DEBUG) | ||||
| } | ||||
|  | ||||
| // Wrapper for (*Logger).LoadConfiguration | ||||
| func LoadConfiguration(filename string) { | ||||
| 	Global.LoadConfiguration(filename) | ||||
| } | ||||
|  | ||||
| // Wrapper for (*Logger).AddFilter | ||||
| func AddFilter(name string, lvl Level, writer LogWriter) { | ||||
| 	Global.AddFilter(name, lvl, writer) | ||||
| } | ||||
|  | ||||
| // Wrapper for (*Logger).Close (closes and removes all logwriters) | ||||
| func Close() { | ||||
| 	Global.Close() | ||||
| } | ||||
|  | ||||
| func Crash(args ...interface{}) { | ||||
| 	if len(args) > 0 { | ||||
| 		Global.intLogf(CRITICAL, strings.Repeat(" %v", len(args))[1:], args...) | ||||
| 	} | ||||
| 	panic(args) | ||||
| } | ||||
|  | ||||
| // Logs the given message and crashes the program | ||||
| func Crashf(format string, args ...interface{}) { | ||||
| 	Global.intLogf(CRITICAL, format, args...) | ||||
| 	Global.Close() // so that hopefully the messages get logged | ||||
| 	panic(fmt.Sprintf(format, args...)) | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Exit(args ...interface{}) { | ||||
| 	if len(args) > 0 { | ||||
| 		Global.intLogf(ERROR, strings.Repeat(" %v", len(args))[1:], args...) | ||||
| 	} | ||||
| 	Global.Close() // so that hopefully the messages get logged | ||||
| 	os.Exit(0) | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Exitf(format string, args ...interface{}) { | ||||
| 	Global.intLogf(ERROR, format, args...) | ||||
| 	Global.Close() // so that hopefully the messages get logged | ||||
| 	os.Exit(0) | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Stderr(args ...interface{}) { | ||||
| 	if len(args) > 0 { | ||||
| 		Global.intLogf(ERROR, strings.Repeat(" %v", len(args))[1:], args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Stderrf(format string, args ...interface{}) { | ||||
| 	Global.intLogf(ERROR, format, args...) | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Stdout(args ...interface{}) { | ||||
| 	if len(args) > 0 { | ||||
| 		Global.intLogf(INFO, strings.Repeat(" %v", len(args))[1:], args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Compatibility with `log` | ||||
| func Stdoutf(format string, args ...interface{}) { | ||||
| 	Global.intLogf(INFO, format, args...) | ||||
| } | ||||
|  | ||||
| // Send a log message manually | ||||
| // Wrapper for (*Logger).Log | ||||
| func Log(lvl Level, source, message string) { | ||||
| 	Global.Log(lvl, source, message) | ||||
| } | ||||
|  | ||||
| // Send a formatted log message easily | ||||
| // Wrapper for (*Logger).Logf | ||||
| func Logf(lvl Level, format string, args ...interface{}) { | ||||
| 	Global.intLogf(lvl, format, args...) | ||||
| } | ||||
|  | ||||
| // Send a closure log message | ||||
| // Wrapper for (*Logger).Logc | ||||
| func Logc(lvl Level, closure func() string) { | ||||
| 	Global.intLogc(lvl, closure) | ||||
| } | ||||
|  | ||||
| // Utility for finest log messages (see Debug() for parameter explanation) | ||||
| // Wrapper for (*Logger).Finest | ||||
| func Finest(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = FINEST | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		Global.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Utility for fine log messages (see Debug() for parameter explanation) | ||||
| // Wrapper for (*Logger).Fine | ||||
| func Fine(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = FINE | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		Global.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Utility for debug log messages | ||||
| // When given a string as the first argument, this behaves like Logf but with the DEBUG log level (e.g. the first argument is interpreted as a format for the latter arguments) | ||||
| // When given a closure of type func()string, this logs the string returned by the closure iff it will be logged.  The closure runs at most one time. | ||||
| // When given anything else, the log message will be each of the arguments formatted with %v and separated by spaces (ala Sprint). | ||||
| // Wrapper for (*Logger).Debug | ||||
| func Debug(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = DEBUG | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		Global.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Utility for trace log messages (see Debug() for parameter explanation) | ||||
| // Wrapper for (*Logger).Trace | ||||
| func Trace(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = TRACE | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		Global.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Utility for info log messages (see Debug() for parameter explanation) | ||||
| // Wrapper for (*Logger).Info | ||||
| func Info(arg0 interface{}, args ...interface{}) { | ||||
| 	const ( | ||||
| 		lvl = INFO | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		Global.intLogc(lvl, first) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(arg0)+strings.Repeat(" %v", len(args)), args...) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Utility for warn log messages (returns an error for easy function returns) (see Debug() for parameter explanation) | ||||
| // These functions will execute a closure exactly once, to build the error message for the return | ||||
| // Wrapper for (*Logger).Warn | ||||
| func Warn(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = WARNING | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 		return errors.New(fmt.Sprintf(first, args...)) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		str := first() | ||||
| 		Global.intLogf(lvl, "%s", str) | ||||
| 		return errors.New(str) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Utility for error log messages (returns an error for easy function returns) (see Debug() for parameter explanation) | ||||
| // These functions will execute a closure exactly once, to build the error message for the return | ||||
| // Wrapper for (*Logger).Error | ||||
| func Error(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = ERROR | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 		return errors.New(fmt.Sprintf(first, args...)) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		str := first() | ||||
| 		Global.intLogf(lvl, "%s", str) | ||||
| 		return errors.New(str) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| // Utility for critical log messages (returns an error for easy function returns) (see Debug() for parameter explanation) | ||||
| // These functions will execute a closure exactly once, to build the error message for the return | ||||
| // Wrapper for (*Logger).Critical | ||||
| func Critical(arg0 interface{}, args ...interface{}) error { | ||||
| 	const ( | ||||
| 		lvl = CRITICAL | ||||
| 	) | ||||
| 	switch first := arg0.(type) { | ||||
| 	case string: | ||||
| 		// Use the string as a format string | ||||
| 		Global.intLogf(lvl, first, args...) | ||||
| 		return errors.New(fmt.Sprintf(first, args...)) | ||||
| 	case func() string: | ||||
| 		// Log the closure (no other arguments used) | ||||
| 		str := first() | ||||
| 		Global.intLogf(lvl, "%s", str) | ||||
| 		return errors.New(str) | ||||
| 	default: | ||||
| 		// Build a format string so that it will be similar to Sprint | ||||
| 		Global.intLogf(lvl, fmt.Sprint(first)+strings.Repeat(" %v", len(args)), args...) | ||||
| 		return errors.New(fmt.Sprint(first) + fmt.Sprintf(strings.Repeat(" %v", len(args)), args...)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										1
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/.gitignore
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,2 +1,3 @@ | ||||
| .idea/ | ||||
| coverage.out | ||||
| tmp/ | ||||
|   | ||||
							
								
								
									
										5
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/.travis.yml
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										5
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/.travis.yml
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,5 +1,6 @@ | ||||
| language: go | ||||
|  | ||||
| go: | ||||
|   - 1.4 | ||||
|   - tip | ||||
|   - '1.10' | ||||
|   - '1.11' | ||||
|   - tip | ||||
|   | ||||
							
								
								
									
										27
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										27
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,10 +3,6 @@ | ||||
| [](http://godoc.org/github.com/go-telegram-bot-api/telegram-bot-api) | ||||
| [](https://travis-ci.org/go-telegram-bot-api/telegram-bot-api) | ||||
|  | ||||
| All methods have been added, and all features should be available. | ||||
| If you want a feature that hasn't been added yet or something is broken, | ||||
| open an issue and I'll see what I can do. | ||||
|  | ||||
| All methods are fairly self explanatory, and reading the godoc page should | ||||
| explain everything. If something isn't clear, open an issue or submit | ||||
| a pull request. | ||||
| @@ -16,14 +12,14 @@ without any additional features. There are other projects for creating | ||||
| something with plugins and command handlers without having to design | ||||
| all that yourself. | ||||
|  | ||||
| Use `github.com/go-telegram-bot-api/telegram-bot-api` for the latest | ||||
| version, or use `gopkg.in/telegram-bot-api.v4` for the stable build. | ||||
|  | ||||
| Join [the development group](https://telegram.me/go_telegram_bot_api) if | ||||
| you want to ask questions or discuss development. | ||||
|  | ||||
| ## Example | ||||
|  | ||||
| First, ensure the library is installed and up to date by running | ||||
| `go get -u github.com/go-telegram-bot-api/telegram-bot-api`. | ||||
|  | ||||
| This is a very simple bot that just displays any gotten updates, | ||||
| then replies it to that chat. | ||||
|  | ||||
| @@ -32,7 +28,8 @@ package main | ||||
|  | ||||
| import ( | ||||
| 	"log" | ||||
| 	"gopkg.in/telegram-bot-api.v4" | ||||
|  | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| ) | ||||
|  | ||||
| func main() { | ||||
| @@ -51,7 +48,7 @@ func main() { | ||||
| 	updates, err := bot.GetUpdatesChan(u) | ||||
|  | ||||
| 	for update := range updates { | ||||
| 		if update.Message == nil { | ||||
| 		if update.Message == nil { // ignore any non-Message Updates | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| @@ -65,6 +62,11 @@ func main() { | ||||
| } | ||||
| ``` | ||||
|  | ||||
| There are more examples on the [wiki](https://github.com/go-telegram-bot-api/telegram-bot-api/wiki) | ||||
| with detailed information on how to do many differen kinds of things. | ||||
| It's a great place to get started on using keyboards, commands, or other | ||||
| kinds of reply markup. | ||||
|  | ||||
| If you need to use webhooks (if you wish to run on Google App Engine), | ||||
| you may use a slightly different method. | ||||
|  | ||||
| @@ -72,9 +74,10 @@ you may use a slightly different method. | ||||
| package main | ||||
|  | ||||
| import ( | ||||
| 	"gopkg.in/telegram-bot-api.v4" | ||||
| 	"log" | ||||
| 	"net/http" | ||||
|  | ||||
| 	"github.com/go-telegram-bot-api/telegram-bot-api" | ||||
| ) | ||||
|  | ||||
| func main() { | ||||
| @@ -96,7 +99,7 @@ func main() { | ||||
| 		log.Fatal(err) | ||||
| 	} | ||||
| 	if info.LastErrorDate != 0 { | ||||
| 		log.Printf("[Telegram callback failed]%s", info.LastErrorMessage) | ||||
| 		log.Printf("Telegram callback failed: %s", info.LastErrorMessage) | ||||
| 	} | ||||
| 	updates := bot.ListenForWebhook("/" + bot.Token) | ||||
| 	go http.ListenAndServeTLS("0.0.0.0:8443", "cert.pem", "key.pem", nil) | ||||
| @@ -114,5 +117,5 @@ properly signed. | ||||
|  | ||||
|     openssl req -x509 -newkey rsa:2048 -keyout key.pem -out cert.pem -days 3560 -subj "//O=Org\CN=Test" -nodes | ||||
|  | ||||
| Now that [Let's Encrypt](https://letsencrypt.org) has entered public beta, | ||||
| Now that [Let's Encrypt](https://letsencrypt.org) is available, | ||||
| you may wish to generate your free TLS certificate there. | ||||
|   | ||||
							
								
								
									
										18
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										18
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/bot.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -9,7 +9,6 @@ import ( | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"log" | ||||
| 	"net/http" | ||||
| 	"net/url" | ||||
| 	"os" | ||||
| @@ -28,6 +27,7 @@ type BotAPI struct { | ||||
|  | ||||
| 	Self   User         `json:"-"` | ||||
| 	Client *http.Client `json:"-"` | ||||
| 	shutdownChannel chan interface{} | ||||
| } | ||||
|  | ||||
| // NewBotAPI creates a new BotAPI instance. | ||||
| @@ -46,6 +46,7 @@ func NewBotAPIWithClient(token string, client *http.Client) (*BotAPI, error) { | ||||
| 		Token:  token, | ||||
| 		Client: client, | ||||
| 		Buffer: 100, | ||||
| 		shutdownChannel: make(chan interface{}), | ||||
| 	} | ||||
|  | ||||
| 	self, err := bot.GetMe() | ||||
| @@ -484,6 +485,12 @@ func (bot *BotAPI) GetUpdatesChan(config UpdateConfig) (UpdatesChannel, error) { | ||||
|  | ||||
| 	go func() { | ||||
| 		for { | ||||
| 			select { | ||||
| 			case <-bot.shutdownChannel: | ||||
| 				return | ||||
| 			default: | ||||
| 			} | ||||
| 			 | ||||
| 			updates, err := bot.GetUpdates(config) | ||||
| 			if err != nil { | ||||
| 				log.Println(err) | ||||
| @@ -505,12 +512,21 @@ func (bot *BotAPI) GetUpdatesChan(config UpdateConfig) (UpdatesChannel, error) { | ||||
| 	return ch, nil | ||||
| } | ||||
|  | ||||
| // StopReceivingUpdates stops the go routine which receives updates | ||||
| func (bot *BotAPI) StopReceivingUpdates() { | ||||
| 	if bot.Debug { | ||||
| 		log.Println("Stopping the update receiver routine...") | ||||
| 	} | ||||
| 	close(bot.shutdownChannel) | ||||
| } | ||||
|  | ||||
| // ListenForWebhook registers a http handler for a webhook. | ||||
| func (bot *BotAPI) ListenForWebhook(pattern string) UpdatesChannel { | ||||
| 	ch := make(chan Update, bot.Buffer) | ||||
|  | ||||
| 	http.HandleFunc(pattern, func(w http.ResponseWriter, r *http.Request) { | ||||
| 		bytes, _ := ioutil.ReadAll(r.Body) | ||||
| 		r.Body.Close() | ||||
|  | ||||
| 		var update Update | ||||
| 		json.Unmarshal(bytes, &update) | ||||
|   | ||||
							
								
								
									
										133
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										133
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/configs.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -243,7 +243,8 @@ func (config ForwardConfig) method() string { | ||||
| // PhotoConfig contains information about a SendPhoto request. | ||||
| type PhotoConfig struct { | ||||
| 	BaseFile | ||||
| 	Caption string | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| } | ||||
|  | ||||
| // Params returns a map[string]string representation of PhotoConfig. | ||||
| @@ -252,6 +253,9 @@ func (config PhotoConfig) params() (map[string]string, error) { | ||||
|  | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| @@ -267,7 +271,11 @@ func (config PhotoConfig) values() (url.Values, error) { | ||||
| 	v.Add(config.name(), config.FileID) | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| } | ||||
|  | ||||
| @@ -285,6 +293,7 @@ func (config PhotoConfig) method() string { | ||||
| type AudioConfig struct { | ||||
| 	BaseFile | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| 	Duration  int | ||||
| 	Performer string | ||||
| 	Title     string | ||||
| @@ -310,6 +319,9 @@ func (config AudioConfig) values() (url.Values, error) { | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| @@ -331,6 +343,9 @@ func (config AudioConfig) params() (map[string]string, error) { | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| @@ -349,7 +364,8 @@ func (config AudioConfig) method() string { | ||||
| // DocumentConfig contains information about a SendDocument request. | ||||
| type DocumentConfig struct { | ||||
| 	BaseFile | ||||
| 	Caption string | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| } | ||||
|  | ||||
| // values returns a url.Values representation of DocumentConfig. | ||||
| @@ -362,6 +378,9 @@ func (config DocumentConfig) values() (url.Values, error) { | ||||
| 	v.Add(config.name(), config.FileID) | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| @@ -373,6 +392,9 @@ func (config DocumentConfig) params() (map[string]string, error) { | ||||
|  | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| @@ -425,8 +447,9 @@ func (config StickerConfig) method() string { | ||||
| // VideoConfig contains information about a SendVideo request. | ||||
| type VideoConfig struct { | ||||
| 	BaseFile | ||||
| 	Duration int | ||||
| 	Caption  string | ||||
| 	Duration  int | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| } | ||||
|  | ||||
| // values returns a url.Values representation of VideoConfig. | ||||
| @@ -442,6 +465,9 @@ func (config VideoConfig) values() (url.Values, error) { | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| @@ -453,6 +479,9 @@ func (config VideoConfig) params() (map[string]string, error) { | ||||
|  | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| @@ -468,6 +497,59 @@ func (config VideoConfig) method() string { | ||||
| 	return "sendVideo" | ||||
| } | ||||
|  | ||||
| // AnimationConfig contains information about a SendAnimation request. | ||||
| type AnimationConfig struct { | ||||
| 	BaseFile | ||||
| 	Duration  int | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| } | ||||
|  | ||||
| // values returns a url.Values representation of AnimationConfig. | ||||
| func (config AnimationConfig) values() (url.Values, error) { | ||||
| 	v, err := config.BaseChat.values() | ||||
| 	if err != nil { | ||||
| 		return v, err | ||||
| 	} | ||||
|  | ||||
| 	v.Add(config.name(), config.FileID) | ||||
| 	if config.Duration != 0 { | ||||
| 		v.Add("duration", strconv.Itoa(config.Duration)) | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| } | ||||
|  | ||||
| // params returns a map[string]string representation of AnimationConfig. | ||||
| func (config AnimationConfig) params() (map[string]string, error) { | ||||
| 	params, _ := config.BaseFile.params() | ||||
|  | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| } | ||||
|  | ||||
| // name returns the field name for the Animation. | ||||
| func (config AnimationConfig) name() string { | ||||
| 	return "animation" | ||||
| } | ||||
|  | ||||
| // method returns Telegram API method name for sending Animation. | ||||
| func (config AnimationConfig) method() string { | ||||
| 	return "sendAnimation" | ||||
| } | ||||
|  | ||||
| // VideoNoteConfig contains information about a SendVideoNote request. | ||||
| type VideoNoteConfig struct { | ||||
| 	BaseFile | ||||
| @@ -522,8 +604,9 @@ func (config VideoNoteConfig) method() string { | ||||
| // VoiceConfig contains information about a SendVoice request. | ||||
| type VoiceConfig struct { | ||||
| 	BaseFile | ||||
| 	Caption  string | ||||
| 	Duration int | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| 	Duration  int | ||||
| } | ||||
|  | ||||
| // values returns a url.Values representation of VoiceConfig. | ||||
| @@ -539,6 +622,9 @@ func (config VoiceConfig) values() (url.Values, error) { | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		v.Add("caption", config.Caption) | ||||
| 		if config.ParseMode != "" { | ||||
| 			v.Add("parse_mode", config.ParseMode) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| @@ -553,6 +639,9 @@ func (config VoiceConfig) params() (map[string]string, error) { | ||||
| 	} | ||||
| 	if config.Caption != "" { | ||||
| 		params["caption"] = config.Caption | ||||
| 		if config.ParseMode != "" { | ||||
| 			params["parse_mode"] = config.ParseMode | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return params, nil | ||||
| @@ -568,6 +657,32 @@ func (config VoiceConfig) method() string { | ||||
| 	return "sendVoice" | ||||
| } | ||||
|  | ||||
| // MediaGroupConfig contains information about a sendMediaGroup request. | ||||
| type MediaGroupConfig struct { | ||||
| 	BaseChat | ||||
| 	InputMedia []interface{} | ||||
| } | ||||
|  | ||||
| func (config MediaGroupConfig) values() (url.Values, error) { | ||||
| 	v, err := config.BaseChat.values() | ||||
| 	if err != nil { | ||||
| 		return v, err | ||||
| 	} | ||||
|  | ||||
| 	data, err := json.Marshal(config.InputMedia) | ||||
| 	if err != nil { | ||||
| 		return v, err | ||||
| 	} | ||||
|  | ||||
| 	v.Add("media", string(data)) | ||||
|  | ||||
| 	return v, nil | ||||
| } | ||||
|  | ||||
| func (config MediaGroupConfig) method() string { | ||||
| 	return "sendMediaGroup" | ||||
| } | ||||
|  | ||||
| // LocationConfig contains information about a SendLocation request. | ||||
| type LocationConfig struct { | ||||
| 	BaseChat | ||||
| @@ -786,13 +901,17 @@ func (config EditMessageTextConfig) method() string { | ||||
| // EditMessageCaptionConfig allows you to modify the caption of a message. | ||||
| type EditMessageCaptionConfig struct { | ||||
| 	BaseEdit | ||||
| 	Caption string | ||||
| 	Caption   string | ||||
| 	ParseMode string | ||||
| } | ||||
|  | ||||
| func (config EditMessageCaptionConfig) values() (url.Values, error) { | ||||
| 	v, _ := config.BaseEdit.values() | ||||
|  | ||||
| 	v.Add("caption", config.Caption) | ||||
| 	if config.ParseMode != "" { | ||||
| 		v.Add("parse_mode", config.ParseMode) | ||||
| 	} | ||||
|  | ||||
| 	return v, nil | ||||
| } | ||||
|   | ||||
							
								
								
									
										128
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										128
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/helpers.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,7 +1,6 @@ | ||||
| package tgbotapi | ||||
|  | ||||
| import ( | ||||
| 	"log" | ||||
| 	"net/url" | ||||
| ) | ||||
|  | ||||
| @@ -14,11 +13,12 @@ func NewMessage(chatID int64, text string) MessageConfig { | ||||
| 			ChatID:           chatID, | ||||
| 			ReplyToMessageID: 0, | ||||
| 		}, | ||||
| 		Text: text, | ||||
| 		Text:                  text, | ||||
| 		DisableWebPagePreview: false, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewDeleteMessage creates a request to delete a message. | ||||
| func NewDeleteMessage(chatID int64, messageID int) DeleteMessageConfig { | ||||
| 	return DeleteMessageConfig{ | ||||
| 		ChatID:    chatID, | ||||
| @@ -201,6 +201,35 @@ func NewVideoShare(chatID int64, fileID string) VideoConfig { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewAnimationUpload creates a new animation uploader. | ||||
| // | ||||
| // chatID is where to send it, file is a string path to the file, | ||||
| // FileReader, or FileBytes. | ||||
| func NewAnimationUpload(chatID int64, file interface{}) AnimationConfig { | ||||
| 	return AnimationConfig{ | ||||
| 		BaseFile: BaseFile{ | ||||
| 			BaseChat:    BaseChat{ChatID: chatID}, | ||||
| 			File:        file, | ||||
| 			UseExisting: false, | ||||
| 		}, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewAnimationShare shares an existing animation. | ||||
| // You may use this to reshare an existing animation without reuploading it. | ||||
| // | ||||
| // chatID is where to send it, fileID is the ID of the animation | ||||
| // already uploaded. | ||||
| func NewAnimationShare(chatID int64, fileID string) AnimationConfig { | ||||
| 	return AnimationConfig{ | ||||
| 		BaseFile: BaseFile{ | ||||
| 			BaseChat:    BaseChat{ChatID: chatID}, | ||||
| 			FileID:      fileID, | ||||
| 			UseExisting: true, | ||||
| 		}, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewVideoNoteUpload creates a new video note uploader. | ||||
| // | ||||
| // chatID is where to send it, file is a string path to the file, | ||||
| @@ -261,6 +290,33 @@ func NewVoiceShare(chatID int64, fileID string) VoiceConfig { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewMediaGroup creates a new media group. Files should be an array of | ||||
| // two to ten InputMediaPhoto or InputMediaVideo. | ||||
| func NewMediaGroup(chatID int64, files []interface{}) MediaGroupConfig { | ||||
| 	return MediaGroupConfig{ | ||||
| 		BaseChat: BaseChat{ | ||||
| 			ChatID: chatID, | ||||
| 		}, | ||||
| 		InputMedia: files, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInputMediaPhoto creates a new InputMediaPhoto. | ||||
| func NewInputMediaPhoto(media string) InputMediaPhoto { | ||||
| 	return InputMediaPhoto{ | ||||
| 		Type:  "photo", | ||||
| 		Media: media, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInputMediaVideo creates a new InputMediaVideo. | ||||
| func NewInputMediaVideo(media string) InputMediaVideo { | ||||
| 	return InputMediaVideo{ | ||||
| 		Type:  "video", | ||||
| 		Media: media, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewContact allows you to send a shared contact. | ||||
| func NewContact(chatID int64, phoneNumber, firstName string) ContactConfig { | ||||
| 	return ContactConfig{ | ||||
| @@ -403,6 +459,15 @@ func NewInlineQueryResultGIF(id, url string) InlineQueryResultGIF { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedGIF create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedGIF(id, gifID string) InlineQueryResultCachedGIF { | ||||
| 	return InlineQueryResultCachedGIF{ | ||||
| 		Type:  "gif", | ||||
| 		ID:    id, | ||||
| 		GifID: gifID, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultMPEG4GIF creates a new inline query MPEG4 GIF. | ||||
| func NewInlineQueryResultMPEG4GIF(id, url string) InlineQueryResultMPEG4GIF { | ||||
| 	return InlineQueryResultMPEG4GIF{ | ||||
| @@ -412,6 +477,15 @@ func NewInlineQueryResultMPEG4GIF(id, url string) InlineQueryResultMPEG4GIF { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedPhoto create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedMPEG4GIF(id, MPEG4GifID string) InlineQueryResultCachedMpeg4Gif { | ||||
| 	return InlineQueryResultCachedMpeg4Gif{ | ||||
| 		Type:   "mpeg4_gif", | ||||
| 		ID:     id, | ||||
| 		MGifID: MPEG4GifID, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultPhoto creates a new inline query photo. | ||||
| func NewInlineQueryResultPhoto(id, url string) InlineQueryResultPhoto { | ||||
| 	return InlineQueryResultPhoto{ | ||||
| @@ -431,6 +505,15 @@ func NewInlineQueryResultPhotoWithThumb(id, url, thumb string) InlineQueryResult | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedPhoto create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedPhoto(id, photoID string) InlineQueryResultCachedPhoto { | ||||
| 	return InlineQueryResultCachedPhoto{ | ||||
| 		Type:    "photo", | ||||
| 		ID:      id, | ||||
| 		PhotoID: photoID, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultVideo creates a new inline query video. | ||||
| func NewInlineQueryResultVideo(id, url string) InlineQueryResultVideo { | ||||
| 	return InlineQueryResultVideo{ | ||||
| @@ -440,6 +523,16 @@ func NewInlineQueryResultVideo(id, url string) InlineQueryResultVideo { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedVideo create a new inline query with cached video. | ||||
| func NewInlineQueryResultCachedVideo(id, videoID, title string) InlineQueryResultCachedVideo { | ||||
| 	return InlineQueryResultCachedVideo{ | ||||
| 		Type:    "video", | ||||
| 		ID:      id, | ||||
| 		VideoID: videoID, | ||||
| 		Title:   title, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultAudio creates a new inline query audio. | ||||
| func NewInlineQueryResultAudio(id, url, title string) InlineQueryResultAudio { | ||||
| 	return InlineQueryResultAudio{ | ||||
| @@ -450,6 +543,15 @@ func NewInlineQueryResultAudio(id, url, title string) InlineQueryResultAudio { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedAudio create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedAudio(id, audioID string) InlineQueryResultCachedAudio { | ||||
| 	return InlineQueryResultCachedAudio{ | ||||
| 		Type:    "audio", | ||||
| 		ID:      id, | ||||
| 		AudioID: audioID, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultVoice creates a new inline query voice. | ||||
| func NewInlineQueryResultVoice(id, url, title string) InlineQueryResultVoice { | ||||
| 	return InlineQueryResultVoice{ | ||||
| @@ -460,6 +562,16 @@ func NewInlineQueryResultVoice(id, url, title string) InlineQueryResultVoice { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedVoice create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedVoice(id, voiceID, title string) InlineQueryResultCachedVoice { | ||||
| 	return InlineQueryResultCachedVoice{ | ||||
| 		Type:    "voice", | ||||
| 		ID:      id, | ||||
| 		VoiceID: voiceID, | ||||
| 		Title:   title, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultDocument creates a new inline query document. | ||||
| func NewInlineQueryResultDocument(id, url, title, mimeType string) InlineQueryResultDocument { | ||||
| 	return InlineQueryResultDocument{ | ||||
| @@ -471,6 +583,16 @@ func NewInlineQueryResultDocument(id, url, title, mimeType string) InlineQueryRe | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultCachedDocument create a new inline query with cached photo. | ||||
| func NewInlineQueryResultCachedDocument(id, documentID, title string) InlineQueryResultCachedDocument { | ||||
| 	return InlineQueryResultCachedDocument{ | ||||
| 		Type:       "document", | ||||
| 		ID:         id, | ||||
| 		DocumentID: documentID, | ||||
| 		Title:      title, | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // NewInlineQueryResultLocation creates a new inline query location. | ||||
| func NewInlineQueryResultLocation(id, title string, latitude, longitude float64) InlineQueryResultLocation { | ||||
| 	return InlineQueryResultLocation{ | ||||
| @@ -500,7 +622,7 @@ func NewEditMessageCaption(chatID int64, messageID int, caption string) EditMess | ||||
| 			ChatID:    chatID, | ||||
| 			MessageID: messageID, | ||||
| 		}, | ||||
| 		Caption: caption, | ||||
| 		Caption:   caption, | ||||
| 	} | ||||
| } | ||||
|  | ||||
|   | ||||
							
								
								
									
										27
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/log.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										27
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/log.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,27 @@ | ||||
| package tgbotapi | ||||
|  | ||||
| import ( | ||||
| 	"errors" | ||||
| 	stdlog "log" | ||||
| 	"os" | ||||
| ) | ||||
|  | ||||
| // BotLogger is an interface that represents the required methods to log data. | ||||
| // | ||||
| // Instead of requiring the standard logger, we can just specify the methods we | ||||
| // use and allow users to pass anything that implements these. | ||||
| type BotLogger interface { | ||||
| 	Println(v ...interface{}) | ||||
| 	Printf(format string, v ...interface{}) | ||||
| } | ||||
|  | ||||
| var log BotLogger = stdlog.New(os.Stderr, "", stdlog.LstdFlags) | ||||
|  | ||||
| // SetLogger specifies the logger that the package should use. | ||||
| func SetLogger(logger BotLogger) error { | ||||
| 	if logger == nil { | ||||
| 		return errors.New("logger is nil") | ||||
| 	} | ||||
| 	log = logger | ||||
| 	return nil | ||||
| } | ||||
							
								
								
									
										315
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/passport.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										315
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/passport.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,315 @@ | ||||
| package tgbotapi | ||||
|  | ||||
| // PassportRequestInfoConfig allows you to request passport info | ||||
| type PassportRequestInfoConfig struct { | ||||
| 	BotID     int            `json:"bot_id"` | ||||
| 	Scope     *PassportScope `json:"scope"` | ||||
| 	Nonce     string         `json:"nonce"` | ||||
| 	PublicKey string         `json:"public_key"` | ||||
| } | ||||
|  | ||||
| // PassportScopeElement supports using one or one of several elements. | ||||
| type PassportScopeElement interface { | ||||
| 	ScopeType() string | ||||
| } | ||||
|  | ||||
| // PassportScope is the requested scopes of data. | ||||
| type PassportScope struct { | ||||
| 	V    int                    `json:"v"` | ||||
| 	Data []PassportScopeElement `json:"data"` | ||||
| } | ||||
|  | ||||
| // PassportScopeElementOneOfSeveral allows you to request any one of the | ||||
| // requested documents. | ||||
| type PassportScopeElementOneOfSeveral struct { | ||||
| } | ||||
|  | ||||
| // ScopeType is the scope type. | ||||
| func (eo *PassportScopeElementOneOfSeveral) ScopeType() string { | ||||
| 	return "one_of" | ||||
| } | ||||
|  | ||||
| // PassportScopeElementOne requires the specified element be provided. | ||||
| type PassportScopeElementOne struct { | ||||
| 	Type        string `json:"type"` // One of “personal_details”, “passport”, “driver_license”, “identity_card”, “internal_passport”, “address”, “utility_bill”, “bank_statement”, “rental_agreement”, “passport_registration”, “temporary_registration”, “phone_number”, “email” | ||||
| 	Selfie      bool   `json:"selfie"` | ||||
| 	Translation bool   `json:"translation"` | ||||
| 	NativeNames bool   `json:"native_name"` | ||||
| } | ||||
|  | ||||
| // ScopeType is the scope type. | ||||
| func (eo *PassportScopeElementOne) ScopeType() string { | ||||
| 	return "one" | ||||
| } | ||||
|  | ||||
| type ( | ||||
| 	// PassportData contains information about Telegram Passport data shared with | ||||
| 	// the bot by the user. | ||||
| 	PassportData struct { | ||||
| 		// Array with information about documents and other Telegram Passport | ||||
| 		// elements that was shared with the bot | ||||
| 		Data []EncryptedPassportElement `json:"data"` | ||||
|  | ||||
| 		// Encrypted credentials required to decrypt the data | ||||
| 		Credentials *EncryptedCredentials `json:"credentials"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportFile represents a file uploaded to Telegram Passport. Currently all | ||||
| 	// Telegram Passport files are in JPEG format when decrypted and don't exceed | ||||
| 	// 10MB. | ||||
| 	PassportFile struct { | ||||
| 		// Unique identifier for this file | ||||
| 		FileID string `json:"file_id"` | ||||
|  | ||||
| 		// File size | ||||
| 		FileSize int `json:"file_size"` | ||||
|  | ||||
| 		// Unix time when the file was uploaded | ||||
| 		FileDate int64 `json:"file_date"` | ||||
| 	} | ||||
|  | ||||
| 	// EncryptedPassportElement contains information about documents or other | ||||
| 	// Telegram Passport elements shared with the bot by the user. | ||||
| 	EncryptedPassportElement struct { | ||||
| 		// Element type. | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Base64-encoded encrypted Telegram Passport element data provided by | ||||
| 		// the user, available for "personal_details", "passport", | ||||
| 		// "driver_license", "identity_card", "identity_passport" and "address" | ||||
| 		// types. Can be decrypted and verified using the accompanying | ||||
| 		// EncryptedCredentials. | ||||
| 		Data string `json:"data,omitempty"` | ||||
|  | ||||
| 		// User's verified phone number, available only for "phone_number" type | ||||
| 		PhoneNumber string `json:"phone_number,omitempty"` | ||||
|  | ||||
| 		// User's verified email address, available only for "email" type | ||||
| 		Email string `json:"email,omitempty"` | ||||
|  | ||||
| 		// Array of encrypted files with documents provided by the user, | ||||
| 		// available for "utility_bill", "bank_statement", "rental_agreement", | ||||
| 		// "passport_registration" and "temporary_registration" types. Files can | ||||
| 		// be decrypted and verified using the accompanying EncryptedCredentials. | ||||
| 		Files []PassportFile `json:"files,omitempty"` | ||||
|  | ||||
| 		// Encrypted file with the front side of the document, provided by the | ||||
| 		// user. Available for "passport", "driver_license", "identity_card" and | ||||
| 		// "internal_passport". The file can be decrypted and verified using the | ||||
| 		// accompanying EncryptedCredentials. | ||||
| 		FrontSide *PassportFile `json:"front_side,omitempty"` | ||||
|  | ||||
| 		// Encrypted file with the reverse side of the document, provided by the | ||||
| 		// user. Available for "driver_license" and "identity_card". The file can | ||||
| 		// be decrypted and verified using the accompanying EncryptedCredentials. | ||||
| 		ReverseSide *PassportFile `json:"reverse_side,omitempty"` | ||||
|  | ||||
| 		// Encrypted file with the selfie of the user holding a document, | ||||
| 		// provided by the user; available for "passport", "driver_license", | ||||
| 		// "identity_card" and "internal_passport". The file can be decrypted | ||||
| 		// and verified using the accompanying EncryptedCredentials. | ||||
| 		Selfie *PassportFile `json:"selfie,omitempty"` | ||||
| 	} | ||||
|  | ||||
| 	// EncryptedCredentials contains data required for decrypting and | ||||
| 	// authenticating EncryptedPassportElement. See the Telegram Passport | ||||
| 	// Documentation for a complete description of the data decryption and | ||||
| 	// authentication processes. | ||||
| 	EncryptedCredentials struct { | ||||
| 		// Base64-encoded encrypted JSON-serialized data with unique user's | ||||
| 		// payload, data hashes and secrets required for EncryptedPassportElement | ||||
| 		// decryption and authentication | ||||
| 		Data string `json:"data"` | ||||
|  | ||||
| 		// Base64-encoded data hash for data authentication | ||||
| 		Hash string `json:"hash"` | ||||
|  | ||||
| 		// Base64-encoded secret, encrypted with the bot's public RSA key, | ||||
| 		// required for data decryption | ||||
| 		Secret string `json:"secret"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementError represents an error in the Telegram Passport element | ||||
| 	// which was submitted that should be resolved by the user. | ||||
| 	PassportElementError interface{} | ||||
|  | ||||
| 	// PassportElementErrorDataField represents an issue in one of the data | ||||
| 	// fields that was provided by the user. The error is considered resolved | ||||
| 	// when the field's value changes. | ||||
| 	PassportElementErrorDataField struct { | ||||
| 		// Error source, must be data | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the error, one | ||||
| 		// of "personal_details", "passport", "driver_license", "identity_card", | ||||
| 		// "internal_passport", "address" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Name of the data field which has the error | ||||
| 		FieldName string `json:"field_name"` | ||||
|  | ||||
| 		// Base64-encoded data hash | ||||
| 		DataHash string `json:"data_hash"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementErrorFrontSide represents an issue with the front side of | ||||
| 	// a document. The error is considered resolved when the file with the front | ||||
| 	// side of the document changes. | ||||
| 	PassportElementErrorFrontSide struct { | ||||
| 		// Error source, must be front_side | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the issue, one | ||||
| 		// of "passport", "driver_license", "identity_card", "internal_passport" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Base64-encoded hash of the file with the front side of the document | ||||
| 		FileHash string `json:"file_hash"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementErrorReverseSide represents an issue with the reverse side | ||||
| 	// of a document. The error is considered resolved when the file with reverse | ||||
| 	// side of the document changes. | ||||
| 	PassportElementErrorReverseSide struct { | ||||
| 		// Error source, must be reverse_side | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the issue, one | ||||
| 		// of "driver_license", "identity_card" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Base64-encoded hash of the file with the reverse side of the document | ||||
| 		FileHash string `json:"file_hash"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementErrorSelfie represents an issue with the selfie with a | ||||
| 	// document. The error is considered resolved when the file with the selfie | ||||
| 	// changes. | ||||
| 	PassportElementErrorSelfie struct { | ||||
| 		// Error source, must be selfie | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the issue, one | ||||
| 		// of "passport", "driver_license", "identity_card", "internal_passport" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Base64-encoded hash of the file with the selfie | ||||
| 		FileHash string `json:"file_hash"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementErrorFile represents an issue with a document scan. The | ||||
| 	// error is considered resolved when the file with the document scan changes. | ||||
| 	PassportElementErrorFile struct { | ||||
| 		// Error source, must be file | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the issue, one | ||||
| 		// of "utility_bill", "bank_statement", "rental_agreement", | ||||
| 		// "passport_registration", "temporary_registration" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// Base64-encoded file hash | ||||
| 		FileHash string `json:"file_hash"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// PassportElementErrorFiles represents an issue with a list of scans. The | ||||
| 	// error is considered resolved when the list of files containing the scans | ||||
| 	// changes. | ||||
| 	PassportElementErrorFiles struct { | ||||
| 		// Error source, must be files | ||||
| 		Source string `json:"source"` | ||||
|  | ||||
| 		// The section of the user's Telegram Passport which has the issue, one | ||||
| 		// of "utility_bill", "bank_statement", "rental_agreement", | ||||
| 		// "passport_registration", "temporary_registration" | ||||
| 		Type string `json:"type"` | ||||
|  | ||||
| 		// List of base64-encoded file hashes | ||||
| 		FileHashes []string `json:"file_hashes"` | ||||
|  | ||||
| 		// Error message | ||||
| 		Message string `json:"message"` | ||||
| 	} | ||||
|  | ||||
| 	// Credentials contains encrypted data. | ||||
| 	Credentials struct { | ||||
| 		Data SecureData `json:"secure_data"` | ||||
| 		// Nonce the same nonce given in the request | ||||
| 		Nonce string `json:"nonce"` | ||||
| 	} | ||||
|  | ||||
| 	// SecureData is a map of the fields and their encrypted values. | ||||
| 	SecureData map[string]*SecureValue | ||||
| 	// PersonalDetails       *SecureValue `json:"personal_details"` | ||||
| 	// Passport              *SecureValue `json:"passport"` | ||||
| 	// InternalPassport      *SecureValue `json:"internal_passport"` | ||||
| 	// DriverLicense         *SecureValue `json:"driver_license"` | ||||
| 	// IdentityCard          *SecureValue `json:"identity_card"` | ||||
| 	// Address               *SecureValue `json:"address"` | ||||
| 	// UtilityBill           *SecureValue `json:"utility_bill"` | ||||
| 	// BankStatement         *SecureValue `json:"bank_statement"` | ||||
| 	// RentalAgreement       *SecureValue `json:"rental_agreement"` | ||||
| 	// PassportRegistration  *SecureValue `json:"passport_registration"` | ||||
| 	// TemporaryRegistration *SecureValue `json:"temporary_registration"` | ||||
|  | ||||
| 	// SecureValue contains encrypted values for a SecureData item. | ||||
| 	SecureValue struct { | ||||
| 		Data        *DataCredentials   `json:"data"` | ||||
| 		FrontSide   *FileCredentials   `json:"front_side"` | ||||
| 		ReverseSide *FileCredentials   `json:"reverse_side"` | ||||
| 		Selfie      *FileCredentials   `json:"selfie"` | ||||
| 		Translation []*FileCredentials `json:"translation"` | ||||
| 		Files       []*FileCredentials `json:"files"` | ||||
| 	} | ||||
|  | ||||
| 	// DataCredentials contains information required to decrypt data. | ||||
| 	DataCredentials struct { | ||||
| 		// DataHash checksum of encrypted data | ||||
| 		DataHash string `json:"data_hash"` | ||||
| 		// Secret of encrypted data | ||||
| 		Secret string `json:"secret"` | ||||
| 	} | ||||
|  | ||||
| 	// FileCredentials contains information required to decrypt files. | ||||
| 	FileCredentials struct { | ||||
| 		// FileHash checksum of encrypted data | ||||
| 		FileHash string `json:"file_hash"` | ||||
| 		// Secret of encrypted data | ||||
| 		Secret string `json:"secret"` | ||||
| 	} | ||||
|  | ||||
| 	// PersonalDetails https://core.telegram.org/passport#personaldetails | ||||
| 	PersonalDetails struct { | ||||
| 		FirstName            string `json:"first_name"` | ||||
| 		LastName             string `json:"last_name"` | ||||
| 		MiddleName           string `json:"middle_name"` | ||||
| 		BirthDate            string `json:"birth_date"` | ||||
| 		Gender               string `json:"gender"` | ||||
| 		CountryCode          string `json:"country_code"` | ||||
| 		ResidenceCountryCode string `json:"residence_country_code"` | ||||
| 		FirstNameNative      string `json:"first_name_native"` | ||||
| 		LastNameNative       string `json:"last_name_native"` | ||||
| 		MiddleNameNative     string `json:"middle_name_native"` | ||||
| 	} | ||||
|  | ||||
| 	// IDDocumentData https://core.telegram.org/passport#iddocumentdata | ||||
| 	IDDocumentData struct { | ||||
| 		DocumentNumber string `json:"document_no"` | ||||
| 		ExpiryDate     string `json:"expiry_date"` | ||||
| 	} | ||||
| ) | ||||
							
								
								
									
										122
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										122
									
								
								vendor/github.com/go-telegram-bot-api/telegram-bot-api/types.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -144,6 +144,7 @@ type Message struct { | ||||
| 	Entities              *[]MessageEntity   `json:"entities"`                // optional | ||||
| 	Audio                 *Audio             `json:"audio"`                   // optional | ||||
| 	Document              *Document          `json:"document"`                // optional | ||||
| 	Animation             *ChatAnimation     `json:"animation"`               // optional | ||||
| 	Game                  *Game              `json:"game"`                    // optional | ||||
| 	Photo                 *[]PhotoSize       `json:"photo"`                   // optional | ||||
| 	Sticker               *Sticker           `json:"sticker"`                 // optional | ||||
| @@ -167,6 +168,7 @@ type Message struct { | ||||
| 	PinnedMessage         *Message           `json:"pinned_message"`          // optional | ||||
| 	Invoice               *Invoice           `json:"invoice"`                 // optional | ||||
| 	SuccessfulPayment     *SuccessfulPayment `json:"successful_payment"`      // optional | ||||
| 	PassportData          *PassportData      `json:"passport_data,omitempty"` // optional | ||||
| } | ||||
|  | ||||
| // Time converts the message timestamp into a Time. | ||||
| @@ -293,6 +295,18 @@ type Sticker struct { | ||||
| 	SetName   string     `json:"set_name"`  // optional | ||||
| } | ||||
|  | ||||
| // ChatAnimation contains information about an animation. | ||||
| type ChatAnimation struct { | ||||
| 	FileID    string     `json:"file_id"` | ||||
| 	Width     int        `json:"width"` | ||||
| 	Height    int        `json:"height"` | ||||
| 	Duration  int        `json:"duration"` | ||||
| 	Thumbnail *PhotoSize `json:"thumb"`     // optional | ||||
| 	FileName  string     `json:"file_name"` // optional | ||||
| 	MimeType  string     `json:"mime_type"` // optional | ||||
| 	FileSize  int        `json:"file_size"` // optional | ||||
| } | ||||
|  | ||||
| // Video contains information about a video. | ||||
| type Video struct { | ||||
| 	FileID    string     `json:"file_id"` | ||||
| @@ -511,6 +525,27 @@ func (info WebhookInfo) IsSet() bool { | ||||
| 	return info.URL != "" | ||||
| } | ||||
|  | ||||
| // InputMediaPhoto contains a photo for displaying as part of a media group. | ||||
| type InputMediaPhoto struct { | ||||
| 	Type      string `json:"type"` | ||||
| 	Media     string `json:"media"` | ||||
| 	Caption   string `json:"caption"` | ||||
| 	ParseMode string `json:"parse_mode"` | ||||
| } | ||||
|  | ||||
| // InputMediaVideo contains a video for displaying as part of a media group. | ||||
| type InputMediaVideo struct { | ||||
| 	Type  string `json:"type"` | ||||
| 	Media string `json:"media"` | ||||
| 	// thumb intentionally missing as it is not currently compatible | ||||
| 	Caption           string `json:"caption"` | ||||
| 	ParseMode         string `json:"parse_mode"` | ||||
| 	Width             int    `json:"width"` | ||||
| 	Height            int    `json:"height"` | ||||
| 	Duration          int    `json:"duration"` | ||||
| 	SupportsStreaming bool   `json:"supports_streaming"` | ||||
| } | ||||
|  | ||||
| // InlineQuery is a Query from Telegram for an inline request. | ||||
| type InlineQuery struct { | ||||
| 	ID       string    `json:"id"` | ||||
| @@ -551,6 +586,19 @@ type InlineQueryResultPhoto struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedPhoto is an inline query response with cached photo. | ||||
| type InlineQueryResultCachedPhoto struct { | ||||
| 	Type                string                `json:"type"`          // required | ||||
| 	ID                  string                `json:"id"`            // required | ||||
| 	PhotoID             string                `json:"photo_file_id"` // required | ||||
| 	Title               string                `json:"title"` | ||||
| 	Description         string                `json:"description"` | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultGIF is an inline query response GIF. | ||||
| type InlineQueryResultGIF struct { | ||||
| 	Type                string                `json:"type"`    // required | ||||
| @@ -566,6 +614,18 @@ type InlineQueryResultGIF struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedGIF is an inline query response with cached gif. | ||||
| type InlineQueryResultCachedGIF struct { | ||||
| 	Type                string                `json:"type"`        // required | ||||
| 	ID                  string                `json:"id"`          // required | ||||
| 	GifID               string                `json:"gif_file_id"` // required | ||||
| 	Title               string                `json:"title"` | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultMPEG4GIF is an inline query response MPEG4 GIF. | ||||
| type InlineQueryResultMPEG4GIF struct { | ||||
| 	Type                string                `json:"type"`      // required | ||||
| @@ -581,6 +641,19 @@ type InlineQueryResultMPEG4GIF struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedMpeg4Gif is an inline query response with cached | ||||
| // H.264/MPEG-4 AVC video without sound gif. | ||||
| type InlineQueryResultCachedMpeg4Gif struct { | ||||
| 	Type                string                `json:"type"`          // required | ||||
| 	ID                  string                `json:"id"`            // required | ||||
| 	MGifID              string                `json:"mpeg4_file_id"` // required | ||||
| 	Title               string                `json:"title"` | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultVideo is an inline query response video. | ||||
| type InlineQueryResultVideo struct { | ||||
| 	Type                string                `json:"type"`      // required | ||||
| @@ -598,6 +671,19 @@ type InlineQueryResultVideo struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedVideo is an inline query response with cached video. | ||||
| type InlineQueryResultCachedVideo struct { | ||||
| 	Type                string                `json:"type"`          // required | ||||
| 	ID                  string                `json:"id"`            // required | ||||
| 	VideoID             string                `json:"video_file_id"` // required | ||||
| 	Title               string                `json:"title"`         // required | ||||
| 	Description         string                `json:"description"` | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultAudio is an inline query response audio. | ||||
| type InlineQueryResultAudio struct { | ||||
| 	Type                string                `json:"type"`      // required | ||||
| @@ -611,6 +697,17 @@ type InlineQueryResultAudio struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedAudio is an inline query response with cached audio. | ||||
| type InlineQueryResultCachedAudio struct { | ||||
| 	Type                string                `json:"type"`          // required | ||||
| 	ID                  string                `json:"id"`            // required | ||||
| 	AudioID             string                `json:"audio_file_id"` // required | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultVoice is an inline query response voice. | ||||
| type InlineQueryResultVoice struct { | ||||
| 	Type                string                `json:"type"`      // required | ||||
| @@ -623,6 +720,18 @@ type InlineQueryResultVoice struct { | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedVoice is an inline query response with cached voice. | ||||
| type InlineQueryResultCachedVoice struct { | ||||
| 	Type                string                `json:"type"`          // required | ||||
| 	ID                  string                `json:"id"`            // required | ||||
| 	VoiceID             string                `json:"voice_file_id"` // required | ||||
| 	Title               string                `json:"title"`         // required | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultDocument is an inline query response document. | ||||
| type InlineQueryResultDocument struct { | ||||
| 	Type                string                `json:"type"`  // required | ||||
| @@ -639,6 +748,19 @@ type InlineQueryResultDocument struct { | ||||
| 	ThumbHeight         int                   `json:"thumb_height"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultCachedDocument is an inline query response with cached document. | ||||
| type InlineQueryResultCachedDocument struct { | ||||
| 	Type                string                `json:"type"`             // required | ||||
| 	ID                  string                `json:"id"`               // required | ||||
| 	DocumentID          string                `json:"document_file_id"` // required | ||||
| 	Title               string                `json:"title"`            // required | ||||
| 	Caption             string                `json:"caption"` | ||||
| 	Description         string                `json:"description"` | ||||
| 	ParseMode           string                `json:"parse_mode"` | ||||
| 	ReplyMarkup         *InlineKeyboardMarkup `json:"reply_markup,omitempty"` | ||||
| 	InputMessageContent interface{}           `json:"input_message_content,omitempty"` | ||||
| } | ||||
|  | ||||
| // InlineQueryResultLocation is an inline query response location. | ||||
| type InlineQueryResultLocation struct { | ||||
| 	Type                string                `json:"type"`      // required | ||||
|   | ||||
							
								
								
									
										57
									
								
								vendor/github.com/google/gops/agent/agent.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										57
									
								
								vendor/github.com/google/gops/agent/agent.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -7,6 +7,8 @@ | ||||
| package agent | ||||
|  | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"encoding/binary" | ||||
| 	"fmt" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| @@ -14,14 +16,13 @@ import ( | ||||
| 	"os" | ||||
| 	gosignal "os/signal" | ||||
| 	"runtime" | ||||
| 	"runtime/debug" | ||||
| 	"runtime/pprof" | ||||
| 	"runtime/trace" | ||||
| 	"strconv" | ||||
| 	"sync" | ||||
| 	"time" | ||||
|  | ||||
| 	"bufio" | ||||
|  | ||||
| 	"github.com/google/gops/internal" | ||||
| 	"github.com/google/gops/signal" | ||||
| 	"github.com/kardianos/osext" | ||||
| @@ -43,10 +44,16 @@ type Options struct { | ||||
| 	// Optional. | ||||
| 	Addr string | ||||
|  | ||||
| 	// NoShutdownCleanup tells the agent not to automatically cleanup | ||||
| 	// resources if the running process receives an interrupt. | ||||
| 	// ConfigDir is the directory to store the configuration file, | ||||
| 	// PID of the gops process, filename, port as well as content. | ||||
| 	// Optional. | ||||
| 	NoShutdownCleanup bool | ||||
| 	ConfigDir string | ||||
|  | ||||
| 	// ShutdownCleanup automatically cleans up resources if the | ||||
| 	// running process receives an interrupt. Otherwise, users | ||||
| 	// can call Close before shutting down. | ||||
| 	// Optional. | ||||
| 	ShutdownCleanup bool | ||||
| } | ||||
|  | ||||
| // Listen starts the gops agent on a host process. Once agent started, users | ||||
| @@ -58,26 +65,29 @@ type Options struct { | ||||
| // Note: The agent exposes an endpoint via a TCP connection that can be used by | ||||
| // any program on the system. Review your security requirements before starting | ||||
| // the agent. | ||||
| func Listen(opts *Options) error { | ||||
| func Listen(opts Options) error { | ||||
| 	mu.Lock() | ||||
| 	defer mu.Unlock() | ||||
|  | ||||
| 	if opts == nil { | ||||
| 		opts = &Options{} | ||||
| 	} | ||||
| 	if portfile != "" { | ||||
| 		return fmt.Errorf("gops: agent already listening at: %v", listener.Addr()) | ||||
| 	} | ||||
|  | ||||
| 	gopsdir, err := internal.ConfigDir() | ||||
| 	// new | ||||
| 	gopsdir := opts.ConfigDir | ||||
| 	if gopsdir == "" { | ||||
| 		cfgDir, err := internal.ConfigDir() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		gopsdir = cfgDir | ||||
| 	} | ||||
|  | ||||
| 	err := os.MkdirAll(gopsdir, os.ModePerm) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	err = os.MkdirAll(gopsdir, os.ModePerm) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	if !opts.NoShutdownCleanup { | ||||
| 	if opts.ShutdownCleanup { | ||||
| 		gracefulShutdown() | ||||
| 	} | ||||
|  | ||||
| @@ -165,7 +175,7 @@ func formatBytes(val uint64) string { | ||||
| 	return fmt.Sprintf("%d bytes", val) | ||||
| } | ||||
|  | ||||
| func handle(conn io.Writer, msg []byte) error { | ||||
| func handle(conn io.ReadWriter, msg []byte) error { | ||||
| 	switch msg[0] { | ||||
| 	case signal.StackTrace: | ||||
| 		return pprof.Lookup("goroutine").WriteTo(conn, 2) | ||||
| @@ -190,13 +200,20 @@ func handle(conn io.Writer, msg []byte) error { | ||||
| 		fmt.Fprintf(conn, "heap-objects: %v\n", s.HeapObjects) | ||||
| 		fmt.Fprintf(conn, "stack-in-use: %v\n", formatBytes(s.StackInuse)) | ||||
| 		fmt.Fprintf(conn, "stack-sys: %v\n", formatBytes(s.StackSys)) | ||||
| 		fmt.Fprintf(conn, "stack-mspan-inuse: %v\n", formatBytes(s.MSpanInuse)) | ||||
| 		fmt.Fprintf(conn, "stack-mspan-sys: %v\n", formatBytes(s.MSpanSys)) | ||||
| 		fmt.Fprintf(conn, "stack-mcache-inuse: %v\n", formatBytes(s.MCacheInuse)) | ||||
| 		fmt.Fprintf(conn, "stack-mcache-sys: %v\n", formatBytes(s.MCacheSys)) | ||||
| 		fmt.Fprintf(conn, "other-sys: %v\n", formatBytes(s.OtherSys)) | ||||
| 		fmt.Fprintf(conn, "gc-sys: %v\n", formatBytes(s.GCSys)) | ||||
| 		fmt.Fprintf(conn, "next-gc: when heap-alloc >= %v\n", formatBytes(s.NextGC)) | ||||
| 		lastGC := "-" | ||||
| 		if s.LastGC != 0 { | ||||
| 			lastGC = fmt.Sprint(time.Unix(0, int64(s.LastGC))) | ||||
| 		} | ||||
| 		fmt.Fprintf(conn, "last-gc: %v\n", lastGC) | ||||
| 		fmt.Fprintf(conn, "gc-pause: %v\n", time.Duration(s.PauseTotalNs)) | ||||
| 		fmt.Fprintf(conn, "gc-pause-total: %v\n", time.Duration(s.PauseTotalNs)) | ||||
| 		fmt.Fprintf(conn, "gc-pause: %v\n", s.PauseNs[(s.NumGC+255)%256]) | ||||
| 		fmt.Fprintf(conn, "num-gc: %v\n", s.NumGC) | ||||
| 		fmt.Fprintf(conn, "enable-gc: %v\n", s.EnableGC) | ||||
| 		fmt.Fprintf(conn, "debug-gc: %v\n", s.DebugGC) | ||||
| @@ -232,6 +249,12 @@ func handle(conn io.Writer, msg []byte) error { | ||||
| 		trace.Start(conn) | ||||
| 		time.Sleep(5 * time.Second) | ||||
| 		trace.Stop() | ||||
| 	case signal.SetGCPercent: | ||||
| 		perc, err := binary.ReadVarint(bufio.NewReader(conn)) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		fmt.Fprintf(conn, "New GC percent set to %v. Previous value was %v.\n", perc, debug.SetGCPercent(int(perc))) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|   | ||||
							
								
								
									
										10
									
								
								vendor/github.com/google/gops/internal/internal.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										10
									
								
								vendor/github.com/google/gops/internal/internal.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,3 +1,7 @@ | ||||
| // Copyright 2017 The Go Authors. All rights reserved. | ||||
| // Use of this source code is governed by a BSD-style | ||||
| // license that can be found in the LICENSE file. | ||||
|  | ||||
| package internal | ||||
|  | ||||
| import ( | ||||
| @@ -11,7 +15,13 @@ import ( | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| const gopsConfigDirEnvKey = "GOPS_CONFIG_DIR" | ||||
|  | ||||
| func ConfigDir() (string, error) { | ||||
| 	if configDir := os.Getenv(gopsConfigDirEnvKey); configDir != "" { | ||||
| 		return configDir, nil | ||||
| 	} | ||||
|  | ||||
| 	if runtime.GOOS == "windows" { | ||||
| 		return filepath.Join(os.Getenv("APPDATA"), "gops"), nil | ||||
| 	} | ||||
|   | ||||
							
								
								
									
										3
									
								
								vendor/github.com/google/gops/signal/signal.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								vendor/github.com/google/gops/signal/signal.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -32,4 +32,7 @@ const ( | ||||
|  | ||||
| 	// BinaryDump returns running binary file. | ||||
| 	BinaryDump = byte(0x9) | ||||
|  | ||||
| 	// SetGCPercent sets the garbage collection target percentage. | ||||
| 	SetGCPercent = byte(0x10) | ||||
| ) | ||||
|   | ||||
							
								
								
									
										54
									
								
								vendor/github.com/gorilla/schema/cache.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										54
									
								
								vendor/github.com/gorilla/schema/cache.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -18,13 +18,9 @@ var invalidPath = errors.New("schema: invalid path") | ||||
| func newCache() *cache { | ||||
| 	c := cache{ | ||||
| 		m:       make(map[reflect.Type]*structInfo), | ||||
| 		conv:    make(map[reflect.Kind]Converter), | ||||
| 		regconv: make(map[reflect.Type]Converter), | ||||
| 		tag:     "schema", | ||||
| 	} | ||||
| 	for k, v := range converters { | ||||
| 		c.conv[k] = v | ||||
| 	} | ||||
| 	return &c | ||||
| } | ||||
|  | ||||
| @@ -32,11 +28,15 @@ func newCache() *cache { | ||||
| type cache struct { | ||||
| 	l       sync.RWMutex | ||||
| 	m       map[reflect.Type]*structInfo | ||||
| 	conv    map[reflect.Kind]Converter | ||||
| 	regconv map[reflect.Type]Converter | ||||
| 	tag     string | ||||
| } | ||||
|  | ||||
| // registerConverter registers a converter function for a custom type. | ||||
| func (c *cache) registerConverter(value interface{}, converterFunc Converter) { | ||||
| 	c.regconv[reflect.TypeOf(value)] = converterFunc | ||||
| } | ||||
|  | ||||
| // parsePath parses a path in dotted notation verifying that it is a valid | ||||
| // path to a struct field. | ||||
| // | ||||
| @@ -63,7 +63,7 @@ func (c *cache) parsePath(p string, t reflect.Type) ([]pathPart, error) { | ||||
| 		} | ||||
| 		// Valid field. Append index. | ||||
| 		path = append(path, field.name) | ||||
| 		if field.ss { | ||||
| 		if field.isSliceOfStructs && (!field.unmarshalerInfo.IsValid || (field.unmarshalerInfo.IsValid && field.unmarshalerInfo.IsSliceElement)) { | ||||
| 			// Parse a special case: slices of structs. | ||||
| 			// i+1 must be the slice index. | ||||
| 			// | ||||
| @@ -142,7 +142,7 @@ func (c *cache) create(t reflect.Type, info *structInfo) *structInfo { | ||||
| 				c.create(ft, info) | ||||
| 				for _, fi := range info.fields[bef:len(info.fields)] { | ||||
| 					// exclude required check because duplicated to embedded field | ||||
| 					fi.required = false | ||||
| 					fi.isRequired = false | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| @@ -162,6 +162,7 @@ func (c *cache) createField(field reflect.StructField, info *structInfo) { | ||||
| 	// First let's get the basic type. | ||||
| 	isSlice, isStruct := false, false | ||||
| 	ft := field.Type | ||||
| 	m := isTextUnmarshaler(reflect.Zero(ft)) | ||||
| 	if ft.Kind() == reflect.Ptr { | ||||
| 		ft = ft.Elem() | ||||
| 	} | ||||
| @@ -178,29 +179,26 @@ func (c *cache) createField(field reflect.StructField, info *structInfo) { | ||||
| 		} | ||||
| 	} | ||||
| 	if isStruct = ft.Kind() == reflect.Struct; !isStruct { | ||||
| 		if conv := c.converter(ft); conv == nil { | ||||
| 		if c.converter(ft) == nil && builtinConverters[ft.Kind()] == nil { | ||||
| 			// Type is not supported. | ||||
| 			return | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	info.fields = append(info.fields, &fieldInfo{ | ||||
| 		typ:      field.Type, | ||||
| 		name:     field.Name, | ||||
| 		ss:       isSlice && isStruct, | ||||
| 		alias:    alias, | ||||
| 		anon:     field.Anonymous, | ||||
| 		required: options.Contains("required"), | ||||
| 		typ:              field.Type, | ||||
| 		name:             field.Name, | ||||
| 		alias:            alias, | ||||
| 		unmarshalerInfo:  m, | ||||
| 		isSliceOfStructs: isSlice && isStruct, | ||||
| 		isAnonymous:      field.Anonymous, | ||||
| 		isRequired:       options.Contains("required"), | ||||
| 	}) | ||||
| } | ||||
|  | ||||
| // converter returns the converter for a type. | ||||
| func (c *cache) converter(t reflect.Type) Converter { | ||||
| 	conv := c.regconv[t] | ||||
| 	if conv == nil { | ||||
| 		conv = c.conv[t.Kind()] | ||||
| 	} | ||||
| 	return conv | ||||
| 	return c.regconv[t] | ||||
| } | ||||
|  | ||||
| // ---------------------------------------------------------------------------- | ||||
| @@ -219,12 +217,18 @@ func (i *structInfo) get(alias string) *fieldInfo { | ||||
| } | ||||
|  | ||||
| type fieldInfo struct { | ||||
| 	typ      reflect.Type | ||||
| 	name     string // field name in the struct. | ||||
| 	ss       bool   // true if this is a slice of structs. | ||||
| 	alias    string | ||||
| 	anon     bool // is an embedded field | ||||
| 	required bool // tag option | ||||
| 	typ reflect.Type | ||||
| 	// name is the field name in the struct. | ||||
| 	name  string | ||||
| 	alias string | ||||
| 	// unmarshalerInfo contains information regarding the | ||||
| 	// encoding.TextUnmarshaler implementation of the field type. | ||||
| 	unmarshalerInfo unmarshaler | ||||
| 	// isSliceOfStructs indicates if the field type is a slice of structs. | ||||
| 	isSliceOfStructs bool | ||||
| 	// isAnonymous indicates whether the field is embedded in the struct. | ||||
| 	isAnonymous bool | ||||
| 	isRequired  bool | ||||
| } | ||||
|  | ||||
| type pathPart struct { | ||||
|   | ||||
							
								
								
									
										2
									
								
								vendor/github.com/gorilla/schema/converter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/gorilla/schema/converter.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -30,7 +30,7 @@ var ( | ||||
| ) | ||||
|  | ||||
| // Default converters for basic types. | ||||
| var converters = map[reflect.Kind]Converter{ | ||||
| var builtinConverters = map[reflect.Kind]Converter{ | ||||
| 	boolType:    convertBool, | ||||
| 	float32Type: convertFloat32, | ||||
| 	float64Type: convertFloat64, | ||||
|   | ||||
							
								
								
									
										159
									
								
								vendor/github.com/gorilla/schema/decoder.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										159
									
								
								vendor/github.com/gorilla/schema/decoder.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -56,7 +56,7 @@ func (d *Decoder) IgnoreUnknownKeys(i bool) { | ||||
|  | ||||
| // RegisterConverter registers a converter function for a custom type. | ||||
| func (d *Decoder) RegisterConverter(value interface{}, converterFunc Converter) { | ||||
| 	d.cache.regconv[reflect.TypeOf(value)] = converterFunc | ||||
| 	d.cache.registerConverter(value, converterFunc) | ||||
| } | ||||
|  | ||||
| // Decode decodes a map[string][]string to a struct. | ||||
| @@ -90,7 +90,7 @@ func (d *Decoder) Decode(dst interface{}, src map[string][]string) error { | ||||
| 	return d.checkRequired(t, src, "") | ||||
| } | ||||
|  | ||||
| // checkRequired checks whether requred field empty | ||||
| // checkRequired checks whether required fields are empty | ||||
| // | ||||
| // check type t recursively if t has struct fields, and prefix is same as parsePath: in dotted notation | ||||
| // | ||||
| @@ -106,7 +106,7 @@ func (d *Decoder) checkRequired(t reflect.Type, src map[string][]string, prefix | ||||
| 		if f.typ.Kind() == reflect.Struct { | ||||
| 			err := d.checkRequired(f.typ, src, prefix+f.alias+".") | ||||
| 			if err != nil { | ||||
| 				if !f.anon { | ||||
| 				if !f.isAnonymous { | ||||
| 					return err | ||||
| 				} | ||||
| 				// check embedded parent field. | ||||
| @@ -116,7 +116,7 @@ func (d *Decoder) checkRequired(t reflect.Type, src map[string][]string, prefix | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
| 		if f.required { | ||||
| 		if f.isRequired { | ||||
| 			key := f.alias | ||||
| 			if prefix != "" { | ||||
| 				key = prefix + key | ||||
| @@ -153,7 +153,6 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
| 		} | ||||
| 		v = v.FieldByName(name) | ||||
| 	} | ||||
|  | ||||
| 	// Don't even bother for unexported fields. | ||||
| 	if !v.CanSet() { | ||||
| 		return nil | ||||
| @@ -185,7 +184,8 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
|  | ||||
| 	// Get the converter early in case there is one for a slice type. | ||||
| 	conv := d.cache.converter(t) | ||||
| 	if conv == nil && t.Kind() == reflect.Slice { | ||||
| 	m := isTextUnmarshaler(v) | ||||
| 	if conv == nil && t.Kind() == reflect.Slice && m.IsSliceElement { | ||||
| 		var items []reflect.Value | ||||
| 		elemT := t.Elem() | ||||
| 		isPtrElem := elemT.Kind() == reflect.Ptr | ||||
| @@ -196,9 +196,12 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
| 		// Try to get a converter for the element type. | ||||
| 		conv := d.cache.converter(elemT) | ||||
| 		if conv == nil { | ||||
| 			// As we are not dealing with slice of structs here, we don't need to check if the type | ||||
| 			// implements TextUnmarshaler interface | ||||
| 			return fmt.Errorf("schema: converter not found for %v", elemT) | ||||
| 			conv = builtinConverters[elemT.Kind()] | ||||
| 			if conv == nil { | ||||
| 				// As we are not dealing with slice of structs here, we don't need to check if the type | ||||
| 				// implements TextUnmarshaler interface | ||||
| 				return fmt.Errorf("schema: converter not found for %v", elemT) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		for key, value := range values { | ||||
| @@ -206,6 +209,26 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
| 				if d.zeroEmpty { | ||||
| 					items = append(items, reflect.Zero(elemT)) | ||||
| 				} | ||||
| 			} else if m.IsValid { | ||||
| 				u := reflect.New(elemT) | ||||
| 				if m.IsSliceElementPtr { | ||||
| 					u = reflect.New(reflect.PtrTo(elemT).Elem()) | ||||
| 				} | ||||
| 				if err := u.Interface().(encoding.TextUnmarshaler).UnmarshalText([]byte(value)); err != nil { | ||||
| 					return ConversionError{ | ||||
| 						Key:   path, | ||||
| 						Type:  t, | ||||
| 						Index: key, | ||||
| 						Err:   err, | ||||
| 					} | ||||
| 				} | ||||
| 				if m.IsSliceElementPtr { | ||||
| 					items = append(items, u.Elem().Addr()) | ||||
| 				} else if u.Kind() == reflect.Ptr { | ||||
| 					items = append(items, u.Elem()) | ||||
| 				} else { | ||||
| 					items = append(items, u) | ||||
| 				} | ||||
| 			} else if item := conv(value); item.IsValid() { | ||||
| 				if isPtrElem { | ||||
| 					ptr := reflect.New(elemT) | ||||
| @@ -260,11 +283,45 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
| 			val = values[len(values)-1] | ||||
| 		} | ||||
|  | ||||
| 		if val == "" { | ||||
| 		if conv != nil { | ||||
| 			if value := conv(val); value.IsValid() { | ||||
| 				v.Set(value.Convert(t)) | ||||
| 			} else { | ||||
| 				return ConversionError{ | ||||
| 					Key:   path, | ||||
| 					Type:  t, | ||||
| 					Index: -1, | ||||
| 				} | ||||
| 			} | ||||
| 		} else if m.IsValid { | ||||
| 			if m.IsPtr { | ||||
| 				u := reflect.New(v.Type()) | ||||
| 				if err := u.Interface().(encoding.TextUnmarshaler).UnmarshalText([]byte(val)); err != nil { | ||||
| 					return ConversionError{ | ||||
| 						Key:   path, | ||||
| 						Type:  t, | ||||
| 						Index: -1, | ||||
| 						Err:   err, | ||||
| 					} | ||||
| 				} | ||||
| 				v.Set(reflect.Indirect(u)) | ||||
| 			} else { | ||||
| 				// If the value implements the encoding.TextUnmarshaler interface | ||||
| 				// apply UnmarshalText as the converter | ||||
| 				if err := m.Unmarshaler.UnmarshalText([]byte(val)); err != nil { | ||||
| 					return ConversionError{ | ||||
| 						Key:   path, | ||||
| 						Type:  t, | ||||
| 						Index: -1, | ||||
| 						Err:   err, | ||||
| 					} | ||||
| 				} | ||||
| 			} | ||||
| 		} else if val == "" { | ||||
| 			if d.zeroEmpty { | ||||
| 				v.Set(reflect.Zero(t)) | ||||
| 			} | ||||
| 		} else if conv != nil { | ||||
| 		} else if conv := builtinConverters[t.Kind()]; conv != nil { | ||||
| 			if value := conv(val); value.IsValid() { | ||||
| 				v.Set(value.Convert(t)) | ||||
| 			} else { | ||||
| @@ -275,31 +332,71 @@ func (d *Decoder) decode(v reflect.Value, path string, parts []pathPart, values | ||||
| 				} | ||||
| 			} | ||||
| 		} else { | ||||
| 			// When there's no registered conversion for the custom type, we will check if the type | ||||
| 			// implements the TextUnmarshaler interface. As the UnmarshalText function should be applied | ||||
| 			// to the pointer of the type, we convert the value to pointer. | ||||
| 			if v.CanAddr() { | ||||
| 				v = v.Addr() | ||||
| 			} | ||||
|  | ||||
| 			if u, ok := v.Interface().(encoding.TextUnmarshaler); ok { | ||||
| 				if err := u.UnmarshalText([]byte(val)); err != nil { | ||||
| 					return ConversionError{ | ||||
| 						Key:   path, | ||||
| 						Type:  t, | ||||
| 						Index: -1, | ||||
| 						Err:   err, | ||||
| 					} | ||||
| 				} | ||||
|  | ||||
| 			} else { | ||||
| 				return fmt.Errorf("schema: converter not found for %v", t) | ||||
| 			} | ||||
| 			return fmt.Errorf("schema: converter not found for %v", t) | ||||
| 		} | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func isTextUnmarshaler(v reflect.Value) unmarshaler { | ||||
| 	// Create a new unmarshaller instance | ||||
| 	m := unmarshaler{} | ||||
| 	if m.Unmarshaler, m.IsValid = v.Interface().(encoding.TextUnmarshaler); m.IsValid { | ||||
| 		return m | ||||
| 	} | ||||
| 	// As the UnmarshalText function should be applied to the pointer of the | ||||
| 	// type, we check that type to see if it implements the necessary | ||||
| 	// method. | ||||
| 	if m.Unmarshaler, m.IsValid = reflect.New(v.Type()).Interface().(encoding.TextUnmarshaler); m.IsValid { | ||||
| 		m.IsPtr = true | ||||
| 		return m | ||||
| 	} | ||||
|  | ||||
| 	// if v is []T or *[]T create new T | ||||
| 	t := v.Type() | ||||
| 	if t.Kind() == reflect.Ptr { | ||||
| 		t = t.Elem() | ||||
| 	} | ||||
| 	if t.Kind() == reflect.Slice { | ||||
| 		// Check if the slice implements encoding.TextUnmarshaller | ||||
| 		if m.Unmarshaler, m.IsValid = v.Interface().(encoding.TextUnmarshaler); m.IsValid { | ||||
| 			return m | ||||
| 		} | ||||
| 		// If t is a pointer slice, check if its elements implement | ||||
| 		// encoding.TextUnmarshaler | ||||
| 		m.IsSliceElement = true | ||||
| 		if t = t.Elem(); t.Kind() == reflect.Ptr { | ||||
| 			t = reflect.PtrTo(t.Elem()) | ||||
| 			v = reflect.Zero(t) | ||||
| 			m.IsSliceElementPtr = true | ||||
| 			m.Unmarshaler, m.IsValid = v.Interface().(encoding.TextUnmarshaler) | ||||
| 			return m | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	v = reflect.New(t) | ||||
| 	m.Unmarshaler, m.IsValid = v.Interface().(encoding.TextUnmarshaler) | ||||
| 	return m | ||||
| } | ||||
|  | ||||
| // TextUnmarshaler helpers ---------------------------------------------------- | ||||
| // unmarshaller contains information about a TextUnmarshaler type | ||||
| type unmarshaler struct { | ||||
| 	Unmarshaler encoding.TextUnmarshaler | ||||
| 	// IsValid indicates whether the resolved type indicated by the other | ||||
| 	// flags implements the encoding.TextUnmarshaler interface. | ||||
| 	IsValid bool | ||||
| 	// IsPtr indicates that the resolved type is the pointer of the original | ||||
| 	// type. | ||||
| 	IsPtr bool | ||||
| 	// IsSliceElement indicates that the resolved type is a slice element of | ||||
| 	// the original type. | ||||
| 	IsSliceElement bool | ||||
| 	// IsSliceElementPtr indicates that the resolved type is a pointer to a | ||||
| 	// slice element of the original type. | ||||
| 	IsSliceElementPtr bool | ||||
| } | ||||
|  | ||||
| // Errors --------------------------------------------------------------------- | ||||
|  | ||||
| // ConversionError stores information about a failed conversion. | ||||
|   | ||||
							
								
								
									
										6
									
								
								vendor/github.com/gorilla/schema/doc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								vendor/github.com/gorilla/schema/doc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -24,7 +24,7 @@ The basic usage is really simple. Given this struct: | ||||
|  | ||||
| This is just a simple example and it doesn't make a lot of sense to create | ||||
| the map manually. Typically it will come from a http.Request object and | ||||
| will be of type url.Values: http.Request.Form or http.Request.MultipartForm: | ||||
| will be of type url.Values, http.Request.Form, or http.Request.MultipartForm: | ||||
|  | ||||
| 	func MyHandler(w http.ResponseWriter, r *http.Request) { | ||||
| 		err := r.ParseForm() | ||||
| @@ -45,7 +45,7 @@ will be of type url.Values: http.Request.Form or http.Request.MultipartForm: | ||||
| 	} | ||||
|  | ||||
| Note: it is a good idea to set a Decoder instance as a package global, | ||||
| because it caches meta-data about structs, and a instance can be shared safely: | ||||
| because it caches meta-data about structs, and an instance can be shared safely: | ||||
|  | ||||
| 	var decoder = schema.NewDecoder() | ||||
|  | ||||
| @@ -121,7 +121,7 @@ field, we could not translate multiple values to it if we did not use an | ||||
| index for the parent struct. | ||||
|  | ||||
| There's also the possibility to create a custom type that implements the | ||||
| TextUnmarshaler interface, and in this case there's no need to registry | ||||
| TextUnmarshaler interface, and in this case there's no need to register | ||||
| a converter, like: | ||||
|  | ||||
| 	type Person struct { | ||||
|   | ||||
							
								
								
									
										40
									
								
								vendor/github.com/gorilla/schema/encoder.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										40
									
								
								vendor/github.com/gorilla/schema/encoder.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -40,6 +40,34 @@ func (e *Encoder) SetAliasTag(tag string) { | ||||
| 	e.cache.tag = tag | ||||
| } | ||||
|  | ||||
| // isValidStructPointer test if input value is a valid struct pointer. | ||||
| func isValidStructPointer(v reflect.Value) bool { | ||||
| 	return v.Type().Kind() == reflect.Ptr && v.Elem().IsValid() && v.Elem().Type().Kind() == reflect.Struct | ||||
| } | ||||
|  | ||||
| func isZero(v reflect.Value) bool { | ||||
| 	switch v.Kind() { | ||||
| 	case reflect.Func: | ||||
| 	case reflect.Map, reflect.Slice: | ||||
| 		return v.IsNil() || v.Len() == 0 | ||||
| 	case reflect.Array: | ||||
| 		z := true | ||||
| 		for i := 0; i < v.Len(); i++ { | ||||
| 			z = z && isZero(v.Index(i)) | ||||
| 		} | ||||
| 		return z | ||||
| 	case reflect.Struct: | ||||
| 		z := true | ||||
| 		for i := 0; i < v.NumField(); i++ { | ||||
| 			z = z && isZero(v.Field(i)) | ||||
| 		} | ||||
| 		return z | ||||
| 	} | ||||
| 	// Compare other types directly: | ||||
| 	z := reflect.Zero(v.Type()) | ||||
| 	return v.Interface() == z.Interface() | ||||
| } | ||||
|  | ||||
| func (e *Encoder) encode(v reflect.Value, dst map[string][]string) error { | ||||
| 	if v.Kind() == reflect.Ptr { | ||||
| 		v = v.Elem() | ||||
| @@ -57,8 +85,9 @@ func (e *Encoder) encode(v reflect.Value, dst map[string][]string) error { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if v.Field(i).Type().Kind() == reflect.Struct { | ||||
| 			e.encode(v.Field(i), dst) | ||||
| 		// Encode struct pointer types if the field is a valid pointer and a struct. | ||||
| 		if isValidStructPointer(v.Field(i)) { | ||||
| 			e.encode(v.Field(i).Elem(), dst) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| @@ -67,7 +96,7 @@ func (e *Encoder) encode(v reflect.Value, dst map[string][]string) error { | ||||
| 		// Encode non-slice types and custom implementations immediately. | ||||
| 		if encFunc != nil { | ||||
| 			value := encFunc(v.Field(i)) | ||||
| 			if value == "" && opts.Contains("omitempty") { | ||||
| 			if opts.Contains("omitempty") && isZero(v.Field(i)) { | ||||
| 				continue | ||||
| 			} | ||||
|  | ||||
| @@ -75,6 +104,11 @@ func (e *Encoder) encode(v reflect.Value, dst map[string][]string) error { | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if v.Field(i).Type().Kind() == reflect.Struct { | ||||
| 			e.encode(v.Field(i), dst) | ||||
| 			continue | ||||
| 		} | ||||
|  | ||||
| 		if v.Field(i).Type().Kind() == reflect.Slice { | ||||
| 			encFunc = typeEncoder(v.Field(i).Type().Elem(), e.regenc) | ||||
| 		} | ||||
|   | ||||
							
								
								
									
										21
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/hashicorp/golang-lru/2q.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -30,9 +30,9 @@ type TwoQueueCache struct { | ||||
| 	size       int | ||||
| 	recentSize int | ||||
|  | ||||
| 	recent      *simplelru.LRU | ||||
| 	frequent    *simplelru.LRU | ||||
| 	recentEvict *simplelru.LRU | ||||
| 	recent      simplelru.LRUCache | ||||
| 	frequent    simplelru.LRUCache | ||||
| 	recentEvict simplelru.LRUCache | ||||
| 	lock        sync.RWMutex | ||||
| } | ||||
|  | ||||
| @@ -84,7 +84,8 @@ func New2QParams(size int, recentRatio float64, ghostRatio float64) (*TwoQueueCa | ||||
| 	return c, nil | ||||
| } | ||||
|  | ||||
| func (c *TwoQueueCache) Get(key interface{}) (interface{}, bool) { | ||||
| // Get looks up a key's value from the cache. | ||||
| func (c *TwoQueueCache) Get(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
|  | ||||
| @@ -105,6 +106,7 @@ func (c *TwoQueueCache) Get(key interface{}) (interface{}, bool) { | ||||
| 	return nil, false | ||||
| } | ||||
|  | ||||
| // Add adds a value to the cache. | ||||
| func (c *TwoQueueCache) Add(key, value interface{}) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
| @@ -160,12 +162,15 @@ func (c *TwoQueueCache) ensureSpace(recentEvict bool) { | ||||
| 	c.frequent.RemoveOldest() | ||||
| } | ||||
|  | ||||
| // Len returns the number of items in the cache. | ||||
| func (c *TwoQueueCache) Len() int { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	return c.recent.Len() + c.frequent.Len() | ||||
| } | ||||
|  | ||||
| // Keys returns a slice of the keys in the cache. | ||||
| // The frequently used keys are first in the returned slice. | ||||
| func (c *TwoQueueCache) Keys() []interface{} { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| @@ -174,6 +179,7 @@ func (c *TwoQueueCache) Keys() []interface{} { | ||||
| 	return append(k1, k2...) | ||||
| } | ||||
|  | ||||
| // Remove removes the provided key from the cache. | ||||
| func (c *TwoQueueCache) Remove(key interface{}) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
| @@ -188,6 +194,7 @@ func (c *TwoQueueCache) Remove(key interface{}) { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| // Purge is used to completely clear the cache. | ||||
| func (c *TwoQueueCache) Purge() { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
| @@ -196,13 +203,17 @@ func (c *TwoQueueCache) Purge() { | ||||
| 	c.recentEvict.Purge() | ||||
| } | ||||
|  | ||||
| // Contains is used to check if the cache contains a key | ||||
| // without updating recency or frequency. | ||||
| func (c *TwoQueueCache) Contains(key interface{}) bool { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	return c.frequent.Contains(key) || c.recent.Contains(key) | ||||
| } | ||||
|  | ||||
| func (c *TwoQueueCache) Peek(key interface{}) (interface{}, bool) { | ||||
| // Peek is used to inspect the cache value of a key | ||||
| // without updating recency or frequency. | ||||
| func (c *TwoQueueCache) Peek(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	if val, ok := c.frequent.Peek(key); ok { | ||||
|   | ||||
							
								
								
									
										16
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										16
									
								
								vendor/github.com/hashicorp/golang-lru/arc.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -18,11 +18,11 @@ type ARCCache struct { | ||||
| 	size int // Size is the total capacity of the cache | ||||
| 	p    int // P is the dynamic preference towards T1 or T2 | ||||
|  | ||||
| 	t1 *simplelru.LRU // T1 is the LRU for recently accessed items | ||||
| 	b1 *simplelru.LRU // B1 is the LRU for evictions from t1 | ||||
| 	t1 simplelru.LRUCache // T1 is the LRU for recently accessed items | ||||
| 	b1 simplelru.LRUCache // B1 is the LRU for evictions from t1 | ||||
|  | ||||
| 	t2 *simplelru.LRU // T2 is the LRU for frequently accessed items | ||||
| 	b2 *simplelru.LRU // B2 is the LRU for evictions from t2 | ||||
| 	t2 simplelru.LRUCache // T2 is the LRU for frequently accessed items | ||||
| 	b2 simplelru.LRUCache // B2 is the LRU for evictions from t2 | ||||
|  | ||||
| 	lock sync.RWMutex | ||||
| } | ||||
| @@ -60,11 +60,11 @@ func NewARC(size int) (*ARCCache, error) { | ||||
| } | ||||
|  | ||||
| // Get looks up a key's value from the cache. | ||||
| func (c *ARCCache) Get(key interface{}) (interface{}, bool) { | ||||
| func (c *ARCCache) Get(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
|  | ||||
| 	// Ff the value is contained in T1 (recent), then | ||||
| 	// If the value is contained in T1 (recent), then | ||||
| 	// promote it to T2 (frequent) | ||||
| 	if val, ok := c.t1.Peek(key); ok { | ||||
| 		c.t1.Remove(key) | ||||
| @@ -153,7 +153,7 @@ func (c *ARCCache) Add(key, value interface{}) { | ||||
| 		// Remove from B2 | ||||
| 		c.b2.Remove(key) | ||||
|  | ||||
| 		// Add the key to the frequntly used list | ||||
| 		// Add the key to the frequently used list | ||||
| 		c.t2.Add(key, value) | ||||
| 		return | ||||
| 	} | ||||
| @@ -247,7 +247,7 @@ func (c *ARCCache) Contains(key interface{}) bool { | ||||
|  | ||||
| // Peek is used to inspect the cache value of a key | ||||
| // without updating recency or frequency. | ||||
| func (c *ARCCache) Peek(key interface{}) (interface{}, bool) { | ||||
| func (c *ARCCache) Peek(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	if val, ok := c.t1.Peek(key); ok { | ||||
|   | ||||
							
								
								
									
										21
									
								
								vendor/github.com/hashicorp/golang-lru/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										21
									
								
								vendor/github.com/hashicorp/golang-lru/doc.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,21 @@ | ||||
| // Package lru provides three different LRU caches of varying sophistication. | ||||
| // | ||||
| // Cache is a simple LRU cache. It is based on the | ||||
| // LRU implementation in groupcache: | ||||
| // https://github.com/golang/groupcache/tree/master/lru | ||||
| // | ||||
| // TwoQueueCache tracks frequently used and recently used entries separately. | ||||
| // This avoids a burst of accesses from taking out frequently used entries, | ||||
| // at the cost of about 2x computational overhead and some extra bookkeeping. | ||||
| // | ||||
| // ARCCache is an adaptive replacement cache. It tracks recent evictions as | ||||
| // well as recent usage in both the frequent and recent caches. Its | ||||
| // computational overhead is comparable to TwoQueueCache, but the memory | ||||
| // overhead is linear with the size of the cache. | ||||
| // | ||||
| // ARC has been patented by IBM, so do not use it if that is problematic for | ||||
| // your program. | ||||
| // | ||||
| // All caches in this package take locks while operating, and are therefore | ||||
| // thread-safe for consumers. | ||||
| package lru | ||||
							
								
								
									
										1
									
								
								vendor/github.com/hashicorp/golang-lru/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								vendor/github.com/hashicorp/golang-lru/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| module github.com/hashicorp/golang-lru | ||||
							
								
								
									
										28
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								vendor/github.com/hashicorp/golang-lru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,6 +1,3 @@ | ||||
| // This package provides a simple LRU cache. It is based on the | ||||
| // LRU implementation in groupcache: | ||||
| // https://github.com/golang/groupcache/tree/master/lru | ||||
| package lru | ||||
|  | ||||
| import ( | ||||
| @@ -11,11 +8,11 @@ import ( | ||||
|  | ||||
| // Cache is a thread-safe fixed size LRU cache. | ||||
| type Cache struct { | ||||
| 	lru  *simplelru.LRU | ||||
| 	lru  simplelru.LRUCache | ||||
| 	lock sync.RWMutex | ||||
| } | ||||
|  | ||||
| // New creates an LRU of the given size | ||||
| // New creates an LRU of the given size. | ||||
| func New(size int) (*Cache, error) { | ||||
| 	return NewWithEvict(size, nil) | ||||
| } | ||||
| @@ -33,7 +30,7 @@ func NewWithEvict(size int, onEvicted func(key interface{}, value interface{})) | ||||
| 	return c, nil | ||||
| } | ||||
|  | ||||
| // Purge is used to completely clear the cache | ||||
| // Purge is used to completely clear the cache. | ||||
| func (c *Cache) Purge() { | ||||
| 	c.lock.Lock() | ||||
| 	c.lru.Purge() | ||||
| @@ -41,30 +38,30 @@ func (c *Cache) Purge() { | ||||
| } | ||||
|  | ||||
| // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||
| func (c *Cache) Add(key, value interface{}) bool { | ||||
| func (c *Cache) Add(key, value interface{}) (evicted bool) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
| 	return c.lru.Add(key, value) | ||||
| } | ||||
|  | ||||
| // Get looks up a key's value from the cache. | ||||
| func (c *Cache) Get(key interface{}) (interface{}, bool) { | ||||
| func (c *Cache) Get(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
| 	return c.lru.Get(key) | ||||
| } | ||||
|  | ||||
| // Check if a key is in the cache, without updating the recent-ness | ||||
| // or deleting it for being stale. | ||||
| // Contains checks if a key is in the cache, without updating the | ||||
| // recent-ness or deleting it for being stale. | ||||
| func (c *Cache) Contains(key interface{}) bool { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	return c.lru.Contains(key) | ||||
| } | ||||
|  | ||||
| // Returns the key value (or undefined if not found) without updating | ||||
| // Peek returns the key value (or undefined if not found) without updating | ||||
| // the "recently used"-ness of the key. | ||||
| func (c *Cache) Peek(key interface{}) (interface{}, bool) { | ||||
| func (c *Cache) Peek(key interface{}) (value interface{}, ok bool) { | ||||
| 	c.lock.RLock() | ||||
| 	defer c.lock.RUnlock() | ||||
| 	return c.lru.Peek(key) | ||||
| @@ -73,16 +70,15 @@ func (c *Cache) Peek(key interface{}) (interface{}, bool) { | ||||
| // ContainsOrAdd checks if a key is in the cache  without updating the | ||||
| // recent-ness or deleting it for being stale,  and if not, adds the value. | ||||
| // Returns whether found and whether an eviction occurred. | ||||
| func (c *Cache) ContainsOrAdd(key, value interface{}) (ok, evict bool) { | ||||
| func (c *Cache) ContainsOrAdd(key, value interface{}) (ok, evicted bool) { | ||||
| 	c.lock.Lock() | ||||
| 	defer c.lock.Unlock() | ||||
|  | ||||
| 	if c.lru.Contains(key) { | ||||
| 		return true, false | ||||
| 	} else { | ||||
| 		evict := c.lru.Add(key, value) | ||||
| 		return false, evict | ||||
| 	} | ||||
| 	evicted = c.lru.Add(key, value) | ||||
| 	return false, evicted | ||||
| } | ||||
|  | ||||
| // Remove removes the provided key from the cache. | ||||
|   | ||||
							
								
								
									
										17
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										17
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -36,7 +36,7 @@ func NewLRU(size int, onEvict EvictCallback) (*LRU, error) { | ||||
| 	return c, nil | ||||
| } | ||||
|  | ||||
| // Purge is used to completely clear the cache | ||||
| // Purge is used to completely clear the cache. | ||||
| func (c *LRU) Purge() { | ||||
| 	for k, v := range c.items { | ||||
| 		if c.onEvict != nil { | ||||
| @@ -48,7 +48,7 @@ func (c *LRU) Purge() { | ||||
| } | ||||
|  | ||||
| // Add adds a value to the cache.  Returns true if an eviction occurred. | ||||
| func (c *LRU) Add(key, value interface{}) bool { | ||||
| func (c *LRU) Add(key, value interface{}) (evicted bool) { | ||||
| 	// Check for existing item | ||||
| 	if ent, ok := c.items[key]; ok { | ||||
| 		c.evictList.MoveToFront(ent) | ||||
| @@ -78,17 +78,18 @@ func (c *LRU) Get(key interface{}) (value interface{}, ok bool) { | ||||
| 	return | ||||
| } | ||||
|  | ||||
| // Check if a key is in the cache, without updating the recent-ness | ||||
| // Contains checks if a key is in the cache, without updating the recent-ness | ||||
| // or deleting it for being stale. | ||||
| func (c *LRU) Contains(key interface{}) (ok bool) { | ||||
| 	_, ok = c.items[key] | ||||
| 	return ok | ||||
| } | ||||
|  | ||||
| // Returns the key value (or undefined if not found) without updating | ||||
| // Peek returns the key value (or undefined if not found) without updating | ||||
| // the "recently used"-ness of the key. | ||||
| func (c *LRU) Peek(key interface{}) (value interface{}, ok bool) { | ||||
| 	if ent, ok := c.items[key]; ok { | ||||
| 	var ent *list.Element | ||||
| 	if ent, ok = c.items[key]; ok { | ||||
| 		return ent.Value.(*entry).value, true | ||||
| 	} | ||||
| 	return nil, ok | ||||
| @@ -96,7 +97,7 @@ func (c *LRU) Peek(key interface{}) (value interface{}, ok bool) { | ||||
|  | ||||
| // Remove removes the provided key from the cache, returning if the | ||||
| // key was contained. | ||||
| func (c *LRU) Remove(key interface{}) bool { | ||||
| func (c *LRU) Remove(key interface{}) (present bool) { | ||||
| 	if ent, ok := c.items[key]; ok { | ||||
| 		c.removeElement(ent) | ||||
| 		return true | ||||
| @@ -105,7 +106,7 @@ func (c *LRU) Remove(key interface{}) bool { | ||||
| } | ||||
|  | ||||
| // RemoveOldest removes the oldest item from the cache. | ||||
| func (c *LRU) RemoveOldest() (interface{}, interface{}, bool) { | ||||
| func (c *LRU) RemoveOldest() (key interface{}, value interface{}, ok bool) { | ||||
| 	ent := c.evictList.Back() | ||||
| 	if ent != nil { | ||||
| 		c.removeElement(ent) | ||||
| @@ -116,7 +117,7 @@ func (c *LRU) RemoveOldest() (interface{}, interface{}, bool) { | ||||
| } | ||||
|  | ||||
| // GetOldest returns the oldest entry | ||||
| func (c *LRU) GetOldest() (interface{}, interface{}, bool) { | ||||
| func (c *LRU) GetOldest() (key interface{}, value interface{}, ok bool) { | ||||
| 	ent := c.evictList.Back() | ||||
| 	if ent != nil { | ||||
| 		kv := ent.Value.(*entry) | ||||
|   | ||||
							
								
								
									
										36
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru_interface.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								vendor/github.com/hashicorp/golang-lru/simplelru/lru_interface.go
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,36 @@ | ||||
| package simplelru | ||||
|  | ||||
| // LRUCache is the interface for simple LRU cache. | ||||
| type LRUCache interface { | ||||
| 	// Adds a value to the cache, returns true if an eviction occurred and | ||||
| 	// updates the "recently used"-ness of the key. | ||||
| 	Add(key, value interface{}) bool | ||||
|  | ||||
| 	// Returns key's value from the cache and | ||||
| 	// updates the "recently used"-ness of the key. #value, isFound | ||||
| 	Get(key interface{}) (value interface{}, ok bool) | ||||
|  | ||||
| 	// Check if a key exsists in cache without updating the recent-ness. | ||||
| 	Contains(key interface{}) (ok bool) | ||||
|  | ||||
| 	// Returns key's value without updating the "recently used"-ness of the key. | ||||
| 	Peek(key interface{}) (value interface{}, ok bool) | ||||
|  | ||||
| 	// Removes a key from the cache. | ||||
| 	Remove(key interface{}) bool | ||||
|  | ||||
| 	// Removes the oldest entry from cache. | ||||
| 	RemoveOldest() (interface{}, interface{}, bool) | ||||
|  | ||||
| 	// Returns the oldest entry from the cache. #key, value, isFound | ||||
| 	GetOldest() (interface{}, interface{}, bool) | ||||
|  | ||||
| 	// Returns a slice of the keys in the cache, from oldest to newest. | ||||
| 	Keys() []interface{} | ||||
|  | ||||
| 	// Returns the number of items in the cache. | ||||
| 	Len() int | ||||
|  | ||||
| 	// Clear all cache entries | ||||
| 	Purge() | ||||
| } | ||||
							
								
								
									
										3
									
								
								vendor/github.com/hashicorp/hcl/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								vendor/github.com/hashicorp/hcl/go.mod
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,3 @@ | ||||
| module github.com/hashicorp/hcl | ||||
|  | ||||
| require github.com/davecgh/go-spew v1.1.1 | ||||
							
								
								
									
										2
									
								
								vendor/github.com/hashicorp/hcl/go.sum
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/hashicorp/hcl/go.sum
									
									
									
										generated
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1,2 @@ | ||||
| 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= | ||||
							
								
								
									
										6
									
								
								vendor/github.com/hashicorp/hcl/hcl/parser/parser.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								vendor/github.com/hashicorp/hcl/hcl/parser/parser.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -205,6 +205,12 @@ func (p *Parser) objectItem() (*ast.ObjectItem, error) { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// key=#comment | ||||
| 	// val | ||||
| 	if p.lineComment != nil { | ||||
| 		o.LineComment, p.lineComment = p.lineComment, nil | ||||
| 	} | ||||
|  | ||||
| 	// do a look-ahead for line comment | ||||
| 	p.scan() | ||||
| 	if len(keys) > 0 && o.Val.Pos().Line == keys[0].Pos().Line && p.lineComment != nil { | ||||
|   | ||||
							
								
								
									
										208
									
								
								vendor/github.com/hashicorp/hcl/hcl/printer/nodes.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										208
									
								
								vendor/github.com/hashicorp/hcl/hcl/printer/nodes.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -252,6 +252,14 @@ func (p *printer) objectItem(o *ast.ObjectItem) []byte { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// If key and val are on different lines, treat line comments like lead comments. | ||||
| 	if o.LineComment != nil && o.Val.Pos().Line != o.Keys[0].Pos().Line { | ||||
| 		for _, comment := range o.LineComment.List { | ||||
| 			buf.WriteString(comment.Text) | ||||
| 			buf.WriteByte(newline) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	for i, k := range o.Keys { | ||||
| 		buf.WriteString(k.Token.Text) | ||||
| 		buf.WriteByte(blank) | ||||
| @@ -265,7 +273,7 @@ func (p *printer) objectItem(o *ast.ObjectItem) []byte { | ||||
|  | ||||
| 	buf.Write(p.output(o.Val)) | ||||
|  | ||||
| 	if o.Val.Pos().Line == o.Keys[0].Pos().Line && o.LineComment != nil { | ||||
| 	if o.LineComment != nil && o.Val.Pos().Line == o.Keys[0].Pos().Line { | ||||
| 		buf.WriteByte(blank) | ||||
| 		for _, comment := range o.LineComment.List { | ||||
| 			buf.WriteString(comment.Text) | ||||
| @@ -509,8 +517,13 @@ func (p *printer) alignedItems(items []*ast.ObjectItem) []byte { | ||||
|  | ||||
| // list returns the printable HCL form of an list type. | ||||
| func (p *printer) list(l *ast.ListType) []byte { | ||||
| 	if p.isSingleLineList(l) { | ||||
| 		return p.singleLineList(l) | ||||
| 	} | ||||
|  | ||||
| 	var buf bytes.Buffer | ||||
| 	buf.WriteString("[") | ||||
| 	buf.WriteByte(newline) | ||||
|  | ||||
| 	var longestLine int | ||||
| 	for _, item := range l.List { | ||||
| @@ -523,115 +536,112 @@ func (p *printer) list(l *ast.ListType) []byte { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	insertSpaceBeforeItem := false | ||||
| 	lastHadLeadComment := false | ||||
| 	haveEmptyLine := false | ||||
| 	for i, item := range l.List { | ||||
| 		// Keep track of whether this item is a heredoc since that has | ||||
| 		// unique behavior. | ||||
| 		heredoc := false | ||||
| 		// If we have a lead comment, then we want to write that first | ||||
| 		leadComment := false | ||||
| 		if lit, ok := item.(*ast.LiteralType); ok && lit.LeadComment != nil { | ||||
| 			leadComment = true | ||||
|  | ||||
| 			// Ensure an empty line before every element with a | ||||
| 			// lead comment (except the first item in a list). | ||||
| 			if !haveEmptyLine && i != 0 { | ||||
| 				buf.WriteByte(newline) | ||||
| 			} | ||||
|  | ||||
| 			for _, comment := range lit.LeadComment.List { | ||||
| 				buf.Write(p.indent([]byte(comment.Text))) | ||||
| 				buf.WriteByte(newline) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		// also indent each line | ||||
| 		val := p.output(item) | ||||
| 		curLen := len(val) | ||||
| 		buf.Write(p.indent(val)) | ||||
|  | ||||
| 		// if this item is a heredoc, then we output the comma on | ||||
| 		// the next line. This is the only case this happens. | ||||
| 		comma := []byte{','} | ||||
| 		if lit, ok := item.(*ast.LiteralType); ok && lit.Token.Type == token.HEREDOC { | ||||
| 			heredoc = true | ||||
| 		} | ||||
|  | ||||
| 		if item.Pos().Line != l.Lbrack.Line { | ||||
| 			// multiline list, add newline before we add each item | ||||
| 			buf.WriteByte(newline) | ||||
| 			insertSpaceBeforeItem = false | ||||
| 			comma = p.indent(comma) | ||||
| 		} | ||||
|  | ||||
| 			// If we have a lead comment, then we want to write that first | ||||
| 			leadComment := false | ||||
| 			if lit, ok := item.(*ast.LiteralType); ok && lit.LeadComment != nil { | ||||
| 				leadComment = true | ||||
| 		buf.Write(comma) | ||||
|  | ||||
| 				// If this isn't the first item and the previous element | ||||
| 				// didn't have a lead comment, then we need to add an extra | ||||
| 				// newline to properly space things out. If it did have a | ||||
| 				// lead comment previously then this would be done | ||||
| 				// automatically. | ||||
| 				if i > 0 && !lastHadLeadComment { | ||||
| 					buf.WriteByte(newline) | ||||
| 				} | ||||
|  | ||||
| 				for _, comment := range lit.LeadComment.List { | ||||
| 					buf.Write(p.indent([]byte(comment.Text))) | ||||
| 					buf.WriteByte(newline) | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			// also indent each line | ||||
| 			val := p.output(item) | ||||
| 			curLen := len(val) | ||||
| 			buf.Write(p.indent(val)) | ||||
|  | ||||
| 			// if this item is a heredoc, then we output the comma on | ||||
| 			// the next line. This is the only case this happens. | ||||
| 			comma := []byte{','} | ||||
| 			if heredoc { | ||||
| 				buf.WriteByte(newline) | ||||
| 				comma = p.indent(comma) | ||||
| 			} | ||||
|  | ||||
| 			buf.Write(comma) | ||||
|  | ||||
| 			if lit, ok := item.(*ast.LiteralType); ok && lit.LineComment != nil { | ||||
| 				// if the next item doesn't have any comments, do not align | ||||
| 				buf.WriteByte(blank) // align one space | ||||
| 				for i := 0; i < longestLine-curLen; i++ { | ||||
| 					buf.WriteByte(blank) | ||||
| 				} | ||||
|  | ||||
| 				for _, comment := range lit.LineComment.List { | ||||
| 					buf.WriteString(comment.Text) | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			lastItem := i == len(l.List)-1 | ||||
| 			if lastItem { | ||||
| 				buf.WriteByte(newline) | ||||
| 			} | ||||
|  | ||||
| 			if leadComment && !lastItem { | ||||
| 				buf.WriteByte(newline) | ||||
| 			} | ||||
|  | ||||
| 			lastHadLeadComment = leadComment | ||||
| 		} else { | ||||
| 			if insertSpaceBeforeItem { | ||||
| 		if lit, ok := item.(*ast.LiteralType); ok && lit.LineComment != nil { | ||||
| 			// if the next item doesn't have any comments, do not align | ||||
| 			buf.WriteByte(blank) // align one space | ||||
| 			for i := 0; i < longestLine-curLen; i++ { | ||||
| 				buf.WriteByte(blank) | ||||
| 				insertSpaceBeforeItem = false | ||||
| 			} | ||||
|  | ||||
| 			// Output the item itself | ||||
| 			// also indent each line | ||||
| 			val := p.output(item) | ||||
| 			curLen := len(val) | ||||
| 			buf.Write(val) | ||||
|  | ||||
| 			// If this is a heredoc item we always have to output a newline | ||||
| 			// so that it parses properly. | ||||
| 			if heredoc { | ||||
| 				buf.WriteByte(newline) | ||||
| 			} | ||||
|  | ||||
| 			// If this isn't the last element, write a comma. | ||||
| 			if i != len(l.List)-1 { | ||||
| 				buf.WriteString(",") | ||||
| 				insertSpaceBeforeItem = true | ||||
| 			} | ||||
|  | ||||
| 			if lit, ok := item.(*ast.LiteralType); ok && lit.LineComment != nil { | ||||
| 				// if the next item doesn't have any comments, do not align | ||||
| 				buf.WriteByte(blank) // align one space | ||||
| 				for i := 0; i < longestLine-curLen; i++ { | ||||
| 					buf.WriteByte(blank) | ||||
| 				} | ||||
|  | ||||
| 				for _, comment := range lit.LineComment.List { | ||||
| 					buf.WriteString(comment.Text) | ||||
| 				} | ||||
| 			for _, comment := range lit.LineComment.List { | ||||
| 				buf.WriteString(comment.Text) | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		buf.WriteByte(newline) | ||||
|  | ||||
| 		// Ensure an empty line after every element with a | ||||
| 		// lead comment (except the first item in a list). | ||||
| 		haveEmptyLine = leadComment && i != len(l.List)-1 | ||||
| 		if haveEmptyLine { | ||||
| 			buf.WriteByte(newline) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	buf.WriteString("]") | ||||
| 	return buf.Bytes() | ||||
| } | ||||
|  | ||||
| // isSingleLineList returns true if: | ||||
| // * they were previously formatted entirely on one line | ||||
| // * they consist entirely of literals | ||||
| // * there are either no heredoc strings or the list has exactly one element | ||||
| // * there are no line comments | ||||
| func (printer) isSingleLineList(l *ast.ListType) bool { | ||||
| 	for _, item := range l.List { | ||||
| 		if item.Pos().Line != l.Lbrack.Line { | ||||
| 			return false | ||||
| 		} | ||||
|  | ||||
| 		lit, ok := item.(*ast.LiteralType) | ||||
| 		if !ok { | ||||
| 			return false | ||||
| 		} | ||||
|  | ||||
| 		if lit.Token.Type == token.HEREDOC && len(l.List) != 1 { | ||||
| 			return false | ||||
| 		} | ||||
|  | ||||
| 		if lit.LineComment != nil { | ||||
| 			return false | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	return true | ||||
| } | ||||
|  | ||||
| // singleLineList prints a simple single line list. | ||||
| // For a definition of "simple", see isSingleLineList above. | ||||
| func (p *printer) singleLineList(l *ast.ListType) []byte { | ||||
| 	buf := &bytes.Buffer{} | ||||
|  | ||||
| 	buf.WriteString("[") | ||||
| 	for i, item := range l.List { | ||||
| 		if i != 0 { | ||||
| 			buf.WriteString(", ") | ||||
| 		} | ||||
|  | ||||
| 		// Output the item itself | ||||
| 		buf.Write(p.output(item)) | ||||
|  | ||||
| 		// The heredoc marker needs to be at the end of line. | ||||
| 		if lit, ok := item.(*ast.LiteralType); ok && lit.Token.Type == token.HEREDOC { | ||||
| 			buf.WriteByte(newline) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	buf.WriteString("]") | ||||
|   | ||||
							
								
								
									
										29
									
								
								vendor/github.com/hashicorp/hcl/hcl/scanner/scanner.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										29
									
								
								vendor/github.com/hashicorp/hcl/hcl/scanner/scanner.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -74,14 +74,6 @@ func (s *Scanner) next() rune { | ||||
| 		return eof | ||||
| 	} | ||||
|  | ||||
| 	if ch == utf8.RuneError && size == 1 { | ||||
| 		s.srcPos.Column++ | ||||
| 		s.srcPos.Offset += size | ||||
| 		s.lastCharLen = size | ||||
| 		s.err("illegal UTF-8 encoding") | ||||
| 		return ch | ||||
| 	} | ||||
|  | ||||
| 	// remember last position | ||||
| 	s.prevPos = s.srcPos | ||||
|  | ||||
| @@ -89,18 +81,27 @@ func (s *Scanner) next() rune { | ||||
| 	s.lastCharLen = size | ||||
| 	s.srcPos.Offset += size | ||||
|  | ||||
| 	if ch == utf8.RuneError && size == 1 { | ||||
| 		s.err("illegal UTF-8 encoding") | ||||
| 		return ch | ||||
| 	} | ||||
|  | ||||
| 	if ch == '\n' { | ||||
| 		s.srcPos.Line++ | ||||
| 		s.lastLineLen = s.srcPos.Column | ||||
| 		s.srcPos.Column = 0 | ||||
| 	} | ||||
|  | ||||
| 	// If we see a null character with data left, then that is an error | ||||
| 	if ch == '\x00' && s.buf.Len() > 0 { | ||||
| 	if ch == '\x00' { | ||||
| 		s.err("unexpected null character (0x00)") | ||||
| 		return eof | ||||
| 	} | ||||
|  | ||||
| 	if ch == '\uE123' { | ||||
| 		s.err("unicode code point U+E123 reserved for internal use") | ||||
| 		return utf8.RuneError | ||||
| 	} | ||||
|  | ||||
| 	// debug | ||||
| 	// fmt.Printf("ch: %q, offset:column: %d:%d\n", ch, s.srcPos.Offset, s.srcPos.Column) | ||||
| 	return ch | ||||
| @@ -432,16 +433,16 @@ func (s *Scanner) scanHeredoc() { | ||||
|  | ||||
| 	// Read the identifier | ||||
| 	identBytes := s.src[offs : s.srcPos.Offset-s.lastCharLen] | ||||
| 	if len(identBytes) == 0 { | ||||
| 	if len(identBytes) == 0 || (len(identBytes) == 1 && identBytes[0] == '-') { | ||||
| 		s.err("zero-length heredoc anchor") | ||||
| 		return | ||||
| 	} | ||||
|  | ||||
| 	var identRegexp *regexp.Regexp | ||||
| 	if identBytes[0] == '-' { | ||||
| 		identRegexp = regexp.MustCompile(fmt.Sprintf(`[[:space:]]*%s\z`, identBytes[1:])) | ||||
| 		identRegexp = regexp.MustCompile(fmt.Sprintf(`^[[:space:]]*%s\r*\z`, identBytes[1:])) | ||||
| 	} else { | ||||
| 		identRegexp = regexp.MustCompile(fmt.Sprintf(`[[:space:]]*%s\z`, identBytes)) | ||||
| 		identRegexp = regexp.MustCompile(fmt.Sprintf(`^[[:space:]]*%s\r*\z`, identBytes)) | ||||
| 	} | ||||
|  | ||||
| 	// Read the actual string value | ||||
| @@ -551,7 +552,7 @@ func (s *Scanner) scanDigits(ch rune, base, n int) rune { | ||||
| 		s.err("illegal char escape") | ||||
| 	} | ||||
|  | ||||
| 	if n != start { | ||||
| 	if n != start && ch != eof { | ||||
| 		// we scanned all digits, put the last non digit char back, | ||||
| 		// only if we read anything at all | ||||
| 		s.unread() | ||||
|   | ||||
							
								
								
									
										2
									
								
								vendor/github.com/jpillora/backoff/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/jpillora/backoff/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -116,4 +116,4 @@ https://godoc.org/github.com/jpillora/backoff | ||||
|  | ||||
| #### Credits | ||||
|  | ||||
| Forked from some JavaScript written by [@tj](https://github.com/tj) | ||||
| Forked from [some JavaScript](https://github.com/segmentio/backo) written by [@tj](https://github.com/tj) | ||||
|   | ||||
							
								
								
									
										13
									
								
								vendor/github.com/jpillora/backoff/backoff.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										13
									
								
								vendor/github.com/jpillora/backoff/backoff.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -71,7 +71,8 @@ func (b *Backoff) ForAttempt(attempt float64) time.Duration { | ||||
| 	//keep within bounds | ||||
| 	if dur < min { | ||||
| 		return min | ||||
| 	} else if dur > max { | ||||
| 	} | ||||
| 	if dur > max { | ||||
| 		return max | ||||
| 	} | ||||
| 	return dur | ||||
| @@ -86,3 +87,13 @@ func (b *Backoff) Reset() { | ||||
| func (b *Backoff) Attempt() float64 { | ||||
| 	return b.attempt | ||||
| } | ||||
|  | ||||
| // Copy returns a backoff with equals constraints as the original | ||||
| func (b *Backoff) Copy() *Backoff { | ||||
| 	return &Backoff{ | ||||
| 		Factor: b.Factor, | ||||
| 		Jitter: b.Jitter, | ||||
| 		Min:    b.Min, | ||||
| 		Max:    b.Max, | ||||
| 	} | ||||
| } | ||||
|   | ||||
							
								
								
									
										2
									
								
								vendor/github.com/labstack/echo/.travis.yml
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/labstack/echo/.travis.yml
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,7 +1,7 @@ | ||||
| language: go | ||||
| go: | ||||
|   - 1.8.x | ||||
|   - 1.9.x | ||||
|   - 1.10.x | ||||
|   - tip | ||||
| install: | ||||
|   - make dependency | ||||
|   | ||||
							
								
								
									
										26
									
								
								vendor/github.com/labstack/echo/Gopkg.lock
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										26
									
								
								vendor/github.com/labstack/echo/Gopkg.lock
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -10,26 +10,26 @@ | ||||
| [[projects]] | ||||
|   name = "github.com/dgrijalva/jwt-go" | ||||
|   packages = ["."] | ||||
|   revision = "d2709f9f1f31ebcda9651b03077758c1f3a0018c" | ||||
|   version = "v3.0.0" | ||||
|   revision = "06ea1031745cb8b3dab3f6a236daf2b0aa468b7e" | ||||
|   version = "v3.2.0" | ||||
|  | ||||
| [[projects]] | ||||
|   name = "github.com/labstack/gommon" | ||||
|   packages = ["bytes","color","log","random"] | ||||
|   revision = "1121fd3e243c202482226a7afe4dcd07ffc4139a" | ||||
|   version = "v0.2.1" | ||||
|   revision = "6fe1405d73ec4bd4cd8a4ac8e2a2b2bf95d03954" | ||||
|   version = "0.2.4" | ||||
|  | ||||
| [[projects]] | ||||
|   name = "github.com/mattn/go-colorable" | ||||
|   packages = ["."] | ||||
|   revision = "d228849504861217f796da67fae4f6e347643f15" | ||||
|   version = "v0.0.7" | ||||
|   revision = "167de6bfdfba052fa6b2d3664c8f5272e23c9072" | ||||
|   version = "v0.0.9" | ||||
|  | ||||
| [[projects]] | ||||
|   name = "github.com/mattn/go-isatty" | ||||
|   packages = ["."] | ||||
|   revision = "fc9e8d8ef48496124e79ae0df75490096eccf6fe" | ||||
|   version = "v0.0.2" | ||||
|   revision = "0360b2af4f38e8d38c7fce2a9f4e702702d73a39" | ||||
|   version = "v0.0.3" | ||||
|  | ||||
| [[projects]] | ||||
|   name = "github.com/pmezard/go-difflib" | ||||
| @@ -40,8 +40,8 @@ | ||||
| [[projects]] | ||||
|   name = "github.com/stretchr/testify" | ||||
|   packages = ["assert"] | ||||
|   revision = "69483b4bd14f5845b5a1e55bca19e954e827f1d0" | ||||
|   version = "v1.1.4" | ||||
|   revision = "12b6f73e6084dad08a7c6e575284b177ecafbc71" | ||||
|   version = "v1.2.1" | ||||
|  | ||||
| [[projects]] | ||||
|   branch = "master" | ||||
| @@ -59,17 +59,17 @@ | ||||
|   branch = "master" | ||||
|   name = "golang.org/x/crypto" | ||||
|   packages = ["acme","acme/autocert"] | ||||
|   revision = "e1a4589e7d3ea14a3352255d04b6f1a418845e5e" | ||||
|   revision = "182114d582623c1caa54f73de9c7224e23a48487" | ||||
|  | ||||
| [[projects]] | ||||
|   branch = "master" | ||||
|   name = "golang.org/x/sys" | ||||
|   packages = ["unix"] | ||||
|   revision = "b90f89a1e7a9c1f6b918820b3daa7f08488c8594" | ||||
|   revision = "c28acc882ebcbfbe8ce9f0f14b9ac26ee138dd51" | ||||
|  | ||||
| [solve-meta] | ||||
|   analyzer-name = "dep" | ||||
|   analyzer-version = 1 | ||||
|   inputs-digest = "5f74a2a2ba5b07475ad0faa1b4c021b973ad40b2ae749e3d94e15fe839bb440e" | ||||
|   inputs-digest = "9c7b45e80fe353405800cf01f429b3a203cfb8d4468a04c64a908e11a98ea764" | ||||
|   solver-name = "gps-cdcl" | ||||
|   solver-version = 1 | ||||
|   | ||||
							
								
								
									
										83
									
								
								vendor/github.com/labstack/echo/Gopkg.toml
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										83
									
								
								vendor/github.com/labstack/echo/Gopkg.toml
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,82 +1,37 @@ | ||||
|  | ||||
| ## Gopkg.toml example (these lines may be deleted) | ||||
|  | ||||
| ## "metadata" defines metadata about the project that could be used by other independent | ||||
| ## systems. The metadata defined here will be ignored by dep. | ||||
| # [metadata] | ||||
| # key1 = "value that convey data to other systems" | ||||
| # system1-data = "value that is used by a system" | ||||
| # system2-data = "value that is used by another system" | ||||
|  | ||||
| ## "required" lists a set of packages (not projects) that must be included in | ||||
| ## Gopkg.lock. This list is merged with the set of packages imported by the current | ||||
| ## project. Use it when your project needs a package it doesn't explicitly import - | ||||
| ## including "main" packages. | ||||
| # Gopkg.toml example | ||||
| # | ||||
| # Refer to https://github.com/golang/dep/blob/master/docs/Gopkg.toml.md | ||||
| # for detailed Gopkg.toml documentation. | ||||
| # | ||||
| # required = ["github.com/user/thing/cmd/thing"] | ||||
|  | ||||
| ## "ignored" lists a set of packages (not projects) that are ignored when | ||||
| ## dep statically analyzes source code. Ignored packages can be in this project, | ||||
| ## or in a dependency. | ||||
| # ignored = ["github.com/user/project/badpkg"] | ||||
|  | ||||
| ## Constraints are rules for how directly imported projects | ||||
| ## may be incorporated into the depgraph. They are respected by | ||||
| ## dep whether coming from the Gopkg.toml of the current project or a dependency. | ||||
| # ignored = ["github.com/user/project/pkgX", "bitbucket.org/user/project/pkgA/pkgY"] | ||||
| # | ||||
| # [[constraint]] | ||||
| ## Required: the root import path of the project being constrained. | ||||
| # name = "github.com/user/project" | ||||
| #   name = "github.com/user/project" | ||||
| #   version = "1.0.0" | ||||
| # | ||||
| ## Recommended: the version constraint to enforce for the project. | ||||
| ## Only one of "branch", "version" or "revision" can be specified. | ||||
| # version = "1.0.0" | ||||
| # branch = "master" | ||||
| # revision = "abc123" | ||||
| # [[constraint]] | ||||
| #   name = "github.com/user/project2" | ||||
| #   branch = "dev" | ||||
| #   source = "github.com/myfork/project2" | ||||
| # | ||||
| ## Optional: an alternate location (URL or import path) for the project's source. | ||||
| # source = "https://github.com/myfork/package.git" | ||||
| # | ||||
| ## "metadata" defines metadata about the dependency or override that could be used | ||||
| ## by other independent systems. The metadata defined here will be ignored by dep. | ||||
| # [metadata] | ||||
| # key1 = "value that convey data to other systems" | ||||
| # system1-data = "value that is used by a system" | ||||
| # system2-data = "value that is used by another system" | ||||
|  | ||||
| ## Overrides have the same structure as [[constraint]], but supersede all | ||||
| ## [[constraint]] declarations from all projects. Only [[override]] from | ||||
| ## the current project's are applied. | ||||
| ## | ||||
| ## Overrides are a sledgehammer. Use them only as a last resort. | ||||
| # [[override]] | ||||
| ## Required: the root import path of the project being constrained. | ||||
| # name = "github.com/user/project" | ||||
| # | ||||
| ## Optional: specifying a version constraint override will cause all other | ||||
| ## constraints on this project to be ignored; only the overridden constraint | ||||
| ## need be satisfied. | ||||
| ## Again, only one of "branch", "version" or "revision" can be specified. | ||||
| # version = "1.0.0" | ||||
| # branch = "master" | ||||
| # revision = "abc123" | ||||
| # | ||||
| ## Optional: specifying an alternate source location as an override will | ||||
| ## enforce that the alternate location is used for that project, regardless of | ||||
| ## what source location any dependent projects specify. | ||||
| # source = "https://github.com/myfork/package.git" | ||||
|  | ||||
| #  name = "github.com/x/y" | ||||
| #  version = "2.4.0" | ||||
|  | ||||
|  | ||||
| [[constraint]] | ||||
|   name = "github.com/dgrijalva/jwt-go" | ||||
|   version = "3.0.0" | ||||
|   version = "3.2.0" | ||||
|  | ||||
| [[constraint]] | ||||
|   name = "github.com/labstack/gommon" | ||||
|   version = "0.2.1" | ||||
|   version = "0.2.4" | ||||
|  | ||||
| [[constraint]] | ||||
|   name = "github.com/stretchr/testify" | ||||
|   version = "1.1.4" | ||||
|   version = "1.2.1" | ||||
|  | ||||
| [[constraint]] | ||||
|   branch = "master" | ||||
| @@ -84,4 +39,4 @@ | ||||
|  | ||||
| [[constraint]] | ||||
|   branch = "master" | ||||
|   name = "golang.org/x/crypto" | ||||
|   name = "golang.org/x/crypto" | ||||
							
								
								
									
										4
									
								
								vendor/github.com/labstack/echo/Makefile
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								vendor/github.com/labstack/echo/Makefile
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -11,3 +11,7 @@ test: | ||||
| 		go test -race -coverprofile=profile.out -covermode=atomic $$d || exit 1; \ | ||||
| 		[ -f profile.out ] && cat profile.out >> coverage.txt && rm profile.out; \ | ||||
| 	done | ||||
|  | ||||
| tag: | ||||
| 	@git tag `grep -P '^\tversion = ' echo.go|cut -f2 -d'"'` | ||||
| 	@git tag|grep -v ^v | ||||
|   | ||||
							
								
								
									
										8
									
								
								vendor/github.com/labstack/echo/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								vendor/github.com/labstack/echo/README.md
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -26,9 +26,13 @@ | ||||
| - Automatic TLS via Let’s Encrypt | ||||
| - HTTP/2 support | ||||
|  | ||||
| ## Performance | ||||
| ## Benchmarks | ||||
|  | ||||
| <img src="https://api.labstack.com/chart/bar?values=20015,39584,7282,11276&labels=Static,GitHub%20API,Parse%20API,Google%20Plus%20API&x_title=Route&y_title=Time%20(ns/op)&colors=c&dpi=100"> | ||||
| Date: 2018/03/15<br> | ||||
| Source: https://github.com/vishr/web-framework-benchmark<br> | ||||
| Lower is better! | ||||
|  | ||||
| <img src="https://api.labstack.com/chart/bar?values=37223,55382,2985,5265|42013,59865,3350,6424&labels=Static,GitHub%20API,Parse%20API,Gplus%20API&titles=Echo,Gin&colors=lightseagreen,goldenrod&x_title=Routes&y_title=ns/op"> | ||||
|  | ||||
| ## [Guide](https://echo.labstack.com/guide) | ||||
|  | ||||
|   | ||||
							
								
								
									
										2
									
								
								vendor/github.com/labstack/echo/bind.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								vendor/github.com/labstack/echo/bind.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -80,7 +80,7 @@ func (b *DefaultBinder) bindData(ptr interface{}, data map[string][]string, tag | ||||
| 	val := reflect.ValueOf(ptr).Elem() | ||||
|  | ||||
| 	if typ.Kind() != reflect.Struct { | ||||
| 		return errors.New("Binding element must be a struct") | ||||
| 		return errors.New("binding element must be a struct") | ||||
| 	} | ||||
|  | ||||
| 	for i := 0; i < typ.NumField(); i++ { | ||||
|   | ||||
							
								
								
									
										24
									
								
								vendor/github.com/labstack/echo/context.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										24
									
								
								vendor/github.com/labstack/echo/context.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -206,6 +206,13 @@ const ( | ||||
| 	indexPage     = "index.html" | ||||
| ) | ||||
|  | ||||
| func (c *context) writeContentType(value string) { | ||||
| 	header := c.Response().Header() | ||||
| 	if header.Get(HeaderContentType) == "" { | ||||
| 		header.Set(HeaderContentType, value) | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func (c *context) Request() *http.Request { | ||||
| 	return c.request | ||||
| } | ||||
| @@ -430,7 +437,7 @@ func (c *context) JSONP(code int, callback string, i interface{}) (err error) { | ||||
| } | ||||
|  | ||||
| func (c *context) JSONPBlob(code int, callback string, b []byte) (err error) { | ||||
| 	c.response.Header().Set(HeaderContentType, MIMEApplicationJavaScriptCharsetUTF8) | ||||
| 	c.writeContentType(MIMEApplicationJavaScriptCharsetUTF8) | ||||
| 	c.response.WriteHeader(code) | ||||
| 	if _, err = c.response.Write([]byte(callback + "(")); err != nil { | ||||
| 		return | ||||
| @@ -463,7 +470,7 @@ func (c *context) XMLPretty(code int, i interface{}, indent string) (err error) | ||||
| } | ||||
|  | ||||
| func (c *context) XMLBlob(code int, b []byte) (err error) { | ||||
| 	c.response.Header().Set(HeaderContentType, MIMEApplicationXMLCharsetUTF8) | ||||
| 	c.writeContentType(MIMEApplicationXMLCharsetUTF8) | ||||
| 	c.response.WriteHeader(code) | ||||
| 	if _, err = c.response.Write([]byte(xml.Header)); err != nil { | ||||
| 		return | ||||
| @@ -473,14 +480,14 @@ func (c *context) XMLBlob(code int, b []byte) (err error) { | ||||
| } | ||||
|  | ||||
| func (c *context) Blob(code int, contentType string, b []byte) (err error) { | ||||
| 	c.response.Header().Set(HeaderContentType, contentType) | ||||
| 	c.writeContentType(contentType) | ||||
| 	c.response.WriteHeader(code) | ||||
| 	_, err = c.response.Write(b) | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func (c *context) Stream(code int, contentType string, r io.Reader) (err error) { | ||||
| 	c.response.Header().Set(HeaderContentType, contentType) | ||||
| 	c.writeContentType(contentType) | ||||
| 	c.response.WriteHeader(code) | ||||
| 	_, err = io.Copy(c.response, r) | ||||
| 	return | ||||
| @@ -509,18 +516,17 @@ func (c *context) File(file string) (err error) { | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func (c *context) Attachment(file, name string) (err error) { | ||||
| func (c *context) Attachment(file, name string) error { | ||||
| 	return c.contentDisposition(file, name, "attachment") | ||||
| } | ||||
|  | ||||
| func (c *context) Inline(file, name string) (err error) { | ||||
| func (c *context) Inline(file, name string) error { | ||||
| 	return c.contentDisposition(file, name, "inline") | ||||
| } | ||||
|  | ||||
| func (c *context) contentDisposition(file, name, dispositionType string) (err error) { | ||||
| func (c *context) contentDisposition(file, name, dispositionType string) error { | ||||
| 	c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%q", dispositionType, name)) | ||||
| 	c.File(file) | ||||
| 	return | ||||
| 	return c.File(file) | ||||
| } | ||||
|  | ||||
| func (c *context) NoContent(code int) error { | ||||
|   | ||||
							
								
								
									
										128
									
								
								vendor/github.com/labstack/echo/echo.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										128
									
								
								vendor/github.com/labstack/echo/echo.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -38,6 +38,7 @@ package echo | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	stdContext "context" | ||||
| 	"crypto/tls" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| @@ -45,6 +46,7 @@ import ( | ||||
| 	stdLog "log" | ||||
| 	"net" | ||||
| 	"net/http" | ||||
| 	"net/url" | ||||
| 	"path" | ||||
| 	"path/filepath" | ||||
| 	"reflect" | ||||
| @@ -81,8 +83,7 @@ type ( | ||||
| 		Binder           Binder | ||||
| 		Validator        Validator | ||||
| 		Renderer         Renderer | ||||
| 		// Mutex            sync.RWMutex | ||||
| 		Logger Logger | ||||
| 		Logger           Logger | ||||
| 	} | ||||
|  | ||||
| 	// Route contains a handler and information for matching against requests. | ||||
| @@ -94,9 +95,9 @@ type ( | ||||
|  | ||||
| 	// HTTPError represents an error that occurred while handling a request. | ||||
| 	HTTPError struct { | ||||
| 		Code    int | ||||
| 		Message interface{} | ||||
| 		Inner   error // Stores the error returned by an external dependency | ||||
| 		Code     int | ||||
| 		Message  interface{} | ||||
| 		Internal error // Stores the error returned by an external dependency | ||||
| 	} | ||||
|  | ||||
| 	// MiddlewareFunc defines a function to process middleware. | ||||
| @@ -129,15 +130,16 @@ type ( | ||||
|  | ||||
| // HTTP methods | ||||
| const ( | ||||
| 	CONNECT = "CONNECT" | ||||
| 	DELETE  = "DELETE" | ||||
| 	GET     = "GET" | ||||
| 	HEAD    = "HEAD" | ||||
| 	OPTIONS = "OPTIONS" | ||||
| 	PATCH   = "PATCH" | ||||
| 	POST    = "POST" | ||||
| 	PUT     = "PUT" | ||||
| 	TRACE   = "TRACE" | ||||
| 	CONNECT  = "CONNECT" | ||||
| 	DELETE   = "DELETE" | ||||
| 	GET      = "GET" | ||||
| 	HEAD     = "HEAD" | ||||
| 	OPTIONS  = "OPTIONS" | ||||
| 	PATCH    = "PATCH" | ||||
| 	POST     = "POST" | ||||
| 	PROPFIND = "PROPFIND" | ||||
| 	PUT      = "PUT" | ||||
| 	TRACE    = "TRACE" | ||||
| ) | ||||
|  | ||||
| // MIME types | ||||
| @@ -191,6 +193,7 @@ const ( | ||||
| 	HeaderXHTTPMethodOverride = "X-HTTP-Method-Override" | ||||
| 	HeaderXRealIP             = "X-Real-IP" | ||||
| 	HeaderXRequestID          = "X-Request-ID" | ||||
| 	HeaderXRequestedWith      = "X-Requested-With" | ||||
| 	HeaderServer              = "Server" | ||||
| 	HeaderOrigin              = "Origin" | ||||
|  | ||||
| @@ -214,7 +217,7 @@ const ( | ||||
| ) | ||||
|  | ||||
| const ( | ||||
| 	version = "3.2.6" | ||||
| 	Version = "3.3.5" | ||||
| 	website = "https://echo.labstack.com" | ||||
| 	// http://patorjk.com/software/taag/#p=display&f=Small%20Slant&t=Echo | ||||
| 	banner = ` | ||||
| @@ -238,6 +241,7 @@ var ( | ||||
| 		OPTIONS, | ||||
| 		PATCH, | ||||
| 		POST, | ||||
| 		PROPFIND, | ||||
| 		PUT, | ||||
| 		TRACE, | ||||
| 	} | ||||
| @@ -251,10 +255,10 @@ var ( | ||||
| 	ErrForbidden                   = NewHTTPError(http.StatusForbidden) | ||||
| 	ErrMethodNotAllowed            = NewHTTPError(http.StatusMethodNotAllowed) | ||||
| 	ErrStatusRequestEntityTooLarge = NewHTTPError(http.StatusRequestEntityTooLarge) | ||||
| 	ErrValidatorNotRegistered      = errors.New("Validator not registered") | ||||
| 	ErrRendererNotRegistered       = errors.New("Renderer not registered") | ||||
| 	ErrInvalidRedirectCode         = errors.New("Invalid redirect status code") | ||||
| 	ErrCookieNotFound              = errors.New("Cookie not found") | ||||
| 	ErrValidatorNotRegistered      = errors.New("validator not registered") | ||||
| 	ErrRendererNotRegistered       = errors.New("renderer not registered") | ||||
| 	ErrInvalidRedirectCode         = errors.New("invalid redirect status code") | ||||
| 	ErrCookieNotFound              = errors.New("cookie not found") | ||||
| ) | ||||
|  | ||||
| // Error handlers | ||||
| @@ -321,8 +325,8 @@ func (e *Echo) DefaultHTTPErrorHandler(err error, c Context) { | ||||
| 	if he, ok := err.(*HTTPError); ok { | ||||
| 		code = he.Code | ||||
| 		msg = he.Message | ||||
| 		if he.Inner != nil { | ||||
| 			msg = fmt.Sprintf("%v, %v", err, he.Inner) | ||||
| 		if he.Internal != nil { | ||||
| 			msg = fmt.Sprintf("%v, %v", err, he.Internal) | ||||
| 		} | ||||
| 	} else if e.Debug { | ||||
| 		msg = err.Error() | ||||
| @@ -443,7 +447,7 @@ func (e *Echo) Static(prefix, root string) *Route { | ||||
|  | ||||
| func static(i i, prefix, root string) *Route { | ||||
| 	h := func(c Context) error { | ||||
| 		p, err := PathUnescape(c.Param("*")) | ||||
| 		p, err := url.PathUnescape(c.Param("*")) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| @@ -530,7 +534,7 @@ func (e *Echo) Reverse(name string, params ...interface{}) string { | ||||
|  | ||||
| // Routes returns the registered routes. | ||||
| func (e *Echo) Routes() []*Route { | ||||
| 	routes := []*Route{} | ||||
| 	routes := make([]*Route, 0, len(e.router.routes)) | ||||
| 	for _, v := range e.router.routes { | ||||
| 		routes = append(routes, v) | ||||
| 	} | ||||
| @@ -551,39 +555,48 @@ func (e *Echo) ReleaseContext(c Context) { | ||||
|  | ||||
| // ServeHTTP implements `http.Handler` interface, which serves HTTP requests. | ||||
| func (e *Echo) ServeHTTP(w http.ResponseWriter, r *http.Request) { | ||||
| 	// Acquire lock | ||||
| 	// e.Mutex.RLock() | ||||
| 	// defer e.Mutex.RUnlock() | ||||
|  | ||||
| 	// Acquire context | ||||
| 	c := e.pool.Get().(*context) | ||||
| 	defer e.pool.Put(c) | ||||
| 	c.Reset(r, w) | ||||
|  | ||||
| 	// Middleware | ||||
| 	h := func(c Context) error { | ||||
| 		method := r.Method | ||||
| 	m := r.Method | ||||
| 	h := NotFoundHandler | ||||
|  | ||||
| 	if e.premiddleware == nil { | ||||
| 		path := r.URL.RawPath | ||||
| 		if path == "" { | ||||
| 			path = r.URL.Path | ||||
| 		} | ||||
| 		e.router.Find(method, path, c) | ||||
| 		h := c.Handler() | ||||
| 		e.router.Find(m, getPath(r), c) | ||||
| 		h = c.Handler() | ||||
| 		for i := len(e.middleware) - 1; i >= 0; i-- { | ||||
| 			h = e.middleware[i](h) | ||||
| 		} | ||||
| 		return h(c) | ||||
| 	} | ||||
|  | ||||
| 	// Premiddleware | ||||
| 	for i := len(e.premiddleware) - 1; i >= 0; i-- { | ||||
| 		h = e.premiddleware[i](h) | ||||
| 	} else { | ||||
| 		h = func(c Context) error { | ||||
| 			path := r.URL.RawPath | ||||
| 			if path == "" { | ||||
| 				path = r.URL.Path | ||||
| 			} | ||||
| 			e.router.Find(m, getPath(r), c) | ||||
| 			h := c.Handler() | ||||
| 			for i := len(e.middleware) - 1; i >= 0; i-- { | ||||
| 				h = e.middleware[i](h) | ||||
| 			} | ||||
| 			return h(c) | ||||
| 		} | ||||
| 		for i := len(e.premiddleware) - 1; i >= 0; i-- { | ||||
| 			h = e.premiddleware[i](h) | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	// Execute chain | ||||
| 	if err := h(c); err != nil { | ||||
| 		e.HTTPErrorHandler(err, c) | ||||
| 	} | ||||
|  | ||||
| 	// Release context | ||||
| 	e.pool.Put(c) | ||||
| } | ||||
|  | ||||
| // Start starts an HTTP server. | ||||
| @@ -609,6 +622,10 @@ func (e *Echo) StartTLS(address string, certFile, keyFile string) (err error) { | ||||
|  | ||||
| // StartAutoTLS starts an HTTPS server using certificates automatically installed from https://letsencrypt.org. | ||||
| func (e *Echo) StartAutoTLS(address string) error { | ||||
| 	if e.Listener == nil { | ||||
| 		go http.ListenAndServe(":http", e.AutoTLSManager.HTTPHandler(nil)) | ||||
| 	} | ||||
|  | ||||
| 	s := e.TLSServer | ||||
| 	s.TLSConfig = new(tls.Config) | ||||
| 	s.TLSConfig.GetCertificate = e.AutoTLSManager.GetCertificate | ||||
| @@ -635,7 +652,7 @@ func (e *Echo) StartServer(s *http.Server) (err error) { | ||||
| 	} | ||||
|  | ||||
| 	if !e.HideBanner { | ||||
| 		e.colorer.Printf(banner, e.colorer.Red("v"+version), e.colorer.Blue(website)) | ||||
| 		e.colorer.Printf(banner, e.colorer.Red("v"+Version), e.colorer.Blue(website)) | ||||
| 	} | ||||
|  | ||||
| 	if s.TLSConfig == nil { | ||||
| @@ -663,6 +680,24 @@ func (e *Echo) StartServer(s *http.Server) (err error) { | ||||
| 	return s.Serve(e.TLSListener) | ||||
| } | ||||
|  | ||||
| // Close immediately stops the server. | ||||
| // It internally calls `http.Server#Close()`. | ||||
| func (e *Echo) Close() error { | ||||
| 	if err := e.TLSServer.Close(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return e.Server.Close() | ||||
| } | ||||
|  | ||||
| // Shutdown stops server the gracefully. | ||||
| // It internally calls `http.Server#Shutdown()`. | ||||
| func (e *Echo) Shutdown(ctx stdContext.Context) error { | ||||
| 	if err := e.TLSServer.Shutdown(ctx); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return e.Server.Shutdown(ctx) | ||||
| } | ||||
|  | ||||
| // NewHTTPError creates a new HTTPError instance. | ||||
| func NewHTTPError(code int, message ...interface{}) *HTTPError { | ||||
| 	he := &HTTPError{Code: code, Message: http.StatusText(code)} | ||||
| @@ -698,6 +733,14 @@ func WrapMiddleware(m func(http.Handler) http.Handler) MiddlewareFunc { | ||||
| 	} | ||||
| } | ||||
|  | ||||
| func getPath(r *http.Request) string { | ||||
| 	path := r.URL.RawPath | ||||
| 	if path == "" { | ||||
| 		path = r.URL.Path | ||||
| 	} | ||||
| 	return path | ||||
| } | ||||
|  | ||||
| func handlerName(h HandlerFunc) string { | ||||
| 	t := reflect.ValueOf(h).Type() | ||||
| 	if t.Kind() == reflect.Func { | ||||
| @@ -706,6 +749,11 @@ func handlerName(h HandlerFunc) string { | ||||
| 	return t.String() | ||||
| } | ||||
|  | ||||
| // // PathUnescape is wraps `url.PathUnescape` | ||||
| // func PathUnescape(s string) (string, error) { | ||||
| // 	return url.PathUnescape(s) | ||||
| // } | ||||
|  | ||||
| // tcpKeepAliveListener sets TCP keep-alive timeouts on accepted | ||||
| // connections. It's used by ListenAndServe and ListenAndServeTLS so | ||||
| // dead TCP connections (e.g. closing laptop mid-download) eventually | ||||
|   | ||||
							
								
								
									
										25
									
								
								vendor/github.com/labstack/echo/echo_go1.8.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										25
									
								
								vendor/github.com/labstack/echo/echo_go1.8.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -1,25 +0,0 @@ | ||||
| // +build go1.8 | ||||
|  | ||||
| package echo | ||||
|  | ||||
| import ( | ||||
| 	stdContext "context" | ||||
| ) | ||||
|  | ||||
| // Close immediately stops the server. | ||||
| // It internally calls `http.Server#Close()`. | ||||
| func (e *Echo) Close() error { | ||||
| 	if err := e.TLSServer.Close(); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return e.Server.Close() | ||||
| } | ||||
|  | ||||
| // Shutdown stops server the gracefully. | ||||
| // It internally calls `http.Server#Shutdown()`. | ||||
| func (e *Echo) Shutdown(ctx stdContext.Context) error { | ||||
| 	if err := e.TLSServer.Shutdown(ctx); err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	return e.Server.Shutdown(ctx) | ||||
| } | ||||
							
								
								
									
										4
									
								
								vendor/github.com/labstack/echo/group.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								vendor/github.com/labstack/echo/group.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -92,7 +92,7 @@ func (g *Group) Match(methods []string, path string, handler HandlerFunc, middle | ||||
|  | ||||
| // Group creates a new sub-group with prefix and optional sub-group-level middleware. | ||||
| func (g *Group) Group(prefix string, middleware ...MiddlewareFunc) *Group { | ||||
| 	m := []MiddlewareFunc{} | ||||
| 	m := make([]MiddlewareFunc, 0, len(g.middleware)+len(middleware)) | ||||
| 	m = append(m, g.middleware...) | ||||
| 	m = append(m, middleware...) | ||||
| 	return g.echo.Group(g.prefix+prefix, m...) | ||||
| @@ -113,7 +113,7 @@ func (g *Group) Add(method, path string, handler HandlerFunc, middleware ...Midd | ||||
| 	// Combine into a new slice to avoid accidentally passing the same slice for | ||||
| 	// multiple routes, which would lead to later add() calls overwriting the | ||||
| 	// middleware from earlier calls. | ||||
| 	m := []MiddlewareFunc{} | ||||
| 	m := make([]MiddlewareFunc, 0, len(g.middleware)+len(middleware)) | ||||
| 	m = append(m, g.middleware...) | ||||
| 	m = append(m, middleware...) | ||||
| 	return g.echo.Add(method, g.prefix+path, handler, m...) | ||||
|   | ||||
							
								
								
									
										6
									
								
								vendor/github.com/labstack/echo/middleware/basic_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										6
									
								
								vendor/github.com/labstack/echo/middleware/basic_auth.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -93,10 +93,8 @@ func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc { | ||||
| 				} | ||||
| 			} | ||||
|  | ||||
| 			realm := "" | ||||
| 			if config.Realm == defaultRealm { | ||||
| 				realm = defaultRealm | ||||
| 			} else { | ||||
| 			realm := defaultRealm | ||||
| 			if config.Realm != defaultRealm { | ||||
| 				realm = strconv.Quote(config.Realm) | ||||
| 			} | ||||
|  | ||||
|   | ||||
							
								
								
									
										3
									
								
								vendor/github.com/labstack/echo/middleware/body_dump.go
									
									
									
										generated
									
									
										vendored
									
									
								
							
							
						
						
									
										3
									
								
								vendor/github.com/labstack/echo/middleware/body_dump.go
									
									
									
										generated
									
									
										vendored
									
									
								
							| @@ -3,12 +3,11 @@ package middleware | ||||
| import ( | ||||
| 	"bufio" | ||||
| 	"bytes" | ||||
| 	"io" | ||||
| 	"io/ioutil" | ||||
| 	"net" | ||||
| 	"net/http" | ||||
|  | ||||
| 	"io" | ||||
|  | ||||
| 	"github.com/labstack/echo" | ||||
| ) | ||||
|  | ||||
|   | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user