mirror of
https://github.com/cwinfo/matterbridge.git
synced 2025-06-27 18:09:26 +00:00
Compare commits
124 Commits
Author | SHA1 | Date | |
---|---|---|---|
222cccf388 | |||
bab308508e | |||
dedb83c867 | |||
723a90cdd6 | |||
67d2398fa8 | |||
5f3b6ec007 | |||
55ab0c12f1 | |||
d1227b5fc9 | |||
6ea368c383 | |||
e92b6de09f | |||
e622587db4 | |||
f2efc06d1f | |||
a2b94452db | |||
4c506f7cc3 | |||
7886f05e88 | |||
f58be0d1c1 | |||
1152394bc1 | |||
a082b5a590 | |||
bae9484df2 | |||
6f78485878 | |||
fd0fe3390b | |||
2522158127 | |||
8be107cecc | |||
5aab158c0b | |||
1d33e60e36 | |||
83c28cb857 | |||
df5bce27b0 | |||
2b15739b48 | |||
3480c88e90 | |||
432cd0f99d | |||
e8b3e9b22d | |||
d4a47671ea | |||
0bcd1e62f3 | |||
80822b7fff | |||
78f1011f52 | |||
67f6257617 | |||
169c614489 | |||
da908c438a | |||
9c9c4bf1f9 | |||
7764493298 | |||
64a20ee61b | |||
62d1af8c37 | |||
0f5274fdf6 | |||
2e2187ebf4 | |||
762c3350f4 | |||
e1a4d7f77e | |||
a7a4554a85 | |||
6bd808ce91 | |||
a5c143bc46 | |||
87c9cac756 | |||
6a047f8722 | |||
6523494e83 | |||
7c6ce8bb90 | |||
dafbfe4021 | |||
a4d5c94d9b | |||
7119e378a7 | |||
e1dc3032c1 | |||
5de03b8921 | |||
7631d43c48 | |||
d0b2ee5c85 | |||
8830a5a1df | |||
ee87626a93 | |||
9f15d38c1c | |||
4a96a977c0 | |||
9a95293bdf | |||
0b3a06d263 | |||
9a6249c4f5 | |||
50bd51e461 | |||
04f8013314 | |||
a0aaf0057a | |||
8e78b3e6be | |||
57a503818d | |||
25d2ff3e9b | |||
31902d3e57 | |||
16f3fa6bae | |||
1f706673cf | |||
fac5f69ad2 | |||
97c944bb63 | |||
d0c4fe78ee | |||
265457b451 | |||
4a4a29c9f6 | |||
0a91b9e1c9 | |||
f56163295c | |||
d1c87c068b | |||
fa20761110 | |||
e4a0e0a0e9 | |||
d30ae19e2a | |||
5c919e6bff | |||
434393d1c3 | |||
af9aa5d7cb | |||
05eb75442a | |||
3496ed0c7e | |||
1b89604c7a | |||
67a9d133e9 | |||
ed9118b346 | |||
59e55cfbd5 | |||
788d3b32ac | |||
1d414cf2fd | |||
cc3c168162 | |||
1ee6837f0e | |||
27dcea7c5b | |||
dcda7f7b8c | |||
e0cbb69a4f | |||
7ec95f786d | |||
1efe40add5 | |||
cbd73ee313 | |||
34227a7a39 | |||
71cb9b2d1d | |||
cd4c9b194f | |||
98762a0235 | |||
2fd1fd9573 | |||
aff3964078 | |||
2778580397 | |||
962062fe44 | |||
0578b21270 | |||
36a800c3f5 | |||
6d21f84187 | |||
f1e9833310 | |||
46f5acc4f9 | |||
95d4dcaeb3 | |||
64c542e614 | |||
13d081ea80 | |||
c0f9d86287 | |||
bcdecdaa73 |
28
.github/ISSUE_TEMPLATE.md
vendored
28
.github/ISSUE_TEMPLATE.md
vendored
@ -1,22 +1,36 @@
|
|||||||
If you have a configuration problem, please first try to create a basic configuration following the instructions on [the wiki](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) before filing an issue.
|
<!-- This is a bug report template. By following the instructions below and
|
||||||
|
filling out the sections with your information, you will help the us to get all
|
||||||
|
the necessary data to fix your issue.
|
||||||
|
|
||||||
Please answer the following questions.
|
You can also preview your report before submitting it.
|
||||||
|
|
||||||
### Which version of matterbridge are you using?
|
Text between <!-- and --> marks will be invisible in the report.
|
||||||
run ```matterbridge -version```
|
-->
|
||||||
|
|
||||||
### If you're having problems with mattermost please specify mattermost version.
|
<!-- If you have a configuration problem, please first try to create a basic configuration following the instructions on [the wiki](https://github.com/42wim/matterbridge/wiki/How-to-create-your-config) before filing an issue. -->
|
||||||
|
|
||||||
|
|
||||||
|
### Environment
|
||||||
|
<!-- run `matterbridge -version` -->
|
||||||
|
<!-- If you're having problems with mattermost also specify the mattermost version. -->
|
||||||
|
Version:
|
||||||
|
|
||||||
|
<!-- What operating system are you using ? (be as specific as possible) -->
|
||||||
|
Operating system:
|
||||||
|
|
||||||
|
<!-- If you compiled matterbridge yourself:
|
||||||
|
* Specify the output of `go version`
|
||||||
|
* Specify the output of `git rev-parse HEAD` -->
|
||||||
|
|
||||||
### Please describe the expected behavior.
|
### Please describe the expected behavior.
|
||||||
|
|
||||||
|
|
||||||
### Please describe the actual behavior.
|
### Please describe the actual behavior.
|
||||||
#### Use logs from running ```matterbridge -debug``` if possible.
|
<!-- Use logs from running `matterbridge -debug` if possible. -->
|
||||||
|
|
||||||
|
|
||||||
### Any steps to reproduce the behavior?
|
### Any steps to reproduce the behavior?
|
||||||
|
|
||||||
|
|
||||||
### Please add your configuration file
|
### Please add your configuration file
|
||||||
#### (be sure to exclude or anonymize private data (tokens/passwords))
|
<!-- (be sure to exclude or anonymize private data (tokens/passwords)) -->
|
||||||
|
@ -34,7 +34,7 @@ before_script:
|
|||||||
# flunk the build and immediately stop. It's sorta like having
|
# flunk the build and immediately stop. It's sorta like having
|
||||||
# set -e enabled in bash.
|
# set -e enabled in bash.
|
||||||
script:
|
script:
|
||||||
- test -z $(gofmt -s -l $GO_FILES) # Fail if a .go file hasn't been formatted with gofmt
|
#- test -z $(gofmt -s -l $GO_FILES) # Fail if a .go file hasn't been formatted with gofmt
|
||||||
- go test -v -race $PKGS # Run all the tests with the race detector enabled
|
- go test -v -race $PKGS # Run all the tests with the race detector enabled
|
||||||
- go vet $PKGS # go vet is the official Go static analyzer
|
- go vet $PKGS # go vet is the official Go static analyzer
|
||||||
- megacheck $PKGS # "go vet on steroids" + linter
|
- megacheck $PKGS # "go vet on steroids" + linter
|
||||||
|
37
README.md
37
README.md
@ -1,17 +1,18 @@
|
|||||||
# matterbridge
|
# matterbridge
|
||||||
Click on one of the badges below to join the chat
|
Click on one of the badges below to join the chat
|
||||||
|
|
||||||
[](https://gitter.im/42wim/matterbridge) [](https://webchat.freenode.net/?channels=matterbridgechat) [](https://discord.gg/AkKPtrQ) [](https://riot.im/app/#/room/#matterbridge:matrix.org) [](https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA) [](https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e) 
|
[](https://gitter.im/42wim/matterbridge) [](https://webchat.freenode.net/?channels=matterbridgechat) [](https://discord.gg/AkKPtrQ) [](https://riot.im/app/#/room/#matterbridge:matrix.org) [](https://join.slack.com/matterbridgechat/shared_invite/MjEwODMxNjU1NDMwLTE0OTk2MTU3NTMtMzZkZmRiNDZhOA) [](https://framateam.org/signup_user_complete/?id=tfqm33ggop8x3qgu4boeieta6e) [](https://inverse.chat) [](https://www.twitch.tv/matterbridge)
|
||||||
|
|
||||||
[](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion)
|
[](https://github.com/42wim/matterbridge/releases/latest) [](https://bintray.com/42wim/nightly/Matterbridge/_latestVersion)
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix and Steam.
|
Simple bridge between Mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat(via xmpp), Matrix, Steam and ssh-chat
|
||||||
Has a REST API.
|
Has a REST API.
|
||||||
|
Minecraft server chat support via [MatterLink](https://github.com/elytra/MatterLink)
|
||||||
|
|
||||||
# Table of Contents
|
# Table of Contents
|
||||||
* [Features](#features)
|
* [Features](https://github.com/42wim/matterbridge/wiki/Features)
|
||||||
* [Requirements](#requirements)
|
* [Requirements](#requirements)
|
||||||
* [Screenshots](https://github.com/42wim/matterbridge/wiki/)
|
* [Screenshots](https://github.com/42wim/matterbridge/wiki/)
|
||||||
* [Installing](#installing)
|
* [Installing](#installing)
|
||||||
@ -27,13 +28,21 @@ Has a REST API.
|
|||||||
* [Thanks](#thanks)
|
* [Thanks](#thanks)
|
||||||
|
|
||||||
# Features
|
# Features
|
||||||
* Relays public channel messages between multiple mattermost, IRC, XMPP, Gitter, Slack, Discord, Telegram, Rocket.Chat, Hipchat (via xmpp), Matrix and Steam.
|
* [Support bridging between any protocols](https://github.com/42wim/matterbridge/wiki/Features#support-bridging-between-any-protocols)
|
||||||
Pick and mix.
|
* [Support multiple gateways(bridges) for your protocols](https://github.com/42wim/matterbridge/wiki/Features#support-multiple-gatewaysbridges-for-your-protocols)
|
||||||
* Support private groups on your mattermost/slack.
|
* [Message edits and deletes](https://github.com/42wim/matterbridge/wiki/Features#message-edits-and-deletes)
|
||||||
* Allow for bridging the same bridges, which means you can eg bridge between multiple mattermosts.
|
* [Attachment / files handling](https://github.com/42wim/matterbridge/wiki/Features#attachment--files-handling)
|
||||||
* The bridge is now a gateway which has support multiple in and out bridges. (and supports multiple gateways).
|
* [Username and avatar spoofing](https://github.com/42wim/matterbridge/wiki/Features#username-and-avatar-spoofing)
|
||||||
* Edits and delete messages across bridges that support it (mattermost,slack,discord,gitter,telegram)
|
* [Private groups](https://github.com/42wim/matterbridge/wiki/Features#private-groups)
|
||||||
* REST API to read/post messages to bridges (WIP).
|
* [API](https://github.com/42wim/matterbridge/wiki/Features#api)
|
||||||
|
|
||||||
|
## API
|
||||||
|
The API is very basic at the moment and rather undocumented.
|
||||||
|
|
||||||
|
Used by at least 2 projects. Feel free to make a PR to add your project to this list.
|
||||||
|
|
||||||
|
* [MatterLink](https://github.com/elytra/MatterLink) (Matterbridge link for Minecraft Server chat)
|
||||||
|
* [pyCord](https://github.com/NikkyAI/pyCord) (crossplatform chatbot)
|
||||||
|
|
||||||
# Requirements
|
# Requirements
|
||||||
Accounts to one of the supported bridges
|
Accounts to one of the supported bridges
|
||||||
@ -48,13 +57,15 @@ Accounts to one of the supported bridges
|
|||||||
* [Rocket.chat](https://rocket.chat)
|
* [Rocket.chat](https://rocket.chat)
|
||||||
* [Matrix](https://matrix.org)
|
* [Matrix](https://matrix.org)
|
||||||
* [Steam](https://store.steampowered.com/)
|
* [Steam](https://store.steampowered.com/)
|
||||||
|
* [Twitch](https://twitch.tv)
|
||||||
|
* [Ssh-chat](https://github.com/shazow/ssh-chat)
|
||||||
|
|
||||||
# Screenshots
|
# Screenshots
|
||||||
See https://github.com/42wim/matterbridge/wiki
|
See https://github.com/42wim/matterbridge/wiki
|
||||||
|
|
||||||
# Installing
|
# Installing
|
||||||
## Binaries
|
## Binaries
|
||||||
* Latest stable release [v1.4.0](https://github.com/42wim/matterbridge/releases/latest)
|
* Latest stable release [v1.8.0](https://github.com/42wim/matterbridge/releases/latest)
|
||||||
* Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)
|
* Development releases (follows master) can be downloaded [here](https://dl.bintray.com/42wim/nightly/)
|
||||||
|
|
||||||
## Building
|
## Building
|
||||||
@ -175,7 +186,7 @@ Matterbridge wouldn't exist without these libraries:
|
|||||||
* echo - https://github.com/labstack/echo
|
* echo - https://github.com/labstack/echo
|
||||||
* gitter - https://github.com/sromku/go-gitter
|
* gitter - https://github.com/sromku/go-gitter
|
||||||
* gops - https://github.com/google/gops
|
* gops - https://github.com/google/gops
|
||||||
* irc - https://github.com/thoj/go-ircevent
|
* irc - https://github.com/lrstanley/girc
|
||||||
* mattermost - https://github.com/mattermost/platform
|
* mattermost - https://github.com/mattermost/platform
|
||||||
* matrix - https://github.com/matrix-org/gomatrix
|
* matrix - https://github.com/matrix-org/gomatrix
|
||||||
* slack - https://github.com/nlopes/slack
|
* slack - https://github.com/nlopes/slack
|
||||||
|
@ -1,21 +1,21 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
|
||||||
"github.com/labstack/echo"
|
"github.com/labstack/echo"
|
||||||
"github.com/labstack/echo/middleware"
|
"github.com/labstack/echo/middleware"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
"github.com/zfjagann/golang-ring"
|
"github.com/zfjagann/golang-ring"
|
||||||
"net/http"
|
"net/http"
|
||||||
"sync"
|
"sync"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Api struct {
|
type Api struct {
|
||||||
Config *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
Messages ring.Ring
|
Messages ring.Ring
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
type ApiMessage struct {
|
type ApiMessage struct {
|
||||||
@ -30,26 +30,30 @@ var flog *log.Entry
|
|||||||
var protocol = "api"
|
var protocol = "api"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Api {
|
func New(cfg *config.BridgeConfig) *Api {
|
||||||
b := &Api{}
|
b := &Api{BridgeConfig: cfg}
|
||||||
e := echo.New()
|
e := echo.New()
|
||||||
|
e.HideBanner = true
|
||||||
|
e.HidePort = true
|
||||||
b.Messages = ring.Ring{}
|
b.Messages = ring.Ring{}
|
||||||
b.Messages.SetCapacity(cfg.Buffer)
|
b.Messages.SetCapacity(b.Config.Buffer)
|
||||||
b.Config = &cfg
|
|
||||||
b.Account = account
|
|
||||||
b.Remote = c
|
|
||||||
if b.Config.Token != "" {
|
if b.Config.Token != "" {
|
||||||
e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) {
|
e.Use(middleware.KeyAuth(func(key string, c echo.Context) (bool, error) {
|
||||||
return key == b.Config.Token, nil
|
return key == b.Config.Token, nil
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
e.GET("/api/messages", b.handleMessages)
|
e.GET("/api/messages", b.handleMessages)
|
||||||
|
e.GET("/api/stream", b.handleStream)
|
||||||
e.POST("/api/message", b.handlePostMessage)
|
e.POST("/api/message", b.handlePostMessage)
|
||||||
go func() {
|
go func() {
|
||||||
flog.Fatal(e.Start(cfg.BindAddress))
|
if b.Config.BindAddress == "" {
|
||||||
|
flog.Fatalf("No BindAddress configured.")
|
||||||
|
}
|
||||||
|
flog.Infof("Listening on %s", b.Config.BindAddress)
|
||||||
|
flog.Fatal(e.Start(b.Config.BindAddress))
|
||||||
}()
|
}()
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
@ -78,21 +82,18 @@ func (b *Api) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Api) handlePostMessage(c echo.Context) error {
|
func (b *Api) handlePostMessage(c echo.Context) error {
|
||||||
message := &ApiMessage{}
|
message := config.Message{}
|
||||||
if err := c.Bind(message); err != nil {
|
if err := c.Bind(&message); err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
// these values are fixed
|
||||||
|
message.Channel = "api"
|
||||||
|
message.Protocol = "api"
|
||||||
|
message.Account = b.Account
|
||||||
|
message.ID = ""
|
||||||
|
message.Timestamp = time.Now()
|
||||||
flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
|
flog.Debugf("Sending message from %s on %s to gateway", message.Username, "api")
|
||||||
b.Remote <- config.Message{
|
b.Remote <- message
|
||||||
Text: message.Text,
|
|
||||||
Username: message.Username,
|
|
||||||
UserID: message.UserID,
|
|
||||||
Channel: "api",
|
|
||||||
Avatar: message.Avatar,
|
|
||||||
Account: b.Account,
|
|
||||||
Gateway: message.Gateway,
|
|
||||||
Protocol: "api",
|
|
||||||
}
|
|
||||||
return c.JSON(http.StatusOK, message)
|
return c.JSON(http.StatusOK, message)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -103,3 +104,24 @@ func (b *Api) handleMessages(c echo.Context) error {
|
|||||||
b.Messages = ring.Ring{}
|
b.Messages = ring.Ring{}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Api) handleStream(c echo.Context) error {
|
||||||
|
c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
||||||
|
c.Response().WriteHeader(http.StatusOK)
|
||||||
|
closeNotifier := c.Response().CloseNotify()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-closeNotifier:
|
||||||
|
return nil
|
||||||
|
default:
|
||||||
|
msg := b.Messages.Dequeue()
|
||||||
|
if msg != nil {
|
||||||
|
if err := json.NewEncoder(c.Response()).Encode(msg); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
c.Response().Flush()
|
||||||
|
}
|
||||||
|
time.Sleep(200 * time.Millisecond)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -10,10 +10,11 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge/mattermost"
|
"github.com/42wim/matterbridge/bridge/mattermost"
|
||||||
"github.com/42wim/matterbridge/bridge/rocketchat"
|
"github.com/42wim/matterbridge/bridge/rocketchat"
|
||||||
"github.com/42wim/matterbridge/bridge/slack"
|
"github.com/42wim/matterbridge/bridge/slack"
|
||||||
|
"github.com/42wim/matterbridge/bridge/sshchat"
|
||||||
"github.com/42wim/matterbridge/bridge/steam"
|
"github.com/42wim/matterbridge/bridge/steam"
|
||||||
"github.com/42wim/matterbridge/bridge/telegram"
|
"github.com/42wim/matterbridge/bridge/telegram"
|
||||||
"github.com/42wim/matterbridge/bridge/xmpp"
|
"github.com/42wim/matterbridge/bridge/xmpp"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
@ -35,6 +36,12 @@ type Bridge struct {
|
|||||||
Joined map[string]bool
|
Joined map[string]bool
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var flog *log.Entry
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flog = log.WithFields(log.Fields{"prefix": "bridge"})
|
||||||
|
}
|
||||||
|
|
||||||
func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge {
|
func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Bridge {
|
||||||
b := new(Bridge)
|
b := new(Bridge)
|
||||||
b.Channels = make(map[string]config.ChannelInfo)
|
b.Channels = make(map[string]config.ChannelInfo)
|
||||||
@ -45,44 +52,49 @@ func New(cfg *config.Config, bridge *config.Bridge, c chan config.Message) *Brid
|
|||||||
b.Protocol = protocol
|
b.Protocol = protocol
|
||||||
b.Account = bridge.Account
|
b.Account = bridge.Account
|
||||||
b.Joined = make(map[string]bool)
|
b.Joined = make(map[string]bool)
|
||||||
|
bridgeConfig := &config.BridgeConfig{General: &cfg.General, Account: bridge.Account, Remote: c}
|
||||||
|
|
||||||
// override config from environment
|
// override config from environment
|
||||||
config.OverrideCfgFromEnv(cfg, protocol, name)
|
config.OverrideCfgFromEnv(cfg, protocol, name)
|
||||||
switch protocol {
|
switch protocol {
|
||||||
case "mattermost":
|
case "mattermost":
|
||||||
b.Config = cfg.Mattermost[name]
|
bridgeConfig.Config = cfg.Mattermost[name]
|
||||||
b.Bridger = bmattermost.New(cfg.Mattermost[name], bridge.Account, c)
|
b.Bridger = bmattermost.New(bridgeConfig)
|
||||||
case "irc":
|
case "irc":
|
||||||
b.Config = cfg.IRC[name]
|
bridgeConfig.Config = cfg.IRC[name]
|
||||||
b.Bridger = birc.New(cfg.IRC[name], bridge.Account, c)
|
b.Bridger = birc.New(bridgeConfig)
|
||||||
case "gitter":
|
case "gitter":
|
||||||
b.Config = cfg.Gitter[name]
|
bridgeConfig.Config = cfg.Gitter[name]
|
||||||
b.Bridger = bgitter.New(cfg.Gitter[name], bridge.Account, c)
|
b.Bridger = bgitter.New(bridgeConfig)
|
||||||
case "slack":
|
case "slack":
|
||||||
b.Config = cfg.Slack[name]
|
bridgeConfig.Config = cfg.Slack[name]
|
||||||
b.Bridger = bslack.New(cfg.Slack[name], bridge.Account, c)
|
b.Bridger = bslack.New(bridgeConfig)
|
||||||
case "xmpp":
|
case "xmpp":
|
||||||
b.Config = cfg.Xmpp[name]
|
bridgeConfig.Config = cfg.Xmpp[name]
|
||||||
b.Bridger = bxmpp.New(cfg.Xmpp[name], bridge.Account, c)
|
b.Bridger = bxmpp.New(bridgeConfig)
|
||||||
case "discord":
|
case "discord":
|
||||||
b.Config = cfg.Discord[name]
|
bridgeConfig.Config = cfg.Discord[name]
|
||||||
b.Bridger = bdiscord.New(cfg.Discord[name], bridge.Account, c)
|
b.Bridger = bdiscord.New(bridgeConfig)
|
||||||
case "telegram":
|
case "telegram":
|
||||||
b.Config = cfg.Telegram[name]
|
bridgeConfig.Config = cfg.Telegram[name]
|
||||||
b.Bridger = btelegram.New(cfg.Telegram[name], bridge.Account, c)
|
b.Bridger = btelegram.New(bridgeConfig)
|
||||||
case "rocketchat":
|
case "rocketchat":
|
||||||
b.Config = cfg.Rocketchat[name]
|
bridgeConfig.Config = cfg.Rocketchat[name]
|
||||||
b.Bridger = brocketchat.New(cfg.Rocketchat[name], bridge.Account, c)
|
b.Bridger = brocketchat.New(bridgeConfig)
|
||||||
case "matrix":
|
case "matrix":
|
||||||
b.Config = cfg.Matrix[name]
|
bridgeConfig.Config = cfg.Matrix[name]
|
||||||
b.Bridger = bmatrix.New(cfg.Matrix[name], bridge.Account, c)
|
b.Bridger = bmatrix.New(bridgeConfig)
|
||||||
case "steam":
|
case "steam":
|
||||||
b.Config = cfg.Steam[name]
|
bridgeConfig.Config = cfg.Steam[name]
|
||||||
b.Bridger = bsteam.New(cfg.Steam[name], bridge.Account, c)
|
b.Bridger = bsteam.New(bridgeConfig)
|
||||||
|
case "sshchat":
|
||||||
|
bridgeConfig.Config = cfg.Sshchat[name]
|
||||||
|
b.Bridger = bsshchat.New(bridgeConfig)
|
||||||
case "api":
|
case "api":
|
||||||
b.Config = cfg.Api[name]
|
bridgeConfig.Config = cfg.Api[name]
|
||||||
b.Bridger = api.New(cfg.Api[name], bridge.Account, c)
|
b.Bridger = api.New(bridgeConfig)
|
||||||
}
|
}
|
||||||
|
b.Config = bridgeConfig.Config
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -94,7 +106,7 @@ func (b *Bridge) JoinChannels() error {
|
|||||||
func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error {
|
func (b *Bridge) joinChannels(channels map[string]config.ChannelInfo, exists map[string]bool) error {
|
||||||
for ID, channel := range channels {
|
for ID, channel := range channels {
|
||||||
if !exists[ID] {
|
if !exists[ID] {
|
||||||
log.Infof("%s: joining %s (%s)", b.Account, channel.Name, ID)
|
flog.Infof("%s: joining %s (ID: %s)", b.Account, channel.Name, ID)
|
||||||
err := b.JoinChannel(channel)
|
err := b.JoinChannel(channel)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
|
@ -10,11 +10,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
EVENT_JOIN_LEAVE = "join_leave"
|
EVENT_JOIN_LEAVE = "join_leave"
|
||||||
EVENT_FAILURE = "failure"
|
EVENT_TOPIC_CHANGE = "topic_change"
|
||||||
EVENT_REJOIN_CHANNELS = "rejoin_channels"
|
EVENT_FAILURE = "failure"
|
||||||
EVENT_USER_ACTION = "user_action"
|
EVENT_FILE_FAILURE_SIZE = "file_failure_size"
|
||||||
EVENT_MSG_DELETE = "msg_delete"
|
EVENT_AVATAR_DOWNLOAD = "avatar_download"
|
||||||
|
EVENT_REJOIN_CHANNELS = "rejoin_channels"
|
||||||
|
EVENT_USER_ACTION = "user_action"
|
||||||
|
EVENT_MSG_DELETE = "msg_delete"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Message struct {
|
type Message struct {
|
||||||
@ -33,8 +36,13 @@ type Message struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type FileInfo struct {
|
type FileInfo struct {
|
||||||
Name string
|
Name string
|
||||||
Data *[]byte
|
Data *[]byte
|
||||||
|
Comment string
|
||||||
|
URL string
|
||||||
|
Size int64
|
||||||
|
Avatar bool
|
||||||
|
SHA string
|
||||||
}
|
}
|
||||||
|
|
||||||
type ChannelInfo struct {
|
type ChannelInfo struct {
|
||||||
@ -51,48 +59,58 @@ type Protocol struct {
|
|||||||
BindAddress string // mattermost, slack // DEPRECATED
|
BindAddress string // mattermost, slack // DEPRECATED
|
||||||
Buffer int // api
|
Buffer int // api
|
||||||
Charset string // irc
|
Charset string // irc
|
||||||
|
Debug bool // general
|
||||||
EditSuffix string // mattermost, slack, discord, telegram, gitter
|
EditSuffix string // mattermost, slack, discord, telegram, gitter
|
||||||
EditDisable bool // mattermost, slack, discord, telegram, gitter
|
EditDisable bool // mattermost, slack, discord, telegram, gitter
|
||||||
IconURL string // mattermost, slack
|
IconURL string // mattermost, slack
|
||||||
IgnoreNicks string // all protocols
|
IgnoreNicks string // all protocols
|
||||||
IgnoreMessages string // all protocols
|
IgnoreMessages string // all protocols
|
||||||
Jid string // xmpp
|
Jid string // xmpp
|
||||||
|
Label string // all protocols
|
||||||
Login string // mattermost, matrix
|
Login string // mattermost, matrix
|
||||||
Muc string // xmpp
|
MediaDownloadSize int // all protocols
|
||||||
Name string // all protocols
|
MediaServerDownload string
|
||||||
Nick string // all protocols
|
MediaServerUpload string
|
||||||
NickFormatter string // mattermost, slack
|
MessageDelay int // IRC, time in millisecond to wait between messages
|
||||||
NickServNick string // IRC
|
MessageFormat string // telegram
|
||||||
NickServUsername string // IRC
|
MessageLength int // IRC, max length of a message allowed
|
||||||
NickServPassword string // IRC
|
MessageQueue int // IRC, size of message queue for flood control
|
||||||
NicksPerRow int // mattermost, slack
|
MessageSplit bool // IRC, split long messages with newlines on MessageLength instead of clipping
|
||||||
NoHomeServerSuffix bool // matrix
|
Muc string // xmpp
|
||||||
NoTLS bool // mattermost
|
Name string // all protocols
|
||||||
Password string // IRC,mattermost,XMPP,matrix
|
Nick string // all protocols
|
||||||
PrefixMessagesWithNick bool // mattemost, slack
|
NickFormatter string // mattermost, slack
|
||||||
Protocol string //all protocols
|
NickServNick string // IRC
|
||||||
MessageQueue int // IRC, size of message queue for flood control
|
NickServUsername string // IRC
|
||||||
MessageDelay int // IRC, time in millisecond to wait between messages
|
NickServPassword string // IRC
|
||||||
MessageLength int // IRC, max length of a message allowed
|
NicksPerRow int // mattermost, slack
|
||||||
MessageFormat string // telegram
|
NoHomeServerSuffix bool // matrix
|
||||||
RemoteNickFormat string // all protocols
|
NoTLS bool // mattermost
|
||||||
Server string // IRC,mattermost,XMPP,discord
|
Password string // IRC,mattermost,XMPP,matrix
|
||||||
ShowJoinPart bool // all protocols
|
PrefixMessagesWithNick bool // mattemost, slack
|
||||||
ShowEmbeds bool // discord
|
Protocol string // all protocols
|
||||||
SkipTLSVerify bool // IRC, mattermost
|
RejoinDelay int // IRC
|
||||||
StripNick bool // all protocols
|
ReplaceMessages [][]string // all protocols
|
||||||
Team string // mattermost
|
ReplaceNicks [][]string // all protocols
|
||||||
Token string // gitter, slack, discord, api
|
RemoteNickFormat string // all protocols
|
||||||
URL string // mattermost, slack // DEPRECATED
|
Server string // IRC,mattermost,XMPP,discord
|
||||||
UseAPI bool // mattermost, slack
|
ShowJoinPart bool // all protocols
|
||||||
UseSASL bool // IRC
|
ShowTopicChange bool // slack
|
||||||
UseTLS bool // IRC
|
ShowEmbeds bool // discord
|
||||||
UseFirstName bool // telegram
|
SkipTLSVerify bool // IRC, mattermost
|
||||||
UseUserName bool // discord
|
StripNick bool // all protocols
|
||||||
UseInsecureURL bool // telegram
|
Team string // mattermost
|
||||||
WebhookBindAddress string // mattermost, slack
|
Token string // gitter, slack, discord, api
|
||||||
WebhookURL string // mattermost, slack
|
URL string // mattermost, slack // DEPRECATED
|
||||||
WebhookUse string // mattermost, slack, discord
|
UseAPI bool // mattermost, slack
|
||||||
|
UseSASL bool // IRC
|
||||||
|
UseTLS bool // IRC
|
||||||
|
UseFirstName bool // telegram
|
||||||
|
UseUserName bool // discord
|
||||||
|
UseInsecureURL bool // telegram
|
||||||
|
WebhookBindAddress string // mattermost, slack
|
||||||
|
WebhookURL string // mattermost, slack
|
||||||
|
WebhookUse string // mattermost, slack, discord
|
||||||
}
|
}
|
||||||
|
|
||||||
type ChannelOptions struct {
|
type ChannelOptions struct {
|
||||||
@ -134,11 +152,19 @@ type Config struct {
|
|||||||
Discord map[string]Protocol
|
Discord map[string]Protocol
|
||||||
Telegram map[string]Protocol
|
Telegram map[string]Protocol
|
||||||
Rocketchat map[string]Protocol
|
Rocketchat map[string]Protocol
|
||||||
|
Sshchat map[string]Protocol
|
||||||
General Protocol
|
General Protocol
|
||||||
Gateway []Gateway
|
Gateway []Gateway
|
||||||
SameChannelGateway []SameChannelGateway
|
SameChannelGateway []SameChannelGateway
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type BridgeConfig struct {
|
||||||
|
Config Protocol
|
||||||
|
General *Protocol
|
||||||
|
Account string
|
||||||
|
Remote chan Message
|
||||||
|
}
|
||||||
|
|
||||||
func NewConfig(cfgfile string) *Config {
|
func NewConfig(cfgfile string) *Config {
|
||||||
var cfg Config
|
var cfg Config
|
||||||
if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil {
|
if _, err := toml.DecodeFile(cfgfile, &cfg); err != nil {
|
||||||
@ -166,6 +192,9 @@ func NewConfig(cfgfile string) *Config {
|
|||||||
if fail {
|
if fail {
|
||||||
log.Fatalf("Fix your config. Please see changelog for more information")
|
log.Fatalf("Fix your config. Please see changelog for more information")
|
||||||
}
|
}
|
||||||
|
if cfg.General.MediaDownloadSize == 0 {
|
||||||
|
cfg.General.MediaDownloadSize = 1000000
|
||||||
|
}
|
||||||
return &cfg
|
return &cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,8 +3,9 @@ package bdiscord
|
|||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/bwmarrin/discordgo"
|
"github.com/bwmarrin/discordgo"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
@ -12,9 +13,6 @@ import (
|
|||||||
|
|
||||||
type bdiscord struct {
|
type bdiscord struct {
|
||||||
c *discordgo.Session
|
c *discordgo.Session
|
||||||
Config *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
Channels []*discordgo.Channel
|
Channels []*discordgo.Channel
|
||||||
Nick string
|
Nick string
|
||||||
UseChannelID bool
|
UseChannelID bool
|
||||||
@ -24,20 +22,18 @@ type bdiscord struct {
|
|||||||
webhookToken string
|
webhookToken string
|
||||||
channelInfoMap map[string]*config.ChannelInfo
|
channelInfoMap map[string]*config.ChannelInfo
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "discord"
|
var protocol = "discord"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *bdiscord {
|
func New(cfg *config.BridgeConfig) *bdiscord {
|
||||||
b := &bdiscord{}
|
b := &bdiscord{BridgeConfig: cfg}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
b.userMemberMap = make(map[string]*discordgo.Member)
|
b.userMemberMap = make(map[string]*discordgo.Member)
|
||||||
b.channelInfoMap = make(map[string]*config.ChannelInfo)
|
b.channelInfoMap = make(map[string]*config.ChannelInfo)
|
||||||
if b.Config.WebhookURL != "" {
|
if b.Config.WebhookURL != "" {
|
||||||
@ -144,6 +140,9 @@ func (b *bdiscord) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.c.ChannelMessageSend(channelID, rmsg.Username+rmsg.Text)
|
||||||
|
}
|
||||||
// check if we have files to upload (from slack, telegram or mattermost)
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
if len(msg.Extra["file"]) > 0 {
|
if len(msg.Extra["file"]) > 0 {
|
||||||
var err error
|
var err error
|
||||||
@ -151,11 +150,12 @@ func (b *bdiscord) Send(msg config.Message) (string, error) {
|
|||||||
fi := f.(config.FileInfo)
|
fi := f.(config.FileInfo)
|
||||||
files := []*discordgo.File{}
|
files := []*discordgo.File{}
|
||||||
files = append(files, &discordgo.File{fi.Name, "", bytes.NewReader(*fi.Data)})
|
files = append(files, &discordgo.File{fi.Name, "", bytes.NewReader(*fi.Data)})
|
||||||
_, err = b.c.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{Content: msg.Text, Files: files})
|
_, err = b.c.ChannelMessageSendComplex(channelID, &discordgo.MessageSend{Content: msg.Username + fi.Comment, Files: files})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("file upload failed: %#v", err)
|
flog.Errorf("file upload failed: %#v", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return "", nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -202,6 +202,7 @@ func (b *bdiscord) messageUpdate(s *discordgo.Session, m *discordgo.MessageUpdat
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
|
||||||
|
var err error
|
||||||
// not relay our own messages
|
// not relay our own messages
|
||||||
if m.Author.Username == b.Nick {
|
if m.Author.Username == b.Nick {
|
||||||
return
|
return
|
||||||
@ -220,12 +221,13 @@ func (b *bdiscord) messageCreate(s *discordgo.Session, m *discordgo.MessageCreat
|
|||||||
var text string
|
var text string
|
||||||
if m.Content != "" {
|
if m.Content != "" {
|
||||||
flog.Debugf("Receiving message %#v", m.Message)
|
flog.Debugf("Receiving message %#v", m.Message)
|
||||||
if len(m.MentionRoles) > 0 {
|
|
||||||
m.Message.Content = b.replaceRoleMentions(m.Message.Content)
|
|
||||||
}
|
|
||||||
m.Message.Content = b.stripCustomoji(m.Message.Content)
|
m.Message.Content = b.stripCustomoji(m.Message.Content)
|
||||||
m.Message.Content = b.replaceChannelMentions(m.Message.Content)
|
m.Message.Content = b.replaceChannelMentions(m.Message.Content)
|
||||||
text = m.ContentWithMentionsReplaced()
|
text, err = m.ContentWithMoreMentionsReplaced(b.c)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("ContentWithMoreMentionsReplaced failed: %s", err)
|
||||||
|
text = m.ContentWithMentionsReplaced()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg",
|
rmsg := config.Message{Account: b.Account, Avatar: "https://cdn.discordapp.com/avatars/" + m.Author.ID + "/" + m.Author.Avatar + ".jpg",
|
||||||
@ -322,18 +324,6 @@ func (b *bdiscord) getChannelName(id string) string {
|
|||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *bdiscord) replaceRoleMentions(text string) string {
|
|
||||||
roles, err := b.c.GuildRoles(b.guildID)
|
|
||||||
if err != nil {
|
|
||||||
flog.Debugf("%#v", string(err.(*discordgo.RESTError).ResponseBody))
|
|
||||||
return text
|
|
||||||
}
|
|
||||||
for _, role := range roles {
|
|
||||||
text = strings.Replace(text, "<@&"+role.ID+">", "@"+role.Name, -1)
|
|
||||||
}
|
|
||||||
return text
|
|
||||||
}
|
|
||||||
|
|
||||||
func (b *bdiscord) replaceChannelMentions(text string) string {
|
func (b *bdiscord) replaceChannelMentions(text string) string {
|
||||||
var err error
|
var err error
|
||||||
re := regexp.MustCompile("<#[0-9]+>")
|
re := regexp.MustCompile("<#[0-9]+>")
|
||||||
@ -369,6 +359,9 @@ func (b *bdiscord) stripCustomoji(text string) string {
|
|||||||
// splitURL splits a webhookURL and returns the id and token
|
// splitURL splits a webhookURL and returns the id and token
|
||||||
func (b *bdiscord) splitURL(url string) (string, string) {
|
func (b *bdiscord) splitURL(url string) (string, string) {
|
||||||
webhookURLSplit := strings.Split(url, "/")
|
webhookURLSplit := strings.Split(url, "/")
|
||||||
|
if len(webhookURLSplit) != 7 {
|
||||||
|
log.Fatalf("%s is no correct discord WebhookURL", url)
|
||||||
|
}
|
||||||
return webhookURLSplit[len(webhookURLSplit)-2], webhookURLSplit[len(webhookURLSplit)-1]
|
return webhookURLSplit[len(webhookURLSplit)-2], webhookURLSplit[len(webhookURLSplit)-1]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -4,33 +4,28 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/go-gitter"
|
"github.com/42wim/go-gitter"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Bgitter struct {
|
type Bgitter struct {
|
||||||
c *gitter.Gitter
|
c *gitter.Gitter
|
||||||
Config *config.Protocol
|
User *gitter.User
|
||||||
Remote chan config.Message
|
Users []gitter.User
|
||||||
Account string
|
Rooms []gitter.Room
|
||||||
User *gitter.User
|
*config.BridgeConfig
|
||||||
Users []gitter.User
|
|
||||||
Rooms []gitter.Room
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "gitter"
|
var protocol = "gitter"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bgitter {
|
func New(cfg *config.BridgeConfig) *Bgitter {
|
||||||
b := &Bgitter{}
|
return &Bgitter{BridgeConfig: cfg}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
return b
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bgitter) Connect() error {
|
func (b *Bgitter) Connect() error {
|
||||||
@ -125,6 +120,29 @@ func (b *Bgitter) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.c.SendMessage(roomID, rmsg.Username+rmsg.Text)
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.Comment != "" {
|
||||||
|
msg.Text += fi.Comment + ": "
|
||||||
|
}
|
||||||
|
if fi.URL != "" {
|
||||||
|
msg.Text = fi.URL
|
||||||
|
}
|
||||||
|
_, err := b.c.SendMessage(roomID, msg.Username+msg.Text)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
resp, err := b.c.SendMessage(roomID, msg.Username+msg.Text)
|
resp, err := b.c.SendMessage(roomID, msg.Username+msg.Text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
|
@ -2,6 +2,8 @@ package helper
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
@ -18,11 +20,44 @@ func DownloadFile(url string) (*[]byte, error) {
|
|||||||
}
|
}
|
||||||
resp, err := client.Do(req)
|
resp, err := client.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
resp.Body.Close()
|
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
io.Copy(&buf, resp.Body)
|
io.Copy(&buf, resp.Body)
|
||||||
data := buf.Bytes()
|
data := buf.Bytes()
|
||||||
resp.Body.Close()
|
|
||||||
return &data, nil
|
return &data, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SplitStringLength(input string, length int) string {
|
||||||
|
a := []rune(input)
|
||||||
|
str := ""
|
||||||
|
for i, r := range a {
|
||||||
|
str = str + string(r)
|
||||||
|
if i > 0 && (i+1)%length == 0 {
|
||||||
|
str += "\n"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return str
|
||||||
|
}
|
||||||
|
|
||||||
|
// handle all the stuff we put into extra
|
||||||
|
func HandleExtra(msg *config.Message, general *config.Protocol) []config.Message {
|
||||||
|
extra := msg.Extra
|
||||||
|
rmsg := []config.Message{}
|
||||||
|
if len(extra[config.EVENT_FILE_FAILURE_SIZE]) > 0 {
|
||||||
|
for _, f := range extra[config.EVENT_FILE_FAILURE_SIZE] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
text := fmt.Sprintf("file %s too big to download (%#v > allowed size: %#v)", fi.Name, fi.Size, general.MediaDownloadSize)
|
||||||
|
rmsg = append(rmsg, config.Message{Text: text, Username: "<system> ", Channel: msg.Channel})
|
||||||
|
}
|
||||||
|
return rmsg
|
||||||
|
}
|
||||||
|
return rmsg
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetAvatar(av map[string]string, userid string, general *config.Protocol) string {
|
||||||
|
if sha, ok := av[userid]; ok {
|
||||||
|
return general.MediaServerDownload + "/" + sha + "/" + userid + ".png"
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
@ -5,11 +5,12 @@ import (
|
|||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/lrstanley/girc"
|
"github.com/lrstanley/girc"
|
||||||
"github.com/paulrosania/go-charset/charset"
|
"github.com/paulrosania/go-charset/charset"
|
||||||
_ "github.com/paulrosania/go-charset/data"
|
_ "github.com/paulrosania/go-charset/data"
|
||||||
"github.com/saintfish/chardet"
|
"github.com/saintfish/chardet"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"net"
|
"net"
|
||||||
@ -18,34 +19,32 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
"unicode/utf8"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Birc struct {
|
type Birc struct {
|
||||||
i *girc.Client
|
i *girc.Client
|
||||||
Nick string
|
Nick string
|
||||||
names map[string][]string
|
names map[string][]string
|
||||||
Config *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
connected chan struct{}
|
connected chan struct{}
|
||||||
Local chan config.Message // local queue for flood control
|
Local chan config.Message // local queue for flood control
|
||||||
Account string
|
|
||||||
FirstConnection bool
|
FirstConnection bool
|
||||||
|
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "irc"
|
var protocol = "irc"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Birc {
|
func New(cfg *config.BridgeConfig) *Birc {
|
||||||
b := &Birc{}
|
b := &Birc{}
|
||||||
b.Config = &cfg
|
b.BridgeConfig = cfg
|
||||||
b.Nick = b.Config.Nick
|
b.Nick = b.Config.Nick
|
||||||
b.Remote = c
|
|
||||||
b.names = make(map[string][]string)
|
b.names = make(map[string][]string)
|
||||||
b.Account = account
|
|
||||||
b.connected = make(chan struct{})
|
b.connected = make(chan struct{})
|
||||||
if b.Config.MessageDelay == 0 {
|
if b.Config.MessageDelay == 0 {
|
||||||
b.Config.MessageDelay = 1300
|
b.Config.MessageDelay = 1300
|
||||||
@ -81,12 +80,22 @@ func (b *Birc) Connect() error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
// fix strict user handling of girc
|
||||||
|
user := b.Config.Nick
|
||||||
|
for !girc.IsValidUser(user) {
|
||||||
|
if len(user) == 1 {
|
||||||
|
user = "matterbridge"
|
||||||
|
break
|
||||||
|
}
|
||||||
|
user = user[1:]
|
||||||
|
}
|
||||||
|
|
||||||
i := girc.New(girc.Config{
|
i := girc.New(girc.Config{
|
||||||
Server: server,
|
Server: server,
|
||||||
ServerPass: b.Config.Password,
|
ServerPass: b.Config.Password,
|
||||||
Port: port,
|
Port: port,
|
||||||
Nick: b.Config.Nick,
|
Nick: b.Config.Nick,
|
||||||
User: b.Config.Nick,
|
User: user,
|
||||||
Name: b.Config.Nick,
|
Name: b.Config.Nick,
|
||||||
SSL: b.Config.UseTLS,
|
SSL: b.Config.UseTLS,
|
||||||
TLSConfig: &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, ServerName: server},
|
TLSConfig: &tls.Config{InsecureSkipVerify: b.Config.SkipTLSVerify, ServerName: server},
|
||||||
@ -168,9 +177,36 @@ func (b *Birc) Send(msg config.Message) (string, error) {
|
|||||||
msg.Text = buf.String()
|
msg.Text = buf.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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
|
||||||
|
}
|
||||||
|
b.Local <- config.Message{Text: msg.Text, Username: msg.Username, Channel: msg.Channel, Event: msg.Event}
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// split long messages on messageLength, to avoid clipped messages #281
|
||||||
|
if b.Config.MessageSplit {
|
||||||
|
msg.Text = helper.SplitStringLength(msg.Text, b.Config.MessageLength)
|
||||||
|
}
|
||||||
for _, text := range strings.Split(msg.Text, "\n") {
|
for _, text := range strings.Split(msg.Text, "\n") {
|
||||||
if len(text) > b.Config.MessageLength {
|
if len(text) > b.Config.MessageLength {
|
||||||
text = text[:b.Config.MessageLength] + " <message clipped>"
|
text = text[:b.Config.MessageLength-len(" <message clipped>")]
|
||||||
|
if r, size := utf8.DecodeLastRuneInString(text); r == utf8.RuneError {
|
||||||
|
text = text[:len(text)-size]
|
||||||
|
}
|
||||||
|
text += " <message clipped>"
|
||||||
}
|
}
|
||||||
if len(b.Local) < b.Config.MessageQueue {
|
if len(b.Local) < b.Config.MessageQueue {
|
||||||
if len(b.Local) == b.Config.MessageQueue-1 {
|
if len(b.Local) == b.Config.MessageQueue-1 {
|
||||||
@ -241,6 +277,7 @@ func (b *Birc) handleJoinPart(client *girc.Client, event girc.Event) {
|
|||||||
channel := event.Params[0]
|
channel := event.Params[0]
|
||||||
if event.Command == "KICK" {
|
if event.Command == "KICK" {
|
||||||
flog.Infof("Got kicked from %s by %s", channel, event.Source.Name)
|
flog.Infof("Got kicked from %s by %s", channel, event.Source.Name)
|
||||||
|
time.Sleep(time.Duration(b.Config.RejoinDelay) * time.Second)
|
||||||
b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS}
|
b.Remote <- config.Message{Username: "system", Text: "rejoin", Channel: channel, Account: b.Account, Event: config.EVENT_REJOIN_CHANNELS}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -296,14 +333,13 @@ func (b *Birc) handlePrivMsg(client *girc.Client, event girc.Event) {
|
|||||||
if event.Source.Name == b.Nick {
|
if event.Source.Name == b.Nick {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
rmsg := config.Message{Username: event.Source.Name, Channel: event.Params[0], Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host}
|
rmsg := config.Message{Username: event.Source.Name, Channel: strings.ToLower(event.Params[0]), Account: b.Account, UserID: event.Source.Ident + "@" + event.Source.Host}
|
||||||
flog.Debugf("handlePrivMsg() %s %s %#v", event.Source.Name, event.Trailing, event)
|
flog.Debugf("handlePrivMsg() %s %s %#v", event.Source.Name, event.Trailing, event)
|
||||||
msg := ""
|
msg := ""
|
||||||
if event.Command == "CTCP_ACTION" {
|
if event.IsAction() {
|
||||||
// msg = event.Source.Name + " "
|
|
||||||
rmsg.Event = config.EVENT_USER_ACTION
|
rmsg.Event = config.EVENT_USER_ACTION
|
||||||
}
|
}
|
||||||
msg += event.Trailing
|
msg += event.StripAction()
|
||||||
// strip IRC colors
|
// strip IRC colors
|
||||||
re := regexp.MustCompile(`[[:cntrl:]](?:\d{1,2}(?:,\d{1,2})?)?`)
|
re := regexp.MustCompile(`[[:cntrl:]](?:\d{1,2}(?:,\d{1,2})?)?`)
|
||||||
msg = re.ReplaceAllString(msg, "")
|
msg = re.ReplaceAllString(msg, "")
|
||||||
|
@ -1,37 +1,36 @@
|
|||||||
package bmatrix
|
package bmatrix
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"mime"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
matrix "github.com/matrix-org/gomatrix"
|
log "github.com/sirupsen/logrus"
|
||||||
|
matrix "github.com/matterbridge/gomatrix"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Bmatrix struct {
|
type Bmatrix struct {
|
||||||
mc *matrix.Client
|
mc *matrix.Client
|
||||||
Config *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
UserID string
|
UserID string
|
||||||
RoomMap map[string]string
|
RoomMap map[string]string
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "matrix"
|
var protocol = "matrix"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bmatrix {
|
func New(cfg *config.BridgeConfig) *Bmatrix {
|
||||||
b := &Bmatrix{}
|
b := &Bmatrix{BridgeConfig: cfg}
|
||||||
b.RoomMap = make(map[string]string)
|
b.RoomMap = make(map[string]string)
|
||||||
b.Config = &cfg
|
|
||||||
b.Account = account
|
|
||||||
b.Remote = c
|
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -76,19 +75,79 @@ func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error {
|
|||||||
|
|
||||||
func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
func (b *Bmatrix) Send(msg config.Message) (string, error) {
|
||||||
flog.Debugf("Receiving %#v", msg)
|
flog.Debugf("Receiving %#v", msg)
|
||||||
|
channel := b.getRoomID(msg.Channel)
|
||||||
// ignore delete messages
|
// ignore delete messages
|
||||||
if msg.Event == config.EVENT_MSG_DELETE {
|
if msg.Event == config.EVENT_MSG_DELETE {
|
||||||
return "", nil
|
if msg.ID == "" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{})
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return resp.EventID, err
|
||||||
}
|
}
|
||||||
channel := b.getRoomID(msg.Channel)
|
|
||||||
flog.Debugf("Sending to channel %s", channel)
|
flog.Debugf("Sending to channel %s", channel)
|
||||||
if msg.Event == config.EVENT_USER_ACTION {
|
if msg.Event == config.EVENT_USER_ACTION {
|
||||||
b.mc.SendMessageEvent(channel, "m.room.message",
|
resp, err := b.mc.SendMessageEvent(channel, "m.room.message",
|
||||||
matrix.TextMessage{"m.emote", msg.Username + msg.Text})
|
matrix.TextMessage{"m.emote", msg.Username + msg.Text})
|
||||||
return "", nil
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return resp.EventID, err
|
||||||
}
|
}
|
||||||
b.mc.SendText(channel, msg.Username+msg.Text)
|
|
||||||
return "", nil
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.mc.SendText(channel, rmsg.Username+rmsg.Text)
|
||||||
|
}
|
||||||
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
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 {
|
||||||
|
flog.Errorf("file comment failed: %#v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
flog.Debugf("uploading file: %s %s", fi.Name, mtype)
|
||||||
|
res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data)))
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("file upload failed: %#v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.Contains(mtype, "video") {
|
||||||
|
flog.Debugf("sendVideo %s", res.ContentURI)
|
||||||
|
_, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("sendVideo failed: %#v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if strings.Contains(mtype, "image") {
|
||||||
|
flog.Debugf("sendImage %s", res.ContentURI)
|
||||||
|
_, err = b.mc.SendImage(channel, fi.Name, res.ContentURI)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("sendImage failed: %#v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
flog.Debugf("result: %#v", res)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := b.mc.SendText(channel, msg.Username+msg.Text)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return resp.EventID, err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) getRoomID(channel string) string {
|
func (b *Bmatrix) getRoomID(channel string) string {
|
||||||
@ -101,31 +160,11 @@ func (b *Bmatrix) getRoomID(channel string) string {
|
|||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bmatrix) handlematrix() error {
|
func (b *Bmatrix) handlematrix() error {
|
||||||
syncer := b.mc.Syncer.(*matrix.DefaultSyncer)
|
syncer := b.mc.Syncer.(*matrix.DefaultSyncer)
|
||||||
syncer.OnEventType("m.room.message", func(ev *matrix.Event) {
|
syncer.OnEventType("m.room.redaction", b.handleEvent)
|
||||||
if (ev.Content["msgtype"].(string) == "m.text" || ev.Content["msgtype"].(string) == "m.notice" || ev.Content["msgtype"].(string) == "m.emote") && ev.Sender != b.UserID {
|
syncer.OnEventType("m.room.message", b.handleEvent)
|
||||||
b.RLock()
|
|
||||||
channel, ok := b.RoomMap[ev.RoomID]
|
|
||||||
b.RUnlock()
|
|
||||||
if !ok {
|
|
||||||
flog.Debugf("Unknown room %s", ev.RoomID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
username := ev.Sender[1:]
|
|
||||||
if b.Config.NoHomeServerSuffix {
|
|
||||||
re := regexp.MustCompile("(.*?):.*")
|
|
||||||
username = re.ReplaceAllString(username, `$1`)
|
|
||||||
}
|
|
||||||
rmsg := config.Message{Username: username, Text: ev.Content["body"].(string), Channel: channel, Account: b.Account, UserID: ev.Sender}
|
|
||||||
if ev.Content["msgtype"].(string) == "m.emote" {
|
|
||||||
rmsg.Event = config.EVENT_USER_ACTION
|
|
||||||
}
|
|
||||||
flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account)
|
|
||||||
b.Remote <- rmsg
|
|
||||||
}
|
|
||||||
flog.Debugf("Received: %#v", ev)
|
|
||||||
})
|
|
||||||
go func() {
|
go func() {
|
||||||
for {
|
for {
|
||||||
if err := b.mc.Sync(); err != nil {
|
if err := b.mc.Sync(); err != nil {
|
||||||
@ -135,3 +174,77 @@ func (b *Bmatrix) handlematrix() error {
|
|||||||
}()
|
}()
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Bmatrix) handleEvent(ev *matrix.Event) {
|
||||||
|
flog.Debugf("Received: %#v", ev)
|
||||||
|
if ev.Sender != b.UserID {
|
||||||
|
b.RLock()
|
||||||
|
channel, ok := b.RoomMap[ev.RoomID]
|
||||||
|
b.RUnlock()
|
||||||
|
if !ok {
|
||||||
|
flog.Debugf("Unknown room %s", ev.RoomID)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
username := ev.Sender[1:]
|
||||||
|
if b.Config.NoHomeServerSuffix {
|
||||||
|
re := regexp.MustCompile("(.*?):.*")
|
||||||
|
username = re.ReplaceAllString(username, `$1`)
|
||||||
|
}
|
||||||
|
var text string
|
||||||
|
text, _ = ev.Content["body"].(string)
|
||||||
|
rmsg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: ev.Sender}
|
||||||
|
rmsg.ID = ev.ID
|
||||||
|
if ev.Type == "m.room.redaction" {
|
||||||
|
rmsg.Event = config.EVENT_MSG_DELETE
|
||||||
|
rmsg.ID = ev.Redacts
|
||||||
|
rmsg.Text = config.EVENT_MSG_DELETE
|
||||||
|
b.Remote <- rmsg
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if ev.Content["msgtype"].(string) == "m.emote" {
|
||||||
|
rmsg.Event = config.EVENT_USER_ACTION
|
||||||
|
}
|
||||||
|
if ev.Content["msgtype"] != nil && ev.Content["msgtype"].(string) == "m.image" ||
|
||||||
|
ev.Content["msgtype"].(string) == "m.video" ||
|
||||||
|
ev.Content["msgtype"].(string) == "m.file" {
|
||||||
|
flog.Debugf("ev: %#v", ev)
|
||||||
|
rmsg.Extra = make(map[string][]interface{})
|
||||||
|
url := ev.Content["url"].(string)
|
||||||
|
url = strings.Replace(url, "mxc://", b.Config.Server+"/_matrix/media/v1/download/", -1)
|
||||||
|
info := ev.Content["info"].(map[string]interface{})
|
||||||
|
size := info["size"].(float64)
|
||||||
|
name := ev.Content["body"].(string)
|
||||||
|
// check if we have an image uploaded without extension
|
||||||
|
if !strings.Contains(name, ".") {
|
||||||
|
if ev.Content["msgtype"].(string) == "m.image" {
|
||||||
|
if mtype, ok := ev.Content["mimetype"].(string); ok {
|
||||||
|
mext, _ := mime.ExtensionsByType(mtype)
|
||||||
|
if len(mext) > 0 {
|
||||||
|
name = name + mext[0]
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// just a default .png extension if we don't have mime info
|
||||||
|
name = name + ".png"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
flog.Debugf("trying to download %#v with size %#v", name, size)
|
||||||
|
if size <= float64(b.General.MediaDownloadSize) {
|
||||||
|
data, err := helper.DownloadFile(url)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("download %s failed %#v", url, err)
|
||||||
|
} else {
|
||||||
|
flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url))
|
||||||
|
rmsg.Extra["file"] = append(rmsg.Extra["file"], config.FileInfo{Name: name, Data: data})
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, b.General.MediaDownloadSize)
|
||||||
|
rmsg.Event = config.EVENT_FILE_FAILURE_SIZE
|
||||||
|
rmsg.Extra[rmsg.Event] = append(rmsg.Extra[rmsg.Event], config.FileInfo{Name: name, Size: int64(size)})
|
||||||
|
}
|
||||||
|
rmsg.Text = ""
|
||||||
|
}
|
||||||
|
flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account)
|
||||||
|
b.Remote <- rmsg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -4,9 +4,10 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/matterclient"
|
"github.com/42wim/matterbridge/matterclient"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -32,24 +33,20 @@ type MMMessage struct {
|
|||||||
type Bmattermost struct {
|
type Bmattermost struct {
|
||||||
MMhook
|
MMhook
|
||||||
MMapi
|
MMapi
|
||||||
Config *config.Protocol
|
TeamId string
|
||||||
Remote chan config.Message
|
*config.BridgeConfig
|
||||||
TeamId string
|
avatarMap map[string]string
|
||||||
Account string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "mattermost"
|
var protocol = "mattermost"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bmattermost {
|
func New(cfg *config.BridgeConfig) *Bmattermost {
|
||||||
b := &Bmattermost{}
|
b := &Bmattermost{BridgeConfig: cfg, avatarMap: make(map[string]string)}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
b.mmMap = make(map[string]string)
|
b.mmMap = make(map[string]string)
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
@ -153,17 +150,46 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
|
|||||||
message := msg.Text
|
message := msg.Text
|
||||||
channel := msg.Channel
|
channel := msg.Channel
|
||||||
|
|
||||||
|
// map the file SHA to our user (caches the avatar)
|
||||||
|
if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
|
||||||
|
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 != "" {
|
||||||
|
flog.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
|
||||||
|
b.avatarMap[msg.UserID] = fi.SHA
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
if b.Config.PrefixMessagesWithNick {
|
if b.Config.PrefixMessagesWithNick {
|
||||||
message = nick + message
|
message = nick + message
|
||||||
}
|
}
|
||||||
if b.Config.WebhookURL != "" {
|
if b.Config.WebhookURL != "" {
|
||||||
|
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: channel, UserName: rmsg.Username,
|
||||||
|
Text: rmsg.Text, Props: make(map[string]interface{})}
|
||||||
|
matterMessage.Props["matterbridge"] = true
|
||||||
|
b.mh.Send(matterMessage)
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.URL != "" {
|
||||||
|
message += fi.URL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
||||||
matterMessage.IconURL = msg.Avatar
|
matterMessage.IconURL = msg.Avatar
|
||||||
matterMessage.Channel = channel
|
matterMessage.Channel = channel
|
||||||
matterMessage.UserName = nick
|
matterMessage.UserName = nick
|
||||||
matterMessage.Type = ""
|
matterMessage.Type = ""
|
||||||
matterMessage.Text = message
|
matterMessage.Text = message
|
||||||
matterMessage.Text = message
|
|
||||||
matterMessage.Props = make(map[string]interface{})
|
matterMessage.Props = make(map[string]interface{})
|
||||||
matterMessage.Props["matterbridge"] = true
|
matterMessage.Props["matterbridge"] = true
|
||||||
err := b.mh.Send(matterMessage)
|
err := b.mh.Send(matterMessage)
|
||||||
@ -180,6 +206,9 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
|
|||||||
return msg.ID, b.mc.DeleteMessage(msg.ID)
|
return msg.ID, b.mc.DeleteMessage(msg.ID)
|
||||||
}
|
}
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.mc.PostMessage(b.mc.GetChannelId(channel, ""), rmsg.Username+rmsg.Text)
|
||||||
|
}
|
||||||
if len(msg.Extra["file"]) > 0 {
|
if len(msg.Extra["file"]) > 0 {
|
||||||
var err error
|
var err error
|
||||||
var res, id string
|
var res, id string
|
||||||
@ -190,9 +219,9 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
|
|||||||
flog.Debugf("ERROR %#v", err)
|
flog.Debugf("ERROR %#v", err)
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
message = "uploaded a file: " + fi.Name
|
message = fi.Comment
|
||||||
if b.Config.PrefixMessagesWithNick {
|
if b.Config.PrefixMessagesWithNick {
|
||||||
message = nick + "uploaded a file: " + fi.Name
|
message = nick + fi.Comment
|
||||||
}
|
}
|
||||||
res, err = b.mc.PostMessageWithFiles(b.mc.GetChannelId(channel, ""), message, []string{id})
|
res, err = b.mc.PostMessageWithFiles(b.mc.GetChannelId(channel, ""), message, []string{id})
|
||||||
}
|
}
|
||||||
@ -219,7 +248,8 @@ func (b *Bmattermost) handleMatter() {
|
|||||||
go b.handleMatterClient(mchan)
|
go b.handleMatterClient(mchan)
|
||||||
}
|
}
|
||||||
for message := range mchan {
|
for message := range mchan {
|
||||||
rmsg := config.Message{Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID, ID: message.ID, Event: message.Event, Extra: message.Extra}
|
avatar := helper.GetAvatar(b.avatarMap, message.UserID, b.General)
|
||||||
|
rmsg := config.Message{Username: message.Username, Channel: message.Channel, Account: b.Account, UserID: message.UserID, ID: message.ID, Event: message.Event, Extra: message.Extra, Avatar: avatar}
|
||||||
text, ok := b.replaceAction(message.Text)
|
text, ok := b.replaceAction(message.Text)
|
||||||
if ok {
|
if ok {
|
||||||
rmsg.Event = config.EVENT_USER_ACTION
|
rmsg.Event = config.EVENT_USER_ACTION
|
||||||
@ -245,6 +275,11 @@ func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// only download avatars if we have a place to upload them (configured mediaserver)
|
||||||
|
if b.General.MediaServerUpload != "" {
|
||||||
|
b.handleDownloadAvatar(message.UserID, message.Channel)
|
||||||
|
}
|
||||||
|
|
||||||
m := &MMMessage{Extra: make(map[string][]interface{})}
|
m := &MMMessage{Extra: make(map[string][]interface{})}
|
||||||
|
|
||||||
props := message.Post.Props
|
props := message.Post.Props
|
||||||
@ -281,8 +316,26 @@ func (b *Bmattermost) handleMatterClient(mchan chan *MMMessage) {
|
|||||||
m.Event = config.EVENT_MSG_DELETE
|
m.Event = config.EVENT_MSG_DELETE
|
||||||
}
|
}
|
||||||
if len(message.Post.FileIds) > 0 {
|
if len(message.Post.FileIds) > 0 {
|
||||||
for _, link := range b.mc.GetFileLinks(message.Post.FileIds) {
|
for _, id := range message.Post.FileIds {
|
||||||
m.Text = m.Text + "\n" + link
|
url, _ := b.mc.Client.GetFileLink(id)
|
||||||
|
finfo, resp := b.mc.Client.GetFileInfo(id)
|
||||||
|
if resp.Error != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
flog.Debugf("trying to download %#v fileid %#v with size %#v", finfo.Name, finfo.Id, finfo.Size)
|
||||||
|
if int(finfo.Size) > b.General.MediaDownloadSize {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", finfo.Name, finfo.Size, b.General.MediaDownloadSize)
|
||||||
|
m.Event = config.EVENT_FILE_FAILURE_SIZE
|
||||||
|
m.Extra[m.Event] = append(m.Extra[m.Event], config.FileInfo{Name: finfo.Name, Comment: message.Text, Size: int64(finfo.Size)})
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
data, resp := b.mc.Client.DownloadFile(id, true)
|
||||||
|
if resp.Error != nil {
|
||||||
|
flog.Errorf("download %s failed %#v", finfo.Name, resp.Error)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
flog.Debugf("download OK %#v %#v", finfo.Name, len(data))
|
||||||
|
m.Extra["file"] = append(m.Extra["file"], config.FileInfo{Name: finfo.Name, Data: &data, URL: url, Comment: message.Text})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
mchan <- m
|
mchan <- m
|
||||||
@ -311,6 +364,9 @@ func (b *Bmattermost) apiLogin() error {
|
|||||||
|
|
||||||
b.mc = matterclient.New(b.Config.Login, password,
|
b.mc = matterclient.New(b.Config.Login, password,
|
||||||
b.Config.Team, b.Config.Server)
|
b.Config.Team, b.Config.Server)
|
||||||
|
if b.General.Debug {
|
||||||
|
b.mc.SetLogLevel("debug")
|
||||||
|
}
|
||||||
b.mc.SkipTLSVerify = b.Config.SkipTLSVerify
|
b.mc.SkipTLSVerify = b.Config.SkipTLSVerify
|
||||||
b.mc.NoTLS = b.Config.NoTLS
|
b.mc.NoTLS = b.Config.NoTLS
|
||||||
flog.Infof("Connecting %s (team: %s) on %s", b.Config.Login, b.Config.Team, b.Config.Server)
|
flog.Infof("Connecting %s (team: %s) on %s", b.Config.Login, b.Config.Team, b.Config.Server)
|
||||||
@ -331,3 +387,27 @@ func (b *Bmattermost) replaceAction(text string) (string, bool) {
|
|||||||
}
|
}
|
||||||
return text, false
|
return text, false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 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) {
|
||||||
|
var name string
|
||||||
|
msg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: userid, Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})}
|
||||||
|
if _, ok := b.avatarMap[userid]; !ok {
|
||||||
|
data, resp := b.mc.Client.GetProfileImage(userid, "")
|
||||||
|
if resp.Error != nil {
|
||||||
|
flog.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error)
|
||||||
|
}
|
||||||
|
if len(data) <= b.General.MediaDownloadSize {
|
||||||
|
name = userid + ".png"
|
||||||
|
flog.Debugf("download OK %#v %#v", name, len(data))
|
||||||
|
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: &data, Avatar: true})
|
||||||
|
flog.Debugf("Sending avatar download message from %#v on %s to gateway", userid, b.Account)
|
||||||
|
flog.Debugf("Message is %#v", msg)
|
||||||
|
b.Remote <- msg
|
||||||
|
} else {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, len(data), b.General.MediaDownloadSize)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -2,9 +2,10 @@ package brocketchat
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/hook/rockethook"
|
"github.com/42wim/matterbridge/hook/rockethook"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type MMhook struct {
|
type MMhook struct {
|
||||||
@ -14,24 +15,18 @@ type MMhook struct {
|
|||||||
|
|
||||||
type Brocketchat struct {
|
type Brocketchat struct {
|
||||||
MMhook
|
MMhook
|
||||||
Config *config.Protocol
|
*config.BridgeConfig
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "rocketchat"
|
var protocol = "rocketchat"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Brocketchat {
|
func New(cfg *config.BridgeConfig) *Brocketchat {
|
||||||
b := &Brocketchat{}
|
return &Brocketchat{BridgeConfig: cfg}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
return b
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Brocketchat) Command(cmd string) string {
|
func (b *Brocketchat) Command(cmd string) string {
|
||||||
@ -63,6 +58,22 @@ func (b *Brocketchat) Send(msg config.Message) (string, error) {
|
|||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
flog.Debugf("Receiving %#v", msg)
|
flog.Debugf("Receiving %#v", msg)
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: rmsg.Channel, UserName: rmsg.Username,
|
||||||
|
Text: rmsg.Text}
|
||||||
|
b.mh.Send(matterMessage)
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.URL != "" {
|
||||||
|
msg.Text += fi.URL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
||||||
matterMessage.Channel = msg.Channel
|
matterMessage.Channel = msg.Channel
|
||||||
matterMessage.UserName = msg.Username
|
matterMessage.UserName = msg.Username
|
||||||
|
@ -5,9 +5,10 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
"github.com/42wim/matterbridge/matterhook"
|
"github.com/42wim/matterbridge/matterhook"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
"github.com/matterbridge/slack"
|
"github.com/nlopes/slack"
|
||||||
"html"
|
"html"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
@ -27,29 +28,23 @@ type MMMessage struct {
|
|||||||
type Bslack struct {
|
type Bslack struct {
|
||||||
mh *matterhook.Client
|
mh *matterhook.Client
|
||||||
sc *slack.Client
|
sc *slack.Client
|
||||||
Config *config.Protocol
|
|
||||||
rtm *slack.RTM
|
rtm *slack.RTM
|
||||||
Plus bool
|
Plus bool
|
||||||
Remote chan config.Message
|
|
||||||
Users []slack.User
|
Users []slack.User
|
||||||
Account string
|
|
||||||
si *slack.Info
|
si *slack.Info
|
||||||
channels []slack.Channel
|
channels []slack.Channel
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "slack"
|
var protocol = "slack"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bslack {
|
func New(cfg *config.BridgeConfig) *Bslack {
|
||||||
b := &Bslack{}
|
return &Bslack{BridgeConfig: cfg}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
return b
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Bslack) Command(cmd string) string {
|
func (b *Bslack) Command(cmd string) string {
|
||||||
@ -113,7 +108,7 @@ func (b *Bslack) Disconnect() error {
|
|||||||
|
|
||||||
func (b *Bslack) JoinChannel(channel config.ChannelInfo) error {
|
func (b *Bslack) JoinChannel(channel config.ChannelInfo) error {
|
||||||
// we can only join channels using the API
|
// we can only join channels using the API
|
||||||
if b.Config.WebhookURL == "" && b.Config.WebhookBindAddress == "" {
|
if b.sc != nil {
|
||||||
if strings.HasPrefix(b.Config.Token, "xoxb") {
|
if strings.HasPrefix(b.Config.Token, "xoxb") {
|
||||||
// TODO check if bot has already joined channel
|
// TODO check if bot has already joined channel
|
||||||
return nil
|
return nil
|
||||||
@ -140,6 +135,22 @@ func (b *Bslack) Send(msg config.Message) (string, error) {
|
|||||||
message = nick + " " + message
|
message = nick + " " + message
|
||||||
}
|
}
|
||||||
if b.Config.WebhookURL != "" {
|
if b.Config.WebhookURL != "" {
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL, Channel: channel, UserName: rmsg.Username,
|
||||||
|
Text: rmsg.Text}
|
||||||
|
b.mh.Send(matterMessage)
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.URL != "" {
|
||||||
|
message += fi.URL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
matterMessage := matterhook.OMessage{IconURL: b.Config.IconURL}
|
||||||
matterMessage.Channel = channel
|
matterMessage.Channel = channel
|
||||||
matterMessage.UserName = nick
|
matterMessage.UserName = nick
|
||||||
@ -161,7 +172,7 @@ func (b *Bslack) Send(msg config.Message) (string, error) {
|
|||||||
np.AsUser = true
|
np.AsUser = true
|
||||||
}
|
}
|
||||||
np.Username = nick
|
np.Username = nick
|
||||||
np.IconURL = config.GetIconURL(&msg, b.Config)
|
np.IconURL = config.GetIconURL(&msg, &b.Config)
|
||||||
if msg.Avatar != "" {
|
if msg.Avatar != "" {
|
||||||
np.IconURL = msg.Avatar
|
np.IconURL = msg.Avatar
|
||||||
}
|
}
|
||||||
@ -189,15 +200,19 @@ func (b *Bslack) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.sc.PostMessage(schannel.ID, rmsg.Username+rmsg.Text, np)
|
||||||
|
}
|
||||||
// check if we have files to upload (from slack, telegram or mattermost)
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
if len(msg.Extra["file"]) > 0 {
|
if len(msg.Extra["file"]) > 0 {
|
||||||
var err error
|
var err error
|
||||||
for _, f := range msg.Extra["file"] {
|
for _, f := range msg.Extra["file"] {
|
||||||
fi := f.(config.FileInfo)
|
fi := f.(config.FileInfo)
|
||||||
_, err = b.sc.UploadFile(slack.FileUploadParameters{
|
_, err = b.sc.UploadFile(slack.FileUploadParameters{
|
||||||
Reader: bytes.NewReader(*fi.Data),
|
Reader: bytes.NewReader(*fi.Data),
|
||||||
Filename: fi.Name,
|
Filename: fi.Name,
|
||||||
Channels: []string{schannel.ID},
|
Channels: []string{schannel.ID},
|
||||||
|
InitialComment: fi.Comment,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("uploadfile %#v", err)
|
flog.Errorf("uploadfile %#v", err)
|
||||||
@ -289,16 +304,29 @@ func (b *Bslack) handleSlack() {
|
|||||||
msg.Event = config.EVENT_MSG_DELETE
|
msg.Event = config.EVENT_MSG_DELETE
|
||||||
msg.ID = "slack " + message.Raw.DeletedTimestamp
|
msg.ID = "slack " + message.Raw.DeletedTimestamp
|
||||||
}
|
}
|
||||||
|
if message.Raw.SubType == "channel_topic" || message.Raw.SubType == "channel_purpose" {
|
||||||
|
msg.Event = config.EVENT_TOPIC_CHANGE
|
||||||
|
}
|
||||||
|
|
||||||
// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
|
// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
|
||||||
if message.Raw.File != nil {
|
if message.Raw.File != nil {
|
||||||
// limit to 1MB for now
|
// limit to 1MB for now
|
||||||
if message.Raw.File.Size <= 1000000 {
|
comment := ""
|
||||||
|
results := regexp.MustCompile(`.*?commented: (.*)`).FindAllStringSubmatch(msg.Text, -1)
|
||||||
|
if len(results) > 0 {
|
||||||
|
comment = results[0][1]
|
||||||
|
}
|
||||||
|
|
||||||
|
if message.Raw.File.Size > b.General.MediaDownloadSize {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", message.Raw.File.Name, message.Raw.File.Size, b.General.MediaDownloadSize)
|
||||||
|
msg.Event = config.EVENT_FILE_FAILURE_SIZE
|
||||||
|
msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: message.Raw.File.Name, Comment: comment, Size: int64(message.Raw.File.Size)})
|
||||||
|
} else {
|
||||||
data, err := b.downloadFile(message.Raw.File.URLPrivateDownload)
|
data, err := b.downloadFile(message.Raw.File.URLPrivateDownload)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("download %s failed %#v", message.Raw.File.URLPrivateDownload, err)
|
flog.Errorf("download %s failed %#v", message.Raw.File.URLPrivateDownload, err)
|
||||||
} else {
|
} else {
|
||||||
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: message.Raw.File.Name, Data: data})
|
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: message.Raw.File.Name, Data: data, Comment: comment})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -309,9 +337,14 @@ func (b *Bslack) handleSlack() {
|
|||||||
|
|
||||||
func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
|
func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
|
||||||
for msg := range b.rtm.IncomingEvents {
|
for msg := range b.rtm.IncomingEvents {
|
||||||
|
if msg.Type != "user_typing" && msg.Type != "latency_report" {
|
||||||
|
flog.Debugf("Receiving from slackclient %#v", msg.Data)
|
||||||
|
}
|
||||||
switch ev := msg.Data.(type) {
|
switch ev := msg.Data.(type) {
|
||||||
case *slack.MessageEvent:
|
case *slack.MessageEvent:
|
||||||
flog.Debugf("Receiving from slackclient %#v", ev)
|
if ev.SubType == "pinned_item" || ev.SubType == "unpinned_item" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
if len(ev.Attachments) > 0 {
|
if len(ev.Attachments) > 0 {
|
||||||
// skip messages we made ourselves
|
// skip messages we made ourselves
|
||||||
if ev.Attachments[0].CallbackID == "matterbridge" {
|
if ev.Attachments[0].CallbackID == "matterbridge" {
|
||||||
@ -335,7 +368,7 @@ func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
m := &MMMessage{}
|
m := &MMMessage{}
|
||||||
if ev.BotID == "" && ev.SubType != "message_deleted" {
|
if ev.BotID == "" && ev.SubType != "message_deleted" && ev.SubType != "file_comment" {
|
||||||
user, err := b.rtm.GetUserInfo(ev.User)
|
user, err := b.rtm.GetUserInfo(ev.User)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
@ -375,6 +408,11 @@ func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
|
|||||||
m.UserID = bot.ID
|
m.UserID = bot.ID
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if ev.SubType == "file_comment" {
|
||||||
|
m.Username = "system"
|
||||||
|
}
|
||||||
|
|
||||||
mchan <- m
|
mchan <- m
|
||||||
case *slack.OutgoingErrorEvent:
|
case *slack.OutgoingErrorEvent:
|
||||||
flog.Debugf("%#v", ev.Error())
|
flog.Debugf("%#v", ev.Error())
|
||||||
@ -394,6 +432,8 @@ func (b *Bslack) handleSlackClient(mchan chan *MMMessage) {
|
|||||||
}
|
}
|
||||||
case *slack.InvalidAuthEvent:
|
case *slack.InvalidAuthEvent:
|
||||||
flog.Fatalf("Invalid Token %#v", ev)
|
flog.Fatalf("Invalid Token %#v", ev)
|
||||||
|
case *slack.ConnectionErrorEvent:
|
||||||
|
flog.Errorf("Connection failed %#v %#v", ev.Error(), ev.ErrorObj)
|
||||||
default:
|
default:
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
139
bridge/sshchat/sshchat.go
Normal file
139
bridge/sshchat/sshchat.go
Normal file
@ -0,0 +1,139 @@
|
|||||||
|
package bsshchat
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"github.com/shazow/ssh-chat/sshd"
|
||||||
|
"io"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Bsshchat struct {
|
||||||
|
r *bufio.Scanner
|
||||||
|
w io.WriteCloser
|
||||||
|
*config.BridgeConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
var flog *log.Entry
|
||||||
|
var protocol = "sshchat"
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(cfg *config.BridgeConfig) *Bsshchat {
|
||||||
|
return &Bsshchat{BridgeConfig: cfg}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bsshchat) Connect() error {
|
||||||
|
var err error
|
||||||
|
flog.Infof("Connecting %s", b.Config.Server)
|
||||||
|
go func() {
|
||||||
|
err = sshd.ConnectShell(b.Config.Server, b.Config.Nick, func(r io.Reader, w io.WriteCloser) error {
|
||||||
|
b.r = bufio.NewScanner(r)
|
||||||
|
b.w = w
|
||||||
|
b.r.Scan()
|
||||||
|
w.Write([]byte("/theme mono\r\n"))
|
||||||
|
b.handleSshChat()
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}()
|
||||||
|
if err != nil {
|
||||||
|
flog.Debugf("%#v", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
flog.Info("Connection succeeded")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bsshchat) Disconnect() error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bsshchat) JoinChannel(channel config.ChannelInfo) error {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bsshchat) Send(msg config.Message) (string, error) {
|
||||||
|
// ignore delete messages
|
||||||
|
if msg.Event == config.EVENT_MSG_DELETE {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
flog.Debugf("Receiving %#v", msg)
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.w.Write([]byte(rmsg.Username + rmsg.Text + "\r\n"))
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
for _, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
if fi.Comment != "" {
|
||||||
|
msg.Text += fi.Comment + ": "
|
||||||
|
}
|
||||||
|
if fi.URL != "" {
|
||||||
|
msg.Text = fi.URL
|
||||||
|
}
|
||||||
|
b.w.Write([]byte(msg.Username + msg.Text))
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
b.w.Write([]byte(msg.Username + msg.Text + "\r\n"))
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
func (b *Bsshchat) sshchatKeepAlive() chan bool {
|
||||||
|
done := make(chan bool)
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(90 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ticker.C:
|
||||||
|
flog.Debugf("PING")
|
||||||
|
err := b.xc.PingC2S("", "")
|
||||||
|
if err != nil {
|
||||||
|
flog.Debugf("PING failed %#v", err)
|
||||||
|
}
|
||||||
|
case <-done:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
return done
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
func stripPrompt(s string) string {
|
||||||
|
pos := strings.LastIndex(s, "\033[K")
|
||||||
|
if pos < 0 {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
return s[pos+3:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Bsshchat) handleSshChat() error {
|
||||||
|
/*
|
||||||
|
done := b.sshchatKeepAlive()
|
||||||
|
defer close(done)
|
||||||
|
*/
|
||||||
|
wait := true
|
||||||
|
for {
|
||||||
|
if b.r.Scan() {
|
||||||
|
res := strings.Split(stripPrompt(b.r.Text()), ":")
|
||||||
|
if res[0] == "-> Set theme" {
|
||||||
|
wait = false
|
||||||
|
log.Debugf("mono found, allowing")
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if !wait {
|
||||||
|
flog.Debugf("message %#v", res)
|
||||||
|
rmsg := config.Message{Username: res[0], Text: strings.Join(res[1:], ":"), Channel: "sshchat", Account: b.Account, UserID: "nick"}
|
||||||
|
b.Remote <- rmsg
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -6,7 +6,7 @@ import (
|
|||||||
"github.com/Philipp15b/go-steam"
|
"github.com/Philipp15b/go-steam"
|
||||||
"github.com/Philipp15b/go-steam/protocol/steamlang"
|
"github.com/Philipp15b/go-steam/protocol/steamlang"
|
||||||
"github.com/Philipp15b/go-steam/steamid"
|
"github.com/Philipp15b/go-steam/steamid"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
//"io/ioutil"
|
//"io/ioutil"
|
||||||
"strconv"
|
"strconv"
|
||||||
"sync"
|
"sync"
|
||||||
@ -16,25 +16,20 @@ import (
|
|||||||
type Bsteam struct {
|
type Bsteam struct {
|
||||||
c *steam.Client
|
c *steam.Client
|
||||||
connected chan struct{}
|
connected chan struct{}
|
||||||
Config *config.Protocol
|
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
userMap map[steamid.SteamId]string
|
userMap map[steamid.SteamId]string
|
||||||
sync.RWMutex
|
sync.RWMutex
|
||||||
|
*config.BridgeConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "steam"
|
var protocol = "steam"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bsteam {
|
func New(cfg *config.BridgeConfig) *Bsteam {
|
||||||
b := &Bsteam{}
|
b := &Bsteam{BridgeConfig: cfg}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
b.userMap = make(map[steamid.SteamId]string)
|
b.userMap = make(map[steamid.SteamId]string)
|
||||||
b.connected = make(chan struct{})
|
b.connected = make(chan struct{})
|
||||||
return b
|
return b
|
||||||
|
@ -3,33 +3,29 @@ package btelegram
|
|||||||
import (
|
import (
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/bridge/helper"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
log "github.com/Sirupsen/logrus"
|
|
||||||
"github.com/go-telegram-bot-api/telegram-bot-api"
|
"github.com/go-telegram-bot-api/telegram-bot-api"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
)
|
)
|
||||||
|
|
||||||
type Btelegram struct {
|
type Btelegram struct {
|
||||||
c *tgbotapi.BotAPI
|
c *tgbotapi.BotAPI
|
||||||
Config *config.Protocol
|
*config.BridgeConfig
|
||||||
Remote chan config.Message
|
avatarMap map[string]string // keep cache of userid and avatar sha
|
||||||
Account string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "telegram"
|
var protocol = "telegram"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Btelegram {
|
func New(cfg *config.BridgeConfig) *Btelegram {
|
||||||
b := &Btelegram{}
|
return &Btelegram{BridgeConfig: cfg, avatarMap: make(map[string]string)}
|
||||||
b.Config = &cfg
|
|
||||||
b.Remote = c
|
|
||||||
b.Account = account
|
|
||||||
return b
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Btelegram) Connect() error {
|
func (b *Btelegram) Connect() error {
|
||||||
@ -40,7 +36,9 @@ func (b *Btelegram) Connect() error {
|
|||||||
flog.Debugf("%#v", err)
|
flog.Debugf("%#v", err)
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
updates, err := b.c.GetUpdatesChan(tgbotapi.NewUpdate(0))
|
u := tgbotapi.NewUpdate(0)
|
||||||
|
u.Timeout = 60
|
||||||
|
updates, err := b.c.GetUpdatesChan(u)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Debugf("%#v", err)
|
flog.Debugf("%#v", err)
|
||||||
return err
|
return err
|
||||||
@ -66,6 +64,18 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// map the file SHA to our user (caches the avatar)
|
||||||
|
if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
|
||||||
|
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 != "" {
|
||||||
|
flog.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
|
||||||
|
b.avatarMap[msg.UserID] = fi.SHA
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
if b.Config.MessageFormat == "HTML" {
|
if b.Config.MessageFormat == "HTML" {
|
||||||
msg.Text = makeHTML(msg.Text)
|
msg.Text = makeHTML(msg.Text)
|
||||||
}
|
}
|
||||||
@ -89,6 +99,14 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text)
|
m := tgbotapi.NewEditMessageText(chatid, msgid, msg.Username+msg.Text)
|
||||||
|
if b.Config.MessageFormat == "HTML" {
|
||||||
|
flog.Debug("Using mode HTML")
|
||||||
|
m.ParseMode = tgbotapi.ModeHTML
|
||||||
|
}
|
||||||
|
if b.Config.MessageFormat == "Markdown" {
|
||||||
|
flog.Debug("Using mode markdown")
|
||||||
|
m.ParseMode = tgbotapi.ModeMarkdown
|
||||||
|
}
|
||||||
_, err = b.c.Send(m)
|
_, err = b.c.Send(m)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
@ -97,6 +115,9 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.sendMessage(chatid, rmsg.Username+rmsg.Text)
|
||||||
|
}
|
||||||
// check if we have files to upload (from slack, telegram or mattermost)
|
// check if we have files to upload (from slack, telegram or mattermost)
|
||||||
if len(msg.Extra["file"]) > 0 {
|
if len(msg.Extra["file"]) > 0 {
|
||||||
var c tgbotapi.Chattable
|
var c tgbotapi.Chattable
|
||||||
@ -113,25 +134,23 @@ func (b *Btelegram) Send(msg config.Message) (string, error) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
log.Errorf("file upload failed: %#v", err)
|
log.Errorf("file upload failed: %#v", err)
|
||||||
}
|
}
|
||||||
|
if fi.Comment != "" {
|
||||||
|
b.sendMessage(chatid, msg.Username+fi.Comment)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
return "", nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return b.sendMessage(chatid, msg.Username+msg.Text)
|
||||||
m := tgbotapi.NewMessage(chatid, msg.Username+msg.Text)
|
|
||||||
if b.Config.MessageFormat == "HTML" {
|
|
||||||
m.ParseMode = tgbotapi.ModeHTML
|
|
||||||
}
|
|
||||||
res, err := b.c.Send(m)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
return strconv.Itoa(res.MessageID), nil
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
||||||
for update := range updates {
|
for update := range updates {
|
||||||
flog.Debugf("Receiving from telegram: %#v", update.Message)
|
flog.Debugf("Receiving from telegram: %#v", update.Message)
|
||||||
|
if update.Message == nil {
|
||||||
|
flog.Error("Getting nil messages, this shouldn't happen.")
|
||||||
|
continue
|
||||||
|
}
|
||||||
var message *tgbotapi.Message
|
var message *tgbotapi.Message
|
||||||
username := ""
|
username := ""
|
||||||
channel := ""
|
channel := ""
|
||||||
@ -167,23 +186,54 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
|||||||
}
|
}
|
||||||
text = message.Text
|
text = message.Text
|
||||||
channel = strconv.FormatInt(message.Chat.ID, 10)
|
channel = strconv.FormatInt(message.Chat.ID, 10)
|
||||||
|
// only download avatars if we have a place to upload them (configured mediaserver)
|
||||||
|
if b.General.MediaServerUpload != "" {
|
||||||
|
b.handleDownloadAvatar(message.From.ID, channel)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if username == "" {
|
if username == "" {
|
||||||
username = "unknown"
|
username = "unknown"
|
||||||
}
|
}
|
||||||
if message.Sticker != nil {
|
if message.Sticker != nil {
|
||||||
b.handleDownload(message.Sticker, &fmsg)
|
b.handleDownload(message.Sticker, message.Caption, &fmsg)
|
||||||
}
|
}
|
||||||
if message.Video != nil {
|
if message.Video != nil {
|
||||||
b.handleDownload(message.Video, &fmsg)
|
b.handleDownload(message.Video, message.Caption, &fmsg)
|
||||||
}
|
}
|
||||||
if message.Photo != nil && b.Config.UseInsecureURL {
|
if message.Photo != nil {
|
||||||
b.handleDownload(message.Photo, &fmsg)
|
b.handleDownload(message.Photo, message.Caption, &fmsg)
|
||||||
}
|
}
|
||||||
if message.Document != nil && b.Config.UseInsecureURL {
|
if message.Document != nil {
|
||||||
b.handleDownload(message.Sticker, &fmsg)
|
b.handleDownload(message.Document, message.Caption, &fmsg)
|
||||||
text = text + " " + message.Document.FileName + " : " + b.getFileDirectURL(message.Document.FileID)
|
}
|
||||||
|
if message.Voice != nil {
|
||||||
|
b.handleDownload(message.Voice, message.Caption, &fmsg)
|
||||||
|
}
|
||||||
|
if message.Audio != nil {
|
||||||
|
b.handleDownload(message.Audio, message.Caption, &fmsg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// If UseInsecureURL is used we'll have a text in fmsg.Text
|
||||||
|
if fmsg.Text != "" {
|
||||||
|
text = text + fmsg.Text
|
||||||
|
}
|
||||||
|
|
||||||
|
if message.ForwardFrom != nil {
|
||||||
|
usernameForward := ""
|
||||||
|
if b.Config.UseFirstName {
|
||||||
|
usernameForward = message.ForwardFrom.FirstName
|
||||||
|
}
|
||||||
|
if usernameForward == "" {
|
||||||
|
usernameForward = message.ForwardFrom.UserName
|
||||||
|
if usernameForward == "" {
|
||||||
|
usernameForward = message.ForwardFrom.FirstName
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if usernameForward == "" {
|
||||||
|
usernameForward = "unknown"
|
||||||
|
}
|
||||||
|
text = "Forwarded from " + usernameForward + ": " + text
|
||||||
}
|
}
|
||||||
|
|
||||||
// quote the previous message
|
// quote the previous message
|
||||||
@ -207,8 +257,9 @@ func (b *Btelegram) handleRecv(updates <-chan tgbotapi.Update) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if text != "" || len(fmsg.Extra) > 0 {
|
if text != "" || len(fmsg.Extra) > 0 {
|
||||||
|
avatar := helper.GetAvatar(b.avatarMap, strconv.Itoa(message.From.ID), b.General)
|
||||||
flog.Debugf("Sending message from %s on %s to gateway", username, b.Account)
|
flog.Debugf("Sending message from %s on %s to gateway", username, b.Account)
|
||||||
msg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID), ID: strconv.Itoa(message.MessageID)}
|
msg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: strconv.Itoa(message.From.ID), ID: strconv.Itoa(message.MessageID), Extra: fmsg.Extra, Avatar: avatar}
|
||||||
flog.Debugf("Message is %#v", msg)
|
flog.Debugf("Message is %#v", msg)
|
||||||
b.Remote <- msg
|
b.Remote <- msg
|
||||||
}
|
}
|
||||||
@ -223,46 +274,129 @@ func (b *Btelegram) getFileDirectURL(id string) string {
|
|||||||
return res
|
return res
|
||||||
}
|
}
|
||||||
|
|
||||||
func (b *Btelegram) handleDownload(file interface{}, msg *config.Message) {
|
// 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) {
|
||||||
|
msg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: strconv.Itoa(userid), Event: config.EVENT_AVATAR_DOWNLOAD, Extra: make(map[string][]interface{})}
|
||||||
|
if _, ok := b.avatarMap[strconv.Itoa(userid)]; !ok {
|
||||||
|
photos, err := b.c.GetUserProfilePhotos(tgbotapi.UserProfilePhotosConfig{UserID: userid, Limit: 1})
|
||||||
|
if err != nil {
|
||||||
|
flog.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"
|
||||||
|
flog.Debugf("trying to download %#v fileid %#v with size %#v", name, photo.FileID, photo.FileSize)
|
||||||
|
if photo.FileSize <= b.General.MediaDownloadSize {
|
||||||
|
data, err := helper.DownloadFile(url)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("download %s failed %#v", url, err)
|
||||||
|
} else {
|
||||||
|
flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url))
|
||||||
|
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, Avatar: true})
|
||||||
|
flog.Debugf("Sending avatar download message from %#v on %s to gateway", userid, b.Account)
|
||||||
|
flog.Debugf("Message is %#v", msg)
|
||||||
|
b.Remote <- msg
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, photo.FileSize, b.General.MediaDownloadSize)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (b *Btelegram) handleDownload(file interface{}, comment string, msg *config.Message) {
|
||||||
size := 0
|
size := 0
|
||||||
url := ""
|
url := ""
|
||||||
name := ""
|
name := ""
|
||||||
text := ""
|
text := ""
|
||||||
|
fileid := ""
|
||||||
switch v := file.(type) {
|
switch v := file.(type) {
|
||||||
|
case *tgbotapi.Audio:
|
||||||
|
size = v.FileSize
|
||||||
|
url = b.getFileDirectURL(v.FileID)
|
||||||
|
urlPart := strings.Split(url, "/")
|
||||||
|
name = urlPart[len(urlPart)-1]
|
||||||
|
text = " " + url
|
||||||
|
fileid = v.FileID
|
||||||
|
case *tgbotapi.Voice:
|
||||||
|
size = v.FileSize
|
||||||
|
url = b.getFileDirectURL(v.FileID)
|
||||||
|
urlPart := strings.Split(url, "/")
|
||||||
|
name = urlPart[len(urlPart)-1]
|
||||||
|
text = " " + url
|
||||||
|
if !strings.HasSuffix(name, ".ogg") {
|
||||||
|
name = name + ".ogg"
|
||||||
|
}
|
||||||
|
fileid = v.FileID
|
||||||
case *tgbotapi.Sticker:
|
case *tgbotapi.Sticker:
|
||||||
size = v.FileSize
|
size = v.FileSize
|
||||||
url = b.getFileDirectURL(v.FileID)
|
url = b.getFileDirectURL(v.FileID)
|
||||||
name = "sticker"
|
urlPart := strings.Split(url, "/")
|
||||||
|
name = urlPart[len(urlPart)-1]
|
||||||
|
if !strings.HasSuffix(name, ".webp") {
|
||||||
|
name = name + ".webp"
|
||||||
|
}
|
||||||
text = " " + url
|
text = " " + url
|
||||||
|
fileid = v.FileID
|
||||||
case *tgbotapi.Video:
|
case *tgbotapi.Video:
|
||||||
size = v.FileSize
|
size = v.FileSize
|
||||||
url = b.getFileDirectURL(v.FileID)
|
url = b.getFileDirectURL(v.FileID)
|
||||||
name = "video"
|
urlPart := strings.Split(url, "/")
|
||||||
|
name = urlPart[len(urlPart)-1]
|
||||||
text = " " + url
|
text = " " + url
|
||||||
|
fileid = v.FileID
|
||||||
case *[]tgbotapi.PhotoSize:
|
case *[]tgbotapi.PhotoSize:
|
||||||
photos := *v
|
photos := *v
|
||||||
size = photos[len(photos)-1].FileSize
|
size = photos[len(photos)-1].FileSize
|
||||||
url = b.getFileDirectURL(photos[len(photos)-1].FileID)
|
url = b.getFileDirectURL(photos[len(photos)-1].FileID)
|
||||||
name = "photo"
|
urlPart := strings.Split(url, "/")
|
||||||
|
name = urlPart[len(urlPart)-1]
|
||||||
text = " " + url
|
text = " " + url
|
||||||
case *tgbotapi.Document:
|
case *tgbotapi.Document:
|
||||||
size = v.FileSize
|
size = v.FileSize
|
||||||
url = b.getFileDirectURL(v.FileID)
|
url = b.getFileDirectURL(v.FileID)
|
||||||
name = v.FileName
|
name = v.FileName
|
||||||
text = " " + v.FileName + " : " + url
|
text = " " + v.FileName + " : " + url
|
||||||
|
fileid = v.FileID
|
||||||
}
|
}
|
||||||
if b.Config.UseInsecureURL {
|
if b.Config.UseInsecureURL {
|
||||||
|
flog.Debugf("Setting message text to :%s", text)
|
||||||
msg.Text = text
|
msg.Text = text
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
|
// if we have a file attached, download it (in memory) and put a pointer to it in msg.Extra
|
||||||
// limit to 1MB for now
|
flog.Debugf("trying to download %#v fileid %#v with size %#v", name, fileid, size)
|
||||||
if size <= 1000000 {
|
if size <= b.General.MediaDownloadSize {
|
||||||
data, err := helper.DownloadFile(url)
|
data, err := helper.DownloadFile(url)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
flog.Errorf("download %s failed %#v", url, err)
|
flog.Errorf("download %s failed %#v", url, err)
|
||||||
} else {
|
} else {
|
||||||
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data})
|
flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url))
|
||||||
|
msg.Extra["file"] = append(msg.Extra["file"], config.FileInfo{Name: name, Data: data, Comment: comment})
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
flog.Errorf("File %#v to large to download (%#v). MediaDownloadSize is %#v", name, size, b.General.MediaDownloadSize)
|
||||||
|
msg.Event = config.EVENT_FILE_FAILURE_SIZE
|
||||||
|
msg.Extra[msg.Event] = append(msg.Extra[msg.Event], config.FileInfo{Name: name, Comment: comment, Size: int64(size)})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (b *Btelegram) sendMessage(chatid int64, text string) (string, error) {
|
||||||
|
m := tgbotapi.NewMessage(chatid, text)
|
||||||
|
if b.Config.MessageFormat == "HTML" {
|
||||||
|
flog.Debug("Using mode HTML")
|
||||||
|
m.ParseMode = tgbotapi.ModeHTML
|
||||||
|
}
|
||||||
|
if b.Config.MessageFormat == "Markdown" {
|
||||||
|
flog.Debug("Using mode markdown")
|
||||||
|
m.ParseMode = tgbotapi.ModeMarkdown
|
||||||
|
}
|
||||||
|
res, err := b.c.Send(m)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return strconv.Itoa(res.MessageID), nil
|
||||||
|
}
|
||||||
|
@ -3,7 +3,8 @@ package bxmpp
|
|||||||
import (
|
import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
"github.com/42wim/matterbridge/bridge/helper"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
"github.com/jpillora/backoff"
|
"github.com/jpillora/backoff"
|
||||||
"github.com/mattn/go-xmpp"
|
"github.com/mattn/go-xmpp"
|
||||||
|
|
||||||
@ -14,24 +15,19 @@ import (
|
|||||||
type Bxmpp struct {
|
type Bxmpp struct {
|
||||||
xc *xmpp.Client
|
xc *xmpp.Client
|
||||||
xmppMap map[string]string
|
xmppMap map[string]string
|
||||||
Config *config.Protocol
|
*config.BridgeConfig
|
||||||
Remote chan config.Message
|
|
||||||
Account string
|
|
||||||
}
|
}
|
||||||
|
|
||||||
var flog *log.Entry
|
var flog *log.Entry
|
||||||
var protocol = "xmpp"
|
var protocol = "xmpp"
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flog = log.WithFields(log.Fields{"module": protocol})
|
flog = log.WithFields(log.Fields{"prefix": protocol})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Protocol, account string, c chan config.Message) *Bxmpp {
|
func New(cfg *config.BridgeConfig) *Bxmpp {
|
||||||
b := &Bxmpp{}
|
b := &Bxmpp{BridgeConfig: cfg}
|
||||||
b.xmppMap = make(map[string]string)
|
b.xmppMap = make(map[string]string)
|
||||||
b.Config = &cfg
|
|
||||||
b.Account = account
|
|
||||||
b.Remote = c
|
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -85,6 +81,25 @@ func (b *Bxmpp) Send(msg config.Message) (string, error) {
|
|||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
flog.Debugf("Receiving %#v", msg)
|
flog.Debugf("Receiving %#v", msg)
|
||||||
|
if msg.Extra != nil {
|
||||||
|
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
|
||||||
|
b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: rmsg.Channel + "@" + b.Config.Muc, Text: 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
|
||||||
|
}
|
||||||
|
b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text})
|
||||||
|
}
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text})
|
b.xc.Send(xmpp.Chat{Type: "groupchat", Remote: msg.Channel + "@" + b.Config.Muc, Text: msg.Username + msg.Text})
|
||||||
return "", nil
|
return "", nil
|
||||||
}
|
}
|
||||||
@ -102,7 +117,7 @@ func (b *Bxmpp) createXMPP() (*xmpp.Client, error) {
|
|||||||
TLSConfig: tc,
|
TLSConfig: tc,
|
||||||
|
|
||||||
//StartTLS: false,
|
//StartTLS: false,
|
||||||
Debug: true,
|
Debug: b.General.Debug,
|
||||||
Session: true,
|
Session: true,
|
||||||
Status: "",
|
Status: "",
|
||||||
StatusMessage: "",
|
StatusMessage: "",
|
||||||
@ -158,7 +173,7 @@ func (b *Bxmpp) handleXmpp() error {
|
|||||||
if len(s) == 2 {
|
if len(s) == 2 {
|
||||||
nick = s[1]
|
nick = s[1]
|
||||||
}
|
}
|
||||||
if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" {
|
if nick != b.Config.Nick && v.Stamp == nodelay && v.Text != "" && !strings.Contains(v.Text, "</subject>") {
|
||||||
rmsg := config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote}
|
rmsg := config.Message{Username: nick, Text: v.Text, Channel: channel, Account: b.Account, UserID: v.Remote}
|
||||||
rmsg.Text, ok = b.replaceAction(rmsg.Text)
|
rmsg.Text, ok = b.replaceAction(rmsg.Text)
|
||||||
if ok {
|
if ok {
|
||||||
|
94
changelog.md
94
changelog.md
@ -1,3 +1,97 @@
|
|||||||
|
# v1.8.0
|
||||||
|
## New features
|
||||||
|
* general: Send chat notification if media is too big to be re-uploaded to MediaServer. See #359
|
||||||
|
* general: Download (and upload) avatar images from mattermost and telegram when mediaserver is configured. Closes #362
|
||||||
|
* general: Add label support in RemoteNickFormat
|
||||||
|
* general: Prettier info/debug log output
|
||||||
|
* mattermost: Download files and reupload to supported bridges (mattermost). Closes #357
|
||||||
|
* slack: Add ShowTopicChange option. Allow/disable topic change messages (currently only from slack). Closes #353
|
||||||
|
* slack: Add support for file comments (slack). Closes #346
|
||||||
|
* telegram: Add comment to file upload from telegram. Show comments on all bridges. Closes #358
|
||||||
|
* telegram: Add markdown support (telegram). #355
|
||||||
|
* api: Give api access to whole config.Message (and events). Closes #374
|
||||||
|
|
||||||
|
## Bugfix
|
||||||
|
* discord: Check for a valid WebhookURL (discord). Closes #367
|
||||||
|
* discord: Fix role mention replace issues
|
||||||
|
* irc: Truncate messages sent to IRC based on byte count (#368)
|
||||||
|
* mattermost: Add file download urls also to mattermost webhooks #356
|
||||||
|
* telegram: Fix panic on nil messages (telegram). Closes #366
|
||||||
|
* telegram: Fix the UseInsecureURL text (telegram). Closes #184
|
||||||
|
|
||||||
|
# v1.7.1
|
||||||
|
## Bugfix
|
||||||
|
* telegram: Enable Long Polling for Telegram. Reduces bandwidth consumption. (#350)
|
||||||
|
|
||||||
|
# v1.7.0
|
||||||
|
## New features
|
||||||
|
* matrix: Add support for deleting messages from/to matrix (matrix). Closes #320
|
||||||
|
* xmpp: Ignore <subject> messages (xmpp). #272
|
||||||
|
* irc: Add twitch support (irc) to README / wiki
|
||||||
|
|
||||||
|
## Bugfix
|
||||||
|
* general: Change RemoteNickFormat replacement order. Closes #336
|
||||||
|
* general: Make edits/delete work for bridges that gets reused. Closes #342
|
||||||
|
* general: Lowercase irc channels in config. Closes #348
|
||||||
|
* matrix: Fix possible panics (matrix). Closes #333
|
||||||
|
* matrix: Add an extension to images without one (matrix). #331
|
||||||
|
* api: Obey the Gateway value from the json (api). Closes #344
|
||||||
|
* xmpp: Print only debug messages when specified (xmpp). Closes #345
|
||||||
|
* xmpp: Allow xmpp to receive the extra messages (file uploads) when text is empty. #295
|
||||||
|
|
||||||
|
# v1.6.3
|
||||||
|
## Bugfix
|
||||||
|
* slack: Fix connection issues
|
||||||
|
* slack: Add more debug messages
|
||||||
|
* irc: Convert received IRC channel names to lowercase. Fixes #329 (#330)
|
||||||
|
|
||||||
|
# v1.6.2
|
||||||
|
## Bugfix
|
||||||
|
* mattermost: Crashes while connecting to Mattermost (regression). Closes #327
|
||||||
|
|
||||||
|
# v1.6.1
|
||||||
|
## Bugfix
|
||||||
|
* general: Display of nicks not longer working (regression). Closes #323
|
||||||
|
|
||||||
|
# v1.6.0
|
||||||
|
## New features
|
||||||
|
* sshchat: New protocol support added (https://github.com/shazow/ssh-chat)
|
||||||
|
* general: Allow specifying maximum download size of media using MediaDownloadSize (slack,telegram,matrix)
|
||||||
|
* api: Add (simple, one listener) long-polling support (api). Closes #307
|
||||||
|
* telegram: Add support for forwarded messages. Closes #313
|
||||||
|
* telegram: Add support for Audio/Voice files (telegram). Closes #314
|
||||||
|
* irc: Add RejoinDelay option. Delay to rejoin after channel kick (irc). Closes #322
|
||||||
|
|
||||||
|
## Bugfix
|
||||||
|
* telegram: Also use HTML in edited messages (telegram). Closes #315
|
||||||
|
* matrix: Fix panic (matrix). Closes #316
|
||||||
|
|
||||||
|
# v1.5.1
|
||||||
|
|
||||||
|
## Bugfix
|
||||||
|
* irc: Fix irc ACTION regression (irc). Closes #306
|
||||||
|
* irc: Split on UTF-8 for MessageSplit (irc). Closes #308
|
||||||
|
|
||||||
|
# v1.5.0
|
||||||
|
## New features
|
||||||
|
* general: remote mediaserver support. See MediaServerDownload and MediaServerUpload in matterbridge.toml.sample
|
||||||
|
more information on https://github.com/42wim/matterbridge/wiki/Mediaserver-setup-%5Badvanced%5D
|
||||||
|
* general: Add support for ReplaceNicks using regexp to replace nicks. Closes #269 (see matterbridge.toml.sample)
|
||||||
|
* general: Add support for ReplaceMessages using regexp to replace messages. #269 (see matterbridge.toml.sample)
|
||||||
|
* irc: Add MessageSplit option to split messages on MessageLength (irc). Closes #281
|
||||||
|
* matrix: Add support for uploading images/video (matrix). Closes #302
|
||||||
|
* matrix: Add support for uploaded images/video (matrix)
|
||||||
|
|
||||||
|
## Bugfix
|
||||||
|
* telegram: Add webp extension to stickers if necessary (telegram)
|
||||||
|
* mattermost: Break when re-login fails (mattermost)
|
||||||
|
|
||||||
|
# v1.4.1
|
||||||
|
## Bugfix
|
||||||
|
* telegram: fix issue with uploading for images/documents/stickers
|
||||||
|
* slack: remove double messages sent to other bridges when uploading files
|
||||||
|
* irc: Fix strict user handling of girc (irc). Closes #298
|
||||||
|
|
||||||
# v1.4.0
|
# v1.4.0
|
||||||
## Breaking changes
|
## Breaking changes
|
||||||
* general: `[general]` settings don't override the specific bridge settings
|
* general: `[general]` settings don't override the specific bridge settings
|
||||||
|
11
docker/arm/Dockerfile
Normal file
11
docker/arm/Dockerfile
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
FROM cmosh/alpine-arm:edge
|
||||||
|
ENTRYPOINT ["/bin/matterbridge"]
|
||||||
|
|
||||||
|
COPY . /go/src/github.com/42wim/matterbridge
|
||||||
|
RUN apk update && apk add go git gcc musl-dev ca-certificates \
|
||||||
|
&& cd /go/src/github.com/42wim/matterbridge \
|
||||||
|
&& export GOPATH=/go \
|
||||||
|
&& go get \
|
||||||
|
&& go build -x -ldflags "-X main.githash=$(git log --pretty=format:'%h' -n 1)" -o /bin/matterbridge \
|
||||||
|
&& rm -rf /go \
|
||||||
|
&& apk del --purge git go gcc musl-dev
|
@ -1,13 +1,16 @@
|
|||||||
package gateway
|
package gateway
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
// "github.com/davecgh/go-spew/spew"
|
// "github.com/davecgh/go-spew/spew"
|
||||||
|
"crypto/sha1"
|
||||||
"github.com/hashicorp/golang-lru"
|
"github.com/hashicorp/golang-lru"
|
||||||
"github.com/peterhellberg/emojilib"
|
"github.com/peterhellberg/emojilib"
|
||||||
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
@ -26,8 +29,15 @@ type Gateway struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type BrMsgID struct {
|
type BrMsgID struct {
|
||||||
br *bridge.Bridge
|
br *bridge.Bridge
|
||||||
ID string
|
ID string
|
||||||
|
ChannelID string
|
||||||
|
}
|
||||||
|
|
||||||
|
var flog *log.Entry
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
flog = log.WithFields(log.Fields{"prefix": "gateway"})
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Gateway, r *Router) *Gateway {
|
func New(cfg config.Gateway, r *Router) *Gateway {
|
||||||
@ -74,10 +84,10 @@ func (gw *Gateway) reconnectBridge(br *bridge.Bridge) {
|
|||||||
br.Disconnect()
|
br.Disconnect()
|
||||||
time.Sleep(time.Second * 5)
|
time.Sleep(time.Second * 5)
|
||||||
RECONNECT:
|
RECONNECT:
|
||||||
log.Infof("Reconnecting %s", br.Account)
|
flog.Infof("Reconnecting %s", br.Account)
|
||||||
err := br.Connect()
|
err := br.Connect()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err)
|
flog.Errorf("Reconnection failed: %s. Trying again in 60 seconds", err)
|
||||||
time.Sleep(time.Second * 60)
|
time.Sleep(time.Second * 60)
|
||||||
goto RECONNECT
|
goto RECONNECT
|
||||||
}
|
}
|
||||||
@ -90,6 +100,10 @@ func (gw *Gateway) mapChannelConfig(cfg []config.Bridge, direction string) {
|
|||||||
if isApi(br.Account) {
|
if isApi(br.Account) {
|
||||||
br.Channel = "api"
|
br.Channel = "api"
|
||||||
}
|
}
|
||||||
|
// make sure to lowercase irc channels in config #348
|
||||||
|
if strings.HasPrefix(br.Account, "irc.") {
|
||||||
|
br.Channel = strings.ToLower(br.Channel)
|
||||||
|
}
|
||||||
ID := br.Channel + br.Account
|
ID := br.Channel + br.Account
|
||||||
if _, ok := gw.Channels[ID]; !ok {
|
if _, ok := gw.Channels[ID]; !ok {
|
||||||
channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account,
|
channel := &config.ChannelInfo{Name: br.Channel, Direction: direction, ID: ID, Options: br.Options, Account: br.Account,
|
||||||
@ -115,6 +129,12 @@ func (gw *Gateway) mapChannels() error {
|
|||||||
|
|
||||||
func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo {
|
func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []config.ChannelInfo {
|
||||||
var channels []config.ChannelInfo
|
var channels []config.ChannelInfo
|
||||||
|
|
||||||
|
// for messages received from the api check that the gateway is the specified one
|
||||||
|
if msg.Protocol == "api" && gw.Name != msg.Gateway {
|
||||||
|
return channels
|
||||||
|
}
|
||||||
|
|
||||||
// if source channel is in only, do nothing
|
// if source channel is in only, do nothing
|
||||||
for _, channel := range gw.Channels {
|
for _, channel := range gw.Channels {
|
||||||
// lookup the channel from the message
|
// lookup the channel from the message
|
||||||
@ -131,7 +151,7 @@ func (gw *Gateway) getDestChannel(msg *config.Message, dest bridge.Bridge) []con
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
// do samechannelgateway logic
|
// do samechannelgateway flogic
|
||||||
if channel.SameChannel[msg.Gateway] {
|
if channel.SameChannel[msg.Gateway] {
|
||||||
if msg.Channel == channel.Name && msg.Account != dest.Account {
|
if msg.Channel == channel.Name && msg.Account != dest.Account {
|
||||||
channels = append(channels, *channel)
|
channels = append(channels, *channel)
|
||||||
@ -152,30 +172,55 @@ func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrM
|
|||||||
// only slack now, check will have to be done in the different bridges.
|
// only slack now, check will have to be done in the different bridges.
|
||||||
// we need to check if we can't use fallback or text in other bridges
|
// we need to check if we can't use fallback or text in other bridges
|
||||||
if msg.Extra != nil {
|
if msg.Extra != nil {
|
||||||
if dest.Protocol != "slack" {
|
if dest.Protocol != "discord" &&
|
||||||
|
dest.Protocol != "slack" &&
|
||||||
|
dest.Protocol != "mattermost" &&
|
||||||
|
dest.Protocol != "telegram" &&
|
||||||
|
dest.Protocol != "matrix" &&
|
||||||
|
dest.Protocol != "xmpp" &&
|
||||||
|
len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) == 0 {
|
||||||
if msg.Text == "" {
|
if msg.Text == "" {
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Avatar downloads are only relevant for telegram and mattermost for now
|
||||||
|
if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
|
||||||
|
if dest.Protocol != "mattermost" &&
|
||||||
|
dest.Protocol != "telegram" {
|
||||||
|
return brMsgIDs
|
||||||
|
}
|
||||||
|
}
|
||||||
// only relay join/part when configged
|
// only relay join/part when configged
|
||||||
if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart {
|
if msg.Event == config.EVENT_JOIN_LEAVE && !gw.Bridges[dest.Account].Config.ShowJoinPart {
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
|
if msg.Event == config.EVENT_TOPIC_CHANGE && !gw.Bridges[dest.Account].Config.ShowTopicChange {
|
||||||
|
return brMsgIDs
|
||||||
|
}
|
||||||
|
|
||||||
// broadcast to every out channel (irc QUIT)
|
// broadcast to every out channel (irc QUIT)
|
||||||
if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE {
|
if msg.Channel == "" && msg.Event != config.EVENT_JOIN_LEAVE {
|
||||||
log.Debug("empty channel")
|
flog.Debug("empty channel")
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
}
|
}
|
||||||
originchannel := msg.Channel
|
originchannel := msg.Channel
|
||||||
origmsg := msg
|
origmsg := msg
|
||||||
channels := gw.getDestChannel(&msg, *dest)
|
channels := gw.getDestChannel(&msg, *dest)
|
||||||
for _, channel := range channels {
|
for _, channel := range channels {
|
||||||
// do not send to ourself
|
// Only send the avatar download event to ourselves.
|
||||||
if channel.ID == getChannelID(origmsg) {
|
if msg.Event == config.EVENT_AVATAR_DOWNLOAD {
|
||||||
continue
|
if channel.ID != getChannelID(origmsg) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// do not send to ourself for any other event
|
||||||
|
if channel.ID == getChannelID(origmsg) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
}
|
}
|
||||||
log.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name)
|
flog.Debugf("Sending %#v from %s (%s) to %s (%s)", msg, msg.Account, originchannel, dest.Account, channel.Name)
|
||||||
msg.Channel = channel.Name
|
msg.Channel = channel.Name
|
||||||
msg.Avatar = gw.modifyAvatar(origmsg, dest)
|
msg.Avatar = gw.modifyAvatar(origmsg, dest)
|
||||||
msg.Username = gw.modifyUsername(origmsg, dest)
|
msg.Username = gw.modifyUsername(origmsg, dest)
|
||||||
@ -183,7 +228,9 @@ func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrM
|
|||||||
if res, ok := gw.Messages.Get(origmsg.ID); ok {
|
if res, ok := gw.Messages.Get(origmsg.ID); ok {
|
||||||
IDs := res.([]*BrMsgID)
|
IDs := res.([]*BrMsgID)
|
||||||
for _, id := range IDs {
|
for _, id := range IDs {
|
||||||
if dest.Protocol == id.br.Protocol {
|
// check protocol, bridge name and channelname
|
||||||
|
// for people that reuse the same bridge multiple times. see #342
|
||||||
|
if dest.Protocol == id.br.Protocol && dest.Name == id.br.Name && channel.ID == id.ChannelID {
|
||||||
msg.ID = id.ID
|
msg.ID = id.ID
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -198,7 +245,7 @@ func (gw *Gateway) handleMessage(msg config.Message, dest *bridge.Bridge) []*BrM
|
|||||||
}
|
}
|
||||||
// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice
|
// append the message ID (mID) from this bridge (dest) to our brMsgIDs slice
|
||||||
if mID != "" {
|
if mID != "" {
|
||||||
brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID})
|
brMsgIDs = append(brMsgIDs, &BrMsgID{dest, mID, channel.ID})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return brMsgIDs
|
return brMsgIDs
|
||||||
@ -210,16 +257,19 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
if msg.Text == "" {
|
if msg.Text == "" {
|
||||||
// we have an attachment
|
// we have an attachment or actual bytes
|
||||||
if msg.Extra != nil && msg.Extra["attachments"] != nil {
|
if msg.Extra != nil &&
|
||||||
|
(msg.Extra["attachments"] != nil ||
|
||||||
|
len(msg.Extra["file"]) > 0 ||
|
||||||
|
len(msg.Extra[config.EVENT_FILE_FAILURE_SIZE]) > 0) {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
log.Debugf("ignoring empty message %#v from %s", msg, msg.Account)
|
flog.Debugf("ignoring empty message %#v from %s", msg, msg.Account)
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) {
|
for _, entry := range strings.Fields(gw.Bridges[msg.Account].Config.IgnoreNicks) {
|
||||||
if msg.Username == entry {
|
if msg.Username == entry {
|
||||||
log.Debugf("ignoring %s from %s", msg.Username, msg.Account)
|
flog.Debugf("ignoring %s from %s", msg.Username, msg.Account)
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -228,11 +278,11 @@ func (gw *Gateway) ignoreMessage(msg *config.Message) bool {
|
|||||||
if entry != "" {
|
if entry != "" {
|
||||||
re, err := regexp.Compile(entry)
|
re, err := regexp.Compile(entry)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Errorf("incorrect regexp %s for %s", entry, msg.Account)
|
flog.Errorf("incorrect regexp %s for %s", entry, msg.Account)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if re.MatchString(msg.Text) {
|
if re.MatchString(msg.Text) {
|
||||||
log.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account)
|
flog.Debugf("matching %s. ignoring %s from %s", entry, msg.Text, msg.Account)
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -251,6 +301,20 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin
|
|||||||
if nick == "" {
|
if nick == "" {
|
||||||
nick = gw.Config.General.RemoteNickFormat
|
nick = gw.Config.General.RemoteNickFormat
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// loop to replace nicks
|
||||||
|
for _, outer := range br.Config.ReplaceNicks {
|
||||||
|
search := outer[0]
|
||||||
|
replace := outer[1]
|
||||||
|
// TODO move compile to bridge init somewhere
|
||||||
|
re, err := regexp.Compile(search)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("regexp in %s failed: %s", msg.Account, err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
msg.Username = re.ReplaceAllString(msg.Username, replace)
|
||||||
|
}
|
||||||
|
|
||||||
if len(msg.Username) > 0 {
|
if len(msg.Username) > 0 {
|
||||||
// fix utf-8 issue #193
|
// fix utf-8 issue #193
|
||||||
i := 0
|
i := 0
|
||||||
@ -263,9 +327,10 @@ func (gw *Gateway) modifyUsername(msg config.Message, dest *bridge.Bridge) strin
|
|||||||
}
|
}
|
||||||
nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1)
|
nick = strings.Replace(nick, "{NOPINGNICK}", msg.Username[:i]+""+msg.Username[i:], -1)
|
||||||
}
|
}
|
||||||
nick = strings.Replace(nick, "{NICK}", msg.Username, -1)
|
|
||||||
nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
|
nick = strings.Replace(nick, "{BRIDGE}", br.Name, -1)
|
||||||
nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
|
nick = strings.Replace(nick, "{PROTOCOL}", br.Protocol, -1)
|
||||||
|
nick = strings.Replace(nick, "{LABEL}", br.Config.Label, -1)
|
||||||
|
nick = strings.Replace(nick, "{NICK}", msg.Username, -1)
|
||||||
return nick
|
return nick
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -284,7 +349,55 @@ func (gw *Gateway) modifyAvatar(msg config.Message, dest *bridge.Bridge) string
|
|||||||
func (gw *Gateway) modifyMessage(msg *config.Message) {
|
func (gw *Gateway) modifyMessage(msg *config.Message) {
|
||||||
// replace :emoji: to unicode
|
// replace :emoji: to unicode
|
||||||
msg.Text = emojilib.Replace(msg.Text)
|
msg.Text = emojilib.Replace(msg.Text)
|
||||||
msg.Gateway = gw.Name
|
br := gw.Bridges[msg.Account]
|
||||||
|
// loop to replace messages
|
||||||
|
for _, outer := range br.Config.ReplaceMessages {
|
||||||
|
search := outer[0]
|
||||||
|
replace := outer[1]
|
||||||
|
// TODO move compile to bridge init somewhere
|
||||||
|
re, err := regexp.Compile(search)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("regexp in %s failed: %s", msg.Account, err)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
msg.Text = re.ReplaceAllString(msg.Text, replace)
|
||||||
|
}
|
||||||
|
|
||||||
|
// messages from api have Gateway specified, don't overwrite
|
||||||
|
if msg.Protocol != "api" {
|
||||||
|
msg.Gateway = gw.Name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (gw *Gateway) handleFiles(msg *config.Message) {
|
||||||
|
if msg.Extra == nil || gw.Config.General.MediaServerUpload == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(msg.Extra["file"]) > 0 {
|
||||||
|
client := &http.Client{
|
||||||
|
Timeout: time.Second * 5,
|
||||||
|
}
|
||||||
|
for i, f := range msg.Extra["file"] {
|
||||||
|
fi := f.(config.FileInfo)
|
||||||
|
sha1sum := fmt.Sprintf("%x", sha1.Sum(*fi.Data))
|
||||||
|
reader := bytes.NewReader(*fi.Data)
|
||||||
|
url := gw.Config.General.MediaServerUpload + "/" + sha1sum + "/" + fi.Name
|
||||||
|
durl := gw.Config.General.MediaServerDownload + "/" + sha1sum + "/" + fi.Name
|
||||||
|
extra := msg.Extra["file"][i].(config.FileInfo)
|
||||||
|
extra.URL = durl
|
||||||
|
req, _ := http.NewRequest("PUT", url, reader)
|
||||||
|
req.Header.Set("Content-Type", "binary/octet-stream")
|
||||||
|
_, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
flog.Errorf("mediaserver upload failed: %#v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
flog.Debugf("mediaserver download URL = %s", durl)
|
||||||
|
// we uploaded the file successfully. Add the SHA
|
||||||
|
extra.SHA = sha1sum
|
||||||
|
msg.Extra["file"][i] = extra
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func getChannelID(msg config.Message) string {
|
func getChannelID(msg config.Message) string {
|
||||||
|
@ -5,7 +5,7 @@ import (
|
|||||||
"github.com/42wim/matterbridge/bridge"
|
"github.com/42wim/matterbridge/bridge"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/gateway/samechannel"
|
"github.com/42wim/matterbridge/gateway/samechannel"
|
||||||
log "github.com/Sirupsen/logrus"
|
//log "github.com/sirupsen/logrus"
|
||||||
// "github.com/davecgh/go-spew/spew"
|
// "github.com/davecgh/go-spew/spew"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@ -42,12 +42,13 @@ func NewRouter(cfg *config.Config) (*Router, error) {
|
|||||||
func (r *Router) Start() error {
|
func (r *Router) Start() error {
|
||||||
m := make(map[string]*bridge.Bridge)
|
m := make(map[string]*bridge.Bridge)
|
||||||
for _, gw := range r.Gateways {
|
for _, gw := range r.Gateways {
|
||||||
|
flog.Infof("Parsing gateway %s", gw.Name)
|
||||||
for _, br := range gw.Bridges {
|
for _, br := range gw.Bridges {
|
||||||
m[br.Account] = br
|
m[br.Account] = br
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
for _, br := range m {
|
for _, br := range m {
|
||||||
log.Infof("Starting bridge: %s ", br.Account)
|
flog.Infof("Starting bridge: %s ", br.Account)
|
||||||
err := br.Connect()
|
err := br.Connect()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
|
return fmt.Errorf("Bridge %s failed to start: %v", br.Account, err)
|
||||||
@ -99,6 +100,7 @@ func (r *Router) handleReceive() {
|
|||||||
if !gw.ignoreMessage(&msg) {
|
if !gw.ignoreMessage(&msg) {
|
||||||
msg.Timestamp = time.Now()
|
msg.Timestamp = time.Now()
|
||||||
gw.modifyMessage(&msg)
|
gw.modifyMessage(&msg)
|
||||||
|
gw.handleFiles(&msg)
|
||||||
for _, br := range gw.Bridges {
|
for _, br := range gw.Bridges {
|
||||||
msgIDs = append(msgIDs, gw.handleMessage(msg, br)...)
|
msgIDs = append(msgIDs, gw.handleMessage(msg, br)...)
|
||||||
}
|
}
|
||||||
|
@ -5,22 +5,21 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"github.com/42wim/matterbridge/bridge/config"
|
"github.com/42wim/matterbridge/bridge/config"
|
||||||
"github.com/42wim/matterbridge/gateway"
|
"github.com/42wim/matterbridge/gateway"
|
||||||
log "github.com/Sirupsen/logrus"
|
|
||||||
"github.com/google/gops/agent"
|
"github.com/google/gops/agent"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
prefixed "github.com/x-cray/logrus-prefixed-formatter"
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
version = "1.4.0"
|
version = "1.8.0"
|
||||||
githash string
|
githash string
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
|
||||||
log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
|
log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: true})
|
||||||
|
flog := log.WithFields(log.Fields{"prefix": "main"})
|
||||||
flagConfig := flag.String("conf", "matterbridge.toml", "config file")
|
flagConfig := flag.String("conf", "matterbridge.toml", "config file")
|
||||||
flagDebug := flag.Bool("debug", false, "enable debug")
|
flagDebug := flag.Bool("debug", false, "enable debug")
|
||||||
flagVersion := flag.Bool("version", false, "show version")
|
flagVersion := flag.Bool("version", false, "show version")
|
||||||
@ -35,22 +34,24 @@ func main() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
if *flagDebug || os.Getenv("DEBUG") == "1" {
|
if *flagDebug || os.Getenv("DEBUG") == "1" {
|
||||||
log.Info("Enabling debug")
|
log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false})
|
||||||
|
flog.Info("Enabling debug")
|
||||||
log.SetLevel(log.DebugLevel)
|
log.SetLevel(log.DebugLevel)
|
||||||
}
|
}
|
||||||
log.Printf("Running version %s %s", version, githash)
|
flog.Printf("Running version %s %s", version, githash)
|
||||||
if strings.Contains(version, "-dev") {
|
if strings.Contains(version, "-dev") {
|
||||||
log.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.")
|
flog.Println("WARNING: THIS IS A DEVELOPMENT VERSION. Things may break.")
|
||||||
}
|
}
|
||||||
cfg := config.NewConfig(*flagConfig)
|
cfg := config.NewConfig(*flagConfig)
|
||||||
|
cfg.General.Debug = *flagDebug
|
||||||
r, err := gateway.NewRouter(cfg)
|
r, err := gateway.NewRouter(cfg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("Starting gateway failed: %s", err)
|
flog.Fatalf("Starting gateway failed: %s", err)
|
||||||
}
|
}
|
||||||
err = r.Start()
|
err = r.Start()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatalf("Starting gateway failed: %s", err)
|
flog.Fatalf("Starting gateway failed: %s", err)
|
||||||
}
|
}
|
||||||
log.Printf("Gateway(s) started succesfully. Now relaying messages")
|
flog.Printf("Gateway(s) started succesfully. Now relaying messages")
|
||||||
select {}
|
select {}
|
||||||
}
|
}
|
||||||
|
@ -80,6 +80,15 @@ MessageQueue=30
|
|||||||
#OPTIONAL (default 400)
|
#OPTIONAL (default 400)
|
||||||
MessageLength=400
|
MessageLength=400
|
||||||
|
|
||||||
|
#Split messages on MessageLength instead of showing the <message clipped>
|
||||||
|
#WARNING: this could lead to flooding
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
MessageSplit=false
|
||||||
|
|
||||||
|
#Delay in seconds to rejoin a channel when kicked
|
||||||
|
#OPTIONAL (default 0)
|
||||||
|
RejoinDelay=0
|
||||||
|
|
||||||
#Nicks you want to ignore.
|
#Nicks you want to ignore.
|
||||||
#Messages from those users will not be sent to other bridges.
|
#Messages from those users will not be sent to other bridges.
|
||||||
#OPTIONAL
|
#OPTIONAL
|
||||||
@ -91,16 +100,38 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#The string "{NOPINGNICK}" (case sensitive) will be replaced by the actual nick / username, but with a ZWSP inside the nick, so the irc user with the same nick won't get pinged. See https://github.com/42wim/matterbridge/issues/175 for more information
|
#The string "{NOPINGNICK}" (case sensitive) will be replaced by the actual nick / username, but with a ZWSP inside the nick, so the irc user with the same nick won't get pinged. See https://github.com/42wim/matterbridge/issues/175 for more information
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -109,6 +140,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#XMPP section
|
#XMPP section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -154,15 +190,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#Messages you want to replace.
|
||||||
|
#It replaces outgoing messages from the bridge.
|
||||||
|
#So you need to place it by the sending bridge definition.
|
||||||
|
#Regular expressions supported
|
||||||
|
#Some examples:
|
||||||
|
#This replaces cat => dog and sleep => awake
|
||||||
|
#ReplaceMessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#This Replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#ReplaceMessages=[ ["[0-9]","number"] ]
|
||||||
|
#OPTIONAL (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#Nicks you want to replace.
|
||||||
|
#See ReplaceMessages for syntaxA
|
||||||
|
#OPTIONAL (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -171,6 +229,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#hipchat section
|
#hipchat section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -208,15 +271,37 @@ IgnoreNicks="spammer1 spammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}/{BRIDGE}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -225,6 +310,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#mattermost section
|
#mattermost section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -322,15 +412,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -339,6 +451,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#Gitter section
|
#Gitter section
|
||||||
#Best to make a dedicated gitter account for the bot.
|
#Best to make a dedicated gitter account for the bot.
|
||||||
@ -366,15 +483,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -383,6 +522,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#slack section
|
#slack section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -418,6 +562,7 @@ WebhookBindAddress="0.0.0.0:9999"
|
|||||||
#Icon that will be showed in slack
|
#Icon that will be showed in slack
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL
|
#OPTIONAL
|
||||||
IconURL="https://robohash.org/{NICK}.png?size=48x48"
|
IconURL="https://robohash.org/{NICK}.png?size=48x48"
|
||||||
@ -457,15 +602,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -474,6 +641,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#discord section
|
#discord section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -525,15 +697,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -542,6 +736,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#telegram section
|
#telegram section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -592,15 +791,37 @@ IgnoreNicks="spammer1 spammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -609,6 +830,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#rocketchat section
|
#rocketchat section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -660,15 +886,37 @@ IgnoreNicks="ircspammer1 ircspammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -677,6 +925,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#matrix section
|
#matrix section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -720,15 +973,37 @@ IgnoreNicks="spammer1 spammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -737,6 +1012,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#steam section
|
#steam section
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -774,15 +1054,37 @@ IgnoreNicks="spammer1 spammer2"
|
|||||||
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
#OPTIONAL (example below ignores messages starting with ~~ or messages containing badword
|
||||||
IgnoreMessages="^~~ badword"
|
IgnoreMessages="^~~ badword"
|
||||||
|
|
||||||
|
#messages you want to replace.
|
||||||
|
#it replaces outgoing messages from the bridge.
|
||||||
|
#so you need to place it by the sending bridge definition.
|
||||||
|
#regular expressions supported
|
||||||
|
#some examples:
|
||||||
|
#this replaces cat => dog and sleep => awake
|
||||||
|
#replacemessages=[ ["cat","dog"], ["sleep","awake"] ]
|
||||||
|
#this replaces every number with number. 123 => numbernumbernumber
|
||||||
|
#replacemessages=[ ["[0-9]","number"] ]
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceMessages=[ ["cat","dog"] ]
|
||||||
|
|
||||||
|
#nicks you want to replace.
|
||||||
|
#see replacemessages for syntaxa
|
||||||
|
#optional (default empty)
|
||||||
|
ReplaceNicks=[ ["user--","user"] ]
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
|
|
||||||
#Enable to show users joins/parts from other bridges
|
#Enable to show users joins/parts from other bridges
|
||||||
#Only works hiding/show messages from irc and mattermost bridge for now
|
#Currently works for messages from the following bridges: irc, mattermost, slack
|
||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
ShowJoinPart=false
|
ShowJoinPart=false
|
||||||
|
|
||||||
@ -791,6 +1093,11 @@ ShowJoinPart=false
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
#Enable to show topic changes from other bridges
|
||||||
|
#Only works hiding/show topic changes from slack bridge for now
|
||||||
|
#OPTIONAL (default false)
|
||||||
|
ShowTopicChange=false
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#API
|
#API
|
||||||
###################################################################
|
###################################################################
|
||||||
@ -812,9 +1119,14 @@ Buffer=1000
|
|||||||
#OPTIONAL (no authorization if token is empty)
|
#OPTIONAL (no authorization if token is empty)
|
||||||
Token="mytoken"
|
Token="mytoken"
|
||||||
|
|
||||||
|
#extra label that can be used in the RemoteNickFormat
|
||||||
|
#optional (default empty)
|
||||||
|
Label=""
|
||||||
|
|
||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="{NICK}"
|
RemoteNickFormat="{NICK}"
|
||||||
@ -829,6 +1141,7 @@ RemoteNickFormat="{NICK}"
|
|||||||
#RemoteNickFormat defines how remote users appear on this bridge
|
#RemoteNickFormat defines how remote users appear on this bridge
|
||||||
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
#The string "{NICK}" (case sensitive) will be replaced by the actual nick / username.
|
||||||
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
#The string "{BRIDGE}" (case sensitive) will be replaced by the sending bridge
|
||||||
|
#The string "{LABEL}" (case sensitive) will be replaced by label= field of the sending bridge
|
||||||
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
#The string "{PROTOCOL}" (case sensitive) will be replaced by the protocol used by the bridge
|
||||||
#OPTIONAL (default empty)
|
#OPTIONAL (default empty)
|
||||||
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
||||||
@ -838,6 +1151,30 @@ RemoteNickFormat="[{PROTOCOL}] <{NICK}> "
|
|||||||
#OPTIONAL (default false)
|
#OPTIONAL (default false)
|
||||||
StripNick=false
|
StripNick=false
|
||||||
|
|
||||||
|
|
||||||
|
#MediaServerUpload and MediaServerDownload are used for uploading images/files/video to
|
||||||
|
#a remote "mediaserver" (a webserver like caddy for example).
|
||||||
|
#When configured images/files uploaded on bridges like mattermost,slack, telegram will be downloaded
|
||||||
|
#and uploaded again to MediaServerUpload URL
|
||||||
|
#The MediaServerDownload will be used so that bridges without native uploading support:
|
||||||
|
#gitter, irc and xmpp will be shown links to the files on MediaServerDownload
|
||||||
|
#
|
||||||
|
#More information https://github.com/42wim/matterbridge/wiki/Mediaserver-setup-%5Badvanced%5D
|
||||||
|
#OPTIONAL (default empty)
|
||||||
|
MediaServerUpload="https://user:pass@yourserver.com/upload"
|
||||||
|
#OPTIONAL (default empty)
|
||||||
|
MediaServerDownload="https://youserver.com/download"
|
||||||
|
|
||||||
|
#MediaDownloadSize is the maximum size of attachments, videos, images
|
||||||
|
#matterbridge will download and upload this file to bridges that also support uploading files.
|
||||||
|
#eg downloading from slack to upload it to mattermost
|
||||||
|
#
|
||||||
|
#It will only download from bridges that don't have public links available, which are for the moment
|
||||||
|
#slack, telegram, matrix and mattermost
|
||||||
|
#
|
||||||
|
#Optional (default 1000000 (1 megabyte))
|
||||||
|
MediaDownloadSize=1000000
|
||||||
|
|
||||||
###################################################################
|
###################################################################
|
||||||
#Gateway configuration
|
#Gateway configuration
|
||||||
###################################################################
|
###################################################################
|
||||||
|
@ -13,7 +13,8 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
log "github.com/Sirupsen/logrus"
|
log "github.com/sirupsen/logrus"
|
||||||
|
prefixed "github.com/x-cray/logrus-prefixed-formatter"
|
||||||
|
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"github.com/hashicorp/golang-lru"
|
"github.com/hashicorp/golang-lru"
|
||||||
@ -73,12 +74,16 @@ type MMClient struct {
|
|||||||
func New(login, pass, team, server string) *MMClient {
|
func New(login, pass, team, server string) *MMClient {
|
||||||
cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server}
|
cred := &Credentials{Login: login, Pass: pass, Team: team, Server: server}
|
||||||
mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)}
|
mmclient := &MMClient{Credentials: cred, MessageChan: make(chan *Message, 100), Users: make(map[string]*model.User)}
|
||||||
mmclient.log = log.WithFields(log.Fields{"module": "matterclient"})
|
log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true})
|
||||||
log.SetFormatter(&log.TextFormatter{FullTimestamp: true})
|
mmclient.log = log.WithFields(log.Fields{"prefix": "matterclient"})
|
||||||
mmclient.lruCache, _ = lru.New(500)
|
mmclient.lruCache, _ = lru.New(500)
|
||||||
return mmclient
|
return mmclient
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (m *MMClient) SetDebugLog() {
|
||||||
|
log.SetFormatter(&prefixed.TextFormatter{PrefixPadding: 13, DisableColors: true, FullTimestamp: false})
|
||||||
|
}
|
||||||
|
|
||||||
func (m *MMClient) SetLogLevel(level string) {
|
func (m *MMClient) SetLogLevel(level string) {
|
||||||
l, err := log.ParseLevel(level)
|
l, err := log.ParseLevel(level)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -585,9 +590,9 @@ func (m *MMClient) UpdateChannelHeader(channelId string, header string) {
|
|||||||
func (m *MMClient) UpdateLastViewed(channelId string) {
|
func (m *MMClient) UpdateLastViewed(channelId string) {
|
||||||
m.log.Debugf("posting lastview %#v", channelId)
|
m.log.Debugf("posting lastview %#v", channelId)
|
||||||
view := &model.ChannelView{ChannelId: channelId}
|
view := &model.ChannelView{ChannelId: channelId}
|
||||||
res, _ := m.Client.ViewChannel(m.User.Id, view)
|
_, resp := m.Client.ViewChannel(m.User.Id, view)
|
||||||
if !res {
|
if resp.Error != nil {
|
||||||
m.log.Errorf("ChannelView update for %s failed", channelId)
|
m.log.Errorf("ChannelView update for %s failed: %s", channelId, resp.Error)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -817,9 +822,14 @@ func (m *MMClient) StatusLoop() {
|
|||||||
backoff = time.Second * 60
|
backoff = time.Second * 60
|
||||||
case <-time.After(time.Second * 5):
|
case <-time.After(time.Second * 5):
|
||||||
if retries > 3 {
|
if retries > 3 {
|
||||||
|
m.log.Debug("StatusLoop() timeout")
|
||||||
m.Logout()
|
m.Logout()
|
||||||
m.WsQuit = false
|
m.WsQuit = false
|
||||||
m.Login()
|
err := m.Login()
|
||||||
|
if err != nil {
|
||||||
|
log.Errorf("Login failed: %#v", err)
|
||||||
|
break
|
||||||
|
}
|
||||||
if m.OnWsConnect != nil {
|
if m.OnWsConnect != nil {
|
||||||
m.OnWsConnect()
|
m.OnWsConnect()
|
||||||
}
|
}
|
||||||
|
30
vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
generated
vendored
30
vendor/github.com/Sirupsen/logrus/examples/hook/hook.go
generated
vendored
@ -1,30 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"github.com/Sirupsen/logrus"
|
|
||||||
"gopkg.in/gemnasium/logrus-airbrake-hook.v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
var log = logrus.New()
|
|
||||||
|
|
||||||
func init() {
|
|
||||||
log.Formatter = new(logrus.TextFormatter) // default
|
|
||||||
log.Hooks.Add(airbrake.NewHook(123, "xyz", "development"))
|
|
||||||
}
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
log.WithFields(logrus.Fields{
|
|
||||||
"animal": "walrus",
|
|
||||||
"size": 10,
|
|
||||||
}).Info("A group of walrus emerges from the ocean")
|
|
||||||
|
|
||||||
log.WithFields(logrus.Fields{
|
|
||||||
"omg": true,
|
|
||||||
"number": 122,
|
|
||||||
}).Warn("The group's number increased tremendously!")
|
|
||||||
|
|
||||||
log.WithFields(logrus.Fields{
|
|
||||||
"omg": true,
|
|
||||||
"number": 100,
|
|
||||||
}).Fatal("The ice breaks!")
|
|
||||||
}
|
|
67
vendor/github.com/Sirupsen/logrus/hooks/test/test.go
generated
vendored
67
vendor/github.com/Sirupsen/logrus/hooks/test/test.go
generated
vendored
@ -1,67 +0,0 @@
|
|||||||
package test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io/ioutil"
|
|
||||||
|
|
||||||
"github.com/Sirupsen/logrus"
|
|
||||||
)
|
|
||||||
|
|
||||||
// test.Hook is a hook designed for dealing with logs in test scenarios.
|
|
||||||
type Hook struct {
|
|
||||||
Entries []*logrus.Entry
|
|
||||||
}
|
|
||||||
|
|
||||||
// Installs a test hook for the global logger.
|
|
||||||
func NewGlobal() *Hook {
|
|
||||||
|
|
||||||
hook := new(Hook)
|
|
||||||
logrus.AddHook(hook)
|
|
||||||
|
|
||||||
return hook
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Installs a test hook for a given local logger.
|
|
||||||
func NewLocal(logger *logrus.Logger) *Hook {
|
|
||||||
|
|
||||||
hook := new(Hook)
|
|
||||||
logger.Hooks.Add(hook)
|
|
||||||
|
|
||||||
return hook
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Creates a discarding logger and installs the test hook.
|
|
||||||
func NewNullLogger() (*logrus.Logger, *Hook) {
|
|
||||||
|
|
||||||
logger := logrus.New()
|
|
||||||
logger.Out = ioutil.Discard
|
|
||||||
|
|
||||||
return logger, NewLocal(logger)
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *Hook) Fire(e *logrus.Entry) error {
|
|
||||||
t.Entries = append(t.Entries, e)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *Hook) Levels() []logrus.Level {
|
|
||||||
return logrus.AllLevels
|
|
||||||
}
|
|
||||||
|
|
||||||
// LastEntry returns the last entry that was logged or nil.
|
|
||||||
func (t *Hook) LastEntry() (l *logrus.Entry) {
|
|
||||||
|
|
||||||
if i := len(t.Entries) - 1; i < 0 {
|
|
||||||
return nil
|
|
||||||
} else {
|
|
||||||
return t.Entries[i]
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Reset removes all Entries from this test hook.
|
|
||||||
func (t *Hook) Reset() {
|
|
||||||
t.Entries = make([]*logrus.Entry, 0)
|
|
||||||
}
|
|
10
vendor/github.com/Sirupsen/logrus/terminal_appengine.go
generated
vendored
10
vendor/github.com/Sirupsen/logrus/terminal_appengine.go
generated
vendored
@ -1,10 +0,0 @@
|
|||||||
// +build appengine
|
|
||||||
|
|
||||||
package logrus
|
|
||||||
|
|
||||||
import "io"
|
|
||||||
|
|
||||||
// IsTerminal returns true if stderr's file descriptor is a terminal.
|
|
||||||
func IsTerminal(f io.Writer) bool {
|
|
||||||
return true
|
|
||||||
}
|
|
10
vendor/github.com/Sirupsen/logrus/terminal_bsd.go
generated
vendored
10
vendor/github.com/Sirupsen/logrus/terminal_bsd.go
generated
vendored
@ -1,10 +0,0 @@
|
|||||||
// +build darwin freebsd openbsd netbsd dragonfly
|
|
||||||
// +build !appengine
|
|
||||||
|
|
||||||
package logrus
|
|
||||||
|
|
||||||
import "syscall"
|
|
||||||
|
|
||||||
const ioctlReadTermios = syscall.TIOCGETA
|
|
||||||
|
|
||||||
type Termios syscall.Termios
|
|
28
vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
generated
vendored
28
vendor/github.com/Sirupsen/logrus/terminal_notwindows.go
generated
vendored
@ -1,28 +0,0 @@
|
|||||||
// Based on ssh/terminal:
|
|
||||||
// Copyright 2011 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.
|
|
||||||
|
|
||||||
// +build linux darwin freebsd openbsd netbsd dragonfly
|
|
||||||
// +build !appengine
|
|
||||||
|
|
||||||
package logrus
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
"os"
|
|
||||||
"syscall"
|
|
||||||
"unsafe"
|
|
||||||
)
|
|
||||||
|
|
||||||
// IsTerminal returns true if stderr's file descriptor is a terminal.
|
|
||||||
func IsTerminal(f io.Writer) bool {
|
|
||||||
var termios Termios
|
|
||||||
switch v := f.(type) {
|
|
||||||
case *os.File:
|
|
||||||
_, _, err := syscall.Syscall6(syscall.SYS_IOCTL, uintptr(v.Fd()), ioctlReadTermios, uintptr(unsafe.Pointer(&termios)), 0, 0, 0)
|
|
||||||
return err == 0
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
21
vendor/github.com/Sirupsen/logrus/terminal_solaris.go
generated
vendored
21
vendor/github.com/Sirupsen/logrus/terminal_solaris.go
generated
vendored
@ -1,21 +0,0 @@
|
|||||||
// +build solaris,!appengine
|
|
||||||
|
|
||||||
package logrus
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
"os"
|
|
||||||
|
|
||||||
"golang.org/x/sys/unix"
|
|
||||||
)
|
|
||||||
|
|
||||||
// IsTerminal returns true if the given file descriptor is a terminal.
|
|
||||||
func IsTerminal(f io.Writer) bool {
|
|
||||||
switch v := f.(type) {
|
|
||||||
case *os.File:
|
|
||||||
_, err := unix.IoctlGetTermios(int(v.Fd()), unix.TCGETA)
|
|
||||||
return err == nil
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
33
vendor/github.com/Sirupsen/logrus/terminal_windows.go
generated
vendored
33
vendor/github.com/Sirupsen/logrus/terminal_windows.go
generated
vendored
@ -1,33 +0,0 @@
|
|||||||
// Based on ssh/terminal:
|
|
||||||
// Copyright 2011 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.
|
|
||||||
|
|
||||||
// +build windows,!appengine
|
|
||||||
|
|
||||||
package logrus
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io"
|
|
||||||
"os"
|
|
||||||
"syscall"
|
|
||||||
"unsafe"
|
|
||||||
)
|
|
||||||
|
|
||||||
var kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
|
||||||
|
|
||||||
var (
|
|
||||||
procGetConsoleMode = kernel32.NewProc("GetConsoleMode")
|
|
||||||
)
|
|
||||||
|
|
||||||
// IsTerminal returns true if stderr's file descriptor is a terminal.
|
|
||||||
func IsTerminal(f io.Writer) bool {
|
|
||||||
switch v := f.(type) {
|
|
||||||
case *os.File:
|
|
||||||
var st uint32
|
|
||||||
r, _, e := syscall.Syscall(procGetConsoleMode.Addr(), 2, uintptr(v.Fd()), uintptr(unsafe.Pointer(&st)), 0)
|
|
||||||
return r != 0 && e == 0
|
|
||||||
default:
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
4
vendor/github.com/bwmarrin/discordgo/discord.go
generated
vendored
4
vendor/github.com/bwmarrin/discordgo/discord.go
generated
vendored
@ -21,7 +21,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
// VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/)
|
// VERSION of DiscordGo, follows Semantic Versioning. (http://semver.org/)
|
||||||
const VERSION = "0.17.0"
|
const VERSION = "0.18.0"
|
||||||
|
|
||||||
// ErrMFA will be risen by New when the user has 2FA.
|
// ErrMFA will be risen by New when the user has 2FA.
|
||||||
var ErrMFA = errors.New("account has 2FA enabled")
|
var ErrMFA = errors.New("account has 2FA enabled")
|
||||||
@ -50,7 +50,7 @@ func New(args ...interface{}) (s *Session, err error) {
|
|||||||
// Create an empty Session interface.
|
// Create an empty Session interface.
|
||||||
s = &Session{
|
s = &Session{
|
||||||
State: NewState(),
|
State: NewState(),
|
||||||
ratelimiter: NewRatelimiter(),
|
Ratelimiter: NewRatelimiter(),
|
||||||
StateEnabled: true,
|
StateEnabled: true,
|
||||||
Compress: true,
|
Compress: true,
|
||||||
ShouldReconnectOnError: true,
|
ShouldReconnectOnError: true,
|
||||||
|
5
vendor/github.com/bwmarrin/discordgo/endpoints.go
generated
vendored
5
vendor/github.com/bwmarrin/discordgo/endpoints.go
generated
vendored
@ -71,7 +71,6 @@ var (
|
|||||||
EndpointUserNotes = func(uID string) string { return EndpointUsers + "@me/notes/" + uID }
|
EndpointUserNotes = func(uID string) string { return EndpointUsers + "@me/notes/" + uID }
|
||||||
|
|
||||||
EndpointGuild = func(gID string) string { return EndpointGuilds + gID }
|
EndpointGuild = func(gID string) string { return EndpointGuilds + gID }
|
||||||
EndpointGuildInivtes = func(gID string) string { return EndpointGuilds + gID + "/invites" }
|
|
||||||
EndpointGuildChannels = func(gID string) string { return EndpointGuilds + gID + "/channels" }
|
EndpointGuildChannels = func(gID string) string { return EndpointGuilds + gID + "/channels" }
|
||||||
EndpointGuildMembers = func(gID string) string { return EndpointGuilds + gID + "/members" }
|
EndpointGuildMembers = func(gID string) string { return EndpointGuilds + gID + "/members" }
|
||||||
EndpointGuildMember = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID }
|
EndpointGuildMember = func(gID, uID string) string { return EndpointGuilds + gID + "/members/" + uID }
|
||||||
@ -98,7 +97,7 @@ var (
|
|||||||
EndpointChannelMessages = func(cID string) string { return EndpointChannels + cID + "/messages" }
|
EndpointChannelMessages = func(cID string) string { return EndpointChannels + cID + "/messages" }
|
||||||
EndpointChannelMessage = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID }
|
EndpointChannelMessage = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID }
|
||||||
EndpointChannelMessageAck = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" }
|
EndpointChannelMessageAck = func(cID, mID string) string { return EndpointChannels + cID + "/messages/" + mID + "/ack" }
|
||||||
EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk_delete" }
|
EndpointChannelMessagesBulkDelete = func(cID string) string { return EndpointChannel(cID) + "/messages/bulk-delete" }
|
||||||
EndpointChannelMessagesPins = func(cID string) string { return EndpointChannel(cID) + "/pins" }
|
EndpointChannelMessagesPins = func(cID string) string { return EndpointChannel(cID) + "/pins" }
|
||||||
EndpointChannelMessagePin = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID }
|
EndpointChannelMessagePin = func(cID, mID string) string { return EndpointChannel(cID) + "/pins/" + mID }
|
||||||
|
|
||||||
@ -122,6 +121,8 @@ var (
|
|||||||
EndpointRelationship = func(uID string) string { return EndpointRelationships() + "/" + uID }
|
EndpointRelationship = func(uID string) string { return EndpointRelationships() + "/" + uID }
|
||||||
EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" }
|
EndpointRelationshipsMutual = func(uID string) string { return EndpointUsers + uID + "/relationships" }
|
||||||
|
|
||||||
|
EndpointGuildCreate = EndpointAPI + "guilds"
|
||||||
|
|
||||||
EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID }
|
EndpointInvite = func(iID string) string { return EndpointAPI + "invite/" + iID }
|
||||||
|
|
||||||
EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" }
|
EndpointIntegrationsJoin = func(iID string) string { return EndpointAPI + "integrations/" + iID + "/join" }
|
||||||
|
2
vendor/github.com/bwmarrin/discordgo/event.go
generated
vendored
2
vendor/github.com/bwmarrin/discordgo/event.go
generated
vendored
@ -6,7 +6,7 @@ type EventHandler interface {
|
|||||||
Type() string
|
Type() string
|
||||||
|
|
||||||
// Handle is called whenever an event of Type() happens.
|
// Handle is called whenever an event of Type() happens.
|
||||||
// It is the recievers responsibility to type assert that the interface
|
// It is the receivers responsibility to type assert that the interface
|
||||||
// is the expected struct.
|
// is the expected struct.
|
||||||
Handle(*Session, interface{})
|
Handle(*Session, interface{})
|
||||||
}
|
}
|
||||||
|
2
vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
generated
vendored
2
vendor/github.com/bwmarrin/discordgo/examples/appmaker/main.go
generated
vendored
@ -79,7 +79,7 @@ func main() {
|
|||||||
ap.Name = Name
|
ap.Name = Name
|
||||||
ap, err = dg.ApplicationCreate(ap)
|
ap, err = dg.ApplicationCreate(ap)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Println("error creating new applicaiton,", err)
|
fmt.Println("error creating new application,", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
28
vendor/github.com/bwmarrin/discordgo/logging.go
generated
vendored
28
vendor/github.com/bwmarrin/discordgo/logging.go
generated
vendored
@ -23,7 +23,7 @@ const (
|
|||||||
LogError int = iota
|
LogError int = iota
|
||||||
|
|
||||||
// LogWarning level is used for very abnormal events and errors that are
|
// LogWarning level is used for very abnormal events and errors that are
|
||||||
// also returend to a calling function.
|
// also returned to a calling function.
|
||||||
LogWarning
|
LogWarning
|
||||||
|
|
||||||
// LogInformational level is used for normal non-error activity
|
// LogInformational level is used for normal non-error activity
|
||||||
@ -34,26 +34,34 @@ const (
|
|||||||
LogDebug
|
LogDebug
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Logger can be used to replace the standard logging for discordgo
|
||||||
|
var Logger func(msgL, caller int, format string, a ...interface{})
|
||||||
|
|
||||||
// msglog provides package wide logging consistancy for discordgo
|
// msglog provides package wide logging consistancy for discordgo
|
||||||
// the format, a... portion this command follows that of fmt.Printf
|
// the format, a... portion this command follows that of fmt.Printf
|
||||||
// msgL : LogLevel of the message
|
// msgL : LogLevel of the message
|
||||||
// caller : 1 + the number of callers away from the message source
|
// caller : 1 + the number of callers away from the message source
|
||||||
// format : Printf style message format
|
// format : Printf style message format
|
||||||
// a ... : comma seperated list of values to pass
|
// a ... : comma separated list of values to pass
|
||||||
func msglog(msgL, caller int, format string, a ...interface{}) {
|
func msglog(msgL, caller int, format string, a ...interface{}) {
|
||||||
|
|
||||||
pc, file, line, _ := runtime.Caller(caller)
|
if Logger != nil {
|
||||||
|
Logger(msgL, caller, format, a...)
|
||||||
|
} else {
|
||||||
|
|
||||||
files := strings.Split(file, "/")
|
pc, file, line, _ := runtime.Caller(caller)
|
||||||
file = files[len(files)-1]
|
|
||||||
|
|
||||||
name := runtime.FuncForPC(pc).Name()
|
files := strings.Split(file, "/")
|
||||||
fns := strings.Split(name, ".")
|
file = files[len(files)-1]
|
||||||
name = fns[len(fns)-1]
|
|
||||||
|
|
||||||
msg := fmt.Sprintf(format, a...)
|
name := runtime.FuncForPC(pc).Name()
|
||||||
|
fns := strings.Split(name, ".")
|
||||||
|
name = fns[len(fns)-1]
|
||||||
|
|
||||||
log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg)
|
msg := fmt.Sprintf(format, a...)
|
||||||
|
|
||||||
|
log.Printf("[DG%d] %s:%d:%s() %s\n", msgL, file, line, name, msg)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// helper function that wraps msglog for the Session struct
|
// helper function that wraps msglog for the Session struct
|
||||||
|
2
vendor/github.com/bwmarrin/discordgo/message.go
generated
vendored
2
vendor/github.com/bwmarrin/discordgo/message.go
generated
vendored
@ -237,7 +237,7 @@ func (m *Message) ContentWithMoreMentionsReplaced(s *Session) (content string, e
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
content = strings.Replace(content, "<&"+role.ID+">", "@"+role.Name, -1)
|
content = strings.Replace(content, "<@&"+role.ID+">", "@"+role.Name, -1)
|
||||||
}
|
}
|
||||||
|
|
||||||
content = patternChannels.ReplaceAllStringFunc(content, func(mention string) string {
|
content = patternChannels.ReplaceAllStringFunc(content, func(mention string) string {
|
||||||
|
49
vendor/github.com/bwmarrin/discordgo/ratelimit.go
generated
vendored
49
vendor/github.com/bwmarrin/discordgo/ratelimit.go
generated
vendored
@ -41,8 +41,8 @@ func NewRatelimiter() *RateLimiter {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// getBucket retrieves or creates a bucket
|
// GetBucket retrieves or creates a bucket
|
||||||
func (r *RateLimiter) getBucket(key string) *Bucket {
|
func (r *RateLimiter) GetBucket(key string) *Bucket {
|
||||||
r.Lock()
|
r.Lock()
|
||||||
defer r.Unlock()
|
defer r.Unlock()
|
||||||
|
|
||||||
@ -51,7 +51,7 @@ func (r *RateLimiter) getBucket(key string) *Bucket {
|
|||||||
}
|
}
|
||||||
|
|
||||||
b := &Bucket{
|
b := &Bucket{
|
||||||
remaining: 1,
|
Remaining: 1,
|
||||||
Key: key,
|
Key: key,
|
||||||
global: r.global,
|
global: r.global,
|
||||||
}
|
}
|
||||||
@ -68,27 +68,37 @@ func (r *RateLimiter) getBucket(key string) *Bucket {
|
|||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
// LockBucket Locks until a request can be made
|
// GetWaitTime returns the duration you should wait for a Bucket
|
||||||
func (r *RateLimiter) LockBucket(bucketID string) *Bucket {
|
func (r *RateLimiter) GetWaitTime(b *Bucket, minRemaining int) time.Duration {
|
||||||
|
|
||||||
b := r.getBucket(bucketID)
|
|
||||||
|
|
||||||
b.Lock()
|
|
||||||
|
|
||||||
// If we ran out of calls and the reset time is still ahead of us
|
// If we ran out of calls and the reset time is still ahead of us
|
||||||
// then we need to take it easy and relax a little
|
// then we need to take it easy and relax a little
|
||||||
if b.remaining < 1 && b.reset.After(time.Now()) {
|
if b.Remaining < minRemaining && b.reset.After(time.Now()) {
|
||||||
time.Sleep(b.reset.Sub(time.Now()))
|
return b.reset.Sub(time.Now())
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Check for global ratelimits
|
// Check for global ratelimits
|
||||||
sleepTo := time.Unix(0, atomic.LoadInt64(r.global))
|
sleepTo := time.Unix(0, atomic.LoadInt64(r.global))
|
||||||
if now := time.Now(); now.Before(sleepTo) {
|
if now := time.Now(); now.Before(sleepTo) {
|
||||||
time.Sleep(sleepTo.Sub(now))
|
return sleepTo.Sub(now)
|
||||||
}
|
}
|
||||||
|
|
||||||
b.remaining--
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// LockBucket Locks until a request can be made
|
||||||
|
func (r *RateLimiter) LockBucket(bucketID string) *Bucket {
|
||||||
|
return r.LockBucketObject(r.GetBucket(bucketID))
|
||||||
|
}
|
||||||
|
|
||||||
|
// LockBucketObject Locks an already resolved bucket until a request can be made
|
||||||
|
func (r *RateLimiter) LockBucketObject(b *Bucket) *Bucket {
|
||||||
|
b.Lock()
|
||||||
|
|
||||||
|
if wait := r.GetWaitTime(b, 1); wait > 0 {
|
||||||
|
time.Sleep(wait)
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Remaining--
|
||||||
return b
|
return b
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -96,13 +106,14 @@ func (r *RateLimiter) LockBucket(bucketID string) *Bucket {
|
|||||||
type Bucket struct {
|
type Bucket struct {
|
||||||
sync.Mutex
|
sync.Mutex
|
||||||
Key string
|
Key string
|
||||||
remaining int
|
Remaining int
|
||||||
limit int
|
limit int
|
||||||
reset time.Time
|
reset time.Time
|
||||||
global *int64
|
global *int64
|
||||||
|
|
||||||
lastReset time.Time
|
lastReset time.Time
|
||||||
customRateLimit *customRateLimit
|
customRateLimit *customRateLimit
|
||||||
|
Userdata interface{}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Release unlocks the bucket and reads the headers to update the buckets ratelimit info
|
// Release unlocks the bucket and reads the headers to update the buckets ratelimit info
|
||||||
@ -113,10 +124,10 @@ func (b *Bucket) Release(headers http.Header) error {
|
|||||||
// Check if the bucket uses a custom ratelimiter
|
// Check if the bucket uses a custom ratelimiter
|
||||||
if rl := b.customRateLimit; rl != nil {
|
if rl := b.customRateLimit; rl != nil {
|
||||||
if time.Now().Sub(b.lastReset) >= rl.reset {
|
if time.Now().Sub(b.lastReset) >= rl.reset {
|
||||||
b.remaining = rl.requests - 1
|
b.Remaining = rl.requests - 1
|
||||||
b.lastReset = time.Now()
|
b.lastReset = time.Now()
|
||||||
}
|
}
|
||||||
if b.remaining < 1 {
|
if b.Remaining < 1 {
|
||||||
b.reset = time.Now().Add(rl.reset)
|
b.reset = time.Now().Add(rl.reset)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
@ -176,7 +187,7 @@ func (b *Bucket) Release(headers http.Header) error {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
b.remaining = int(parsedRemaining)
|
b.Remaining = int(parsedRemaining)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
|
54
vendor/github.com/bwmarrin/discordgo/restapi.go
generated
vendored
54
vendor/github.com/bwmarrin/discordgo/restapi.go
generated
vendored
@ -65,9 +65,11 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID
|
|||||||
if bucketID == "" {
|
if bucketID == "" {
|
||||||
bucketID = strings.SplitN(urlStr, "?", 2)[0]
|
bucketID = strings.SplitN(urlStr, "?", 2)[0]
|
||||||
}
|
}
|
||||||
|
return s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucket(bucketID), sequence)
|
||||||
|
}
|
||||||
|
|
||||||
bucket := s.ratelimiter.LockBucket(bucketID)
|
// RequestWithLockedBucket makes a request using a bucket that's already been locked
|
||||||
|
func (s *Session) RequestWithLockedBucket(method, urlStr, contentType string, b []byte, bucket *Bucket, sequence int) (response []byte, err error) {
|
||||||
if s.Debug {
|
if s.Debug {
|
||||||
log.Printf("API REQUEST %8s :: %s\n", method, urlStr)
|
log.Printf("API REQUEST %8s :: %s\n", method, urlStr)
|
||||||
log.Printf("API REQUEST PAYLOAD :: [%s]\n", string(b))
|
log.Printf("API REQUEST PAYLOAD :: [%s]\n", string(b))
|
||||||
@ -139,7 +141,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID
|
|||||||
if sequence < s.MaxRestRetries {
|
if sequence < s.MaxRestRetries {
|
||||||
|
|
||||||
s.log(LogInformational, "%s Failed (%s), Retrying...", urlStr, resp.Status)
|
s.log(LogInformational, "%s Failed (%s), Retrying...", urlStr, resp.Status)
|
||||||
response, err = s.request(method, urlStr, contentType, b, bucketID, sequence+1)
|
response, err = s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucketObject(bucket), sequence+1)
|
||||||
} else {
|
} else {
|
||||||
err = fmt.Errorf("Exceeded Max retries HTTP %s, %s", resp.Status, response)
|
err = fmt.Errorf("Exceeded Max retries HTTP %s, %s", resp.Status, response)
|
||||||
}
|
}
|
||||||
@ -158,7 +160,7 @@ func (s *Session) request(method, urlStr, contentType string, b []byte, bucketID
|
|||||||
// we can make the above smarter
|
// we can make the above smarter
|
||||||
// this method can cause longer delays than required
|
// this method can cause longer delays than required
|
||||||
|
|
||||||
response, err = s.request(method, urlStr, contentType, b, bucketID, sequence)
|
response, err = s.RequestWithLockedBucket(method, urlStr, contentType, b, s.Ratelimiter.LockBucketObject(bucket), sequence)
|
||||||
|
|
||||||
default: // Error condition
|
default: // Error condition
|
||||||
err = newRestError(req, resp, response)
|
err = newRestError(req, resp, response)
|
||||||
@ -585,7 +587,7 @@ func (s *Session) GuildCreate(name string) (st *Guild, err error) {
|
|||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
}{name}
|
}{name}
|
||||||
|
|
||||||
body, err := s.RequestWithBucketID("POST", EndpointGuilds, data, EndpointGuilds)
|
body, err := s.RequestWithBucketID("POST", EndpointGuildCreate, data, EndpointGuildCreate)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -907,7 +909,7 @@ func (s *Session) GuildChannelsReorder(guildID string, channels []*Channel) (err
|
|||||||
// GuildInvites returns an array of Invite structures for the given guild
|
// GuildInvites returns an array of Invite structures for the given guild
|
||||||
// guildID : The ID of a Guild.
|
// guildID : The ID of a Guild.
|
||||||
func (s *Session) GuildInvites(guildID string) (st []*Invite, err error) {
|
func (s *Session) GuildInvites(guildID string) (st []*Invite, err error) {
|
||||||
body, err := s.RequestWithBucketID("GET", EndpointGuildInvites(guildID), nil, EndpointGuildInivtes(guildID))
|
body, err := s.RequestWithBucketID("GET", EndpointGuildInvites(guildID), nil, EndpointGuildInvites(guildID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -957,6 +959,7 @@ func (s *Session) GuildRoleEdit(guildID, roleID, name string, color int, hoist b
|
|||||||
// Prevent sending a color int that is too big.
|
// Prevent sending a color int that is too big.
|
||||||
if color > 0xFFFFFF {
|
if color > 0xFFFFFF {
|
||||||
err = fmt.Errorf("color value cannot be larger than 0xFFFFFF")
|
err = fmt.Errorf("color value cannot be larger than 0xFFFFFF")
|
||||||
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
data := struct {
|
data := struct {
|
||||||
@ -1020,6 +1023,9 @@ func (s *Session) GuildPruneCount(guildID string, days uint32) (count uint32, er
|
|||||||
|
|
||||||
uri := EndpointGuildPrune(guildID) + fmt.Sprintf("?days=%d", days)
|
uri := EndpointGuildPrune(guildID) + fmt.Sprintf("?days=%d", days)
|
||||||
body, err := s.RequestWithBucketID("GET", uri, nil, EndpointGuildPrune(guildID))
|
body, err := s.RequestWithBucketID("GET", uri, nil, EndpointGuildPrune(guildID))
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
err = unmarshal(body, &p)
|
err = unmarshal(body, &p)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -1204,7 +1210,7 @@ func (s *Session) GuildEmbedEdit(guildID string, enabled bool, channelID string)
|
|||||||
// Functions specific to Discord Channels
|
// Functions specific to Discord Channels
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
// Channel returns a Channel strucutre of a specific Channel.
|
// Channel returns a Channel structure of a specific Channel.
|
||||||
// channelID : The ID of the Channel you want returned.
|
// channelID : The ID of the Channel you want returned.
|
||||||
func (s *Session) Channel(channelID string) (st *Channel, err error) {
|
func (s *Session) Channel(channelID string) (st *Channel, err error) {
|
||||||
body, err := s.RequestWithBucketID("GET", EndpointChannel(channelID), nil, EndpointChannel(channelID))
|
body, err := s.RequestWithBucketID("GET", EndpointChannel(channelID), nil, EndpointChannel(channelID))
|
||||||
@ -1219,12 +1225,16 @@ func (s *Session) Channel(channelID string) (st *Channel, err error) {
|
|||||||
// ChannelEdit edits the given channel
|
// ChannelEdit edits the given channel
|
||||||
// channelID : The ID of a Channel
|
// channelID : The ID of a Channel
|
||||||
// name : The new name to assign the channel.
|
// name : The new name to assign the channel.
|
||||||
func (s *Session) ChannelEdit(channelID, name string) (st *Channel, err error) {
|
func (s *Session) ChannelEdit(channelID, name string) (*Channel, error) {
|
||||||
|
return s.ChannelEditComplex(channelID, &ChannelEdit{
|
||||||
data := struct {
|
Name: name,
|
||||||
Name string `json:"name"`
|
})
|
||||||
}{name}
|
}
|
||||||
|
|
||||||
|
// ChannelEditComplex edits an existing channel, replacing the parameters entirely with ChannelEdit struct
|
||||||
|
// channelID : The ID of a Channel
|
||||||
|
// data : The channel struct to send
|
||||||
|
func (s *Session) ChannelEditComplex(channelID string, data *ChannelEdit) (st *Channel, err error) {
|
||||||
body, err := s.RequestWithBucketID("PATCH", EndpointChannel(channelID), data, EndpointChannel(channelID))
|
body, err := s.RequestWithBucketID("PATCH", EndpointChannel(channelID), data, EndpointChannel(channelID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
@ -1476,7 +1486,7 @@ func (s *Session) ChannelMessageDelete(channelID, messageID string) (err error)
|
|||||||
}
|
}
|
||||||
|
|
||||||
// ChannelMessagesBulkDelete bulk deletes the messages from the channel for the provided messageIDs.
|
// ChannelMessagesBulkDelete bulk deletes the messages from the channel for the provided messageIDs.
|
||||||
// If only one messageID is in the slice call channelMessageDelete funciton.
|
// If only one messageID is in the slice call channelMessageDelete function.
|
||||||
// If the slice is empty do nothing.
|
// If the slice is empty do nothing.
|
||||||
// channelID : The ID of the channel for the messages to delete.
|
// channelID : The ID of the channel for the messages to delete.
|
||||||
// messages : The IDs of the messages to be deleted. A slice of string IDs. A maximum of 100 messages.
|
// messages : The IDs of the messages to be deleted. A slice of string IDs. A maximum of 100 messages.
|
||||||
@ -1569,16 +1579,14 @@ func (s *Session) ChannelInvites(channelID string) (st []*Invite, err error) {
|
|||||||
|
|
||||||
// ChannelInviteCreate creates a new invite for the given channel.
|
// ChannelInviteCreate creates a new invite for the given channel.
|
||||||
// channelID : The ID of a Channel
|
// channelID : The ID of a Channel
|
||||||
// i : An Invite struct with the values MaxAge, MaxUses, Temporary,
|
// i : An Invite struct with the values MaxAge, MaxUses and Temporary defined.
|
||||||
// and XkcdPass defined.
|
|
||||||
func (s *Session) ChannelInviteCreate(channelID string, i Invite) (st *Invite, err error) {
|
func (s *Session) ChannelInviteCreate(channelID string, i Invite) (st *Invite, err error) {
|
||||||
|
|
||||||
data := struct {
|
data := struct {
|
||||||
MaxAge int `json:"max_age"`
|
MaxAge int `json:"max_age"`
|
||||||
MaxUses int `json:"max_uses"`
|
MaxUses int `json:"max_uses"`
|
||||||
Temporary bool `json:"temporary"`
|
Temporary bool `json:"temporary"`
|
||||||
XKCDPass string `json:"xkcdpass"`
|
}{i.MaxAge, i.MaxUses, i.Temporary}
|
||||||
}{i.MaxAge, i.MaxUses, i.Temporary, i.XkcdPass}
|
|
||||||
|
|
||||||
body, err := s.RequestWithBucketID("POST", EndpointChannelInvites(channelID), data, EndpointChannelInvites(channelID))
|
body, err := s.RequestWithBucketID("POST", EndpointChannelInvites(channelID), data, EndpointChannelInvites(channelID))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -1618,7 +1626,7 @@ func (s *Session) ChannelPermissionDelete(channelID, targetID string) (err error
|
|||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
// Invite returns an Invite structure of the given invite
|
// Invite returns an Invite structure of the given invite
|
||||||
// inviteID : The invite code (or maybe xkcdpass?)
|
// inviteID : The invite code
|
||||||
func (s *Session) Invite(inviteID string) (st *Invite, err error) {
|
func (s *Session) Invite(inviteID string) (st *Invite, err error) {
|
||||||
|
|
||||||
body, err := s.RequestWithBucketID("GET", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
body, err := s.RequestWithBucketID("GET", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
||||||
@ -1631,7 +1639,7 @@ func (s *Session) Invite(inviteID string) (st *Invite, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// InviteDelete deletes an existing invite
|
// InviteDelete deletes an existing invite
|
||||||
// inviteID : the code (or maybe xkcdpass?) of an invite
|
// inviteID : the code of an invite
|
||||||
func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) {
|
func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) {
|
||||||
|
|
||||||
body, err := s.RequestWithBucketID("DELETE", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
body, err := s.RequestWithBucketID("DELETE", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
||||||
@ -1644,7 +1652,7 @@ func (s *Session) InviteDelete(inviteID string) (st *Invite, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// InviteAccept accepts an Invite to a Guild or Channel
|
// InviteAccept accepts an Invite to a Guild or Channel
|
||||||
// inviteID : The invite code (or maybe xkcdpass?)
|
// inviteID : The invite code
|
||||||
func (s *Session) InviteAccept(inviteID string) (st *Invite, err error) {
|
func (s *Session) InviteAccept(inviteID string) (st *Invite, err error) {
|
||||||
|
|
||||||
body, err := s.RequestWithBucketID("POST", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
body, err := s.RequestWithBucketID("POST", EndpointInvite(inviteID), nil, EndpointInvite(""))
|
||||||
|
9
vendor/github.com/bwmarrin/discordgo/state.go
generated
vendored
9
vendor/github.com/bwmarrin/discordgo/state.go
generated
vendored
@ -531,7 +531,7 @@ func (s *State) PrivateChannel(channelID string) (*Channel, error) {
|
|||||||
return s.Channel(channelID)
|
return s.Channel(channelID)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Channel gets a channel by ID, it will look in all guilds an private channels.
|
// Channel gets a channel by ID, it will look in all guilds and private channels.
|
||||||
func (s *State) Channel(channelID string) (*Channel, error) {
|
func (s *State) Channel(channelID string) (*Channel, error) {
|
||||||
if s == nil {
|
if s == nil {
|
||||||
return nil, ErrNilState
|
return nil, ErrNilState
|
||||||
@ -816,6 +816,13 @@ func (s *State) OnInterface(se *Session, i interface{}) (err error) {
|
|||||||
if s.TrackMembers {
|
if s.TrackMembers {
|
||||||
err = s.MemberRemove(t.Member)
|
err = s.MemberRemove(t.Member)
|
||||||
}
|
}
|
||||||
|
case *GuildMembersChunk:
|
||||||
|
if s.TrackMembers {
|
||||||
|
for i := range t.Members {
|
||||||
|
t.Members[i].GuildID = t.GuildID
|
||||||
|
err = s.MemberAdd(t.Members[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
case *GuildRoleCreate:
|
case *GuildRoleCreate:
|
||||||
if s.TrackRoles {
|
if s.TrackRoles {
|
||||||
err = s.RoleAdd(t.GuildID, t.Role)
|
err = s.RoleAdd(t.GuildID, t.Role)
|
||||||
|
98
vendor/github.com/bwmarrin/discordgo/structs.go
generated
vendored
98
vendor/github.com/bwmarrin/discordgo/structs.go
generated
vendored
@ -14,7 +14,6 @@ package discordgo
|
|||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strconv"
|
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -85,6 +84,9 @@ type Session struct {
|
|||||||
// Stores the last HeartbeatAck that was recieved (in UTC)
|
// Stores the last HeartbeatAck that was recieved (in UTC)
|
||||||
LastHeartbeatAck time.Time
|
LastHeartbeatAck time.Time
|
||||||
|
|
||||||
|
// used to deal with rate limits
|
||||||
|
Ratelimiter *RateLimiter
|
||||||
|
|
||||||
// Event handlers
|
// Event handlers
|
||||||
handlersMu sync.RWMutex
|
handlersMu sync.RWMutex
|
||||||
handlers map[string][]*eventHandlerInstance
|
handlers map[string][]*eventHandlerInstance
|
||||||
@ -96,9 +98,6 @@ type Session struct {
|
|||||||
// When nil, the session is not listening.
|
// When nil, the session is not listening.
|
||||||
listening chan interface{}
|
listening chan interface{}
|
||||||
|
|
||||||
// used to deal with rate limits
|
|
||||||
ratelimiter *RateLimiter
|
|
||||||
|
|
||||||
// sequence tracks the current gateway api websocket sequence number
|
// sequence tracks the current gateway api websocket sequence number
|
||||||
sequence *int64
|
sequence *int64
|
||||||
|
|
||||||
@ -143,9 +142,9 @@ type Invite struct {
|
|||||||
MaxAge int `json:"max_age"`
|
MaxAge int `json:"max_age"`
|
||||||
Uses int `json:"uses"`
|
Uses int `json:"uses"`
|
||||||
MaxUses int `json:"max_uses"`
|
MaxUses int `json:"max_uses"`
|
||||||
XkcdPass string `json:"xkcdpass"`
|
|
||||||
Revoked bool `json:"revoked"`
|
Revoked bool `json:"revoked"`
|
||||||
Temporary bool `json:"temporary"`
|
Temporary bool `json:"temporary"`
|
||||||
|
Unique bool `json:"unique"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// ChannelType is the type of a Channel
|
// ChannelType is the type of a Channel
|
||||||
@ -171,9 +170,22 @@ type Channel struct {
|
|||||||
NSFW bool `json:"nsfw"`
|
NSFW bool `json:"nsfw"`
|
||||||
Position int `json:"position"`
|
Position int `json:"position"`
|
||||||
Bitrate int `json:"bitrate"`
|
Bitrate int `json:"bitrate"`
|
||||||
Recipients []*User `json:"recipient"`
|
Recipients []*User `json:"recipients"`
|
||||||
Messages []*Message `json:"-"`
|
Messages []*Message `json:"-"`
|
||||||
PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"`
|
PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites"`
|
||||||
|
ParentID string `json:"parent_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// A ChannelEdit holds Channel Feild data for a channel edit.
|
||||||
|
type ChannelEdit struct {
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Topic string `json:"topic,omitempty"`
|
||||||
|
NSFW bool `json:"nsfw,omitempty"`
|
||||||
|
Position int `json:"position"`
|
||||||
|
Bitrate int `json:"bitrate,omitempty"`
|
||||||
|
UserLimit int `json:"user_limit,omitempty"`
|
||||||
|
PermissionOverwrites []*PermissionOverwrite `json:"permission_overwrites,omitempty"`
|
||||||
|
ParentID string `json:"parent_id,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// A PermissionOverwrite holds permission overwrite data for a Channel
|
// A PermissionOverwrite holds permission overwrite data for a Channel
|
||||||
@ -191,6 +203,7 @@ type Emoji struct {
|
|||||||
Roles []string `json:"roles"`
|
Roles []string `json:"roles"`
|
||||||
Managed bool `json:"managed"`
|
Managed bool `json:"managed"`
|
||||||
RequireColons bool `json:"require_colons"`
|
RequireColons bool `json:"require_colons"`
|
||||||
|
Animated bool `json:"animated"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// APIName returns an correctly formatted API name for use in the MessageReactions endpoints.
|
// APIName returns an correctly formatted API name for use in the MessageReactions endpoints.
|
||||||
@ -204,7 +217,7 @@ func (e *Emoji) APIName() string {
|
|||||||
return e.ID
|
return e.ID
|
||||||
}
|
}
|
||||||
|
|
||||||
// VerificationLevel type defination
|
// VerificationLevel type definition
|
||||||
type VerificationLevel int
|
type VerificationLevel int
|
||||||
|
|
||||||
// Constants for VerificationLevel levels from 0 to 3 inclusive
|
// Constants for VerificationLevel levels from 0 to 3 inclusive
|
||||||
@ -314,45 +327,58 @@ type Presence struct {
|
|||||||
Since *int `json:"since"`
|
Since *int `json:"since"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GameType is the type of "game" (see GameType* consts) in the Game struct
|
||||||
|
type GameType int
|
||||||
|
|
||||||
|
// Valid GameType values
|
||||||
|
const (
|
||||||
|
GameTypeGame GameType = iota
|
||||||
|
GameTypeStreaming
|
||||||
|
)
|
||||||
|
|
||||||
// A Game struct holds the name of the "playing .." game for a user
|
// A Game struct holds the name of the "playing .." game for a user
|
||||||
type Game struct {
|
type Game struct {
|
||||||
Name string `json:"name"`
|
Name string `json:"name"`
|
||||||
Type int `json:"type"`
|
Type GameType `json:"type"`
|
||||||
URL string `json:"url,omitempty"`
|
URL string `json:"url,omitempty"`
|
||||||
|
Details string `json:"details,omitempty"`
|
||||||
|
State string `json:"state,omitempty"`
|
||||||
|
TimeStamps TimeStamps `json:"timestamps,omitempty"`
|
||||||
|
Assets Assets `json:"assets,omitempty"`
|
||||||
|
ApplicationID string `json:"application_id,omitempty"`
|
||||||
|
Instance int8 `json:"instance,omitempty"`
|
||||||
|
// TODO: Party and Secrets (unknown structure)
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnmarshalJSON unmarshals json to Game struct
|
// A TimeStamps struct contains start and end times used in the rich presence "playing .." Game
|
||||||
func (g *Game) UnmarshalJSON(bytes []byte) error {
|
type TimeStamps struct {
|
||||||
temp := &struct {
|
EndTimestamp int64 `json:"end,omitempty"`
|
||||||
Name json.Number `json:"name"`
|
StartTimestamp int64 `json:"start,omitempty"`
|
||||||
Type json.RawMessage `json:"type"`
|
}
|
||||||
URL string `json:"url"`
|
|
||||||
|
// UnmarshalJSON unmarshals JSON into TimeStamps struct
|
||||||
|
func (t *TimeStamps) UnmarshalJSON(b []byte) error {
|
||||||
|
temp := struct {
|
||||||
|
End float64 `json:"end,omitempty"`
|
||||||
|
Start float64 `json:"start,omitempty"`
|
||||||
}{}
|
}{}
|
||||||
err := json.Unmarshal(bytes, temp)
|
err := json.Unmarshal(b, &temp)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
g.URL = temp.URL
|
t.EndTimestamp = int64(temp.End)
|
||||||
g.Name = temp.Name.String()
|
t.StartTimestamp = int64(temp.Start)
|
||||||
|
|
||||||
if temp.Type != nil {
|
|
||||||
err = json.Unmarshal(temp.Type, &g.Type)
|
|
||||||
if err == nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
s := ""
|
|
||||||
err = json.Unmarshal(temp.Type, &s)
|
|
||||||
if err == nil {
|
|
||||||
g.Type, err = strconv.Atoi(s)
|
|
||||||
}
|
|
||||||
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// An Assets struct contains assets and labels used in the rich presence "playing .." Game
|
||||||
|
type Assets struct {
|
||||||
|
LargeImageID string `json:"large_image,omitempty"`
|
||||||
|
SmallImageID string `json:"small_image,omitempty"`
|
||||||
|
LargeText string `json:"large_text,omitempty"`
|
||||||
|
SmallText string `json:"small_text,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
// A Member stores user information for Guild members.
|
// A Member stores user information for Guild members.
|
||||||
type Member struct {
|
type Member struct {
|
||||||
GuildID string `json:"guild_id"`
|
GuildID string `json:"guild_id"`
|
||||||
@ -383,7 +409,7 @@ type Settings struct {
|
|||||||
DeveloperMode bool `json:"developer_mode"`
|
DeveloperMode bool `json:"developer_mode"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// Status type defination
|
// Status type definition
|
||||||
type Status string
|
type Status string
|
||||||
|
|
||||||
// Constants for Status with the different current available status
|
// Constants for Status with the different current available status
|
||||||
|
9
vendor/github.com/bwmarrin/discordgo/user.go
generated
vendored
9
vendor/github.com/bwmarrin/discordgo/user.go
generated
vendored
@ -29,7 +29,9 @@ func (u *User) Mention() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// AvatarURL returns a URL to the user's avatar.
|
// AvatarURL returns a URL to the user's avatar.
|
||||||
// size: The size of the user's avatar as a power of two
|
// size: The size of the user's avatar as a power of two
|
||||||
|
// if size is an empty string, no size parameter will
|
||||||
|
// be added to the URL.
|
||||||
func (u *User) AvatarURL(size string) string {
|
func (u *User) AvatarURL(size string) string {
|
||||||
var URL string
|
var URL string
|
||||||
if strings.HasPrefix(u.Avatar, "a_") {
|
if strings.HasPrefix(u.Avatar, "a_") {
|
||||||
@ -38,5 +40,8 @@ func (u *User) AvatarURL(size string) string {
|
|||||||
URL = EndpointUserAvatar(u.ID, u.Avatar)
|
URL = EndpointUserAvatar(u.ID, u.Avatar)
|
||||||
}
|
}
|
||||||
|
|
||||||
return URL + "?size=" + size
|
if size != "" {
|
||||||
|
return URL + "?size=" + size
|
||||||
|
}
|
||||||
|
return URL
|
||||||
}
|
}
|
||||||
|
11
vendor/github.com/bwmarrin/discordgo/voice.go
generated
vendored
11
vendor/github.com/bwmarrin/discordgo/voice.go
generated
vendored
@ -13,7 +13,6 @@ import (
|
|||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
|
||||||
"net"
|
"net"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
@ -69,7 +68,7 @@ type VoiceConnection struct {
|
|||||||
voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler
|
voiceSpeakingUpdateHandlers []VoiceSpeakingUpdateHandler
|
||||||
}
|
}
|
||||||
|
|
||||||
// VoiceSpeakingUpdateHandler type provides a function defination for the
|
// VoiceSpeakingUpdateHandler type provides a function definition for the
|
||||||
// VoiceSpeakingUpdate event
|
// VoiceSpeakingUpdate event
|
||||||
type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate)
|
type VoiceSpeakingUpdateHandler func(vc *VoiceConnection, vs *VoiceSpeakingUpdate)
|
||||||
|
|
||||||
@ -104,7 +103,7 @@ func (v *VoiceConnection) Speaking(b bool) (err error) {
|
|||||||
defer v.Unlock()
|
defer v.Unlock()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
v.speaking = false
|
v.speaking = false
|
||||||
log.Println("Speaking() write json error:", err)
|
v.log(LogError, "Speaking() write json error:", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -181,7 +180,7 @@ func (v *VoiceConnection) Close() {
|
|||||||
v.log(LogInformational, "closing udp")
|
v.log(LogInformational, "closing udp")
|
||||||
err := v.udpConn.Close()
|
err := v.udpConn.Close()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("error closing udp connection: ", err)
|
v.log(LogError, "error closing udp connection: ", err)
|
||||||
}
|
}
|
||||||
v.udpConn = nil
|
v.udpConn = nil
|
||||||
}
|
}
|
||||||
@ -247,7 +246,7 @@ type voiceOP2 struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// WaitUntilConnected waits for the Voice Connection to
|
// WaitUntilConnected waits for the Voice Connection to
|
||||||
// become ready, if it does not become ready it retuns an err
|
// become ready, if it does not become ready it returns an err
|
||||||
func (v *VoiceConnection) waitUntilConnected() error {
|
func (v *VoiceConnection) waitUntilConnected() error {
|
||||||
|
|
||||||
v.log(LogInformational, "called")
|
v.log(LogInformational, "called")
|
||||||
@ -858,7 +857,7 @@ func (v *VoiceConnection) reconnect() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if v.session.DataReady == false || v.session.wsConn == nil {
|
if v.session.DataReady == false || v.session.wsConn == nil {
|
||||||
v.log(LogInformational, "cannot reconenct to channel %s with unready session", v.ChannelID)
|
v.log(LogInformational, "cannot reconnect to channel %s with unready session", v.ChannelID)
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
255
vendor/github.com/bwmarrin/discordgo/wsapi.go
generated
vendored
255
vendor/github.com/bwmarrin/discordgo/wsapi.go
generated
vendored
@ -15,6 +15,7 @@ import (
|
|||||||
"compress/zlib"
|
"compress/zlib"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"runtime"
|
"runtime"
|
||||||
@ -45,19 +46,114 @@ type resumePacket struct {
|
|||||||
} `json:"d"`
|
} `json:"d"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// Open opens a websocket connection to Discord.
|
// Open creates a websocket connection to Discord.
|
||||||
func (s *Session) Open() (err error) {
|
// See: https://discordapp.com/developers/docs/topics/gateway#connecting
|
||||||
|
func (s *Session) Open() error {
|
||||||
s.log(LogInformational, "called")
|
s.log(LogInformational, "called")
|
||||||
|
|
||||||
|
var err error
|
||||||
|
|
||||||
|
// Prevent Open or other major Session functions from
|
||||||
|
// being called while Open is still running.
|
||||||
s.Lock()
|
s.Lock()
|
||||||
defer func() {
|
defer s.Unlock()
|
||||||
|
|
||||||
|
// If the websock is already open, bail out here.
|
||||||
|
if s.wsConn != nil {
|
||||||
|
return ErrWSAlreadyOpen
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get the gateway to use for the Websocket connection
|
||||||
|
if s.gateway == "" {
|
||||||
|
s.gateway, err = s.Gateway()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.Unlock()
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add the version and encoding to the URL
|
||||||
|
s.gateway = s.gateway + "?v=" + APIVersion + "&encoding=json"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Connect to the Gateway
|
||||||
|
s.log(LogInformational, "connecting to gateway %s", s.gateway)
|
||||||
|
header := http.Header{}
|
||||||
|
header.Add("accept-encoding", "zlib")
|
||||||
|
s.wsConn, _, err = websocket.DefaultDialer.Dial(s.gateway, header)
|
||||||
|
if err != nil {
|
||||||
|
s.log(LogWarning, "error connecting to gateway %s, %s", s.gateway, err)
|
||||||
|
s.gateway = "" // clear cached gateway
|
||||||
|
s.wsConn = nil // Just to be safe.
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
// because of this, all code below must set err to the error
|
||||||
|
// when exiting with an error :) Maybe someone has a better
|
||||||
|
// way :)
|
||||||
|
if err != nil {
|
||||||
|
s.wsConn.Close()
|
||||||
|
s.wsConn = nil
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
|
|
||||||
|
// The first response from Discord should be an Op 10 (Hello) Packet.
|
||||||
|
// When processed by onEvent the heartbeat goroutine will be started.
|
||||||
|
mt, m, err := s.wsConn.ReadMessage()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
e, err := s.onEvent(mt, m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Operation != 10 {
|
||||||
|
err = fmt.Errorf("expecting Op 10, got Op %d instead", e.Operation)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s.log(LogInformational, "Op 10 Hello Packet received from Discord")
|
||||||
|
s.LastHeartbeatAck = time.Now().UTC()
|
||||||
|
var h helloOp
|
||||||
|
if err = json.Unmarshal(e.RawData, &h); err != nil {
|
||||||
|
err = fmt.Errorf("error unmarshalling helloOp, %s", err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Now we send either an Op 2 Identity if this is a brand new
|
||||||
|
// connection or Op 6 Resume if we are resuming an existing connection.
|
||||||
|
sequence := atomic.LoadInt64(s.sequence)
|
||||||
|
if s.sessionID == "" && sequence == 0 {
|
||||||
|
|
||||||
|
// Send Op 2 Identity Packet
|
||||||
|
err = s.identify()
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("error sending identify packet to gateway, %s, %s", s.gateway, err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// Send Op 6 Resume Packet
|
||||||
|
p := resumePacket{}
|
||||||
|
p.Op = 6
|
||||||
|
p.Data.Token = s.Token
|
||||||
|
p.Data.SessionID = s.sessionID
|
||||||
|
p.Data.Sequence = sequence
|
||||||
|
|
||||||
|
s.log(LogInformational, "sending resume packet to gateway")
|
||||||
|
s.wsMutex.Lock()
|
||||||
|
err = s.wsConn.WriteJSON(p)
|
||||||
|
s.wsMutex.Unlock()
|
||||||
|
if err != nil {
|
||||||
|
err = fmt.Errorf("error sending gateway resume packet, %s, %s", s.gateway, err)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
// A basic state is a hard requirement for Voice.
|
// A basic state is a hard requirement for Voice.
|
||||||
|
// We create it here so the below READY/RESUMED packet can populate
|
||||||
|
// the state :)
|
||||||
|
// XXX: Move to New() func?
|
||||||
if s.State == nil {
|
if s.State == nil {
|
||||||
state := NewState()
|
state := NewState()
|
||||||
state.TrackChannels = false
|
state.TrackChannels = false
|
||||||
@ -68,77 +164,42 @@ func (s *Session) Open() (err error) {
|
|||||||
s.State = state
|
s.State = state
|
||||||
}
|
}
|
||||||
|
|
||||||
if s.wsConn != nil {
|
// Now Discord should send us a READY or RESUMED packet.
|
||||||
err = ErrWSAlreadyOpen
|
mt, m, err = s.wsConn.ReadMessage()
|
||||||
return
|
if err != nil {
|
||||||
|
return err
|
||||||
}
|
}
|
||||||
|
e, err = s.onEvent(mt, m)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if e.Type != `READY` && e.Type != `RESUMED` {
|
||||||
|
// This is not fatal, but it does not follow their API documentation.
|
||||||
|
s.log(LogWarning, "Expected READY/RESUMED, instead got:\n%#v\n", e)
|
||||||
|
}
|
||||||
|
s.log(LogInformational, "First Packet:\n%#v\n", e)
|
||||||
|
|
||||||
|
s.log(LogInformational, "We are now connected to Discord, emitting connect event")
|
||||||
|
s.handleEvent(connectEventType, &Connect{})
|
||||||
|
|
||||||
|
// A VoiceConnections map is a hard requirement for Voice.
|
||||||
|
// XXX: can this be moved to when opening a voice connection?
|
||||||
if s.VoiceConnections == nil {
|
if s.VoiceConnections == nil {
|
||||||
s.log(LogInformational, "creating new VoiceConnections map")
|
s.log(LogInformational, "creating new VoiceConnections map")
|
||||||
s.VoiceConnections = make(map[string]*VoiceConnection)
|
s.VoiceConnections = make(map[string]*VoiceConnection)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get the gateway to use for the Websocket connection
|
// Create listening chan outside of listen, as it needs to happen inside the
|
||||||
if s.gateway == "" {
|
// mutex lock and needs to exist before calling heartbeat and listen
|
||||||
s.gateway, err = s.Gateway()
|
// go rountines.
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add the version and encoding to the URL
|
|
||||||
s.gateway = s.gateway + "?v=" + APIVersion + "&encoding=json"
|
|
||||||
}
|
|
||||||
|
|
||||||
header := http.Header{}
|
|
||||||
header.Add("accept-encoding", "zlib")
|
|
||||||
|
|
||||||
s.log(LogInformational, "connecting to gateway %s", s.gateway)
|
|
||||||
s.wsConn, _, err = websocket.DefaultDialer.Dial(s.gateway, header)
|
|
||||||
if err != nil {
|
|
||||||
s.log(LogWarning, "error connecting to gateway %s, %s", s.gateway, err)
|
|
||||||
s.gateway = "" // clear cached gateway
|
|
||||||
// TODO: should we add a retry block here?
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
sequence := atomic.LoadInt64(s.sequence)
|
|
||||||
if s.sessionID != "" && sequence > 0 {
|
|
||||||
|
|
||||||
p := resumePacket{}
|
|
||||||
p.Op = 6
|
|
||||||
p.Data.Token = s.Token
|
|
||||||
p.Data.SessionID = s.sessionID
|
|
||||||
p.Data.Sequence = sequence
|
|
||||||
|
|
||||||
s.log(LogInformational, "sending resume packet to gateway")
|
|
||||||
err = s.wsConn.WriteJSON(p)
|
|
||||||
if err != nil {
|
|
||||||
s.log(LogWarning, "error sending gateway resume packet, %s, %s", s.gateway, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
} else {
|
|
||||||
|
|
||||||
err = s.identify()
|
|
||||||
if err != nil {
|
|
||||||
s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create listening outside of listen, as it needs to happen inside the mutex
|
|
||||||
// lock.
|
|
||||||
s.listening = make(chan interface{})
|
s.listening = make(chan interface{})
|
||||||
|
|
||||||
|
// Start sending heartbeats and reading messages from Discord.
|
||||||
|
go s.heartbeat(s.wsConn, s.listening, h.HeartbeatInterval)
|
||||||
go s.listen(s.wsConn, s.listening)
|
go s.listen(s.wsConn, s.listening)
|
||||||
s.LastHeartbeatAck = time.Now().UTC()
|
|
||||||
|
|
||||||
s.Unlock()
|
|
||||||
|
|
||||||
s.log(LogInformational, "emit connect event")
|
|
||||||
s.handleEvent(connectEventType, &Connect{})
|
|
||||||
|
|
||||||
s.log(LogInformational, "exiting")
|
s.log(LogInformational, "exiting")
|
||||||
return
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// listen polls the websocket connection for events, it will stop when the
|
// listen polls the websocket connection for events, it will stop when the
|
||||||
@ -249,7 +310,8 @@ func (s *Session) heartbeat(wsConn *websocket.Conn, listening <-chan interface{}
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
type updateStatusData struct {
|
// UpdateStatusData ia provided to UpdateStatusComplex()
|
||||||
|
type UpdateStatusData struct {
|
||||||
IdleSince *int `json:"since"`
|
IdleSince *int `json:"since"`
|
||||||
Game *Game `json:"game"`
|
Game *Game `json:"game"`
|
||||||
AFK bool `json:"afk"`
|
AFK bool `json:"afk"`
|
||||||
@ -258,7 +320,7 @@ type updateStatusData struct {
|
|||||||
|
|
||||||
type updateStatusOp struct {
|
type updateStatusOp struct {
|
||||||
Op int `json:"op"`
|
Op int `json:"op"`
|
||||||
Data updateStatusData `json:"d"`
|
Data UpdateStatusData `json:"d"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateStreamingStatus is used to update the user's streaming status.
|
// UpdateStreamingStatus is used to update the user's streaming status.
|
||||||
@ -270,13 +332,7 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err
|
|||||||
|
|
||||||
s.log(LogInformational, "called")
|
s.log(LogInformational, "called")
|
||||||
|
|
||||||
s.RLock()
|
usd := UpdateStatusData{
|
||||||
defer s.RUnlock()
|
|
||||||
if s.wsConn == nil {
|
|
||||||
return ErrWSNotFound
|
|
||||||
}
|
|
||||||
|
|
||||||
usd := updateStatusData{
|
|
||||||
Status: "online",
|
Status: "online",
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -285,9 +341,9 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err
|
|||||||
}
|
}
|
||||||
|
|
||||||
if game != "" {
|
if game != "" {
|
||||||
gameType := 0
|
gameType := GameTypeGame
|
||||||
if url != "" {
|
if url != "" {
|
||||||
gameType = 1
|
gameType = GameTypeStreaming
|
||||||
}
|
}
|
||||||
usd.Game = &Game{
|
usd.Game = &Game{
|
||||||
Name: game,
|
Name: game,
|
||||||
@ -296,6 +352,18 @@ func (s *Session) UpdateStreamingStatus(idle int, game string, url string) (err
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return s.UpdateStatusComplex(usd)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateStatusComplex allows for sending the raw status update data untouched by discordgo.
|
||||||
|
func (s *Session) UpdateStatusComplex(usd UpdateStatusData) (err error) {
|
||||||
|
|
||||||
|
s.RLock()
|
||||||
|
defer s.RUnlock()
|
||||||
|
if s.wsConn == nil {
|
||||||
|
return ErrWSNotFound
|
||||||
|
}
|
||||||
|
|
||||||
s.wsMutex.Lock()
|
s.wsMutex.Lock()
|
||||||
err = s.wsConn.WriteJSON(updateStatusOp{3, usd})
|
err = s.wsConn.WriteJSON(updateStatusOp{3, usd})
|
||||||
s.wsMutex.Unlock()
|
s.wsMutex.Unlock()
|
||||||
@ -357,9 +425,7 @@ func (s *Session) RequestGuildMembers(guildID, query string, limit int) (err err
|
|||||||
//
|
//
|
||||||
// If you use the AddHandler() function to register a handler for the
|
// If you use the AddHandler() function to register a handler for the
|
||||||
// "OnEvent" event then all events will be passed to that handler.
|
// "OnEvent" event then all events will be passed to that handler.
|
||||||
//
|
func (s *Session) onEvent(messageType int, message []byte) (*Event, error) {
|
||||||
// TODO: You may also register a custom event handler entirely using...
|
|
||||||
func (s *Session) onEvent(messageType int, message []byte) {
|
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
var reader io.Reader
|
var reader io.Reader
|
||||||
@ -371,7 +437,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
z, err2 := zlib.NewReader(reader)
|
z, err2 := zlib.NewReader(reader)
|
||||||
if err2 != nil {
|
if err2 != nil {
|
||||||
s.log(LogError, "error uncompressing websocket message, %s", err)
|
s.log(LogError, "error uncompressing websocket message, %s", err)
|
||||||
return
|
return nil, err2
|
||||||
}
|
}
|
||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
@ -389,7 +455,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
decoder := json.NewDecoder(reader)
|
decoder := json.NewDecoder(reader)
|
||||||
if err = decoder.Decode(&e); err != nil {
|
if err = decoder.Decode(&e); err != nil {
|
||||||
s.log(LogError, "error decoding websocket message, %s", err)
|
s.log(LogError, "error decoding websocket message, %s", err)
|
||||||
return
|
return e, err
|
||||||
}
|
}
|
||||||
|
|
||||||
s.log(LogDebug, "Op: %d, Seq: %d, Type: %s, Data: %s\n\n", e.Operation, e.Sequence, e.Type, string(e.RawData))
|
s.log(LogDebug, "Op: %d, Seq: %d, Type: %s, Data: %s\n\n", e.Operation, e.Sequence, e.Type, string(e.RawData))
|
||||||
@ -403,10 +469,10 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
s.wsMutex.Unlock()
|
s.wsMutex.Unlock()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.log(LogError, "error sending heartbeat in response to Op1")
|
s.log(LogError, "error sending heartbeat in response to Op1")
|
||||||
return
|
return e, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Reconnect
|
// Reconnect
|
||||||
@ -415,7 +481,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
s.log(LogInformational, "Closing and reconnecting in response to Op7")
|
s.log(LogInformational, "Closing and reconnecting in response to Op7")
|
||||||
s.Close()
|
s.Close()
|
||||||
s.reconnect()
|
s.reconnect()
|
||||||
return
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Invalid Session
|
// Invalid Session
|
||||||
@ -427,20 +493,15 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
err = s.identify()
|
err = s.identify()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err)
|
s.log(LogWarning, "error sending gateway identify packet, %s, %s", s.gateway, err)
|
||||||
return
|
return e, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if e.Operation == 10 {
|
if e.Operation == 10 {
|
||||||
var h helloOp
|
// Op10 is handled by Open()
|
||||||
if err = json.Unmarshal(e.RawData, &h); err != nil {
|
return e, nil
|
||||||
s.log(LogError, "error unmarshalling helloOp, %s", err)
|
|
||||||
} else {
|
|
||||||
go s.heartbeat(s.wsConn, s.listening, h.HeartbeatInterval)
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if e.Operation == 11 {
|
if e.Operation == 11 {
|
||||||
@ -448,7 +509,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
s.LastHeartbeatAck = time.Now().UTC()
|
s.LastHeartbeatAck = time.Now().UTC()
|
||||||
s.Unlock()
|
s.Unlock()
|
||||||
s.log(LogInformational, "got heartbeat ACK")
|
s.log(LogInformational, "got heartbeat ACK")
|
||||||
return
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Do not try to Dispatch a non-Dispatch Message
|
// Do not try to Dispatch a non-Dispatch Message
|
||||||
@ -456,7 +517,7 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
// But we probably should be doing something with them.
|
// But we probably should be doing something with them.
|
||||||
// TEMP
|
// TEMP
|
||||||
s.log(LogWarning, "unknown Op: %d, Seq: %d, Type: %s, Data: %s, message: %s", e.Operation, e.Sequence, e.Type, string(e.RawData), string(message))
|
s.log(LogWarning, "unknown Op: %d, Seq: %d, Type: %s, Data: %s, message: %s", e.Operation, e.Sequence, e.Type, string(e.RawData), string(message))
|
||||||
return
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Store the message sequence
|
// Store the message sequence
|
||||||
@ -485,6 +546,8 @@ func (s *Session) onEvent(messageType int, message []byte) {
|
|||||||
|
|
||||||
// For legacy reasons, we send the raw event also, this could be useful for handling unknown events.
|
// For legacy reasons, we send the raw event also, this could be useful for handling unknown events.
|
||||||
s.handleEvent(eventEventType, e)
|
s.handleEvent(eventEventType, e)
|
||||||
|
|
||||||
|
return e, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -610,7 +673,7 @@ func (s *Session) onVoiceServerUpdate(st *VoiceServerUpdate) {
|
|||||||
voice.GuildID = st.GuildID
|
voice.GuildID = st.GuildID
|
||||||
voice.Unlock()
|
voice.Unlock()
|
||||||
|
|
||||||
// Open a conenction to the voice server
|
// Open a connection to the voice server
|
||||||
err := voice.open()
|
err := voice.open()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.log(LogError, "onVoiceServerUpdate voice.open, %s", err)
|
s.log(LogError, "onVoiceServerUpdate voice.open, %s", err)
|
||||||
|
77
vendor/github.com/gorilla/websocket/proxy.go
generated
vendored
Normal file
77
vendor/github.com/gorilla/websocket/proxy.go
generated
vendored
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
// Copyright 2017 The Gorilla WebSocket 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 websocket
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"encoding/base64"
|
||||||
|
"errors"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type netDialerFunc func(netowrk, addr string) (net.Conn, error)
|
||||||
|
|
||||||
|
func (fn netDialerFunc) Dial(network, addr string) (net.Conn, error) {
|
||||||
|
return fn(network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
proxy_RegisterDialerType("http", func(proxyURL *url.URL, forwardDialer proxy_Dialer) (proxy_Dialer, error) {
|
||||||
|
return &httpProxyDialer{proxyURL: proxyURL, fowardDial: forwardDialer.Dial}, nil
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type httpProxyDialer struct {
|
||||||
|
proxyURL *url.URL
|
||||||
|
fowardDial func(network, addr string) (net.Conn, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (hpd *httpProxyDialer) Dial(network string, addr string) (net.Conn, error) {
|
||||||
|
hostPort, _ := hostPortNoPort(hpd.proxyURL)
|
||||||
|
conn, err := hpd.fowardDial(network, hostPort)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
connectHeader := make(http.Header)
|
||||||
|
if user := hpd.proxyURL.User; user != nil {
|
||||||
|
proxyUser := user.Username()
|
||||||
|
if proxyPassword, passwordSet := user.Password(); passwordSet {
|
||||||
|
credential := base64.StdEncoding.EncodeToString([]byte(proxyUser + ":" + proxyPassword))
|
||||||
|
connectHeader.Set("Proxy-Authorization", "Basic "+credential)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
connectReq := &http.Request{
|
||||||
|
Method: "CONNECT",
|
||||||
|
URL: &url.URL{Opaque: addr},
|
||||||
|
Host: addr,
|
||||||
|
Header: connectHeader,
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := connectReq.Write(conn); err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Read response. It's OK to use and discard buffered reader here becaue
|
||||||
|
// the remote server does not speak until spoken to.
|
||||||
|
br := bufio.NewReader(conn)
|
||||||
|
resp, err := http.ReadResponse(br, connectReq)
|
||||||
|
if err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if resp.StatusCode != 200 {
|
||||||
|
conn.Close()
|
||||||
|
f := strings.SplitN(resp.Status, " ", 2)
|
||||||
|
return nil, errors.New(f[1])
|
||||||
|
}
|
||||||
|
return conn, nil
|
||||||
|
}
|
473
vendor/github.com/gorilla/websocket/x_net_proxy.go
generated
vendored
Normal file
473
vendor/github.com/gorilla/websocket/x_net_proxy.go
generated
vendored
Normal file
@ -0,0 +1,473 @@
|
|||||||
|
// Code generated by golang.org/x/tools/cmd/bundle. DO NOT EDIT.
|
||||||
|
//go:generate bundle -o x_net_proxy.go golang.org/x/net/proxy
|
||||||
|
|
||||||
|
// Package proxy provides support for a variety of protocols to proxy network
|
||||||
|
// data.
|
||||||
|
//
|
||||||
|
|
||||||
|
package websocket
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
type proxy_direct struct{}
|
||||||
|
|
||||||
|
// Direct is a direct proxy: one that makes network connections directly.
|
||||||
|
var proxy_Direct = proxy_direct{}
|
||||||
|
|
||||||
|
func (proxy_direct) Dial(network, addr string) (net.Conn, error) {
|
||||||
|
return net.Dial(network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// A PerHost directs connections to a default Dialer unless the host name
|
||||||
|
// requested matches one of a number of exceptions.
|
||||||
|
type proxy_PerHost struct {
|
||||||
|
def, bypass proxy_Dialer
|
||||||
|
|
||||||
|
bypassNetworks []*net.IPNet
|
||||||
|
bypassIPs []net.IP
|
||||||
|
bypassZones []string
|
||||||
|
bypassHosts []string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewPerHost returns a PerHost Dialer that directs connections to either
|
||||||
|
// defaultDialer or bypass, depending on whether the connection matches one of
|
||||||
|
// the configured rules.
|
||||||
|
func proxy_NewPerHost(defaultDialer, bypass proxy_Dialer) *proxy_PerHost {
|
||||||
|
return &proxy_PerHost{
|
||||||
|
def: defaultDialer,
|
||||||
|
bypass: bypass,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dial connects to the address addr on the given network through either
|
||||||
|
// defaultDialer or bypass.
|
||||||
|
func (p *proxy_PerHost) Dial(network, addr string) (c net.Conn, err error) {
|
||||||
|
host, _, err := net.SplitHostPort(addr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return p.dialerForRequest(host).Dial(network, addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *proxy_PerHost) dialerForRequest(host string) proxy_Dialer {
|
||||||
|
if ip := net.ParseIP(host); ip != nil {
|
||||||
|
for _, net := range p.bypassNetworks {
|
||||||
|
if net.Contains(ip) {
|
||||||
|
return p.bypass
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, bypassIP := range p.bypassIPs {
|
||||||
|
if bypassIP.Equal(ip) {
|
||||||
|
return p.bypass
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return p.def
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, zone := range p.bypassZones {
|
||||||
|
if strings.HasSuffix(host, zone) {
|
||||||
|
return p.bypass
|
||||||
|
}
|
||||||
|
if host == zone[1:] {
|
||||||
|
// For a zone ".example.com", we match "example.com"
|
||||||
|
// too.
|
||||||
|
return p.bypass
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for _, bypassHost := range p.bypassHosts {
|
||||||
|
if bypassHost == host {
|
||||||
|
return p.bypass
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return p.def
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddFromString parses a string that contains comma-separated values
|
||||||
|
// specifying hosts that should use the bypass proxy. Each value is either an
|
||||||
|
// IP address, a CIDR range, a zone (*.example.com) or a host name
|
||||||
|
// (localhost). A best effort is made to parse the string and errors are
|
||||||
|
// ignored.
|
||||||
|
func (p *proxy_PerHost) AddFromString(s string) {
|
||||||
|
hosts := strings.Split(s, ",")
|
||||||
|
for _, host := range hosts {
|
||||||
|
host = strings.TrimSpace(host)
|
||||||
|
if len(host) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.Contains(host, "/") {
|
||||||
|
// We assume that it's a CIDR address like 127.0.0.0/8
|
||||||
|
if _, net, err := net.ParseCIDR(host); err == nil {
|
||||||
|
p.AddNetwork(net)
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if ip := net.ParseIP(host); ip != nil {
|
||||||
|
p.AddIP(ip)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if strings.HasPrefix(host, "*.") {
|
||||||
|
p.AddZone(host[1:])
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
p.AddHost(host)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddIP specifies an IP address that will use the bypass proxy. Note that
|
||||||
|
// this will only take effect if a literal IP address is dialed. A connection
|
||||||
|
// to a named host will never match an IP.
|
||||||
|
func (p *proxy_PerHost) AddIP(ip net.IP) {
|
||||||
|
p.bypassIPs = append(p.bypassIPs, ip)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddNetwork specifies an IP range that will use the bypass proxy. Note that
|
||||||
|
// this will only take effect if a literal IP address is dialed. A connection
|
||||||
|
// to a named host will never match.
|
||||||
|
func (p *proxy_PerHost) AddNetwork(net *net.IPNet) {
|
||||||
|
p.bypassNetworks = append(p.bypassNetworks, net)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddZone specifies a DNS suffix that will use the bypass proxy. A zone of
|
||||||
|
// "example.com" matches "example.com" and all of its subdomains.
|
||||||
|
func (p *proxy_PerHost) AddZone(zone string) {
|
||||||
|
if strings.HasSuffix(zone, ".") {
|
||||||
|
zone = zone[:len(zone)-1]
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(zone, ".") {
|
||||||
|
zone = "." + zone
|
||||||
|
}
|
||||||
|
p.bypassZones = append(p.bypassZones, zone)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddHost specifies a host name that will use the bypass proxy.
|
||||||
|
func (p *proxy_PerHost) AddHost(host string) {
|
||||||
|
if strings.HasSuffix(host, ".") {
|
||||||
|
host = host[:len(host)-1]
|
||||||
|
}
|
||||||
|
p.bypassHosts = append(p.bypassHosts, host)
|
||||||
|
}
|
||||||
|
|
||||||
|
// A Dialer is a means to establish a connection.
|
||||||
|
type proxy_Dialer interface {
|
||||||
|
// Dial connects to the given address via the proxy.
|
||||||
|
Dial(network, addr string) (c net.Conn, err error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Auth contains authentication parameters that specific Dialers may require.
|
||||||
|
type proxy_Auth struct {
|
||||||
|
User, Password string
|
||||||
|
}
|
||||||
|
|
||||||
|
// FromEnvironment returns the dialer specified by the proxy related variables in
|
||||||
|
// the environment.
|
||||||
|
func proxy_FromEnvironment() proxy_Dialer {
|
||||||
|
allProxy := proxy_allProxyEnv.Get()
|
||||||
|
if len(allProxy) == 0 {
|
||||||
|
return proxy_Direct
|
||||||
|
}
|
||||||
|
|
||||||
|
proxyURL, err := url.Parse(allProxy)
|
||||||
|
if err != nil {
|
||||||
|
return proxy_Direct
|
||||||
|
}
|
||||||
|
proxy, err := proxy_FromURL(proxyURL, proxy_Direct)
|
||||||
|
if err != nil {
|
||||||
|
return proxy_Direct
|
||||||
|
}
|
||||||
|
|
||||||
|
noProxy := proxy_noProxyEnv.Get()
|
||||||
|
if len(noProxy) == 0 {
|
||||||
|
return proxy
|
||||||
|
}
|
||||||
|
|
||||||
|
perHost := proxy_NewPerHost(proxy, proxy_Direct)
|
||||||
|
perHost.AddFromString(noProxy)
|
||||||
|
return perHost
|
||||||
|
}
|
||||||
|
|
||||||
|
// proxySchemes is a map from URL schemes to a function that creates a Dialer
|
||||||
|
// from a URL with such a scheme.
|
||||||
|
var proxy_proxySchemes map[string]func(*url.URL, proxy_Dialer) (proxy_Dialer, error)
|
||||||
|
|
||||||
|
// RegisterDialerType takes a URL scheme and a function to generate Dialers from
|
||||||
|
// a URL with that scheme and a forwarding Dialer. Registered schemes are used
|
||||||
|
// by FromURL.
|
||||||
|
func proxy_RegisterDialerType(scheme string, f func(*url.URL, proxy_Dialer) (proxy_Dialer, error)) {
|
||||||
|
if proxy_proxySchemes == nil {
|
||||||
|
proxy_proxySchemes = make(map[string]func(*url.URL, proxy_Dialer) (proxy_Dialer, error))
|
||||||
|
}
|
||||||
|
proxy_proxySchemes[scheme] = f
|
||||||
|
}
|
||||||
|
|
||||||
|
// FromURL returns a Dialer given a URL specification and an underlying
|
||||||
|
// Dialer for it to make network requests.
|
||||||
|
func proxy_FromURL(u *url.URL, forward proxy_Dialer) (proxy_Dialer, error) {
|
||||||
|
var auth *proxy_Auth
|
||||||
|
if u.User != nil {
|
||||||
|
auth = new(proxy_Auth)
|
||||||
|
auth.User = u.User.Username()
|
||||||
|
if p, ok := u.User.Password(); ok {
|
||||||
|
auth.Password = p
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch u.Scheme {
|
||||||
|
case "socks5":
|
||||||
|
return proxy_SOCKS5("tcp", u.Host, auth, forward)
|
||||||
|
}
|
||||||
|
|
||||||
|
// If the scheme doesn't match any of the built-in schemes, see if it
|
||||||
|
// was registered by another package.
|
||||||
|
if proxy_proxySchemes != nil {
|
||||||
|
if f, ok := proxy_proxySchemes[u.Scheme]; ok {
|
||||||
|
return f(u, forward)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errors.New("proxy: unknown scheme: " + u.Scheme)
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
proxy_allProxyEnv = &proxy_envOnce{
|
||||||
|
names: []string{"ALL_PROXY", "all_proxy"},
|
||||||
|
}
|
||||||
|
proxy_noProxyEnv = &proxy_envOnce{
|
||||||
|
names: []string{"NO_PROXY", "no_proxy"},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// envOnce looks up an environment variable (optionally by multiple
|
||||||
|
// names) once. It mitigates expensive lookups on some platforms
|
||||||
|
// (e.g. Windows).
|
||||||
|
// (Borrowed from net/http/transport.go)
|
||||||
|
type proxy_envOnce struct {
|
||||||
|
names []string
|
||||||
|
once sync.Once
|
||||||
|
val string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *proxy_envOnce) Get() string {
|
||||||
|
e.once.Do(e.init)
|
||||||
|
return e.val
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *proxy_envOnce) init() {
|
||||||
|
for _, n := range e.names {
|
||||||
|
e.val = os.Getenv(n)
|
||||||
|
if e.val != "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SOCKS5 returns a Dialer that makes SOCKSv5 connections to the given address
|
||||||
|
// with an optional username and password. See RFC 1928 and RFC 1929.
|
||||||
|
func proxy_SOCKS5(network, addr string, auth *proxy_Auth, forward proxy_Dialer) (proxy_Dialer, error) {
|
||||||
|
s := &proxy_socks5{
|
||||||
|
network: network,
|
||||||
|
addr: addr,
|
||||||
|
forward: forward,
|
||||||
|
}
|
||||||
|
if auth != nil {
|
||||||
|
s.user = auth.User
|
||||||
|
s.password = auth.Password
|
||||||
|
}
|
||||||
|
|
||||||
|
return s, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type proxy_socks5 struct {
|
||||||
|
user, password string
|
||||||
|
network, addr string
|
||||||
|
forward proxy_Dialer
|
||||||
|
}
|
||||||
|
|
||||||
|
const proxy_socks5Version = 5
|
||||||
|
|
||||||
|
const (
|
||||||
|
proxy_socks5AuthNone = 0
|
||||||
|
proxy_socks5AuthPassword = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
const proxy_socks5Connect = 1
|
||||||
|
|
||||||
|
const (
|
||||||
|
proxy_socks5IP4 = 1
|
||||||
|
proxy_socks5Domain = 3
|
||||||
|
proxy_socks5IP6 = 4
|
||||||
|
)
|
||||||
|
|
||||||
|
var proxy_socks5Errors = []string{
|
||||||
|
"",
|
||||||
|
"general failure",
|
||||||
|
"connection forbidden",
|
||||||
|
"network unreachable",
|
||||||
|
"host unreachable",
|
||||||
|
"connection refused",
|
||||||
|
"TTL expired",
|
||||||
|
"command not supported",
|
||||||
|
"address type not supported",
|
||||||
|
}
|
||||||
|
|
||||||
|
// Dial connects to the address addr on the given network via the SOCKS5 proxy.
|
||||||
|
func (s *proxy_socks5) Dial(network, addr string) (net.Conn, error) {
|
||||||
|
switch network {
|
||||||
|
case "tcp", "tcp6", "tcp4":
|
||||||
|
default:
|
||||||
|
return nil, errors.New("proxy: no support for SOCKS5 proxy connections of type " + network)
|
||||||
|
}
|
||||||
|
|
||||||
|
conn, err := s.forward.Dial(s.network, s.addr)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err := s.connect(conn, addr); err != nil {
|
||||||
|
conn.Close()
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return conn, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// connect takes an existing connection to a socks5 proxy server,
|
||||||
|
// and commands the server to extend that connection to target,
|
||||||
|
// which must be a canonical address with a host and port.
|
||||||
|
func (s *proxy_socks5) connect(conn net.Conn, target string) error {
|
||||||
|
host, portStr, err := net.SplitHostPort(target)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
port, err := strconv.Atoi(portStr)
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("proxy: failed to parse port number: " + portStr)
|
||||||
|
}
|
||||||
|
if port < 1 || port > 0xffff {
|
||||||
|
return errors.New("proxy: port number out of range: " + portStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// the size here is just an estimate
|
||||||
|
buf := make([]byte, 0, 6+len(host))
|
||||||
|
|
||||||
|
buf = append(buf, proxy_socks5Version)
|
||||||
|
if len(s.user) > 0 && len(s.user) < 256 && len(s.password) < 256 {
|
||||||
|
buf = append(buf, 2 /* num auth methods */, proxy_socks5AuthNone, proxy_socks5AuthPassword)
|
||||||
|
} else {
|
||||||
|
buf = append(buf, 1 /* num auth methods */, proxy_socks5AuthNone)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := conn.Write(buf); err != nil {
|
||||||
|
return errors.New("proxy: failed to write greeting to SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := io.ReadFull(conn, buf[:2]); err != nil {
|
||||||
|
return errors.New("proxy: failed to read greeting from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
if buf[0] != 5 {
|
||||||
|
return errors.New("proxy: SOCKS5 proxy at " + s.addr + " has unexpected version " + strconv.Itoa(int(buf[0])))
|
||||||
|
}
|
||||||
|
if buf[1] == 0xff {
|
||||||
|
return errors.New("proxy: SOCKS5 proxy at " + s.addr + " requires authentication")
|
||||||
|
}
|
||||||
|
|
||||||
|
// See RFC 1929
|
||||||
|
if buf[1] == proxy_socks5AuthPassword {
|
||||||
|
buf = buf[:0]
|
||||||
|
buf = append(buf, 1 /* password protocol version */)
|
||||||
|
buf = append(buf, uint8(len(s.user)))
|
||||||
|
buf = append(buf, s.user...)
|
||||||
|
buf = append(buf, uint8(len(s.password)))
|
||||||
|
buf = append(buf, s.password...)
|
||||||
|
|
||||||
|
if _, err := conn.Write(buf); err != nil {
|
||||||
|
return errors.New("proxy: failed to write authentication request to SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := io.ReadFull(conn, buf[:2]); err != nil {
|
||||||
|
return errors.New("proxy: failed to read authentication reply from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if buf[1] != 0 {
|
||||||
|
return errors.New("proxy: SOCKS5 proxy at " + s.addr + " rejected username/password")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
buf = buf[:0]
|
||||||
|
buf = append(buf, proxy_socks5Version, proxy_socks5Connect, 0 /* reserved */)
|
||||||
|
|
||||||
|
if ip := net.ParseIP(host); ip != nil {
|
||||||
|
if ip4 := ip.To4(); ip4 != nil {
|
||||||
|
buf = append(buf, proxy_socks5IP4)
|
||||||
|
ip = ip4
|
||||||
|
} else {
|
||||||
|
buf = append(buf, proxy_socks5IP6)
|
||||||
|
}
|
||||||
|
buf = append(buf, ip...)
|
||||||
|
} else {
|
||||||
|
if len(host) > 255 {
|
||||||
|
return errors.New("proxy: destination host name too long: " + host)
|
||||||
|
}
|
||||||
|
buf = append(buf, proxy_socks5Domain)
|
||||||
|
buf = append(buf, byte(len(host)))
|
||||||
|
buf = append(buf, host...)
|
||||||
|
}
|
||||||
|
buf = append(buf, byte(port>>8), byte(port))
|
||||||
|
|
||||||
|
if _, err := conn.Write(buf); err != nil {
|
||||||
|
return errors.New("proxy: failed to write connect request to SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := io.ReadFull(conn, buf[:4]); err != nil {
|
||||||
|
return errors.New("proxy: failed to read connect reply from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
failure := "unknown error"
|
||||||
|
if int(buf[1]) < len(proxy_socks5Errors) {
|
||||||
|
failure = proxy_socks5Errors[buf[1]]
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(failure) > 0 {
|
||||||
|
return errors.New("proxy: SOCKS5 proxy at " + s.addr + " failed to connect: " + failure)
|
||||||
|
}
|
||||||
|
|
||||||
|
bytesToDiscard := 0
|
||||||
|
switch buf[3] {
|
||||||
|
case proxy_socks5IP4:
|
||||||
|
bytesToDiscard = net.IPv4len
|
||||||
|
case proxy_socks5IP6:
|
||||||
|
bytesToDiscard = net.IPv6len
|
||||||
|
case proxy_socks5Domain:
|
||||||
|
_, err := io.ReadFull(conn, buf[:1])
|
||||||
|
if err != nil {
|
||||||
|
return errors.New("proxy: failed to read domain length from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
bytesToDiscard = int(buf[0])
|
||||||
|
default:
|
||||||
|
return errors.New("proxy: got unknown address type " + strconv.Itoa(int(buf[3])) + " from SOCKS5 proxy at " + s.addr)
|
||||||
|
}
|
||||||
|
|
||||||
|
if cap(buf) < bytesToDiscard {
|
||||||
|
buf = make([]byte, bytesToDiscard)
|
||||||
|
} else {
|
||||||
|
buf = buf[:bytesToDiscard]
|
||||||
|
}
|
||||||
|
if _, err := io.ReadFull(conn, buf); err != nil {
|
||||||
|
return errors.New("proxy: failed to read address from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also need to discard the port number
|
||||||
|
if _, err := io.ReadFull(conn, buf[:2]); err != nil {
|
||||||
|
return errors.New("proxy: failed to read port from SOCKS5 proxy at " + s.addr + ": " + err.Error())
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
18
vendor/github.com/labstack/echo/context.go
generated
vendored
18
vendor/github.com/labstack/echo/context.go
generated
vendored
@ -274,13 +274,6 @@ func (c *context) Param(name string) string {
|
|||||||
if n == name {
|
if n == name {
|
||||||
return c.pvalues[i]
|
return c.pvalues[i]
|
||||||
}
|
}
|
||||||
|
|
||||||
// Param name with aliases
|
|
||||||
for _, p := range strings.Split(n, ",") {
|
|
||||||
if p == name {
|
|
||||||
return c.pvalues[i]
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return ""
|
return ""
|
||||||
@ -494,14 +487,9 @@ func (c *context) Stream(code int, contentType string, r io.Reader) (err error)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *context) File(file string) (err error) {
|
func (c *context) File(file string) (err error) {
|
||||||
file, err = url.QueryUnescape(file) // Issue #839
|
|
||||||
if err != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
f, err := os.Open(file)
|
f, err := os.Open(file)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ErrNotFound
|
return NotFoundHandler(c)
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
|
|
||||||
@ -510,7 +498,7 @@ func (c *context) File(file string) (err error) {
|
|||||||
file = filepath.Join(file, indexPage)
|
file = filepath.Join(file, indexPage)
|
||||||
f, err = os.Open(file)
|
f, err = os.Open(file)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ErrNotFound
|
return NotFoundHandler(c)
|
||||||
}
|
}
|
||||||
defer f.Close()
|
defer f.Close()
|
||||||
if fi, err = f.Stat(); err != nil {
|
if fi, err = f.Stat(); err != nil {
|
||||||
@ -530,7 +518,7 @@ func (c *context) Inline(file, name string) (err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (c *context) contentDisposition(file, name, dispositionType string) (err error) {
|
func (c *context) contentDisposition(file, name, dispositionType string) (err error) {
|
||||||
c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%s", dispositionType, name))
|
c.response.Header().Set(HeaderContentDisposition, fmt.Sprintf("%s; filename=%q", dispositionType, name))
|
||||||
c.File(file)
|
c.File(file)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
145
vendor/github.com/labstack/echo/echo.go
generated
vendored
145
vendor/github.com/labstack/echo/echo.go
generated
vendored
@ -72,29 +72,31 @@ type (
|
|||||||
TLSServer *http.Server
|
TLSServer *http.Server
|
||||||
Listener net.Listener
|
Listener net.Listener
|
||||||
TLSListener net.Listener
|
TLSListener net.Listener
|
||||||
|
AutoTLSManager autocert.Manager
|
||||||
DisableHTTP2 bool
|
DisableHTTP2 bool
|
||||||
Debug bool
|
Debug bool
|
||||||
HideBanner bool
|
HideBanner bool
|
||||||
|
HidePort bool
|
||||||
HTTPErrorHandler HTTPErrorHandler
|
HTTPErrorHandler HTTPErrorHandler
|
||||||
Binder Binder
|
Binder Binder
|
||||||
Validator Validator
|
Validator Validator
|
||||||
Renderer Renderer
|
Renderer Renderer
|
||||||
AutoTLSManager autocert.Manager
|
|
||||||
// Mutex sync.RWMutex
|
// Mutex sync.RWMutex
|
||||||
Logger Logger
|
Logger Logger
|
||||||
}
|
}
|
||||||
|
|
||||||
// Route contains a handler and information for matching against requests.
|
// Route contains a handler and information for matching against requests.
|
||||||
Route struct {
|
Route struct {
|
||||||
Method string `json:"method"`
|
Method string `json:"method"`
|
||||||
Path string `json:"path"`
|
Path string `json:"path"`
|
||||||
Handler string `json:"handler"`
|
Name string `json:"name"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// HTTPError represents an error that occurred while handling a request.
|
// HTTPError represents an error that occurred while handling a request.
|
||||||
HTTPError struct {
|
HTTPError struct {
|
||||||
Code int
|
Code int
|
||||||
Message interface{}
|
Message interface{}
|
||||||
|
Inner error // Stores the error returned by an external dependency
|
||||||
}
|
}
|
||||||
|
|
||||||
// MiddlewareFunc defines a function to process middleware.
|
// MiddlewareFunc defines a function to process middleware.
|
||||||
@ -121,7 +123,7 @@ type (
|
|||||||
|
|
||||||
// i is the interface for Echo and Group.
|
// i is the interface for Echo and Group.
|
||||||
i interface {
|
i interface {
|
||||||
GET(string, HandlerFunc, ...MiddlewareFunc)
|
GET(string, HandlerFunc, ...MiddlewareFunc) *Route
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -212,7 +214,7 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
version = "3.1.0"
|
version = "3.2.6"
|
||||||
website = "https://echo.labstack.com"
|
website = "https://echo.labstack.com"
|
||||||
// http://patorjk.com/software/taag/#p=display&f=Small%20Slant&t=Echo
|
// http://patorjk.com/software/taag/#p=display&f=Small%20Slant&t=Echo
|
||||||
banner = `
|
banner = `
|
||||||
@ -282,7 +284,7 @@ func New() (e *Echo) {
|
|||||||
e.TLSServer.Handler = e
|
e.TLSServer.Handler = e
|
||||||
e.HTTPErrorHandler = e.DefaultHTTPErrorHandler
|
e.HTTPErrorHandler = e.DefaultHTTPErrorHandler
|
||||||
e.Binder = &DefaultBinder{}
|
e.Binder = &DefaultBinder{}
|
||||||
e.Logger.SetLevel(log.OFF)
|
e.Logger.SetLevel(log.ERROR)
|
||||||
e.stdLogger = stdLog.New(e.Logger.Output(), e.Logger.Prefix()+": ", 0)
|
e.stdLogger = stdLog.New(e.Logger.Output(), e.Logger.Prefix()+": ", 0)
|
||||||
e.pool.New = func() interface{} {
|
e.pool.New = func() interface{} {
|
||||||
return e.NewContext(nil, nil)
|
return e.NewContext(nil, nil)
|
||||||
@ -319,6 +321,9 @@ func (e *Echo) DefaultHTTPErrorHandler(err error, c Context) {
|
|||||||
if he, ok := err.(*HTTPError); ok {
|
if he, ok := err.(*HTTPError); ok {
|
||||||
code = he.Code
|
code = he.Code
|
||||||
msg = he.Message
|
msg = he.Message
|
||||||
|
if he.Inner != nil {
|
||||||
|
msg = fmt.Sprintf("%v, %v", err, he.Inner)
|
||||||
|
}
|
||||||
} else if e.Debug {
|
} else if e.Debug {
|
||||||
msg = err.Error()
|
msg = err.Error()
|
||||||
} else {
|
} else {
|
||||||
@ -328,19 +333,19 @@ func (e *Echo) DefaultHTTPErrorHandler(err error, c Context) {
|
|||||||
msg = Map{"message": msg}
|
msg = Map{"message": msg}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
e.Logger.Error(err)
|
||||||
|
|
||||||
|
// Send response
|
||||||
if !c.Response().Committed {
|
if !c.Response().Committed {
|
||||||
if c.Request().Method == HEAD { // Issue #608
|
if c.Request().Method == HEAD { // Issue #608
|
||||||
if err := c.NoContent(code); err != nil {
|
err = c.NoContent(code)
|
||||||
goto ERROR
|
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
if err := c.JSON(code, msg); err != nil {
|
err = c.JSON(code, msg)
|
||||||
goto ERROR
|
}
|
||||||
}
|
if err != nil {
|
||||||
|
e.Logger.Error(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
ERROR:
|
|
||||||
e.Logger.Error(err)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Pre adds middleware to the chain which is run before router.
|
// Pre adds middleware to the chain which is run before router.
|
||||||
@ -355,104 +360,114 @@ func (e *Echo) Use(middleware ...MiddlewareFunc) {
|
|||||||
|
|
||||||
// CONNECT registers a new CONNECT route for a path with matching handler in the
|
// CONNECT registers a new CONNECT route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(CONNECT, path, h, m...)
|
return e.Add(CONNECT, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// DELETE registers a new DELETE route for a path with matching handler in the router
|
// DELETE registers a new DELETE route for a path with matching handler in the router
|
||||||
// with optional route-level middleware.
|
// with optional route-level middleware.
|
||||||
func (e *Echo) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(DELETE, path, h, m...)
|
return e.Add(DELETE, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GET registers a new GET route for a path with matching handler in the router
|
// GET registers a new GET route for a path with matching handler in the router
|
||||||
// with optional route-level middleware.
|
// with optional route-level middleware.
|
||||||
func (e *Echo) GET(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) GET(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(GET, path, h, m...)
|
return e.Add(GET, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// HEAD registers a new HEAD route for a path with matching handler in the
|
// HEAD registers a new HEAD route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(HEAD, path, h, m...)
|
return e.Add(HEAD, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// OPTIONS registers a new OPTIONS route for a path with matching handler in the
|
// OPTIONS registers a new OPTIONS route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(OPTIONS, path, h, m...)
|
return e.Add(OPTIONS, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PATCH registers a new PATCH route for a path with matching handler in the
|
// PATCH registers a new PATCH route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(PATCH, path, h, m...)
|
return e.Add(PATCH, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// POST registers a new POST route for a path with matching handler in the
|
// POST registers a new POST route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) POST(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) POST(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(POST, path, h, m...)
|
return e.Add(POST, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PUT registers a new PUT route for a path with matching handler in the
|
// PUT registers a new PUT route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(PUT, path, h, m...)
|
return e.Add(PUT, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TRACE registers a new TRACE route for a path with matching handler in the
|
// TRACE registers a new TRACE route for a path with matching handler in the
|
||||||
// router with optional route-level middleware.
|
// router with optional route-level middleware.
|
||||||
func (e *Echo) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (e *Echo) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
e.add(TRACE, path, h, m...)
|
return e.Add(TRACE, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Any registers a new route for all HTTP methods and path with matching handler
|
// Any registers a new route for all HTTP methods and path with matching handler
|
||||||
// in the router with optional route-level middleware.
|
// in the router with optional route-level middleware.
|
||||||
func (e *Echo) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
func (e *Echo) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||||||
for _, m := range methods {
|
routes := make([]*Route, len(methods))
|
||||||
e.add(m, path, handler, middleware...)
|
for i, m := range methods {
|
||||||
|
routes[i] = e.Add(m, path, handler, middleware...)
|
||||||
}
|
}
|
||||||
|
return routes
|
||||||
}
|
}
|
||||||
|
|
||||||
// Match registers a new route for multiple HTTP methods and path with matching
|
// Match registers a new route for multiple HTTP methods and path with matching
|
||||||
// handler in the router with optional route-level middleware.
|
// handler in the router with optional route-level middleware.
|
||||||
func (e *Echo) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
func (e *Echo) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||||||
for _, m := range methods {
|
routes := make([]*Route, len(methods))
|
||||||
e.add(m, path, handler, middleware...)
|
for i, m := range methods {
|
||||||
|
routes[i] = e.Add(m, path, handler, middleware...)
|
||||||
}
|
}
|
||||||
|
return routes
|
||||||
}
|
}
|
||||||
|
|
||||||
// Static registers a new route with path prefix to serve static files from the
|
// Static registers a new route with path prefix to serve static files from the
|
||||||
// provided root directory.
|
// provided root directory.
|
||||||
func (e *Echo) Static(prefix, root string) {
|
func (e *Echo) Static(prefix, root string) *Route {
|
||||||
if root == "" {
|
if root == "" {
|
||||||
root = "." // For security we want to restrict to CWD.
|
root = "." // For security we want to restrict to CWD.
|
||||||
}
|
}
|
||||||
static(e, prefix, root)
|
return static(e, prefix, root)
|
||||||
}
|
}
|
||||||
|
|
||||||
func static(i i, prefix, root string) {
|
func static(i i, prefix, root string) *Route {
|
||||||
h := func(c Context) error {
|
h := func(c Context) error {
|
||||||
name := filepath.Join(root, path.Clean("/"+c.Param("*"))) // "/"+ for security
|
p, err := PathUnescape(c.Param("*"))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
name := filepath.Join(root, path.Clean("/"+p)) // "/"+ for security
|
||||||
return c.File(name)
|
return c.File(name)
|
||||||
}
|
}
|
||||||
i.GET(prefix, h)
|
i.GET(prefix, h)
|
||||||
if prefix == "/" {
|
if prefix == "/" {
|
||||||
i.GET(prefix+"*", h)
|
return i.GET(prefix+"*", h)
|
||||||
} else {
|
|
||||||
i.GET(prefix+"/*", h)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return i.GET(prefix+"/*", h)
|
||||||
}
|
}
|
||||||
|
|
||||||
// File registers a new route with path to serve a static file.
|
// File registers a new route with path to serve a static file.
|
||||||
func (e *Echo) File(path, file string) {
|
func (e *Echo) File(path, file string) *Route {
|
||||||
e.GET(path, func(c Context) error {
|
return e.GET(path, func(c Context) error {
|
||||||
return c.File(file)
|
return c.File(file)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func (e *Echo) add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
// Add registers a new route for an HTTP method and path with matching handler
|
||||||
|
// in the router with optional route-level middleware.
|
||||||
|
func (e *Echo) Add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) *Route {
|
||||||
name := handlerName(handler)
|
name := handlerName(handler)
|
||||||
e.router.Add(method, path, func(c Context) error {
|
e.router.Add(method, path, func(c Context) error {
|
||||||
h := handler
|
h := handler
|
||||||
@ -463,11 +478,12 @@ func (e *Echo) add(method, path string, handler HandlerFunc, middleware ...Middl
|
|||||||
return h(c)
|
return h(c)
|
||||||
})
|
})
|
||||||
r := &Route{
|
r := &Route{
|
||||||
Method: method,
|
Method: method,
|
||||||
Path: path,
|
Path: path,
|
||||||
Handler: name,
|
Name: name,
|
||||||
}
|
}
|
||||||
e.router.routes[method+path] = r
|
e.router.routes[method+path] = r
|
||||||
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
// Group creates a new router group with prefix and optional group-level middleware.
|
// Group creates a new router group with prefix and optional group-level middleware.
|
||||||
@ -479,12 +495,22 @@ func (e *Echo) Group(prefix string, m ...MiddlewareFunc) (g *Group) {
|
|||||||
|
|
||||||
// URI generates a URI from handler.
|
// URI generates a URI from handler.
|
||||||
func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string {
|
func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string {
|
||||||
|
name := handlerName(handler)
|
||||||
|
return e.Reverse(name, params...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// URL is an alias for `URI` function.
|
||||||
|
func (e *Echo) URL(h HandlerFunc, params ...interface{}) string {
|
||||||
|
return e.URI(h, params...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reverse generates an URL from route name and provided parameters.
|
||||||
|
func (e *Echo) Reverse(name string, params ...interface{}) string {
|
||||||
uri := new(bytes.Buffer)
|
uri := new(bytes.Buffer)
|
||||||
ln := len(params)
|
ln := len(params)
|
||||||
n := 0
|
n := 0
|
||||||
name := handlerName(handler)
|
|
||||||
for _, r := range e.router.routes {
|
for _, r := range e.router.routes {
|
||||||
if r.Handler == name {
|
if r.Name == name {
|
||||||
for i, l := 0, len(r.Path); i < l; i++ {
|
for i, l := 0, len(r.Path); i < l; i++ {
|
||||||
if r.Path[i] == ':' && n < ln {
|
if r.Path[i] == ':' && n < ln {
|
||||||
for ; i < l && r.Path[i] != '/'; i++ {
|
for ; i < l && r.Path[i] != '/'; i++ {
|
||||||
@ -502,11 +528,6 @@ func (e *Echo) URI(handler HandlerFunc, params ...interface{}) string {
|
|||||||
return uri.String()
|
return uri.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// URL is an alias for `URI` function.
|
|
||||||
func (e *Echo) URL(h HandlerFunc, params ...interface{}) string {
|
|
||||||
return e.URI(h, params...)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Routes returns the registered routes.
|
// Routes returns the registered routes.
|
||||||
func (e *Echo) Routes() []*Route {
|
func (e *Echo) Routes() []*Route {
|
||||||
routes := []*Route{}
|
routes := []*Route{}
|
||||||
@ -624,7 +645,7 @@ func (e *Echo) StartServer(s *http.Server) (err error) {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !e.HideBanner {
|
if !e.HidePort {
|
||||||
e.colorer.Printf("⇨ http server started on %s\n", e.colorer.Green(e.Listener.Addr()))
|
e.colorer.Printf("⇨ http server started on %s\n", e.colorer.Green(e.Listener.Addr()))
|
||||||
}
|
}
|
||||||
return s.Serve(e.Listener)
|
return s.Serve(e.Listener)
|
||||||
@ -636,7 +657,7 @@ func (e *Echo) StartServer(s *http.Server) (err error) {
|
|||||||
}
|
}
|
||||||
e.TLSListener = tls.NewListener(l, s.TLSConfig)
|
e.TLSListener = tls.NewListener(l, s.TLSConfig)
|
||||||
}
|
}
|
||||||
if !e.HideBanner {
|
if !e.HidePort {
|
||||||
e.colorer.Printf("⇨ https server started on %s\n", e.colorer.Green(e.TLSListener.Addr()))
|
e.colorer.Printf("⇨ https server started on %s\n", e.colorer.Green(e.TLSListener.Addr()))
|
||||||
}
|
}
|
||||||
return s.Serve(e.TLSListener)
|
return s.Serve(e.TLSListener)
|
||||||
@ -653,7 +674,7 @@ func NewHTTPError(code int, message ...interface{}) *HTTPError {
|
|||||||
|
|
||||||
// Error makes it compatible with `error` interface.
|
// Error makes it compatible with `error` interface.
|
||||||
func (he *HTTPError) Error() string {
|
func (he *HTTPError) Error() string {
|
||||||
return fmt.Sprintf("code=%d, message=%s", he.Code, he.Message)
|
return fmt.Sprintf("code=%d, message=%v", he.Code, he.Message)
|
||||||
}
|
}
|
||||||
|
|
||||||
// WrapHandler wraps `http.Handler` into `echo.HandlerFunc`.
|
// WrapHandler wraps `http.Handler` into `echo.HandlerFunc`.
|
||||||
|
65
vendor/github.com/labstack/echo/group.go
generated
vendored
65
vendor/github.com/labstack/echo/group.go
generated
vendored
@ -20,68 +20,74 @@ func (g *Group) Use(middleware ...MiddlewareFunc) {
|
|||||||
g.middleware = append(g.middleware, middleware...)
|
g.middleware = append(g.middleware, middleware...)
|
||||||
// Allow all requests to reach the group as they might get dropped if router
|
// Allow all requests to reach the group as they might get dropped if router
|
||||||
// doesn't find a match, making none of the group middleware process.
|
// doesn't find a match, making none of the group middleware process.
|
||||||
g.echo.Any(path.Clean(g.prefix+"/*"), func(c Context) error {
|
for _, p := range []string{"", "/*"} {
|
||||||
return ErrNotFound
|
g.echo.Any(path.Clean(g.prefix+p), func(c Context) error {
|
||||||
}, g.middleware...)
|
return NotFoundHandler(c)
|
||||||
|
}, g.middleware...)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// CONNECT implements `Echo#CONNECT()` for sub-routes within the Group.
|
// CONNECT implements `Echo#CONNECT()` for sub-routes within the Group.
|
||||||
func (g *Group) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) CONNECT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(CONNECT, path, h, m...)
|
return g.Add(CONNECT, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// DELETE implements `Echo#DELETE()` for sub-routes within the Group.
|
// DELETE implements `Echo#DELETE()` for sub-routes within the Group.
|
||||||
func (g *Group) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) DELETE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(DELETE, path, h, m...)
|
return g.Add(DELETE, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// GET implements `Echo#GET()` for sub-routes within the Group.
|
// GET implements `Echo#GET()` for sub-routes within the Group.
|
||||||
func (g *Group) GET(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) GET(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(GET, path, h, m...)
|
return g.Add(GET, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// HEAD implements `Echo#HEAD()` for sub-routes within the Group.
|
// HEAD implements `Echo#HEAD()` for sub-routes within the Group.
|
||||||
func (g *Group) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) HEAD(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(HEAD, path, h, m...)
|
return g.Add(HEAD, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// OPTIONS implements `Echo#OPTIONS()` for sub-routes within the Group.
|
// OPTIONS implements `Echo#OPTIONS()` for sub-routes within the Group.
|
||||||
func (g *Group) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) OPTIONS(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(OPTIONS, path, h, m...)
|
return g.Add(OPTIONS, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PATCH implements `Echo#PATCH()` for sub-routes within the Group.
|
// PATCH implements `Echo#PATCH()` for sub-routes within the Group.
|
||||||
func (g *Group) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) PATCH(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(PATCH, path, h, m...)
|
return g.Add(PATCH, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// POST implements `Echo#POST()` for sub-routes within the Group.
|
// POST implements `Echo#POST()` for sub-routes within the Group.
|
||||||
func (g *Group) POST(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) POST(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(POST, path, h, m...)
|
return g.Add(POST, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// PUT implements `Echo#PUT()` for sub-routes within the Group.
|
// PUT implements `Echo#PUT()` for sub-routes within the Group.
|
||||||
func (g *Group) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) PUT(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(PUT, path, h, m...)
|
return g.Add(PUT, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TRACE implements `Echo#TRACE()` for sub-routes within the Group.
|
// TRACE implements `Echo#TRACE()` for sub-routes within the Group.
|
||||||
func (g *Group) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) {
|
func (g *Group) TRACE(path string, h HandlerFunc, m ...MiddlewareFunc) *Route {
|
||||||
g.add(TRACE, path, h, m...)
|
return g.Add(TRACE, path, h, m...)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Any implements `Echo#Any()` for sub-routes within the Group.
|
// Any implements `Echo#Any()` for sub-routes within the Group.
|
||||||
func (g *Group) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
func (g *Group) Any(path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||||||
for _, m := range methods {
|
routes := make([]*Route, len(methods))
|
||||||
g.add(m, path, handler, middleware...)
|
for i, m := range methods {
|
||||||
|
routes[i] = g.Add(m, path, handler, middleware...)
|
||||||
}
|
}
|
||||||
|
return routes
|
||||||
}
|
}
|
||||||
|
|
||||||
// Match implements `Echo#Match()` for sub-routes within the Group.
|
// Match implements `Echo#Match()` for sub-routes within the Group.
|
||||||
func (g *Group) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
func (g *Group) Match(methods []string, path string, handler HandlerFunc, middleware ...MiddlewareFunc) []*Route {
|
||||||
for _, m := range methods {
|
routes := make([]*Route, len(methods))
|
||||||
g.add(m, path, handler, middleware...)
|
for i, m := range methods {
|
||||||
|
routes[i] = g.Add(m, path, handler, middleware...)
|
||||||
}
|
}
|
||||||
|
return routes
|
||||||
}
|
}
|
||||||
|
|
||||||
// Group creates a new sub-group with prefix and optional sub-group-level middleware.
|
// Group creates a new sub-group with prefix and optional sub-group-level middleware.
|
||||||
@ -102,12 +108,13 @@ func (g *Group) File(path, file string) {
|
|||||||
g.echo.File(g.prefix+path, file)
|
g.echo.File(g.prefix+path, file)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (g *Group) add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) {
|
// Add implements `Echo#Add()` for sub-routes within the Group.
|
||||||
|
func (g *Group) Add(method, path string, handler HandlerFunc, middleware ...MiddlewareFunc) *Route {
|
||||||
// Combine into a new slice to avoid accidentally passing the same slice for
|
// Combine into a new slice to avoid accidentally passing the same slice for
|
||||||
// multiple routes, which would lead to later add() calls overwriting the
|
// multiple routes, which would lead to later add() calls overwriting the
|
||||||
// middleware from earlier calls.
|
// middleware from earlier calls.
|
||||||
m := []MiddlewareFunc{}
|
m := []MiddlewareFunc{}
|
||||||
m = append(m, g.middleware...)
|
m = append(m, g.middleware...)
|
||||||
m = append(m, middleware...)
|
m = append(m, middleware...)
|
||||||
g.echo.add(method, g.prefix+path, handler, m...)
|
return g.echo.Add(method, g.prefix+path, handler, m...)
|
||||||
}
|
}
|
||||||
|
8
vendor/github.com/labstack/echo/middleware/basic_auth.go
generated
vendored
8
vendor/github.com/labstack/echo/middleware/basic_auth.go
generated
vendored
@ -3,6 +3,7 @@ package middleware
|
|||||||
import (
|
import (
|
||||||
"encoding/base64"
|
"encoding/base64"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/labstack/echo"
|
"github.com/labstack/echo"
|
||||||
)
|
)
|
||||||
@ -27,7 +28,7 @@ type (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
basic = "Basic"
|
basic = "basic"
|
||||||
defaultRealm = "Restricted"
|
defaultRealm = "Restricted"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -54,7 +55,7 @@ func BasicAuth(fn BasicAuthValidator) echo.MiddlewareFunc {
|
|||||||
func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc {
|
func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
// Defaults
|
||||||
if config.Validator == nil {
|
if config.Validator == nil {
|
||||||
panic("basic-auth middleware requires a validator function")
|
panic("echo: basic-auth middleware requires a validator function")
|
||||||
}
|
}
|
||||||
if config.Skipper == nil {
|
if config.Skipper == nil {
|
||||||
config.Skipper = DefaultBasicAuthConfig.Skipper
|
config.Skipper = DefaultBasicAuthConfig.Skipper
|
||||||
@ -72,7 +73,7 @@ func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc {
|
|||||||
auth := c.Request().Header.Get(echo.HeaderAuthorization)
|
auth := c.Request().Header.Get(echo.HeaderAuthorization)
|
||||||
l := len(basic)
|
l := len(basic)
|
||||||
|
|
||||||
if len(auth) > l+1 && auth[:l] == basic {
|
if len(auth) > l+1 && strings.ToLower(auth[:l]) == basic {
|
||||||
b, err := base64.StdEncoding.DecodeString(auth[l+1:])
|
b, err := base64.StdEncoding.DecodeString(auth[l+1:])
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
@ -87,6 +88,7 @@ func BasicAuthWithConfig(config BasicAuthConfig) echo.MiddlewareFunc {
|
|||||||
} else if valid {
|
} else if valid {
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
112
vendor/github.com/labstack/echo/middleware/body_dump.go
generated
vendored
Normal file
112
vendor/github.com/labstack/echo/middleware/body_dump.go
generated
vendored
Normal file
@ -0,0 +1,112 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"io/ioutil"
|
||||||
|
"net"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"io"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// BodyDumpConfig defines the config for BodyDump middleware.
|
||||||
|
BodyDumpConfig struct {
|
||||||
|
// Skipper defines a function to skip middleware.
|
||||||
|
Skipper Skipper
|
||||||
|
|
||||||
|
// Handler receives request and response payload.
|
||||||
|
// Required.
|
||||||
|
Handler BodyDumpHandler
|
||||||
|
}
|
||||||
|
|
||||||
|
// BodyDumpHandler receives the request and response payload.
|
||||||
|
BodyDumpHandler func(echo.Context, []byte, []byte)
|
||||||
|
|
||||||
|
bodyDumpResponseWriter struct {
|
||||||
|
io.Writer
|
||||||
|
http.ResponseWriter
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// DefaultBodyDumpConfig is the default BodyDump middleware config.
|
||||||
|
DefaultBodyDumpConfig = BodyDumpConfig{
|
||||||
|
Skipper: DefaultSkipper,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// BodyDump returns a BodyDump middleware.
|
||||||
|
//
|
||||||
|
// BodyLimit middleware captures the request and response payload and calls the
|
||||||
|
// registered handler.
|
||||||
|
func BodyDump(handler BodyDumpHandler) echo.MiddlewareFunc {
|
||||||
|
c := DefaultBodyDumpConfig
|
||||||
|
c.Handler = handler
|
||||||
|
return BodyDumpWithConfig(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// BodyDumpWithConfig returns a BodyDump middleware with config.
|
||||||
|
// See: `BodyDump()`.
|
||||||
|
func BodyDumpWithConfig(config BodyDumpConfig) echo.MiddlewareFunc {
|
||||||
|
// Defaults
|
||||||
|
if config.Handler == nil {
|
||||||
|
panic("echo: body-dump middleware requires a handler function")
|
||||||
|
}
|
||||||
|
if config.Skipper == nil {
|
||||||
|
config.Skipper = DefaultBodyDumpConfig.Skipper
|
||||||
|
}
|
||||||
|
|
||||||
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) (err error) {
|
||||||
|
if config.Skipper(c) {
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Request
|
||||||
|
reqBody := []byte{}
|
||||||
|
if c.Request().Body != nil { // Read
|
||||||
|
reqBody, _ = ioutil.ReadAll(c.Request().Body)
|
||||||
|
}
|
||||||
|
c.Request().Body = ioutil.NopCloser(bytes.NewBuffer(reqBody)) // Reset
|
||||||
|
|
||||||
|
// Response
|
||||||
|
resBody := new(bytes.Buffer)
|
||||||
|
mw := io.MultiWriter(c.Response().Writer, resBody)
|
||||||
|
writer := &bodyDumpResponseWriter{Writer: mw, ResponseWriter: c.Response().Writer}
|
||||||
|
c.Response().Writer = writer
|
||||||
|
|
||||||
|
if err = next(c); err != nil {
|
||||||
|
c.Error(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Callback
|
||||||
|
config.Handler(c, reqBody, resBody.Bytes())
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *bodyDumpResponseWriter) WriteHeader(code int) {
|
||||||
|
w.ResponseWriter.WriteHeader(code)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *bodyDumpResponseWriter) Write(b []byte) (int, error) {
|
||||||
|
return w.Writer.Write(b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *bodyDumpResponseWriter) Flush() {
|
||||||
|
w.ResponseWriter.(http.Flusher).Flush()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *bodyDumpResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
||||||
|
return w.ResponseWriter.(http.Hijacker).Hijack()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *bodyDumpResponseWriter) CloseNotify() <-chan bool {
|
||||||
|
return w.ResponseWriter.(http.CloseNotifier).CloseNotify()
|
||||||
|
}
|
6
vendor/github.com/labstack/echo/middleware/body_limit.go
generated
vendored
6
vendor/github.com/labstack/echo/middleware/body_limit.go
generated
vendored
@ -17,7 +17,7 @@ type (
|
|||||||
|
|
||||||
// Maximum allowed size for a request body, it can be specified
|
// Maximum allowed size for a request body, it can be specified
|
||||||
// as `4x` or `4xB`, where x is one of the multiple from K, M, G, T or P.
|
// as `4x` or `4xB`, where x is one of the multiple from K, M, G, T or P.
|
||||||
Limit string `json:"limit"`
|
Limit string `yaml:"limit"`
|
||||||
limit int64
|
limit int64
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -30,7 +30,7 @@ type (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
// DefaultBodyLimitConfig is the default Gzip middleware config.
|
// DefaultBodyLimitConfig is the default BodyLimit middleware config.
|
||||||
DefaultBodyLimitConfig = BodyLimitConfig{
|
DefaultBodyLimitConfig = BodyLimitConfig{
|
||||||
Skipper: DefaultSkipper,
|
Skipper: DefaultSkipper,
|
||||||
}
|
}
|
||||||
@ -60,7 +60,7 @@ func BodyLimitWithConfig(config BodyLimitConfig) echo.MiddlewareFunc {
|
|||||||
|
|
||||||
limit, err := bytes.Parse(config.Limit)
|
limit, err := bytes.Parse(config.Limit)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(fmt.Errorf("invalid body-limit=%s", config.Limit))
|
panic(fmt.Errorf("echo: invalid body-limit=%s", config.Limit))
|
||||||
}
|
}
|
||||||
config.limit = limit
|
config.limit = limit
|
||||||
pool := limitedReaderPool(config)
|
pool := limitedReaderPool(config)
|
||||||
|
5
vendor/github.com/labstack/echo/middleware/compress.go
generated
vendored
5
vendor/github.com/labstack/echo/middleware/compress.go
generated
vendored
@ -20,7 +20,7 @@ type (
|
|||||||
|
|
||||||
// Gzip compression level.
|
// Gzip compression level.
|
||||||
// Optional. Default value -1.
|
// Optional. Default value -1.
|
||||||
Level int `json:"level"`
|
Level int `yaml:"level"`
|
||||||
}
|
}
|
||||||
|
|
||||||
gzipResponseWriter struct {
|
gzipResponseWriter struct {
|
||||||
@ -67,7 +67,7 @@ func GzipWithConfig(config GzipConfig) echo.MiddlewareFunc {
|
|||||||
res := c.Response()
|
res := c.Response()
|
||||||
res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding)
|
res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding)
|
||||||
if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) {
|
if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) {
|
||||||
res.Header().Add(echo.HeaderContentEncoding, gzipScheme) // Issue #806
|
res.Header().Set(echo.HeaderContentEncoding, gzipScheme) // Issue #806
|
||||||
rw := res.Writer
|
rw := res.Writer
|
||||||
w, err := gzip.NewWriterLevel(rw, config.Level)
|
w, err := gzip.NewWriterLevel(rw, config.Level)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -98,6 +98,7 @@ func (w *gzipResponseWriter) WriteHeader(code int) {
|
|||||||
if code == http.StatusNoContent { // Issue #489
|
if code == http.StatusNoContent { // Issue #489
|
||||||
w.ResponseWriter.Header().Del(echo.HeaderContentEncoding)
|
w.ResponseWriter.Header().Del(echo.HeaderContentEncoding)
|
||||||
}
|
}
|
||||||
|
w.Header().Del(echo.HeaderContentLength) // Issue #444
|
||||||
w.ResponseWriter.WriteHeader(code)
|
w.ResponseWriter.WriteHeader(code)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
12
vendor/github.com/labstack/echo/middleware/cors.go
generated
vendored
12
vendor/github.com/labstack/echo/middleware/cors.go
generated
vendored
@ -16,34 +16,34 @@ type (
|
|||||||
|
|
||||||
// AllowOrigin defines a list of origins that may access the resource.
|
// AllowOrigin defines a list of origins that may access the resource.
|
||||||
// Optional. Default value []string{"*"}.
|
// Optional. Default value []string{"*"}.
|
||||||
AllowOrigins []string `json:"allow_origins"`
|
AllowOrigins []string `yaml:"allow_origins"`
|
||||||
|
|
||||||
// AllowMethods defines a list methods allowed when accessing the resource.
|
// AllowMethods defines a list methods allowed when accessing the resource.
|
||||||
// This is used in response to a preflight request.
|
// This is used in response to a preflight request.
|
||||||
// Optional. Default value DefaultCORSConfig.AllowMethods.
|
// Optional. Default value DefaultCORSConfig.AllowMethods.
|
||||||
AllowMethods []string `json:"allow_methods"`
|
AllowMethods []string `yaml:"allow_methods"`
|
||||||
|
|
||||||
// AllowHeaders defines a list of request headers that can be used when
|
// AllowHeaders defines a list of request headers that can be used when
|
||||||
// making the actual request. This in response to a preflight request.
|
// making the actual request. This in response to a preflight request.
|
||||||
// Optional. Default value []string{}.
|
// Optional. Default value []string{}.
|
||||||
AllowHeaders []string `json:"allow_headers"`
|
AllowHeaders []string `yaml:"allow_headers"`
|
||||||
|
|
||||||
// AllowCredentials indicates whether or not the response to the request
|
// AllowCredentials indicates whether or not the response to the request
|
||||||
// can be exposed when the credentials flag is true. When used as part of
|
// can be exposed when the credentials flag is true. When used as part of
|
||||||
// a response to a preflight request, this indicates whether or not the
|
// a response to a preflight request, this indicates whether or not the
|
||||||
// actual request can be made using credentials.
|
// actual request can be made using credentials.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
AllowCredentials bool `json:"allow_credentials"`
|
AllowCredentials bool `yaml:"allow_credentials"`
|
||||||
|
|
||||||
// ExposeHeaders defines a whitelist headers that clients are allowed to
|
// ExposeHeaders defines a whitelist headers that clients are allowed to
|
||||||
// access.
|
// access.
|
||||||
// Optional. Default value []string{}.
|
// Optional. Default value []string{}.
|
||||||
ExposeHeaders []string `json:"expose_headers"`
|
ExposeHeaders []string `yaml:"expose_headers"`
|
||||||
|
|
||||||
// MaxAge indicates how long (in seconds) the results of a preflight request
|
// MaxAge indicates how long (in seconds) the results of a preflight request
|
||||||
// can be cached.
|
// can be cached.
|
||||||
// Optional. Default value 0.
|
// Optional. Default value 0.
|
||||||
MaxAge int `json:"max_age"`
|
MaxAge int `yaml:"max_age"`
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
18
vendor/github.com/labstack/echo/middleware/csrf.go
generated
vendored
18
vendor/github.com/labstack/echo/middleware/csrf.go
generated
vendored
@ -18,7 +18,7 @@ type (
|
|||||||
Skipper Skipper
|
Skipper Skipper
|
||||||
|
|
||||||
// TokenLength is the length of the generated token.
|
// TokenLength is the length of the generated token.
|
||||||
TokenLength uint8 `json:"token_length"`
|
TokenLength uint8 `yaml:"token_length"`
|
||||||
// Optional. Default value 32.
|
// Optional. Default value 32.
|
||||||
|
|
||||||
// TokenLookup is a string in the form of "<source>:<key>" that is used
|
// TokenLookup is a string in the form of "<source>:<key>" that is used
|
||||||
@ -28,35 +28,35 @@ type (
|
|||||||
// - "header:<name>"
|
// - "header:<name>"
|
||||||
// - "form:<name>"
|
// - "form:<name>"
|
||||||
// - "query:<name>"
|
// - "query:<name>"
|
||||||
TokenLookup string `json:"token_lookup"`
|
TokenLookup string `yaml:"token_lookup"`
|
||||||
|
|
||||||
// Context key to store generated CSRF token into context.
|
// Context key to store generated CSRF token into context.
|
||||||
// Optional. Default value "csrf".
|
// Optional. Default value "csrf".
|
||||||
ContextKey string `json:"context_key"`
|
ContextKey string `yaml:"context_key"`
|
||||||
|
|
||||||
// Name of the CSRF cookie. This cookie will store CSRF token.
|
// Name of the CSRF cookie. This cookie will store CSRF token.
|
||||||
// Optional. Default value "csrf".
|
// Optional. Default value "csrf".
|
||||||
CookieName string `json:"cookie_name"`
|
CookieName string `yaml:"cookie_name"`
|
||||||
|
|
||||||
// Domain of the CSRF cookie.
|
// Domain of the CSRF cookie.
|
||||||
// Optional. Default value none.
|
// Optional. Default value none.
|
||||||
CookieDomain string `json:"cookie_domain"`
|
CookieDomain string `yaml:"cookie_domain"`
|
||||||
|
|
||||||
// Path of the CSRF cookie.
|
// Path of the CSRF cookie.
|
||||||
// Optional. Default value none.
|
// Optional. Default value none.
|
||||||
CookiePath string `json:"cookie_path"`
|
CookiePath string `yaml:"cookie_path"`
|
||||||
|
|
||||||
// Max age (in seconds) of the CSRF cookie.
|
// Max age (in seconds) of the CSRF cookie.
|
||||||
// Optional. Default value 86400 (24hr).
|
// Optional. Default value 86400 (24hr).
|
||||||
CookieMaxAge int `json:"cookie_max_age"`
|
CookieMaxAge int `yaml:"cookie_max_age"`
|
||||||
|
|
||||||
// Indicates if CSRF cookie is secure.
|
// Indicates if CSRF cookie is secure.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
CookieSecure bool `json:"cookie_secure"`
|
CookieSecure bool `yaml:"cookie_secure"`
|
||||||
|
|
||||||
// Indicates if CSRF cookie is HTTP only.
|
// Indicates if CSRF cookie is HTTP only.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
CookieHTTPOnly bool `json:"cookie_http_only"`
|
CookieHTTPOnly bool `yaml:"cookie_http_only"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// csrfTokenExtractor defines a function that takes `echo.Context` and returns
|
// csrfTokenExtractor defines a function that takes `echo.Context` and returns
|
||||||
|
26
vendor/github.com/labstack/echo/middleware/jwt.go
generated
vendored
26
vendor/github.com/labstack/echo/middleware/jwt.go
generated
vendored
@ -1,7 +1,6 @@
|
|||||||
package middleware
|
package middleware
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"reflect"
|
"reflect"
|
||||||
@ -57,6 +56,12 @@ const (
|
|||||||
AlgorithmHS256 = "HS256"
|
AlgorithmHS256 = "HS256"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Errors
|
||||||
|
var (
|
||||||
|
ErrJWTMissing = echo.NewHTTPError(http.StatusBadRequest, "Missing or malformed jwt")
|
||||||
|
ErrJWTInvalid = echo.NewHTTPError(http.StatusUnauthorized, "Invalid or expired jwt")
|
||||||
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
// DefaultJWTConfig is the default JWT auth middleware config.
|
// DefaultJWTConfig is the default JWT auth middleware config.
|
||||||
DefaultJWTConfig = JWTConfig{
|
DefaultJWTConfig = JWTConfig{
|
||||||
@ -77,7 +82,7 @@ var (
|
|||||||
//
|
//
|
||||||
// See: https://jwt.io/introduction
|
// See: https://jwt.io/introduction
|
||||||
// See `JWTConfig.TokenLookup`
|
// See `JWTConfig.TokenLookup`
|
||||||
func JWT(key []byte) echo.MiddlewareFunc {
|
func JWT(key interface{}) echo.MiddlewareFunc {
|
||||||
c := DefaultJWTConfig
|
c := DefaultJWTConfig
|
||||||
c.SigningKey = key
|
c.SigningKey = key
|
||||||
return JWTWithConfig(c)
|
return JWTWithConfig(c)
|
||||||
@ -134,14 +139,15 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc {
|
|||||||
|
|
||||||
auth, err := extractor(c)
|
auth, err := extractor(c)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusBadRequest, err.Error())
|
return err
|
||||||
}
|
}
|
||||||
token := new(jwt.Token)
|
token := new(jwt.Token)
|
||||||
// Issue #647, #656
|
// Issue #647, #656
|
||||||
if _, ok := config.Claims.(jwt.MapClaims); ok {
|
if _, ok := config.Claims.(jwt.MapClaims); ok {
|
||||||
token, err = jwt.Parse(auth, config.keyFunc)
|
token, err = jwt.Parse(auth, config.keyFunc)
|
||||||
} else {
|
} else {
|
||||||
claims := reflect.ValueOf(config.Claims).Interface().(jwt.Claims)
|
t := reflect.ValueOf(config.Claims).Type().Elem()
|
||||||
|
claims := reflect.New(t).Interface().(jwt.Claims)
|
||||||
token, err = jwt.ParseWithClaims(auth, claims, config.keyFunc)
|
token, err = jwt.ParseWithClaims(auth, claims, config.keyFunc)
|
||||||
}
|
}
|
||||||
if err == nil && token.Valid {
|
if err == nil && token.Valid {
|
||||||
@ -149,7 +155,11 @@ func JWTWithConfig(config JWTConfig) echo.MiddlewareFunc {
|
|||||||
c.Set(config.ContextKey, token)
|
c.Set(config.ContextKey, token)
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
return echo.ErrUnauthorized
|
return &echo.HTTPError{
|
||||||
|
Code: ErrJWTInvalid.Code,
|
||||||
|
Message: ErrJWTInvalid.Message,
|
||||||
|
Inner: err,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -162,7 +172,7 @@ func jwtFromHeader(header string, authScheme string) jwtExtractor {
|
|||||||
if len(auth) > l+1 && auth[:l] == authScheme {
|
if len(auth) > l+1 && auth[:l] == authScheme {
|
||||||
return auth[l+1:], nil
|
return auth[l+1:], nil
|
||||||
}
|
}
|
||||||
return "", errors.New("Missing or invalid jwt in the request header")
|
return "", ErrJWTMissing
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -171,7 +181,7 @@ func jwtFromQuery(param string) jwtExtractor {
|
|||||||
return func(c echo.Context) (string, error) {
|
return func(c echo.Context) (string, error) {
|
||||||
token := c.QueryParam(param)
|
token := c.QueryParam(param)
|
||||||
if token == "" {
|
if token == "" {
|
||||||
return "", errors.New("Missing jwt in the query string")
|
return "", ErrJWTMissing
|
||||||
}
|
}
|
||||||
return token, nil
|
return token, nil
|
||||||
}
|
}
|
||||||
@ -182,7 +192,7 @@ func jwtFromCookie(name string) jwtExtractor {
|
|||||||
return func(c echo.Context) (string, error) {
|
return func(c echo.Context) (string, error) {
|
||||||
cookie, err := c.Cookie(name)
|
cookie, err := c.Cookie(name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", errors.New("Missing jwt in the cookie")
|
return "", ErrJWTMissing
|
||||||
}
|
}
|
||||||
return cookie.Value, nil
|
return cookie.Value, nil
|
||||||
}
|
}
|
||||||
|
18
vendor/github.com/labstack/echo/middleware/key_auth.go
generated
vendored
18
vendor/github.com/labstack/echo/middleware/key_auth.go
generated
vendored
@ -20,7 +20,8 @@ type (
|
|||||||
// Possible values:
|
// Possible values:
|
||||||
// - "header:<name>"
|
// - "header:<name>"
|
||||||
// - "query:<name>"
|
// - "query:<name>"
|
||||||
KeyLookup string `json:"key_lookup"`
|
// - "form:<name>"
|
||||||
|
KeyLookup string `yaml:"key_lookup"`
|
||||||
|
|
||||||
// AuthScheme to be used in the Authorization header.
|
// AuthScheme to be used in the Authorization header.
|
||||||
// Optional. Default value "Bearer".
|
// Optional. Default value "Bearer".
|
||||||
@ -72,7 +73,7 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc {
|
|||||||
config.KeyLookup = DefaultKeyAuthConfig.KeyLookup
|
config.KeyLookup = DefaultKeyAuthConfig.KeyLookup
|
||||||
}
|
}
|
||||||
if config.Validator == nil {
|
if config.Validator == nil {
|
||||||
panic("key-auth middleware requires a validator function")
|
panic("echo: key-auth middleware requires a validator function")
|
||||||
}
|
}
|
||||||
|
|
||||||
// Initialize
|
// Initialize
|
||||||
@ -81,6 +82,8 @@ func KeyAuthWithConfig(config KeyAuthConfig) echo.MiddlewareFunc {
|
|||||||
switch parts[0] {
|
switch parts[0] {
|
||||||
case "query":
|
case "query":
|
||||||
extractor = keyFromQuery(parts[1])
|
extractor = keyFromQuery(parts[1])
|
||||||
|
case "form":
|
||||||
|
extractor = keyFromForm(parts[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
@ -134,3 +137,14 @@ func keyFromQuery(param string) keyExtractor {
|
|||||||
return key, nil
|
return key, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// keyFromForm returns a `keyExtractor` that extracts key from the form.
|
||||||
|
func keyFromForm(param string) keyExtractor {
|
||||||
|
return func(c echo.Context) (string, error) {
|
||||||
|
key := c.FormValue(param)
|
||||||
|
if key == "" {
|
||||||
|
return "", errors.New("Missing key in the form")
|
||||||
|
}
|
||||||
|
return key, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
9
vendor/github.com/labstack/echo/middleware/logger.go
generated
vendored
9
vendor/github.com/labstack/echo/middleware/logger.go
generated
vendored
@ -26,6 +26,7 @@ type (
|
|||||||
// - time_unix_nano
|
// - time_unix_nano
|
||||||
// - time_rfc3339
|
// - time_rfc3339
|
||||||
// - time_rfc3339_nano
|
// - time_rfc3339_nano
|
||||||
|
// - time_custom
|
||||||
// - id (Request ID)
|
// - id (Request ID)
|
||||||
// - remote_ip
|
// - remote_ip
|
||||||
// - uri
|
// - uri
|
||||||
@ -46,7 +47,10 @@ type (
|
|||||||
// Example "${remote_ip} ${status}"
|
// Example "${remote_ip} ${status}"
|
||||||
//
|
//
|
||||||
// Optional. Default value DefaultLoggerConfig.Format.
|
// Optional. Default value DefaultLoggerConfig.Format.
|
||||||
Format string `json:"format"`
|
Format string `yaml:"format"`
|
||||||
|
|
||||||
|
// Optional. Default value DefaultLoggerConfig.CustomTimeFormat.
|
||||||
|
CustomTimeFormat string `yaml:"custom_time_format"`
|
||||||
|
|
||||||
// Output is a writer where logs in JSON format are written.
|
// Output is a writer where logs in JSON format are written.
|
||||||
// Optional. Default value os.Stdout.
|
// Optional. Default value os.Stdout.
|
||||||
@ -66,6 +70,7 @@ var (
|
|||||||
`"method":"${method}","uri":"${uri}","status":${status}, "latency":${latency},` +
|
`"method":"${method}","uri":"${uri}","status":${status}, "latency":${latency},` +
|
||||||
`"latency_human":"${latency_human}","bytes_in":${bytes_in},` +
|
`"latency_human":"${latency_human}","bytes_in":${bytes_in},` +
|
||||||
`"bytes_out":${bytes_out}}` + "\n",
|
`"bytes_out":${bytes_out}}` + "\n",
|
||||||
|
CustomTimeFormat:"2006-01-02 15:04:05.00000",
|
||||||
Output: os.Stdout,
|
Output: os.Stdout,
|
||||||
colorer: color.New(),
|
colorer: color.New(),
|
||||||
}
|
}
|
||||||
@ -126,6 +131,8 @@ func LoggerWithConfig(config LoggerConfig) echo.MiddlewareFunc {
|
|||||||
return buf.WriteString(time.Now().Format(time.RFC3339))
|
return buf.WriteString(time.Now().Format(time.RFC3339))
|
||||||
case "time_rfc3339_nano":
|
case "time_rfc3339_nano":
|
||||||
return buf.WriteString(time.Now().Format(time.RFC3339Nano))
|
return buf.WriteString(time.Now().Format(time.RFC3339Nano))
|
||||||
|
case "time_custom":
|
||||||
|
return buf.WriteString(time.Now().Format(config.CustomTimeFormat))
|
||||||
case "id":
|
case "id":
|
||||||
id := req.Header.Get(echo.HeaderXRequestID)
|
id := req.Header.Get(echo.HeaderXRequestID)
|
||||||
if id == "" {
|
if id == "" {
|
||||||
|
23
vendor/github.com/labstack/echo/middleware/middleware.go
generated
vendored
23
vendor/github.com/labstack/echo/middleware/middleware.go
generated
vendored
@ -1,6 +1,12 @@
|
|||||||
package middleware
|
package middleware
|
||||||
|
|
||||||
import "github.com/labstack/echo"
|
import (
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
// Skipper defines a function to skip middleware. Returning true skips processing
|
// Skipper defines a function to skip middleware. Returning true skips processing
|
||||||
@ -8,6 +14,21 @@ type (
|
|||||||
Skipper func(c echo.Context) bool
|
Skipper func(c echo.Context) bool
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func captureTokens(pattern *regexp.Regexp, input string) *strings.Replacer {
|
||||||
|
groups := pattern.FindAllStringSubmatch(input, -1)
|
||||||
|
if groups == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
values := groups[0][1:]
|
||||||
|
replace := make([]string, 2*len(values))
|
||||||
|
for i, v := range values {
|
||||||
|
j := 2 * i
|
||||||
|
replace[j] = "$" + strconv.Itoa(i+1)
|
||||||
|
replace[j+1] = v
|
||||||
|
}
|
||||||
|
return strings.NewReplacer(replace...)
|
||||||
|
}
|
||||||
|
|
||||||
// DefaultSkipper returns false which processes the middleware.
|
// DefaultSkipper returns false which processes the middleware.
|
||||||
func DefaultSkipper(echo.Context) bool {
|
func DefaultSkipper(echo.Context) bool {
|
||||||
return false
|
return false
|
||||||
|
168
vendor/github.com/labstack/echo/middleware/proxy.go
generated
vendored
168
vendor/github.com/labstack/echo/middleware/proxy.go
generated
vendored
@ -1,7 +1,6 @@
|
|||||||
package middleware
|
package middleware
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
@ -9,6 +8,9 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httputil"
|
"net/http/httputil"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
"sync/atomic"
|
"sync/atomic"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -25,33 +27,56 @@ type (
|
|||||||
|
|
||||||
// Balancer defines a load balancing technique.
|
// Balancer defines a load balancing technique.
|
||||||
// Required.
|
// Required.
|
||||||
// Possible values:
|
|
||||||
// - RandomBalancer
|
|
||||||
// - RoundRobinBalancer
|
|
||||||
Balancer ProxyBalancer
|
Balancer ProxyBalancer
|
||||||
|
|
||||||
|
// Rewrite defines URL path rewrite rules. The values captured in asterisk can be
|
||||||
|
// retrieved by index e.g. $1, $2 and so on.
|
||||||
|
// Examples:
|
||||||
|
// "/old": "/new",
|
||||||
|
// "/api/*": "/$1",
|
||||||
|
// "/js/*": "/public/javascripts/$1",
|
||||||
|
// "/users/*/orders/*": "/user/$1/order/$2",
|
||||||
|
Rewrite map[string]string
|
||||||
|
|
||||||
|
rewriteRegex map[*regexp.Regexp]string
|
||||||
}
|
}
|
||||||
|
|
||||||
// ProxyTarget defines the upstream target.
|
// ProxyTarget defines the upstream target.
|
||||||
ProxyTarget struct {
|
ProxyTarget struct {
|
||||||
URL *url.URL
|
Name string
|
||||||
}
|
URL *url.URL
|
||||||
|
|
||||||
// RandomBalancer implements a random load balancing technique.
|
|
||||||
RandomBalancer struct {
|
|
||||||
Targets []*ProxyTarget
|
|
||||||
random *rand.Rand
|
|
||||||
}
|
|
||||||
|
|
||||||
// RoundRobinBalancer implements a round-robin load balancing technique.
|
|
||||||
RoundRobinBalancer struct {
|
|
||||||
Targets []*ProxyTarget
|
|
||||||
i uint32
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ProxyBalancer defines an interface to implement a load balancing technique.
|
// ProxyBalancer defines an interface to implement a load balancing technique.
|
||||||
ProxyBalancer interface {
|
ProxyBalancer interface {
|
||||||
|
AddTarget(*ProxyTarget) bool
|
||||||
|
RemoveTarget(string) bool
|
||||||
Next() *ProxyTarget
|
Next() *ProxyTarget
|
||||||
}
|
}
|
||||||
|
|
||||||
|
commonBalancer struct {
|
||||||
|
targets []*ProxyTarget
|
||||||
|
mutex sync.RWMutex
|
||||||
|
}
|
||||||
|
|
||||||
|
// RandomBalancer implements a random load balancing technique.
|
||||||
|
randomBalancer struct {
|
||||||
|
*commonBalancer
|
||||||
|
random *rand.Rand
|
||||||
|
}
|
||||||
|
|
||||||
|
// RoundRobinBalancer implements a round-robin load balancing technique.
|
||||||
|
roundRobinBalancer struct {
|
||||||
|
*commonBalancer
|
||||||
|
i uint32
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// DefaultProxyConfig is the default Proxy middleware config.
|
||||||
|
DefaultProxyConfig = ProxyConfig{
|
||||||
|
Skipper: DefaultSkipper,
|
||||||
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
func proxyHTTP(t *ProxyTarget) http.Handler {
|
func proxyHTTP(t *ProxyTarget) http.Handler {
|
||||||
@ -60,29 +85,25 @@ func proxyHTTP(t *ProxyTarget) http.Handler {
|
|||||||
|
|
||||||
func proxyRaw(t *ProxyTarget, c echo.Context) http.Handler {
|
func proxyRaw(t *ProxyTarget, c echo.Context) http.Handler {
|
||||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
h, ok := w.(http.Hijacker)
|
in, _, err := c.Response().Hijack()
|
||||||
if !ok {
|
|
||||||
c.Error(errors.New("proxy raw, not a hijacker"))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
in, _, err := h.Hijack()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.Error(fmt.Errorf("proxy raw, hijack error=%v, url=%s", r.URL, err))
|
c.Error(fmt.Errorf("proxy raw, hijack error=%v, url=%s", t.URL, err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer in.Close()
|
defer in.Close()
|
||||||
|
|
||||||
out, err := net.Dial("tcp", t.URL.Host)
|
out, err := net.Dial("tcp", t.URL.Host)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
he := echo.NewHTTPError(http.StatusBadGateway, fmt.Sprintf("proxy raw, dial error=%v, url=%s", r.URL, err))
|
he := echo.NewHTTPError(http.StatusBadGateway, fmt.Sprintf("proxy raw, dial error=%v, url=%s", t.URL, err))
|
||||||
c.Error(he)
|
c.Error(he)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer out.Close()
|
defer out.Close()
|
||||||
|
|
||||||
|
// Write header
|
||||||
err = r.Write(out)
|
err = r.Write(out)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
he := echo.NewHTTPError(http.StatusBadGateway, fmt.Sprintf("proxy raw, request copy error=%v, url=%s", r.URL, err))
|
he := echo.NewHTTPError(http.StatusBadGateway, fmt.Sprintf("proxy raw, request header copy error=%v, url=%s", t.URL, err))
|
||||||
c.Error(he)
|
c.Error(he)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -97,29 +118,81 @@ func proxyRaw(t *ProxyTarget, c echo.Context) http.Handler {
|
|||||||
go cp(in, out)
|
go cp(in, out)
|
||||||
err = <-errc
|
err = <-errc
|
||||||
if err != nil && err != io.EOF {
|
if err != nil && err != io.EOF {
|
||||||
c.Logger().Errorf("proxy raw, error=%v, url=%s", r.URL, err)
|
c.Logger().Errorf("proxy raw, copy body error=%v, url=%s", t.URL, err)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Next randomly returns an upstream target.
|
// NewRandomBalancer returns a random proxy balancer.
|
||||||
func (r *RandomBalancer) Next() *ProxyTarget {
|
func NewRandomBalancer(targets []*ProxyTarget) ProxyBalancer {
|
||||||
if r.random == nil {
|
b := &randomBalancer{commonBalancer: new(commonBalancer)}
|
||||||
r.random = rand.New(rand.NewSource(int64(time.Now().Nanosecond())))
|
b.targets = targets
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewRoundRobinBalancer returns a round-robin proxy balancer.
|
||||||
|
func NewRoundRobinBalancer(targets []*ProxyTarget) ProxyBalancer {
|
||||||
|
b := &roundRobinBalancer{commonBalancer: new(commonBalancer)}
|
||||||
|
b.targets = targets
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddTarget adds an upstream target to the list.
|
||||||
|
func (b *commonBalancer) AddTarget(target *ProxyTarget) bool {
|
||||||
|
for _, t := range b.targets {
|
||||||
|
if t.Name == target.Name {
|
||||||
|
return false
|
||||||
|
}
|
||||||
}
|
}
|
||||||
return r.Targets[r.random.Intn(len(r.Targets))]
|
b.mutex.Lock()
|
||||||
|
defer b.mutex.Unlock()
|
||||||
|
b.targets = append(b.targets, target)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveTarget removes an upstream target from the list.
|
||||||
|
func (b *commonBalancer) RemoveTarget(name string) bool {
|
||||||
|
b.mutex.Lock()
|
||||||
|
defer b.mutex.Unlock()
|
||||||
|
for i, t := range b.targets {
|
||||||
|
if t.Name == name {
|
||||||
|
b.targets = append(b.targets[:i], b.targets[i+1:]...)
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// Next randomly returns an upstream target.
|
||||||
|
func (b *randomBalancer) Next() *ProxyTarget {
|
||||||
|
if b.random == nil {
|
||||||
|
b.random = rand.New(rand.NewSource(int64(time.Now().Nanosecond())))
|
||||||
|
}
|
||||||
|
b.mutex.RLock()
|
||||||
|
defer b.mutex.RUnlock()
|
||||||
|
return b.targets[b.random.Intn(len(b.targets))]
|
||||||
}
|
}
|
||||||
|
|
||||||
// Next returns an upstream target using round-robin technique.
|
// Next returns an upstream target using round-robin technique.
|
||||||
func (r *RoundRobinBalancer) Next() *ProxyTarget {
|
func (b *roundRobinBalancer) Next() *ProxyTarget {
|
||||||
r.i = r.i % uint32(len(r.Targets))
|
b.i = b.i % uint32(len(b.targets))
|
||||||
t := r.Targets[r.i]
|
t := b.targets[b.i]
|
||||||
atomic.AddUint32(&r.i, 1)
|
atomic.AddUint32(&b.i, 1)
|
||||||
return t
|
return t
|
||||||
}
|
}
|
||||||
|
|
||||||
// Proxy returns an HTTP/WebSocket reverse proxy middleware.
|
// Proxy returns a Proxy middleware.
|
||||||
func Proxy(config ProxyConfig) echo.MiddlewareFunc {
|
//
|
||||||
|
// Proxy middleware forwards the request to upstream server using a configured load balancing technique.
|
||||||
|
func Proxy(balancer ProxyBalancer) echo.MiddlewareFunc {
|
||||||
|
c := DefaultProxyConfig
|
||||||
|
c.Balancer = balancer
|
||||||
|
return ProxyWithConfig(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProxyWithConfig returns a Proxy middleware with config.
|
||||||
|
// See: `Proxy()`
|
||||||
|
func ProxyWithConfig(config ProxyConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
// Defaults
|
||||||
if config.Skipper == nil {
|
if config.Skipper == nil {
|
||||||
config.Skipper = DefaultLoggerConfig.Skipper
|
config.Skipper = DefaultLoggerConfig.Skipper
|
||||||
@ -127,13 +200,32 @@ func Proxy(config ProxyConfig) echo.MiddlewareFunc {
|
|||||||
if config.Balancer == nil {
|
if config.Balancer == nil {
|
||||||
panic("echo: proxy middleware requires balancer")
|
panic("echo: proxy middleware requires balancer")
|
||||||
}
|
}
|
||||||
|
config.rewriteRegex = map[*regexp.Regexp]string{}
|
||||||
|
|
||||||
|
// Initialize
|
||||||
|
for k, v := range config.Rewrite {
|
||||||
|
k = strings.Replace(k, "*", "(\\S*)", -1)
|
||||||
|
config.rewriteRegex[regexp.MustCompile(k)] = v
|
||||||
|
}
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
return func(c echo.Context) (err error) {
|
return func(c echo.Context) (err error) {
|
||||||
|
if config.Skipper(c) {
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
|
||||||
req := c.Request()
|
req := c.Request()
|
||||||
res := c.Response()
|
res := c.Response()
|
||||||
tgt := config.Balancer.Next()
|
tgt := config.Balancer.Next()
|
||||||
|
|
||||||
|
// Rewrite
|
||||||
|
for k, v := range config.rewriteRegex {
|
||||||
|
replacer := captureTokens(k, req.URL.Path)
|
||||||
|
if replacer != nil {
|
||||||
|
req.URL.Path = replacer.Replace(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Fix header
|
// Fix header
|
||||||
if req.Header.Get(echo.HeaderXRealIP) == "" {
|
if req.Header.Get(echo.HeaderXRealIP) == "" {
|
||||||
req.Header.Set(echo.HeaderXRealIP, c.RealIP())
|
req.Header.Set(echo.HeaderXRealIP, c.RealIP())
|
||||||
|
16
vendor/github.com/labstack/echo/middleware/recover.go
generated
vendored
16
vendor/github.com/labstack/echo/middleware/recover.go
generated
vendored
@ -5,7 +5,6 @@ import (
|
|||||||
"runtime"
|
"runtime"
|
||||||
|
|
||||||
"github.com/labstack/echo"
|
"github.com/labstack/echo"
|
||||||
"github.com/labstack/gommon/color"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
@ -16,16 +15,16 @@ type (
|
|||||||
|
|
||||||
// Size of the stack to be printed.
|
// Size of the stack to be printed.
|
||||||
// Optional. Default value 4KB.
|
// Optional. Default value 4KB.
|
||||||
StackSize int `json:"stack_size"`
|
StackSize int `yaml:"stack_size"`
|
||||||
|
|
||||||
// DisableStackAll disables formatting stack traces of all other goroutines
|
// DisableStackAll disables formatting stack traces of all other goroutines
|
||||||
// into buffer after the trace for the current goroutine.
|
// into buffer after the trace for the current goroutine.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
DisableStackAll bool `json:"disable_stack_all"`
|
DisableStackAll bool `yaml:"disable_stack_all"`
|
||||||
|
|
||||||
// DisablePrintStack disables printing stack trace.
|
// DisablePrintStack disables printing stack trace.
|
||||||
// Optional. Default value as false.
|
// Optional. Default value as false.
|
||||||
DisablePrintStack bool `json:"disable_print_stack"`
|
DisablePrintStack bool `yaml:"disable_print_stack"`
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -64,17 +63,14 @@ func RecoverWithConfig(config RecoverConfig) echo.MiddlewareFunc {
|
|||||||
|
|
||||||
defer func() {
|
defer func() {
|
||||||
if r := recover(); r != nil {
|
if r := recover(); r != nil {
|
||||||
var err error
|
err, ok := r.(error)
|
||||||
switch r := r.(type) {
|
if !ok {
|
||||||
case error:
|
|
||||||
err = r
|
|
||||||
default:
|
|
||||||
err = fmt.Errorf("%v", r)
|
err = fmt.Errorf("%v", r)
|
||||||
}
|
}
|
||||||
stack := make([]byte, config.StackSize)
|
stack := make([]byte, config.StackSize)
|
||||||
length := runtime.Stack(stack, !config.DisableStackAll)
|
length := runtime.Stack(stack, !config.DisableStackAll)
|
||||||
if !config.DisablePrintStack {
|
if !config.DisablePrintStack {
|
||||||
c.Logger().Printf("[%s] %s %s\n", color.Red("PANIC RECOVER"), err, stack[:length])
|
c.Logger().Printf("[PANIC RECOVER] %v %s\n", err, stack[:length])
|
||||||
}
|
}
|
||||||
c.Error(err)
|
c.Error(err)
|
||||||
}
|
}
|
||||||
|
170
vendor/github.com/labstack/echo/middleware/redirect.go
generated
vendored
170
vendor/github.com/labstack/echo/middleware/redirect.go
generated
vendored
@ -6,29 +6,28 @@ import (
|
|||||||
"github.com/labstack/echo"
|
"github.com/labstack/echo"
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
// RedirectConfig defines the config for Redirect middleware.
|
||||||
// RedirectConfig defines the config for Redirect middleware.
|
type RedirectConfig struct {
|
||||||
RedirectConfig struct {
|
// Skipper defines a function to skip middleware.
|
||||||
// Skipper defines a function to skip middleware.
|
Skipper
|
||||||
Skipper Skipper
|
|
||||||
|
|
||||||
// Status code to be used when redirecting the request.
|
// Status code to be used when redirecting the request.
|
||||||
// Optional. Default value http.StatusMovedPermanently.
|
// Optional. Default value http.StatusMovedPermanently.
|
||||||
Code int `json:"code"`
|
Code int `yaml:"code"`
|
||||||
}
|
}
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
// redirectLogic represents a function that given a scheme, host and uri
|
||||||
www = "www"
|
// can both: 1) determine if redirect is needed (will set ok accordingly) and
|
||||||
)
|
// 2) return the appropriate redirect url.
|
||||||
|
type redirectLogic func(scheme, host, uri string) (ok bool, url string)
|
||||||
|
|
||||||
var (
|
const www = "www"
|
||||||
// DefaultRedirectConfig is the default Redirect middleware config.
|
|
||||||
DefaultRedirectConfig = RedirectConfig{
|
// DefaultRedirectConfig is the default Redirect middleware config.
|
||||||
Skipper: DefaultSkipper,
|
var DefaultRedirectConfig = RedirectConfig{
|
||||||
Code: http.StatusMovedPermanently,
|
Skipper: DefaultSkipper,
|
||||||
}
|
Code: http.StatusMovedPermanently,
|
||||||
)
|
}
|
||||||
|
|
||||||
// HTTPSRedirect redirects http requests to https.
|
// HTTPSRedirect redirects http requests to https.
|
||||||
// For example, http://labstack.com will be redirect to https://labstack.com.
|
// For example, http://labstack.com will be redirect to https://labstack.com.
|
||||||
@ -41,29 +40,12 @@ func HTTPSRedirect() echo.MiddlewareFunc {
|
|||||||
// HTTPSRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
// HTTPSRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
||||||
// See `HTTPSRedirect()`.
|
// See `HTTPSRedirect()`.
|
||||||
func HTTPSRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
func HTTPSRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
return redirect(config, func(scheme, host, uri string) (ok bool, url string) {
|
||||||
if config.Skipper == nil {
|
if ok = scheme != "https"; ok {
|
||||||
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
url = "https://" + host + uri
|
||||||
}
|
|
||||||
if config.Code == 0 {
|
|
||||||
config.Code = DefaultRedirectConfig.Code
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
||||||
return func(c echo.Context) error {
|
|
||||||
if config.Skipper(c) {
|
|
||||||
return next(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
req := c.Request()
|
|
||||||
host := req.Host
|
|
||||||
uri := req.RequestURI
|
|
||||||
if !c.IsTLS() {
|
|
||||||
return c.Redirect(config.Code, "https://"+host+uri)
|
|
||||||
}
|
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
}
|
return
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// HTTPSWWWRedirect redirects http requests to https www.
|
// HTTPSWWWRedirect redirects http requests to https www.
|
||||||
@ -77,29 +59,12 @@ func HTTPSWWWRedirect() echo.MiddlewareFunc {
|
|||||||
// HTTPSWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
// HTTPSWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
||||||
// See `HTTPSWWWRedirect()`.
|
// See `HTTPSWWWRedirect()`.
|
||||||
func HTTPSWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
func HTTPSWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
return redirect(config, func(scheme, host, uri string) (ok bool, url string) {
|
||||||
if config.Skipper == nil {
|
if ok = scheme != "https" && host[:3] != www; ok {
|
||||||
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
url = "https://www." + host + uri
|
||||||
}
|
|
||||||
if config.Code == 0 {
|
|
||||||
config.Code = DefaultRedirectConfig.Code
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
||||||
return func(c echo.Context) error {
|
|
||||||
if config.Skipper(c) {
|
|
||||||
return next(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
req := c.Request()
|
|
||||||
host := req.Host
|
|
||||||
uri := req.RequestURI
|
|
||||||
if !c.IsTLS() && host[:3] != www {
|
|
||||||
return c.Redirect(config.Code, "https://www."+host+uri)
|
|
||||||
}
|
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
}
|
return
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// HTTPSNonWWWRedirect redirects http requests to https non www.
|
// HTTPSNonWWWRedirect redirects http requests to https non www.
|
||||||
@ -113,32 +78,15 @@ func HTTPSNonWWWRedirect() echo.MiddlewareFunc {
|
|||||||
// HTTPSNonWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
// HTTPSNonWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
||||||
// See `HTTPSNonWWWRedirect()`.
|
// See `HTTPSNonWWWRedirect()`.
|
||||||
func HTTPSNonWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
func HTTPSNonWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
return redirect(config, func(scheme, host, uri string) (ok bool, url string) {
|
||||||
if config.Skipper == nil {
|
if ok = scheme != "https"; ok {
|
||||||
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
if host[:3] == www {
|
||||||
}
|
host = host[4:]
|
||||||
if config.Code == 0 {
|
|
||||||
config.Code = DefaultRedirectConfig.Code
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
||||||
return func(c echo.Context) error {
|
|
||||||
if config.Skipper(c) {
|
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
|
url = "https://" + host + uri
|
||||||
req := c.Request()
|
|
||||||
host := req.Host
|
|
||||||
uri := req.RequestURI
|
|
||||||
if !c.IsTLS() {
|
|
||||||
if host[:3] == www {
|
|
||||||
return c.Redirect(config.Code, "https://"+host[4:]+uri)
|
|
||||||
}
|
|
||||||
return c.Redirect(config.Code, "https://"+host+uri)
|
|
||||||
}
|
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
}
|
return
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// WWWRedirect redirects non www requests to www.
|
// WWWRedirect redirects non www requests to www.
|
||||||
@ -152,30 +100,12 @@ func WWWRedirect() echo.MiddlewareFunc {
|
|||||||
// WWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
// WWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
||||||
// See `WWWRedirect()`.
|
// See `WWWRedirect()`.
|
||||||
func WWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
func WWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
||||||
// Defaults
|
return redirect(config, func(scheme, host, uri string) (ok bool, url string) {
|
||||||
if config.Skipper == nil {
|
if ok = host[:3] != www; ok {
|
||||||
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
url = scheme + "://www." + host + uri
|
||||||
}
|
|
||||||
if config.Code == 0 {
|
|
||||||
config.Code = DefaultRedirectConfig.Code
|
|
||||||
}
|
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
||||||
return func(c echo.Context) error {
|
|
||||||
if config.Skipper(c) {
|
|
||||||
return next(c)
|
|
||||||
}
|
|
||||||
|
|
||||||
req := c.Request()
|
|
||||||
scheme := c.Scheme()
|
|
||||||
host := req.Host
|
|
||||||
if host[:3] != www {
|
|
||||||
uri := req.RequestURI
|
|
||||||
return c.Redirect(config.Code, scheme+"://www."+host+uri)
|
|
||||||
}
|
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
}
|
return
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// NonWWWRedirect redirects www requests to non www.
|
// NonWWWRedirect redirects www requests to non www.
|
||||||
@ -189,6 +119,15 @@ func NonWWWRedirect() echo.MiddlewareFunc {
|
|||||||
// NonWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
// NonWWWRedirectWithConfig returns an HTTPSRedirect middleware with config.
|
||||||
// See `NonWWWRedirect()`.
|
// See `NonWWWRedirect()`.
|
||||||
func NonWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
func NonWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
||||||
|
return redirect(config, func(scheme, host, uri string) (ok bool, url string) {
|
||||||
|
if ok = host[:3] == www; ok {
|
||||||
|
url = scheme + "://" + host[4:] + uri
|
||||||
|
}
|
||||||
|
return
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func redirect(config RedirectConfig, cb redirectLogic) echo.MiddlewareFunc {
|
||||||
if config.Skipper == nil {
|
if config.Skipper == nil {
|
||||||
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
config.Skipper = DefaultTrailingSlashConfig.Skipper
|
||||||
}
|
}
|
||||||
@ -202,13 +141,12 @@ func NonWWWRedirectWithConfig(config RedirectConfig) echo.MiddlewareFunc {
|
|||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
|
|
||||||
req := c.Request()
|
req, scheme := c.Request(), c.Scheme()
|
||||||
scheme := c.Scheme()
|
|
||||||
host := req.Host
|
host := req.Host
|
||||||
if host[:3] == www {
|
if ok, url := cb(scheme, host, req.RequestURI); ok {
|
||||||
uri := req.RequestURI
|
return c.Redirect(config.Code, url)
|
||||||
return c.Redirect(config.Code, scheme+"://"+host[4:]+uri)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
83
vendor/github.com/labstack/echo/middleware/rewrite.go
generated
vendored
Normal file
83
vendor/github.com/labstack/echo/middleware/rewrite.go
generated
vendored
Normal file
@ -0,0 +1,83 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/labstack/echo"
|
||||||
|
)
|
||||||
|
|
||||||
|
type (
|
||||||
|
// RewriteConfig defines the config for Rewrite middleware.
|
||||||
|
RewriteConfig struct {
|
||||||
|
// Skipper defines a function to skip middleware.
|
||||||
|
Skipper Skipper
|
||||||
|
|
||||||
|
// Rules defines the URL path rewrite rules. The values captured in asterisk can be
|
||||||
|
// retrieved by index e.g. $1, $2 and so on.
|
||||||
|
// Example:
|
||||||
|
// "/old": "/new",
|
||||||
|
// "/api/*": "/$1",
|
||||||
|
// "/js/*": "/public/javascripts/$1",
|
||||||
|
// "/users/*/orders/*": "/user/$1/order/$2",
|
||||||
|
// Required.
|
||||||
|
Rules map[string]string `yaml:"rules"`
|
||||||
|
|
||||||
|
rulesRegex map[*regexp.Regexp]string
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// DefaultRewriteConfig is the default Rewrite middleware config.
|
||||||
|
DefaultRewriteConfig = RewriteConfig{
|
||||||
|
Skipper: DefaultSkipper,
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// Rewrite returns a Rewrite middleware.
|
||||||
|
//
|
||||||
|
// Rewrite middleware rewrites the URL path based on the provided rules.
|
||||||
|
func Rewrite(rules map[string]string) echo.MiddlewareFunc {
|
||||||
|
c := DefaultRewriteConfig
|
||||||
|
c.Rules = rules
|
||||||
|
return RewriteWithConfig(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RewriteWithConfig returns a Rewrite middleware with config.
|
||||||
|
// See: `Rewrite()`.
|
||||||
|
func RewriteWithConfig(config RewriteConfig) echo.MiddlewareFunc {
|
||||||
|
// Defaults
|
||||||
|
if config.Rules == nil {
|
||||||
|
panic("echo: rewrite middleware requires url path rewrite rules")
|
||||||
|
}
|
||||||
|
if config.Skipper == nil {
|
||||||
|
config.Skipper = DefaultBodyDumpConfig.Skipper
|
||||||
|
}
|
||||||
|
config.rulesRegex = map[*regexp.Regexp]string{}
|
||||||
|
|
||||||
|
// Initialize
|
||||||
|
for k, v := range config.Rules {
|
||||||
|
k = strings.Replace(k, "*", "(\\S*)", -1)
|
||||||
|
config.rulesRegex[regexp.MustCompile(k)] = v
|
||||||
|
}
|
||||||
|
|
||||||
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(c echo.Context) (err error) {
|
||||||
|
if config.Skipper(c) {
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
|
||||||
|
req := c.Request()
|
||||||
|
|
||||||
|
// Rewrite
|
||||||
|
for k, v := range config.rulesRegex {
|
||||||
|
replacer := captureTokens(k, req.URL.Path)
|
||||||
|
if replacer != nil {
|
||||||
|
req.URL.Path = replacer.Replace(v)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return next(c)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
12
vendor/github.com/labstack/echo/middleware/secure.go
generated
vendored
12
vendor/github.com/labstack/echo/middleware/secure.go
generated
vendored
@ -15,12 +15,12 @@ type (
|
|||||||
// XSSProtection provides protection against cross-site scripting attack (XSS)
|
// XSSProtection provides protection against cross-site scripting attack (XSS)
|
||||||
// by setting the `X-XSS-Protection` header.
|
// by setting the `X-XSS-Protection` header.
|
||||||
// Optional. Default value "1; mode=block".
|
// Optional. Default value "1; mode=block".
|
||||||
XSSProtection string `json:"xss_protection"`
|
XSSProtection string `yaml:"xss_protection"`
|
||||||
|
|
||||||
// ContentTypeNosniff provides protection against overriding Content-Type
|
// ContentTypeNosniff provides protection against overriding Content-Type
|
||||||
// header by setting the `X-Content-Type-Options` header.
|
// header by setting the `X-Content-Type-Options` header.
|
||||||
// Optional. Default value "nosniff".
|
// Optional. Default value "nosniff".
|
||||||
ContentTypeNosniff string `json:"content_type_nosniff"`
|
ContentTypeNosniff string `yaml:"content_type_nosniff"`
|
||||||
|
|
||||||
// XFrameOptions can be used to indicate whether or not a browser should
|
// XFrameOptions can be used to indicate whether or not a browser should
|
||||||
// be allowed to render a page in a <frame>, <iframe> or <object> .
|
// be allowed to render a page in a <frame>, <iframe> or <object> .
|
||||||
@ -32,27 +32,27 @@ type (
|
|||||||
// - "SAMEORIGIN" - The page can only be displayed in a frame on the same origin as the page itself.
|
// - "SAMEORIGIN" - The page can only be displayed in a frame on the same origin as the page itself.
|
||||||
// - "DENY" - The page cannot be displayed in a frame, regardless of the site attempting to do so.
|
// - "DENY" - The page cannot be displayed in a frame, regardless of the site attempting to do so.
|
||||||
// - "ALLOW-FROM uri" - The page can only be displayed in a frame on the specified origin.
|
// - "ALLOW-FROM uri" - The page can only be displayed in a frame on the specified origin.
|
||||||
XFrameOptions string `json:"x_frame_options"`
|
XFrameOptions string `yaml:"x_frame_options"`
|
||||||
|
|
||||||
// HSTSMaxAge sets the `Strict-Transport-Security` header to indicate how
|
// HSTSMaxAge sets the `Strict-Transport-Security` header to indicate how
|
||||||
// long (in seconds) browsers should remember that this site is only to
|
// long (in seconds) browsers should remember that this site is only to
|
||||||
// be accessed using HTTPS. This reduces your exposure to some SSL-stripping
|
// be accessed using HTTPS. This reduces your exposure to some SSL-stripping
|
||||||
// man-in-the-middle (MITM) attacks.
|
// man-in-the-middle (MITM) attacks.
|
||||||
// Optional. Default value 0.
|
// Optional. Default value 0.
|
||||||
HSTSMaxAge int `json:"hsts_max_age"`
|
HSTSMaxAge int `yaml:"hsts_max_age"`
|
||||||
|
|
||||||
// HSTSExcludeSubdomains won't include subdomains tag in the `Strict Transport Security`
|
// HSTSExcludeSubdomains won't include subdomains tag in the `Strict Transport Security`
|
||||||
// header, excluding all subdomains from security policy. It has no effect
|
// header, excluding all subdomains from security policy. It has no effect
|
||||||
// unless HSTSMaxAge is set to a non-zero value.
|
// unless HSTSMaxAge is set to a non-zero value.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
HSTSExcludeSubdomains bool `json:"hsts_exclude_subdomains"`
|
HSTSExcludeSubdomains bool `yaml:"hsts_exclude_subdomains"`
|
||||||
|
|
||||||
// ContentSecurityPolicy sets the `Content-Security-Policy` header providing
|
// ContentSecurityPolicy sets the `Content-Security-Policy` header providing
|
||||||
// security against cross-site scripting (XSS), clickjacking and other code
|
// security against cross-site scripting (XSS), clickjacking and other code
|
||||||
// injection attacks resulting from execution of malicious content in the
|
// injection attacks resulting from execution of malicious content in the
|
||||||
// trusted web page context.
|
// trusted web page context.
|
||||||
// Optional. Default value "".
|
// Optional. Default value "".
|
||||||
ContentSecurityPolicy string `json:"content_security_policy"`
|
ContentSecurityPolicy string `yaml:"content_security_policy"`
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
2
vendor/github.com/labstack/echo/middleware/slash.go
generated
vendored
2
vendor/github.com/labstack/echo/middleware/slash.go
generated
vendored
@ -12,7 +12,7 @@ type (
|
|||||||
|
|
||||||
// Status code to be used when redirecting the request.
|
// Status code to be used when redirecting the request.
|
||||||
// Optional, but when provided the request is redirected using this code.
|
// Optional, but when provided the request is redirected using this code.
|
||||||
RedirectCode int `json:"redirect_code"`
|
RedirectCode int `yaml:"redirect_code"`
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
41
vendor/github.com/labstack/echo/middleware/static.go
generated
vendored
41
vendor/github.com/labstack/echo/middleware/static.go
generated
vendored
@ -2,6 +2,7 @@ package middleware
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"path"
|
"path"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
@ -18,20 +19,20 @@ type (
|
|||||||
|
|
||||||
// Root directory from where the static content is served.
|
// Root directory from where the static content is served.
|
||||||
// Required.
|
// Required.
|
||||||
Root string `json:"root"`
|
Root string `yaml:"root"`
|
||||||
|
|
||||||
// Index file for serving a directory.
|
// Index file for serving a directory.
|
||||||
// Optional. Default value "index.html".
|
// Optional. Default value "index.html".
|
||||||
Index string `json:"index"`
|
Index string `yaml:"index"`
|
||||||
|
|
||||||
// Enable HTML5 mode by forwarding all not-found requests to root so that
|
// Enable HTML5 mode by forwarding all not-found requests to root so that
|
||||||
// SPA (single-page application) can handle the routing.
|
// SPA (single-page application) can handle the routing.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
HTML5 bool `json:"html5"`
|
HTML5 bool `yaml:"html5"`
|
||||||
|
|
||||||
// Enable directory browsing.
|
// Enable directory browsing.
|
||||||
// Optional. Default value false.
|
// Optional. Default value false.
|
||||||
Browse bool `json:"browse"`
|
Browse bool `yaml:"browse"`
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -66,7 +67,7 @@ func StaticWithConfig(config StaticConfig) echo.MiddlewareFunc {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
return func(c echo.Context) error {
|
return func(c echo.Context) (err error) {
|
||||||
if config.Skipper(c) {
|
if config.Skipper(c) {
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
@ -75,17 +76,25 @@ func StaticWithConfig(config StaticConfig) echo.MiddlewareFunc {
|
|||||||
if strings.HasSuffix(c.Path(), "*") { // When serving from a group, e.g. `/static*`.
|
if strings.HasSuffix(c.Path(), "*") { // When serving from a group, e.g. `/static*`.
|
||||||
p = c.Param("*")
|
p = c.Param("*")
|
||||||
}
|
}
|
||||||
|
p, err = echo.PathUnescape(p)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
name := filepath.Join(config.Root, path.Clean("/"+p)) // "/"+ for security
|
name := filepath.Join(config.Root, path.Clean("/"+p)) // "/"+ for security
|
||||||
|
|
||||||
fi, err := os.Stat(name)
|
fi, err := os.Stat(name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
if config.HTML5 && path.Ext(p) == "" {
|
if err = next(c); err != nil {
|
||||||
return c.File(filepath.Join(config.Root, config.Index))
|
if he, ok := err.(*echo.HTTPError); ok {
|
||||||
|
if config.HTML5 && he.Code == http.StatusNotFound {
|
||||||
|
return c.File(filepath.Join(config.Root, config.Index))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
}
|
}
|
||||||
return next(c)
|
|
||||||
}
|
}
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if fi.IsDir() {
|
if fi.IsDir() {
|
||||||
@ -99,7 +108,7 @@ func StaticWithConfig(config StaticConfig) echo.MiddlewareFunc {
|
|||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.File(index)
|
return c.File(index)
|
||||||
@ -110,20 +119,20 @@ func StaticWithConfig(config StaticConfig) echo.MiddlewareFunc {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func listDir(name string, res *echo.Response) error {
|
func listDir(name string, res *echo.Response) (err error) {
|
||||||
dir, err := os.Open(name)
|
dir, err := os.Open(name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
dirs, err := dir.Readdir(-1)
|
dirs, err := dir.Readdir(-1)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create a directory index
|
// Create a directory index
|
||||||
res.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
|
res.Header().Set(echo.HeaderContentType, echo.MIMETextHTMLCharsetUTF8)
|
||||||
if _, err = fmt.Fprintf(res, "<pre>\n"); err != nil {
|
if _, err = fmt.Fprintf(res, "<pre>\n"); err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
for _, d := range dirs {
|
for _, d := range dirs {
|
||||||
name := d.Name()
|
name := d.Name()
|
||||||
@ -133,9 +142,9 @@ func listDir(name string, res *echo.Response) error {
|
|||||||
name += "/"
|
name += "/"
|
||||||
}
|
}
|
||||||
if _, err = fmt.Fprintf(res, "<a href=\"%s\" style=\"color: %s;\">%s</a>\n", name, color, name); err != nil {
|
if _, err = fmt.Fprintf(res, "<a href=\"%s\" style=\"color: %s;\">%s</a>\n", name, color, name); err != nil {
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
_, err = fmt.Fprintf(res, "</pre>\n")
|
_, err = fmt.Fprintf(res, "</pre>\n")
|
||||||
return err
|
return
|
||||||
}
|
}
|
||||||
|
37
vendor/github.com/labstack/echo/response.go
generated
vendored
37
vendor/github.com/labstack/echo/response.go
generated
vendored
@ -4,6 +4,7 @@ import (
|
|||||||
"bufio"
|
"bufio"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strconv"
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
@ -11,11 +12,14 @@ type (
|
|||||||
// by an HTTP handler to construct an HTTP response.
|
// by an HTTP handler to construct an HTTP response.
|
||||||
// See: https://golang.org/pkg/net/http/#ResponseWriter
|
// See: https://golang.org/pkg/net/http/#ResponseWriter
|
||||||
Response struct {
|
Response struct {
|
||||||
Writer http.ResponseWriter
|
echo *Echo
|
||||||
Status int
|
contentLength int64
|
||||||
Size int64
|
beforeFuncs []func()
|
||||||
Committed bool
|
afterFuncs []func()
|
||||||
echo *Echo
|
Writer http.ResponseWriter
|
||||||
|
Status int
|
||||||
|
Size int64
|
||||||
|
Committed bool
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -34,6 +38,17 @@ func (r *Response) Header() http.Header {
|
|||||||
return r.Writer.Header()
|
return r.Writer.Header()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Before registers a function which is called just before the response is written.
|
||||||
|
func (r *Response) Before(fn func()) {
|
||||||
|
r.beforeFuncs = append(r.beforeFuncs, fn)
|
||||||
|
}
|
||||||
|
|
||||||
|
// After registers a function which is called just after the response is written.
|
||||||
|
// If the `Content-Length` is unknown, none of the after function is executed.
|
||||||
|
func (r *Response) After(fn func()) {
|
||||||
|
r.afterFuncs = append(r.afterFuncs, fn)
|
||||||
|
}
|
||||||
|
|
||||||
// WriteHeader sends an HTTP response header with status code. If WriteHeader is
|
// WriteHeader sends an HTTP response header with status code. If WriteHeader is
|
||||||
// not called explicitly, the first call to Write will trigger an implicit
|
// not called explicitly, the first call to Write will trigger an implicit
|
||||||
// WriteHeader(http.StatusOK). Thus explicit calls to WriteHeader are mainly
|
// WriteHeader(http.StatusOK). Thus explicit calls to WriteHeader are mainly
|
||||||
@ -43,9 +58,13 @@ func (r *Response) WriteHeader(code int) {
|
|||||||
r.echo.Logger.Warn("response already committed")
|
r.echo.Logger.Warn("response already committed")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
for _, fn := range r.beforeFuncs {
|
||||||
|
fn()
|
||||||
|
}
|
||||||
r.Status = code
|
r.Status = code
|
||||||
r.Writer.WriteHeader(code)
|
r.Writer.WriteHeader(code)
|
||||||
r.Committed = true
|
r.Committed = true
|
||||||
|
r.contentLength, _ = strconv.ParseInt(r.Header().Get(HeaderContentLength), 10, 0)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Write writes the data to the connection as part of an HTTP reply.
|
// Write writes the data to the connection as part of an HTTP reply.
|
||||||
@ -55,6 +74,11 @@ func (r *Response) Write(b []byte) (n int, err error) {
|
|||||||
}
|
}
|
||||||
n, err = r.Writer.Write(b)
|
n, err = r.Writer.Write(b)
|
||||||
r.Size += int64(n)
|
r.Size += int64(n)
|
||||||
|
if r.Size == r.contentLength {
|
||||||
|
for _, fn := range r.afterFuncs {
|
||||||
|
fn()
|
||||||
|
}
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -82,6 +106,9 @@ func (r *Response) CloseNotify() <-chan bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (r *Response) reset(w http.ResponseWriter) {
|
func (r *Response) reset(w http.ResponseWriter) {
|
||||||
|
r.contentLength = 0
|
||||||
|
r.beforeFuncs = nil
|
||||||
|
r.afterFuncs = nil
|
||||||
r.Writer = w
|
r.Writer = w
|
||||||
r.Size = 0
|
r.Size = 0
|
||||||
r.Status = http.StatusOK
|
r.Status = http.StatusOK
|
||||||
|
10
vendor/github.com/labstack/echo/router.go
generated
vendored
10
vendor/github.com/labstack/echo/router.go
generated
vendored
@ -1,7 +1,5 @@
|
|||||||
package echo
|
package echo
|
||||||
|
|
||||||
import "strings"
|
|
||||||
|
|
||||||
type (
|
type (
|
||||||
// Router is the registry of all registered routes for an `Echo` instance for
|
// Router is the registry of all registered routes for an `Echo` instance for
|
||||||
// request matching and URL path parameter parsing.
|
// request matching and URL path parameter parsing.
|
||||||
@ -175,12 +173,6 @@ func (r *Router) insert(method, path string, h HandlerFunc, t kind, ppath string
|
|||||||
if len(cn.pnames) == 0 { // Issue #729
|
if len(cn.pnames) == 0 { // Issue #729
|
||||||
cn.pnames = pnames
|
cn.pnames = pnames
|
||||||
}
|
}
|
||||||
for i, n := range pnames {
|
|
||||||
// Param name aliases
|
|
||||||
if i < len(cn.pnames) && !strings.Contains(cn.pnames[i], n) {
|
|
||||||
cn.pnames[i] += "," + n
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
@ -394,7 +386,7 @@ func (r *Router) Find(method, path string, c Context) {
|
|||||||
if cn = cn.findChildByKind(akind); cn == nil {
|
if cn = cn.findChildByKind(akind); cn == nil {
|
||||||
if nn != nil {
|
if nn != nil {
|
||||||
cn = nn
|
cn = nn
|
||||||
nn = nil // Next
|
nn = cn.parent // Next (Issue #954)
|
||||||
search = ns
|
search = ns
|
||||||
if nk == pkind {
|
if nk == pkind {
|
||||||
goto Param
|
goto Param
|
||||||
|
12
vendor/github.com/labstack/echo/util_go17.go
generated
vendored
Normal file
12
vendor/github.com/labstack/echo/util_go17.go
generated
vendored
Normal file
@ -0,0 +1,12 @@
|
|||||||
|
// +build go1.7, !go1.8
|
||||||
|
|
||||||
|
package echo
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/url"
|
||||||
|
)
|
||||||
|
|
||||||
|
// PathUnescape is wraps `url.QueryUnescape`
|
||||||
|
func PathUnescape(s string) (string, error) {
|
||||||
|
return url.QueryUnescape(s)
|
||||||
|
}
|
10
vendor/github.com/labstack/echo/util_go18.go
generated
vendored
Normal file
10
vendor/github.com/labstack/echo/util_go18.go
generated
vendored
Normal file
@ -0,0 +1,10 @@
|
|||||||
|
// +build go1.8
|
||||||
|
|
||||||
|
package echo
|
||||||
|
|
||||||
|
import "net/url"
|
||||||
|
|
||||||
|
// PathUnescape is wraps `url.PathUnescape`
|
||||||
|
func PathUnescape(s string) (string, error) {
|
||||||
|
return url.PathUnescape(s)
|
||||||
|
}
|
80
vendor/github.com/lrstanley/girc/builtin.go
generated
vendored
80
vendor/github.com/lrstanley/girc/builtin.go
generated
vendored
@ -16,64 +16,62 @@ func (c *Client) registerBuiltins() {
|
|||||||
c.Handlers.mu.Lock()
|
c.Handlers.mu.Lock()
|
||||||
|
|
||||||
// Built-in things that should always be supported.
|
// Built-in things that should always be supported.
|
||||||
c.Handlers.register(true, RPL_WELCOME, HandlerFunc(func(c *Client, e Event) {
|
c.Handlers.register(true, true, RPL_WELCOME, HandlerFunc(handleConnect))
|
||||||
go handleConnect(c, e)
|
c.Handlers.register(true, false, PING, HandlerFunc(handlePING))
|
||||||
}))
|
c.Handlers.register(true, false, PONG, HandlerFunc(handlePONG))
|
||||||
c.Handlers.register(true, PING, HandlerFunc(handlePING))
|
|
||||||
c.Handlers.register(true, PONG, HandlerFunc(handlePONG))
|
|
||||||
|
|
||||||
if !c.Config.disableTracking {
|
if !c.Config.disableTracking {
|
||||||
// Joins/parts/anything that may add/remove/rename users.
|
// Joins/parts/anything that may add/remove/rename users.
|
||||||
c.Handlers.register(true, JOIN, HandlerFunc(handleJOIN))
|
c.Handlers.register(true, false, JOIN, HandlerFunc(handleJOIN))
|
||||||
c.Handlers.register(true, PART, HandlerFunc(handlePART))
|
c.Handlers.register(true, false, PART, HandlerFunc(handlePART))
|
||||||
c.Handlers.register(true, KICK, HandlerFunc(handleKICK))
|
c.Handlers.register(true, false, KICK, HandlerFunc(handleKICK))
|
||||||
c.Handlers.register(true, QUIT, HandlerFunc(handleQUIT))
|
c.Handlers.register(true, false, QUIT, HandlerFunc(handleQUIT))
|
||||||
c.Handlers.register(true, NICK, HandlerFunc(handleNICK))
|
c.Handlers.register(true, false, NICK, HandlerFunc(handleNICK))
|
||||||
c.Handlers.register(true, RPL_NAMREPLY, HandlerFunc(handleNAMES))
|
c.Handlers.register(true, false, RPL_NAMREPLY, HandlerFunc(handleNAMES))
|
||||||
|
|
||||||
// Modes.
|
// Modes.
|
||||||
c.Handlers.register(true, MODE, HandlerFunc(handleMODE))
|
c.Handlers.register(true, false, MODE, HandlerFunc(handleMODE))
|
||||||
c.Handlers.register(true, RPL_CHANNELMODEIS, HandlerFunc(handleMODE))
|
c.Handlers.register(true, false, RPL_CHANNELMODEIS, HandlerFunc(handleMODE))
|
||||||
|
|
||||||
// WHO/WHOX responses.
|
// WHO/WHOX responses.
|
||||||
c.Handlers.register(true, RPL_WHOREPLY, HandlerFunc(handleWHO))
|
c.Handlers.register(true, false, RPL_WHOREPLY, HandlerFunc(handleWHO))
|
||||||
c.Handlers.register(true, RPL_WHOSPCRPL, HandlerFunc(handleWHO))
|
c.Handlers.register(true, false, RPL_WHOSPCRPL, HandlerFunc(handleWHO))
|
||||||
|
|
||||||
// Other misc. useful stuff.
|
// Other misc. useful stuff.
|
||||||
c.Handlers.register(true, TOPIC, HandlerFunc(handleTOPIC))
|
c.Handlers.register(true, false, TOPIC, HandlerFunc(handleTOPIC))
|
||||||
c.Handlers.register(true, RPL_TOPIC, HandlerFunc(handleTOPIC))
|
c.Handlers.register(true, false, RPL_TOPIC, HandlerFunc(handleTOPIC))
|
||||||
c.Handlers.register(true, RPL_MYINFO, HandlerFunc(handleMYINFO))
|
c.Handlers.register(true, false, RPL_MYINFO, HandlerFunc(handleMYINFO))
|
||||||
c.Handlers.register(true, RPL_ISUPPORT, HandlerFunc(handleISUPPORT))
|
c.Handlers.register(true, false, RPL_ISUPPORT, HandlerFunc(handleISUPPORT))
|
||||||
c.Handlers.register(true, RPL_MOTDSTART, HandlerFunc(handleMOTD))
|
c.Handlers.register(true, false, RPL_MOTDSTART, HandlerFunc(handleMOTD))
|
||||||
c.Handlers.register(true, RPL_MOTD, HandlerFunc(handleMOTD))
|
c.Handlers.register(true, false, RPL_MOTD, HandlerFunc(handleMOTD))
|
||||||
|
|
||||||
// Keep users lastactive times up to date.
|
// Keep users lastactive times up to date.
|
||||||
c.Handlers.register(true, PRIVMSG, HandlerFunc(updateLastActive))
|
c.Handlers.register(true, false, PRIVMSG, HandlerFunc(updateLastActive))
|
||||||
c.Handlers.register(true, NOTICE, HandlerFunc(updateLastActive))
|
c.Handlers.register(true, false, NOTICE, HandlerFunc(updateLastActive))
|
||||||
c.Handlers.register(true, TOPIC, HandlerFunc(updateLastActive))
|
c.Handlers.register(true, false, TOPIC, HandlerFunc(updateLastActive))
|
||||||
c.Handlers.register(true, KICK, HandlerFunc(updateLastActive))
|
c.Handlers.register(true, false, KICK, HandlerFunc(updateLastActive))
|
||||||
|
|
||||||
// CAP IRCv3-specific tracking and functionality.
|
// CAP IRCv3-specific tracking and functionality.
|
||||||
c.Handlers.register(true, CAP, HandlerFunc(handleCAP))
|
c.Handlers.register(true, false, CAP, HandlerFunc(handleCAP))
|
||||||
c.Handlers.register(true, CAP_CHGHOST, HandlerFunc(handleCHGHOST))
|
c.Handlers.register(true, false, CAP_CHGHOST, HandlerFunc(handleCHGHOST))
|
||||||
c.Handlers.register(true, CAP_AWAY, HandlerFunc(handleAWAY))
|
c.Handlers.register(true, false, CAP_AWAY, HandlerFunc(handleAWAY))
|
||||||
c.Handlers.register(true, CAP_ACCOUNT, HandlerFunc(handleACCOUNT))
|
c.Handlers.register(true, false, CAP_ACCOUNT, HandlerFunc(handleACCOUNT))
|
||||||
c.Handlers.register(true, ALL_EVENTS, HandlerFunc(handleTags))
|
c.Handlers.register(true, false, ALL_EVENTS, HandlerFunc(handleTags))
|
||||||
|
|
||||||
// SASL IRCv3 support.
|
// SASL IRCv3 support.
|
||||||
c.Handlers.register(true, AUTHENTICATE, HandlerFunc(handleSASL))
|
c.Handlers.register(true, false, AUTHENTICATE, HandlerFunc(handleSASL))
|
||||||
c.Handlers.register(true, RPL_SASLSUCCESS, HandlerFunc(handleSASL))
|
c.Handlers.register(true, false, RPL_SASLSUCCESS, HandlerFunc(handleSASL))
|
||||||
c.Handlers.register(true, RPL_NICKLOCKED, HandlerFunc(handleSASLError))
|
c.Handlers.register(true, false, RPL_NICKLOCKED, HandlerFunc(handleSASLError))
|
||||||
c.Handlers.register(true, ERR_SASLFAIL, HandlerFunc(handleSASLError))
|
c.Handlers.register(true, false, ERR_SASLFAIL, HandlerFunc(handleSASLError))
|
||||||
c.Handlers.register(true, ERR_SASLTOOLONG, HandlerFunc(handleSASLError))
|
c.Handlers.register(true, false, ERR_SASLTOOLONG, HandlerFunc(handleSASLError))
|
||||||
c.Handlers.register(true, ERR_SASLABORTED, HandlerFunc(handleSASLError))
|
c.Handlers.register(true, false, ERR_SASLABORTED, HandlerFunc(handleSASLError))
|
||||||
c.Handlers.register(true, RPL_SASLMECHS, HandlerFunc(handleSASLError))
|
c.Handlers.register(true, false, RPL_SASLMECHS, HandlerFunc(handleSASLError))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Nickname collisions.
|
// Nickname collisions.
|
||||||
c.Handlers.register(true, ERR_NICKNAMEINUSE, HandlerFunc(nickCollisionHandler))
|
c.Handlers.register(true, false, ERR_NICKNAMEINUSE, HandlerFunc(nickCollisionHandler))
|
||||||
c.Handlers.register(true, ERR_NICKCOLLISION, HandlerFunc(nickCollisionHandler))
|
c.Handlers.register(true, false, ERR_NICKCOLLISION, HandlerFunc(nickCollisionHandler))
|
||||||
c.Handlers.register(true, ERR_UNAVAILRESOURCE, HandlerFunc(nickCollisionHandler))
|
c.Handlers.register(true, false, ERR_UNAVAILRESOURCE, HandlerFunc(nickCollisionHandler))
|
||||||
|
|
||||||
c.Handlers.mu.Unlock()
|
c.Handlers.mu.Unlock()
|
||||||
}
|
}
|
||||||
@ -389,7 +387,7 @@ func handleISUPPORT(c *Client, e Event) {
|
|||||||
c.state.Lock()
|
c.state.Lock()
|
||||||
// Skip the first parameter, as it's our nickname.
|
// Skip the first parameter, as it's our nickname.
|
||||||
for i := 1; i < len(e.Params); i++ {
|
for i := 1; i < len(e.Params); i++ {
|
||||||
j := strings.IndexByte(e.Params[i], 0x3D) // =
|
j := strings.IndexByte(e.Params[i], '=')
|
||||||
|
|
||||||
if j < 1 || (j+1) == len(e.Params[i]) {
|
if j < 1 || (j+1) == len(e.Params[i]) {
|
||||||
c.state.serverOptions[e.Params[i]] = ""
|
c.state.serverOptions[e.Params[i]] = ""
|
||||||
|
16
vendor/github.com/lrstanley/girc/cap.go
generated
vendored
16
vendor/github.com/lrstanley/girc/cap.go
generated
vendored
@ -136,7 +136,7 @@ func handleCAP(c *Client, e Event) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Let them know which ones we'd like to enable.
|
// Let them know which ones we'd like to enable.
|
||||||
c.write(&Event{Command: CAP, Params: []string{CAP_REQ}, Trailing: strings.Join(c.state.tmpCap, " ")})
|
c.write(&Event{Command: CAP, Params: []string{CAP_REQ}, Trailing: strings.Join(c.state.tmpCap, " "), EmptyTrailing: true})
|
||||||
|
|
||||||
// Re-initialize the tmpCap, so if we get multiple 'CAP LS' requests
|
// Re-initialize the tmpCap, so if we get multiple 'CAP LS' requests
|
||||||
// due to cap-notify, we can re-evaluate what we can support.
|
// due to cap-notify, we can re-evaluate what we can support.
|
||||||
@ -375,11 +375,11 @@ func handleTags(c *Client, e Event) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
prefixTag byte = 0x40 // @
|
prefixTag byte = '@'
|
||||||
prefixTagValue byte = 0x3D // =
|
prefixTagValue byte = '='
|
||||||
prefixUserTag byte = 0x2B // +
|
prefixUserTag byte = '+'
|
||||||
tagSeparator byte = 0x3B // ;
|
tagSeparator byte = ';'
|
||||||
maxTagLength int = 511 // 510 + @ and " " (space), though space usually not included.
|
maxTagLength int = 511 // 510 + @ and " " (space), though space usually not included.
|
||||||
)
|
)
|
||||||
|
|
||||||
// Tags represents the key-value pairs in IRCv3 message tags. The map contains
|
// Tags represents the key-value pairs in IRCv3 message tags. The map contains
|
||||||
@ -618,7 +618,7 @@ func validTag(name string) bool {
|
|||||||
|
|
||||||
for i := 0; i < len(name); i++ {
|
for i := 0; i < len(name); i++ {
|
||||||
// A-Z, a-z, 0-9, -/._
|
// A-Z, a-z, 0-9, -/._
|
||||||
if (name[i] < 0x41 || name[i] > 0x5A) && (name[i] < 0x61 || name[i] > 0x7A) && (name[i] < 0x2D || name[i] > 0x39) && name[i] != 0x5F {
|
if (name[i] < 'A' || name[i] > 'Z') && (name[i] < 'a' || name[i] > 'z') && (name[i] < '-' || name[i] > '9') && name[i] != '_' {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -631,7 +631,7 @@ func validTag(name string) bool {
|
|||||||
func validTagValue(value string) bool {
|
func validTagValue(value string) bool {
|
||||||
for i := 0; i < len(value); i++ {
|
for i := 0; i < len(value); i++ {
|
||||||
// Don't allow any invisible chars within the tag, or semicolons.
|
// Don't allow any invisible chars within the tag, or semicolons.
|
||||||
if value[i] < 0x21 || value[i] > 0x7E || value[i] == 0x3B {
|
if value[i] < '!' || value[i] > '~' || value[i] == ';' {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
105
vendor/github.com/lrstanley/girc/client.go
generated
vendored
105
vendor/github.com/lrstanley/girc/client.go
generated
vendored
@ -191,18 +191,6 @@ func (conf *Config) isValid() error {
|
|||||||
// connected.
|
// connected.
|
||||||
var ErrNotConnected = errors.New("client is not connected to server")
|
var ErrNotConnected = errors.New("client is not connected to server")
|
||||||
|
|
||||||
// ErrDisconnected is called when Config.Retries is less than 1, and we
|
|
||||||
// non-intentionally disconnected from the server.
|
|
||||||
var ErrDisconnected = errors.New("unexpectedly disconnected")
|
|
||||||
|
|
||||||
// ErrInvalidTarget should be returned if the target which you are
|
|
||||||
// attempting to send an event to is invalid or doesn't match RFC spec.
|
|
||||||
type ErrInvalidTarget struct {
|
|
||||||
Target string
|
|
||||||
}
|
|
||||||
|
|
||||||
func (e *ErrInvalidTarget) Error() string { return "invalid target: " + e.Target }
|
|
||||||
|
|
||||||
// New creates a new IRC client with the specified server, name and config.
|
// New creates a new IRC client with the specified server, name and config.
|
||||||
func New(config Config) *Client {
|
func New(config Config) *Client {
|
||||||
c := &Client{
|
c := &Client{
|
||||||
@ -253,6 +241,37 @@ func (c *Client) String() string {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TLSConnectionState returns the TLS connection state from tls.Conn{}, which
|
||||||
|
// is useful to return needed TLS fingerprint info, certificates, verify cert
|
||||||
|
// expiration dates, etc. Will only return an error if the underlying
|
||||||
|
// connection wasn't established using TLS (see ErrConnNotTLS), or if the
|
||||||
|
// client isn't connected.
|
||||||
|
func (c *Client) TLSConnectionState() (*tls.ConnectionState, error) {
|
||||||
|
c.mu.RLock()
|
||||||
|
defer c.mu.RUnlock()
|
||||||
|
if c.conn == nil {
|
||||||
|
return nil, ErrNotConnected
|
||||||
|
}
|
||||||
|
|
||||||
|
c.conn.mu.RLock()
|
||||||
|
defer c.conn.mu.RUnlock()
|
||||||
|
|
||||||
|
if !c.conn.connected {
|
||||||
|
return nil, ErrNotConnected
|
||||||
|
}
|
||||||
|
|
||||||
|
if tlsConn, ok := c.conn.sock.(*tls.Conn); ok {
|
||||||
|
cs := tlsConn.ConnectionState()
|
||||||
|
return &cs, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, ErrConnNotTLS
|
||||||
|
}
|
||||||
|
|
||||||
|
// ErrConnNotTLS is returned when Client.TLSConnectionState() is called, and
|
||||||
|
// the connection to the server wasn't made with TLS.
|
||||||
|
var ErrConnNotTLS = errors.New("underlying connection is not tls")
|
||||||
|
|
||||||
// Close closes the network connection to the server, and sends a STOPPED
|
// Close closes the network connection to the server, and sends a STOPPED
|
||||||
// event. This should cause Connect() to return with nil. This should be
|
// event. This should cause Connect() to return with nil. This should be
|
||||||
// safe to call multiple times. See Connect()'s documentation on how
|
// safe to call multiple times. See Connect()'s documentation on how
|
||||||
@ -387,7 +406,7 @@ func (c *Client) ConnSince() (since *time.Duration, err error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// IsConnected returns true if the client is connected to the server.
|
// IsConnected returns true if the client is connected to the server.
|
||||||
func (c *Client) IsConnected() (connected bool) {
|
func (c *Client) IsConnected() bool {
|
||||||
c.mu.RLock()
|
c.mu.RLock()
|
||||||
if c.conn == nil {
|
if c.conn == nil {
|
||||||
c.mu.RUnlock()
|
c.mu.RUnlock()
|
||||||
@ -395,7 +414,7 @@ func (c *Client) IsConnected() (connected bool) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
c.conn.mu.RLock()
|
c.conn.mu.RLock()
|
||||||
connected = c.conn.connected
|
connected := c.conn.connected
|
||||||
c.conn.mu.RUnlock()
|
c.conn.mu.RUnlock()
|
||||||
c.mu.RUnlock()
|
c.mu.RUnlock()
|
||||||
|
|
||||||
@ -445,9 +464,9 @@ func (c *Client) GetHost() string {
|
|||||||
return c.state.host
|
return c.state.host
|
||||||
}
|
}
|
||||||
|
|
||||||
// Channels returns the active list of channels that the client is in.
|
// ChannelList returns the active list of channel names that the client is in.
|
||||||
// Panics if tracking is disabled.
|
// Panics if tracking is disabled.
|
||||||
func (c *Client) Channels() []string {
|
func (c *Client) ChannelList() []string {
|
||||||
c.panicIfNotTracking()
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
c.state.RLock()
|
c.state.RLock()
|
||||||
@ -463,9 +482,26 @@ func (c *Client) Channels() []string {
|
|||||||
return channels
|
return channels
|
||||||
}
|
}
|
||||||
|
|
||||||
// Users returns the active list of users that the client is tracking across
|
// Channels returns the active channels that the client is in. Panics if
|
||||||
// all files. Panics if tracking is disabled.
|
// tracking is disabled.
|
||||||
func (c *Client) Users() []string {
|
func (c *Client) Channels() []*Channel {
|
||||||
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
|
c.state.RLock()
|
||||||
|
channels := make([]*Channel, len(c.state.channels))
|
||||||
|
var i int
|
||||||
|
for channel := range c.state.channels {
|
||||||
|
channels[i] = c.state.channels[channel].Copy()
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
c.state.RUnlock()
|
||||||
|
|
||||||
|
return channels
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserList returns the active list of nicknames that the client is tracking
|
||||||
|
// across all networks. Panics if tracking is disabled.
|
||||||
|
func (c *Client) UserList() []string {
|
||||||
c.panicIfNotTracking()
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
c.state.RLock()
|
c.state.RLock()
|
||||||
@ -481,6 +517,23 @@ func (c *Client) Users() []string {
|
|||||||
return users
|
return users
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Users returns the active users that the client is tracking across all
|
||||||
|
// networks. Panics if tracking is disabled.
|
||||||
|
func (c *Client) Users() []*User {
|
||||||
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
|
c.state.RLock()
|
||||||
|
users := make([]*User, len(c.state.users))
|
||||||
|
var i int
|
||||||
|
for user := range c.state.users {
|
||||||
|
users[i] = c.state.users[user].Copy()
|
||||||
|
i++
|
||||||
|
}
|
||||||
|
c.state.RUnlock()
|
||||||
|
|
||||||
|
return users
|
||||||
|
}
|
||||||
|
|
||||||
// LookupChannel looks up a given channel in state. If the channel doesn't
|
// LookupChannel looks up a given channel in state. If the channel doesn't
|
||||||
// exist, nil is returned. Panics if tracking is disabled.
|
// exist, nil is returned. Panics if tracking is disabled.
|
||||||
func (c *Client) LookupChannel(name string) *Channel {
|
func (c *Client) LookupChannel(name string) *Channel {
|
||||||
@ -562,30 +615,30 @@ func (c *Client) NetworkName() (name string) {
|
|||||||
// supplied this information during connection. May be empty if the server
|
// supplied this information during connection. May be empty if the server
|
||||||
// does not support RPL_MYINFO. Will panic if used when tracking has been
|
// does not support RPL_MYINFO. Will panic if used when tracking has been
|
||||||
// disabled.
|
// disabled.
|
||||||
func (c *Client) ServerVersion() (version string) {
|
func (c *Client) ServerVersion() string {
|
||||||
c.panicIfNotTracking()
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
version, _ = c.GetServerOption("VERSION")
|
version, _ := c.GetServerOption("VERSION")
|
||||||
|
|
||||||
return version
|
return version
|
||||||
}
|
}
|
||||||
|
|
||||||
// ServerMOTD returns the servers message of the day, if the server has sent
|
// ServerMOTD returns the servers message of the day, if the server has sent
|
||||||
// it upon connect. Will panic if used when tracking has been disabled.
|
// it upon connect. Will panic if used when tracking has been disabled.
|
||||||
func (c *Client) ServerMOTD() (motd string) {
|
func (c *Client) ServerMOTD() string {
|
||||||
c.panicIfNotTracking()
|
c.panicIfNotTracking()
|
||||||
|
|
||||||
c.state.RLock()
|
c.state.RLock()
|
||||||
motd = c.state.motd
|
motd := c.state.motd
|
||||||
c.state.RUnlock()
|
c.state.RUnlock()
|
||||||
|
|
||||||
return motd
|
return motd
|
||||||
}
|
}
|
||||||
|
|
||||||
// Lag is the latency between the server and the client. This is measured by
|
// Latency is the latency between the server and the client. This is measured
|
||||||
// determining the difference in time between when we ping the server, and
|
// by determining the difference in time between when we ping the server, and
|
||||||
// when we receive a pong.
|
// when we receive a pong.
|
||||||
func (c *Client) Lag() time.Duration {
|
func (c *Client) Latency() time.Duration {
|
||||||
c.mu.RLock()
|
c.mu.RLock()
|
||||||
c.conn.mu.RLock()
|
c.conn.mu.RLock()
|
||||||
delta := c.conn.lastPong.Sub(c.conn.lastPing)
|
delta := c.conn.lastPong.Sub(c.conn.lastPing)
|
||||||
|
10
vendor/github.com/lrstanley/girc/cmdhandler/cmd.go
generated
vendored
10
vendor/github.com/lrstanley/girc/cmdhandler/cmd.go
generated
vendored
@ -12,8 +12,9 @@ import (
|
|||||||
|
|
||||||
// Input is a wrapper for events, based around private messages.
|
// Input is a wrapper for events, based around private messages.
|
||||||
type Input struct {
|
type Input struct {
|
||||||
Origin *girc.Event
|
Origin *girc.Event
|
||||||
Args []string
|
Args []string
|
||||||
|
RawArgs string
|
||||||
}
|
}
|
||||||
|
|
||||||
// Command is an IRC command, supporting aliases, help documentation and easy
|
// Command is an IRC command, supporting aliases, help documentation and easy
|
||||||
@ -189,8 +190,9 @@ func (ch *CmdHandler) Execute(client *girc.Client, event girc.Event) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
in := &Input{
|
in := &Input{
|
||||||
Origin: &event,
|
Origin: &event,
|
||||||
Args: args,
|
Args: args,
|
||||||
|
RawArgs: parsed[2],
|
||||||
}
|
}
|
||||||
|
|
||||||
go cmd.Fn(client, in)
|
go cmd.Fn(client, in)
|
||||||
|
261
vendor/github.com/lrstanley/girc/commands.go
generated
vendored
261
vendor/github.com/lrstanley/girc/commands.go
generated
vendored
@ -16,18 +16,13 @@ type Commands struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Nick changes the client nickname.
|
// Nick changes the client nickname.
|
||||||
func (cmd *Commands) Nick(name string) error {
|
func (cmd *Commands) Nick(name string) {
|
||||||
if !IsValidNick(name) {
|
|
||||||
return &ErrInvalidTarget{Target: name}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: NICK, Params: []string{name}})
|
cmd.c.Send(&Event{Command: NICK, Params: []string{name}})
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Join attempts to enter a list of IRC channels, at bulk if possible to
|
// Join attempts to enter a list of IRC channels, at bulk if possible to
|
||||||
// prevent sending extensive JOIN commands.
|
// prevent sending extensive JOIN commands.
|
||||||
func (cmd *Commands) Join(channels ...string) error {
|
func (cmd *Commands) Join(channels ...string) {
|
||||||
// We can join multiple channels at once, however we need to ensure that
|
// We can join multiple channels at once, however we need to ensure that
|
||||||
// we are not exceeding the line length. (see maxLength)
|
// we are not exceeding the line length. (see maxLength)
|
||||||
max := maxLength - len(JOIN) - 1
|
max := maxLength - len(JOIN) - 1
|
||||||
@ -35,10 +30,6 @@ func (cmd *Commands) Join(channels ...string) error {
|
|||||||
var buffer string
|
var buffer string
|
||||||
|
|
||||||
for i := 0; i < len(channels); i++ {
|
for i := 0; i < len(channels); i++ {
|
||||||
if !IsValidChannel(channels[i]) {
|
|
||||||
return &ErrInvalidTarget{Target: channels[i]}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(buffer+","+channels[i]) > max {
|
if len(buffer+","+channels[i]) > max {
|
||||||
cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}})
|
cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}})
|
||||||
buffer = ""
|
buffer = ""
|
||||||
@ -53,91 +44,74 @@ func (cmd *Commands) Join(channels ...string) error {
|
|||||||
|
|
||||||
if i == len(channels)-1 {
|
if i == len(channels)-1 {
|
||||||
cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}})
|
cmd.c.Send(&Event{Command: JOIN, Params: []string{buffer}})
|
||||||
return nil
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// JoinKey attempts to enter an IRC channel with a password.
|
// JoinKey attempts to enter an IRC channel with a password.
|
||||||
func (cmd *Commands) JoinKey(channel, password string) error {
|
func (cmd *Commands) JoinKey(channel, password string) {
|
||||||
if !IsValidChannel(channel) {
|
|
||||||
return &ErrInvalidTarget{Target: channel}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: JOIN, Params: []string{channel, password}})
|
cmd.c.Send(&Event{Command: JOIN, Params: []string{channel, password}})
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Part leaves an IRC channel.
|
// Part leaves an IRC channel.
|
||||||
func (cmd *Commands) Part(channel, message string) error {
|
func (cmd *Commands) Part(channels ...string) {
|
||||||
if !IsValidChannel(channel) {
|
for i := 0; i < len(channels); i++ {
|
||||||
return &ErrInvalidTarget{Target: channel}
|
cmd.c.Send(&Event{Command: PART, Params: []string{channels[i]}})
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: JOIN, Params: []string{channel}})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// PartMessage leaves an IRC channel with a specified leave message.
|
// PartMessage leaves an IRC channel with a specified leave message.
|
||||||
func (cmd *Commands) PartMessage(channel, message string) error {
|
func (cmd *Commands) PartMessage(channel, message string) {
|
||||||
if !IsValidChannel(channel) {
|
cmd.c.Send(&Event{Command: PART, Params: []string{channel}, Trailing: message, EmptyTrailing: true})
|
||||||
return &ErrInvalidTarget{Target: channel}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: JOIN, Params: []string{channel}, Trailing: message})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendCTCP sends a CTCP request to target. Note that this method uses
|
// SendCTCP sends a CTCP request to target. Note that this method uses
|
||||||
// PRIVMSG specifically.
|
// PRIVMSG specifically. ctcpType is the CTCP command, e.g. "FINGER", "TIME",
|
||||||
func (cmd *Commands) SendCTCP(target, ctcpType, message string) error {
|
// "VERSION", etc.
|
||||||
|
func (cmd *Commands) SendCTCP(target, ctcpType, message string) {
|
||||||
out := encodeCTCPRaw(ctcpType, message)
|
out := encodeCTCPRaw(ctcpType, message)
|
||||||
if out == "" {
|
if out == "" {
|
||||||
return errors.New("invalid CTCP")
|
panic(fmt.Sprintf("invalid CTCP: %s -> %s: %s", target, ctcpType, message))
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmd.Message(target, out)
|
cmd.Message(target, out)
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendCTCPf sends a CTCP request to target using a specific format. Note that
|
// SendCTCPf sends a CTCP request to target using a specific format. Note that
|
||||||
// this method uses PRIVMSG specifically.
|
// this method uses PRIVMSG specifically. ctcpType is the CTCP command, e.g.
|
||||||
func (cmd *Commands) SendCTCPf(target, ctcpType, format string, a ...interface{}) error {
|
// "FINGER", "TIME", "VERSION", etc.
|
||||||
return cmd.SendCTCP(target, ctcpType, fmt.Sprintf(format, a...))
|
func (cmd *Commands) SendCTCPf(target, ctcpType, format string, a ...interface{}) {
|
||||||
|
cmd.SendCTCP(target, ctcpType, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendCTCPReplyf sends a CTCP response to target using a specific format.
|
// SendCTCPReplyf sends a CTCP response to target using a specific format.
|
||||||
// Note that this method uses NOTICE specifically.
|
// Note that this method uses NOTICE specifically. ctcpType is the CTCP
|
||||||
func (cmd *Commands) SendCTCPReplyf(target, ctcpType, format string, a ...interface{}) error {
|
// command, e.g. "FINGER", "TIME", "VERSION", etc.
|
||||||
return cmd.SendCTCPReply(target, ctcpType, fmt.Sprintf(format, a...))
|
func (cmd *Commands) SendCTCPReplyf(target, ctcpType, format string, a ...interface{}) {
|
||||||
|
cmd.SendCTCPReply(target, ctcpType, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendCTCPReply sends a CTCP response to target. Note that this method uses
|
// SendCTCPReply sends a CTCP response to target. Note that this method uses
|
||||||
// NOTICE specifically.
|
// NOTICE specifically.
|
||||||
func (cmd *Commands) SendCTCPReply(target, ctcpType, message string) error {
|
func (cmd *Commands) SendCTCPReply(target, ctcpType, message string) {
|
||||||
out := encodeCTCPRaw(ctcpType, message)
|
out := encodeCTCPRaw(ctcpType, message)
|
||||||
if out == "" {
|
if out == "" {
|
||||||
return errors.New("invalid CTCP")
|
panic(fmt.Sprintf("invalid CTCP: %s -> %s: %s", target, ctcpType, message))
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmd.Notice(target, out)
|
cmd.Notice(target, out)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Message sends a PRIVMSG to target (either channel, service, or user).
|
// Message sends a PRIVMSG to target (either channel, service, or user).
|
||||||
func (cmd *Commands) Message(target, message string) error {
|
func (cmd *Commands) Message(target, message string) {
|
||||||
if !IsValidNick(target) && !IsValidChannel(target) {
|
cmd.c.Send(&Event{Command: PRIVMSG, Params: []string{target}, Trailing: message, EmptyTrailing: true})
|
||||||
return &ErrInvalidTarget{Target: target}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: PRIVMSG, Params: []string{target}, Trailing: message})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Messagef sends a formated PRIVMSG to target (either channel, service, or
|
// Messagef sends a formated PRIVMSG to target (either channel, service, or
|
||||||
// user).
|
// user).
|
||||||
func (cmd *Commands) Messagef(target, format string, a ...interface{}) error {
|
func (cmd *Commands) Messagef(target, format string, a ...interface{}) {
|
||||||
return cmd.Message(target, fmt.Sprintf(format, a...))
|
cmd.Message(target, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// ErrInvalidSource is returned when a method needs to know the origin of an
|
// ErrInvalidSource is returned when a method needs to know the origin of an
|
||||||
@ -146,94 +120,95 @@ func (cmd *Commands) Messagef(target, format string, a ...interface{}) error {
|
|||||||
var ErrInvalidSource = errors.New("event has nil or invalid source address")
|
var ErrInvalidSource = errors.New("event has nil or invalid source address")
|
||||||
|
|
||||||
// Reply sends a reply to channel or user, based on where the supplied event
|
// Reply sends a reply to channel or user, based on where the supplied event
|
||||||
// originated from. See also ReplyTo().
|
// originated from. See also ReplyTo(). Panics if the incoming event has no
|
||||||
func (cmd *Commands) Reply(event Event, message string) error {
|
// source.
|
||||||
|
func (cmd *Commands) Reply(event Event, message string) {
|
||||||
if event.Source == nil {
|
if event.Source == nil {
|
||||||
return ErrInvalidSource
|
panic(ErrInvalidSource)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(event.Params) > 0 && IsValidChannel(event.Params[0]) {
|
if len(event.Params) > 0 && IsValidChannel(event.Params[0]) {
|
||||||
return cmd.Message(event.Params[0], message)
|
cmd.Message(event.Params[0], message)
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmd.Message(event.Source.Name, message)
|
cmd.Message(event.Source.Name, message)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Replyf sends a reply to channel or user with a format string, based on
|
// Replyf sends a reply to channel or user with a format string, based on
|
||||||
// where the supplied event originated from. See also ReplyTof().
|
// where the supplied event originated from. See also ReplyTof(). Panics if
|
||||||
func (cmd *Commands) Replyf(event Event, format string, a ...interface{}) error {
|
// the incoming event has no source.
|
||||||
return cmd.Reply(event, fmt.Sprintf(format, a...))
|
func (cmd *Commands) Replyf(event Event, format string, a ...interface{}) {
|
||||||
|
cmd.Reply(event, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReplyTo sends a reply to a channel or user, based on where the supplied
|
// ReplyTo sends a reply to a channel or user, based on where the supplied
|
||||||
// event originated from. ReplyTo(), when originating from a channel will
|
// event originated from. ReplyTo(), when originating from a channel will
|
||||||
// default to replying with "<user>, <message>". See also Reply().
|
// default to replying with "<user>, <message>". See also Reply(). Panics if
|
||||||
func (cmd *Commands) ReplyTo(event Event, message string) error {
|
// the incoming event has no source.
|
||||||
|
func (cmd *Commands) ReplyTo(event Event, message string) {
|
||||||
if event.Source == nil {
|
if event.Source == nil {
|
||||||
return ErrInvalidSource
|
panic(ErrInvalidSource)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(event.Params) > 0 && IsValidChannel(event.Params[0]) {
|
if len(event.Params) > 0 && IsValidChannel(event.Params[0]) {
|
||||||
return cmd.Message(event.Params[0], event.Source.Name+", "+message)
|
cmd.Message(event.Params[0], event.Source.Name+", "+message)
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmd.Message(event.Source.Name, message)
|
cmd.Message(event.Source.Name, message)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReplyTof sends a reply to a channel or user with a format string, based
|
// ReplyTof sends a reply to a channel or user with a format string, based
|
||||||
// on where the supplied event originated from. ReplyTo(), when originating
|
// on where the supplied event originated from. ReplyTo(), when originating
|
||||||
// from a channel will default to replying with "<user>, <message>". See
|
// from a channel will default to replying with "<user>, <message>". See
|
||||||
// also Replyf().
|
// also Replyf(). Panics if the incoming event has no source.
|
||||||
func (cmd *Commands) ReplyTof(event Event, format string, a ...interface{}) error {
|
func (cmd *Commands) ReplyTof(event Event, format string, a ...interface{}) {
|
||||||
return cmd.ReplyTo(event, fmt.Sprintf(format, a...))
|
cmd.ReplyTo(event, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Action sends a PRIVMSG ACTION (/me) to target (either channel, service,
|
// Action sends a PRIVMSG ACTION (/me) to target (either channel, service,
|
||||||
// or user).
|
// or user).
|
||||||
func (cmd *Commands) Action(target, message string) error {
|
func (cmd *Commands) Action(target, message string) {
|
||||||
if !IsValidNick(target) && !IsValidChannel(target) {
|
|
||||||
return &ErrInvalidTarget{Target: target}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{
|
cmd.c.Send(&Event{
|
||||||
Command: PRIVMSG,
|
Command: PRIVMSG,
|
||||||
Params: []string{target},
|
Params: []string{target},
|
||||||
Trailing: fmt.Sprintf("\001ACTION %s\001", message),
|
Trailing: fmt.Sprintf("\001ACTION %s\001", message),
|
||||||
})
|
})
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Actionf sends a formated PRIVMSG ACTION (/me) to target (either channel,
|
// Actionf sends a formated PRIVMSG ACTION (/me) to target (either channel,
|
||||||
// service, or user).
|
// service, or user).
|
||||||
func (cmd *Commands) Actionf(target, format string, a ...interface{}) error {
|
func (cmd *Commands) Actionf(target, format string, a ...interface{}) {
|
||||||
return cmd.Action(target, fmt.Sprintf(format, a...))
|
cmd.Action(target, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// Notice sends a NOTICE to target (either channel, service, or user).
|
// Notice sends a NOTICE to target (either channel, service, or user).
|
||||||
func (cmd *Commands) Notice(target, message string) error {
|
func (cmd *Commands) Notice(target, message string) {
|
||||||
if !IsValidNick(target) && !IsValidChannel(target) {
|
cmd.c.Send(&Event{Command: NOTICE, Params: []string{target}, Trailing: message, EmptyTrailing: true})
|
||||||
return &ErrInvalidTarget{Target: target}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: NOTICE, Params: []string{target}, Trailing: message})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Noticef sends a formated NOTICE to target (either channel, service, or
|
// Noticef sends a formated NOTICE to target (either channel, service, or
|
||||||
// user).
|
// user).
|
||||||
func (cmd *Commands) Noticef(target, format string, a ...interface{}) error {
|
func (cmd *Commands) Noticef(target, format string, a ...interface{}) {
|
||||||
return cmd.Notice(target, fmt.Sprintf(format, a...))
|
cmd.Notice(target, fmt.Sprintf(format, a...))
|
||||||
}
|
}
|
||||||
|
|
||||||
// SendRaw sends a raw string back to the server, without carriage returns
|
// SendRaw sends a raw string (or multiple) to the server, without carriage
|
||||||
// or newlines.
|
// returns or newlines. Returns an error if one of the raw strings cannot be
|
||||||
func (cmd *Commands) SendRaw(raw string) error {
|
// properly parsed.
|
||||||
e := ParseEvent(raw)
|
func (cmd *Commands) SendRaw(raw ...string) error {
|
||||||
if e == nil {
|
var event *Event
|
||||||
return errors.New("invalid event: " + raw)
|
|
||||||
|
for i := 0; i < len(raw); i++ {
|
||||||
|
event = ParseEvent(raw[i])
|
||||||
|
if event == nil {
|
||||||
|
return errors.New("invalid event: " + raw[i])
|
||||||
|
}
|
||||||
|
|
||||||
|
cmd.c.Send(event)
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.c.Send(e)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -246,31 +221,26 @@ func (cmd *Commands) SendRawf(format string, a ...interface{}) error {
|
|||||||
// Topic sets the topic of channel to message. Does not verify the length
|
// Topic sets the topic of channel to message. Does not verify the length
|
||||||
// of the topic.
|
// of the topic.
|
||||||
func (cmd *Commands) Topic(channel, message string) {
|
func (cmd *Commands) Topic(channel, message string) {
|
||||||
cmd.c.Send(&Event{Command: TOPIC, Params: []string{channel}, Trailing: message})
|
cmd.c.Send(&Event{Command: TOPIC, Params: []string{channel}, Trailing: message, EmptyTrailing: true})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Who sends a WHO query to the server, which will attempt WHOX by default.
|
// Who sends a WHO query to the server, which will attempt WHOX by default.
|
||||||
// See http://faerion.sourceforge.net/doc/irc/whox.var for more details. This
|
// See http://faerion.sourceforge.net/doc/irc/whox.var for more details. This
|
||||||
// sends "%tcuhnr,2" per default. Do not use "1" as this will conflict with
|
// sends "%tcuhnr,2" per default. Do not use "1" as this will conflict with
|
||||||
// girc's builtin tracking functionality.
|
// girc's builtin tracking functionality.
|
||||||
func (cmd *Commands) Who(target string) error {
|
func (cmd *Commands) Who(users ...string) {
|
||||||
if !IsValidNick(target) && !IsValidChannel(target) && !IsValidUser(target) {
|
for i := 0; i < len(users); i++ {
|
||||||
return &ErrInvalidTarget{Target: target}
|
cmd.c.Send(&Event{Command: WHO, Params: []string{users[i], "%tcuhnr,2"}})
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: WHO, Params: []string{target, "%tcuhnr,2"}})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Whois sends a WHOIS query to the server, targeted at a specific user.
|
// Whois sends a WHOIS query to the server, targeted at a specific user (or
|
||||||
// as WHOIS is a bit slower, you may want to use WHO for brief user info.
|
// set of users). As WHOIS is a bit slower, you may want to use WHO for brief
|
||||||
func (cmd *Commands) Whois(nick string) error {
|
// user info.
|
||||||
if !IsValidNick(nick) {
|
func (cmd *Commands) Whois(users ...string) {
|
||||||
return &ErrInvalidTarget{Target: nick}
|
for i := 0; i < len(users); i++ {
|
||||||
|
cmd.c.Send(&Event{Command: WHOIS, Params: []string{users[i]}})
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: WHOIS, Params: []string{nick}})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ping sends a PING query to the server, with a specific identifier that
|
// Ping sends a PING query to the server, with a specific identifier that
|
||||||
@ -294,36 +264,40 @@ func (cmd *Commands) Oper(user, pass string) {
|
|||||||
// Kick sends a KICK query to the server, attempting to kick nick from
|
// Kick sends a KICK query to the server, attempting to kick nick from
|
||||||
// channel, with reason. If reason is blank, one will not be sent to the
|
// channel, with reason. If reason is blank, one will not be sent to the
|
||||||
// server.
|
// server.
|
||||||
func (cmd *Commands) Kick(channel, nick, reason string) error {
|
func (cmd *Commands) Kick(channel, user, reason string) {
|
||||||
if !IsValidChannel(channel) {
|
|
||||||
return &ErrInvalidTarget{Target: channel}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !IsValidNick(nick) {
|
|
||||||
return &ErrInvalidTarget{Target: nick}
|
|
||||||
}
|
|
||||||
|
|
||||||
if reason != "" {
|
if reason != "" {
|
||||||
cmd.c.Send(&Event{Command: KICK, Params: []string{channel, nick}, Trailing: reason})
|
cmd.c.Send(&Event{Command: KICK, Params: []string{channel, user}, Trailing: reason, EmptyTrailing: true})
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: KICK, Params: []string{channel, nick}})
|
cmd.c.Send(&Event{Command: KICK, Params: []string{channel, user}})
|
||||||
return nil
|
}
|
||||||
|
|
||||||
|
// Ban adds the +b mode on the given mask on a channel.
|
||||||
|
func (cmd *Commands) Ban(channel, mask string) {
|
||||||
|
cmd.Mode(channel, "+b", mask)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Unban removes the +b mode on the given mask on a channel.
|
||||||
|
func (cmd *Commands) Unban(channel, mask string) {
|
||||||
|
cmd.Mode(channel, "-b", mask)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mode sends a mode change to the server which should be applied to target
|
||||||
|
// (usually a channel or user), along with a set of modes (generally "+m",
|
||||||
|
// "+mmmm", or "-m", where "m" is the mode you want to change). Params is only
|
||||||
|
// needed if the mode change requires a parameter (ban or invite-only exclude.)
|
||||||
|
func (cmd *Commands) Mode(target, modes string, params ...string) {
|
||||||
|
out := []string{target, modes}
|
||||||
|
out = append(out, params...)
|
||||||
|
|
||||||
|
cmd.c.Send(&Event{Command: MODE, Params: out})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Invite sends a INVITE query to the server, to invite nick to channel.
|
// Invite sends a INVITE query to the server, to invite nick to channel.
|
||||||
func (cmd *Commands) Invite(channel, nick string) error {
|
func (cmd *Commands) Invite(channel string, users ...string) {
|
||||||
if !IsValidChannel(channel) {
|
for i := 0; i < len(users); i++ {
|
||||||
return &ErrInvalidTarget{Target: channel}
|
cmd.c.Send(&Event{Command: INVITE, Params: []string{users[i], channel}})
|
||||||
}
|
}
|
||||||
|
|
||||||
if !IsValidNick(nick) {
|
|
||||||
return &ErrInvalidTarget{Target: nick}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: INVITE, Params: []string{nick, channel}})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Away sends a AWAY query to the server, suggesting that the client is no
|
// Away sends a AWAY query to the server, suggesting that the client is no
|
||||||
@ -348,10 +322,10 @@ func (cmd *Commands) Back() {
|
|||||||
// Supports multiple channels at once, in hopes it will reduce extensive
|
// Supports multiple channels at once, in hopes it will reduce extensive
|
||||||
// LIST queries to the server. Supply no channels to run a list against the
|
// LIST queries to the server. Supply no channels to run a list against the
|
||||||
// entire server (warning, that may mean LOTS of channels!)
|
// entire server (warning, that may mean LOTS of channels!)
|
||||||
func (cmd *Commands) List(channels ...string) error {
|
func (cmd *Commands) List(channels ...string) {
|
||||||
if len(channels) == 0 {
|
if len(channels) == 0 {
|
||||||
cmd.c.Send(&Event{Command: LIST})
|
cmd.c.Send(&Event{Command: LIST})
|
||||||
return nil
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// We can LIST multiple channels at once, however we need to ensure that
|
// We can LIST multiple channels at once, however we need to ensure that
|
||||||
@ -361,10 +335,6 @@ func (cmd *Commands) List(channels ...string) error {
|
|||||||
var buffer string
|
var buffer string
|
||||||
|
|
||||||
for i := 0; i < len(channels); i++ {
|
for i := 0; i < len(channels); i++ {
|
||||||
if !IsValidChannel(channels[i]) {
|
|
||||||
return &ErrInvalidTarget{Target: channels[i]}
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(buffer+","+channels[i]) > max {
|
if len(buffer+","+channels[i]) > max {
|
||||||
cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}})
|
cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}})
|
||||||
buffer = ""
|
buffer = ""
|
||||||
@ -379,20 +349,13 @@ func (cmd *Commands) List(channels ...string) error {
|
|||||||
|
|
||||||
if i == len(channels)-1 {
|
if i == len(channels)-1 {
|
||||||
cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}})
|
cmd.c.Send(&Event{Command: LIST, Params: []string{buffer}})
|
||||||
return nil
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Whowas sends a WHOWAS query to the server. amount is the amount of results
|
// Whowas sends a WHOWAS query to the server. amount is the amount of results
|
||||||
// you want back.
|
// you want back.
|
||||||
func (cmd *Commands) Whowas(nick string, amount int) error {
|
func (cmd *Commands) Whowas(user string, amount int) {
|
||||||
if !IsValidNick(nick) {
|
cmd.c.Send(&Event{Command: WHOWAS, Params: []string{user, string(amount)}})
|
||||||
return &ErrInvalidTarget{Target: nick}
|
|
||||||
}
|
|
||||||
|
|
||||||
cmd.c.Send(&Event{Command: WHOWAS, Params: []string{nick, string(amount)}})
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
6
vendor/github.com/lrstanley/girc/ctcp.go
generated
vendored
6
vendor/github.com/lrstanley/girc/ctcp.go
generated
vendored
@ -58,7 +58,7 @@ func decodeCTCP(e *Event) *CTCPEvent {
|
|||||||
if s < 0 {
|
if s < 0 {
|
||||||
for i := 0; i < len(text); i++ {
|
for i := 0; i < len(text); i++ {
|
||||||
// Check for A-Z, 0-9.
|
// Check for A-Z, 0-9.
|
||||||
if (text[i] < 0x41 || text[i] > 0x5A) && (text[i] < 0x30 || text[i] > 0x39) {
|
if (text[i] < 'A' || text[i] > 'Z') && (text[i] < '0' || text[i] > '9') {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -74,7 +74,7 @@ func decodeCTCP(e *Event) *CTCPEvent {
|
|||||||
// Loop through checking the tag first.
|
// Loop through checking the tag first.
|
||||||
for i := 0; i < s; i++ {
|
for i := 0; i < s; i++ {
|
||||||
// Check for A-Z, 0-9.
|
// Check for A-Z, 0-9.
|
||||||
if (text[i] < 0x41 || text[i] > 0x5A) && (text[i] < 0x30 || text[i] > 0x39) {
|
if (text[i] < 'A' || text[i] > 'Z') && (text[i] < '0' || text[i] > '9') {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -168,7 +168,7 @@ func (c *CTCP) parseCMD(cmd string) string {
|
|||||||
|
|
||||||
for i := 0; i < len(cmd); i++ {
|
for i := 0; i < len(cmd); i++ {
|
||||||
// Check for A-Z, 0-9.
|
// Check for A-Z, 0-9.
|
||||||
if (cmd[i] < 0x41 || cmd[i] > 0x5A) && (cmd[i] < 0x30 || cmd[i] > 0x39) {
|
if (cmd[i] < 'A' || cmd[i] > 'Z') && (cmd[i] < '0' || cmd[i] > '9') {
|
||||||
return ""
|
return ""
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
12
vendor/github.com/lrstanley/girc/event.go
generated
vendored
12
vendor/github.com/lrstanley/girc/event.go
generated
vendored
@ -11,8 +11,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
eventSpace byte = 0x20 // Separator.
|
eventSpace byte = ' ' // Separator.
|
||||||
maxLength = 510 // Maximum length is 510 (2 for line endings).
|
maxLength = 510 // Maximum length is 510 (2 for line endings).
|
||||||
)
|
)
|
||||||
|
|
||||||
// cutCRFunc is used to trim CR characters from prefixes/messages.
|
// cutCRFunc is used to trim CR characters from prefixes/messages.
|
||||||
@ -256,7 +256,7 @@ func (e *Event) Bytes() []byte {
|
|||||||
|
|
||||||
// Strip newlines and carriage returns.
|
// Strip newlines and carriage returns.
|
||||||
for i := 0; i < len(out); i++ {
|
for i := 0; i < len(out); i++ {
|
||||||
if out[i] == 0x0A || out[i] == 0x0D {
|
if out[i] == '\n' || out[i] == '\r' {
|
||||||
out = append(out[:i], out[i+1:]...)
|
out = append(out[:i], out[i+1:]...)
|
||||||
i-- // Decrease the index so we can pick up where we left off.
|
i-- // Decrease the index so we can pick up where we left off.
|
||||||
}
|
}
|
||||||
@ -432,9 +432,9 @@ func (e *Event) StripAction() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
messagePrefix byte = 0x3A // ":" -- prefix or last argument
|
messagePrefix byte = ':' // Prefix or last argument.
|
||||||
prefixIdent byte = 0x21 // "!" -- username
|
prefixIdent byte = '!' // Username.
|
||||||
prefixHost byte = 0x40 // "@" -- hostname
|
prefixHost byte = '@' // Hostname.
|
||||||
)
|
)
|
||||||
|
|
||||||
// Source represents the sender of an IRC event, see RFC1459 section 2.3.1.
|
// Source represents the sender of an IRC event, see RFC1459 section 2.3.1.
|
||||||
|
46
vendor/github.com/lrstanley/girc/format.go
generated
vendored
46
vendor/github.com/lrstanley/girc/format.go
generated
vendored
@ -12,8 +12,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
fmtOpenChar = 0x7B // {
|
fmtOpenChar = '{'
|
||||||
fmtCloseChar = 0x7D // }
|
fmtCloseChar = '}'
|
||||||
)
|
)
|
||||||
|
|
||||||
var fmtColors = map[string]int{
|
var fmtColors = map[string]int{
|
||||||
@ -113,7 +113,7 @@ func Fmt(text string) string {
|
|||||||
|
|
||||||
if last > -1 {
|
if last > -1 {
|
||||||
// A-Z, a-z, and ","
|
// A-Z, a-z, and ","
|
||||||
if text[i] != 0x2c && (text[i] <= 0x41 || text[i] >= 0x5a) && (text[i] <= 0x61 || text[i] >= 0x7a) {
|
if text[i] != ',' && (text[i] <= 'A' || text[i] >= 'Z') && (text[i] <= 'a' || text[i] >= 'z') {
|
||||||
last = -1
|
last = -1
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@ -127,10 +127,10 @@ func Fmt(text string) string {
|
|||||||
// See Fmt() for more information.
|
// See Fmt() for more information.
|
||||||
func TrimFmt(text string) string {
|
func TrimFmt(text string) string {
|
||||||
for color := range fmtColors {
|
for color := range fmtColors {
|
||||||
text = strings.Replace(text, "{"+color+"}", "", -1)
|
text = strings.Replace(text, string(fmtOpenChar)+color+string(fmtCloseChar), "", -1)
|
||||||
}
|
}
|
||||||
for code := range fmtCodes {
|
for code := range fmtCodes {
|
||||||
text = strings.Replace(text, "{"+code+"}", "", -1)
|
text = strings.Replace(text, string(fmtOpenChar)+code+string(fmtCloseChar), "", -1)
|
||||||
}
|
}
|
||||||
|
|
||||||
return text
|
return text
|
||||||
@ -175,9 +175,10 @@ func IsValidChannel(channel string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// #, +, !<channelid>, or &
|
// #, +, !<channelid>, ~, or &
|
||||||
// Including "*" in the prefix list, as this is commonly used (e.g. ZNC)
|
// Including "*" and "~" in the prefix list, as these are commonly used
|
||||||
if bytes.IndexByte([]byte{0x21, 0x23, 0x26, 0x2A, 0x2B}, channel[0]) == -1 {
|
// (e.g. ZNC.)
|
||||||
|
if bytes.IndexByte([]byte{'!', '#', '&', '*', '~', '+'}, channel[0]) == -1 {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -186,14 +187,14 @@ func IsValidChannel(channel string) bool {
|
|||||||
// 1 (prefix) + 5 (id) + 1 (+, channel name)
|
// 1 (prefix) + 5 (id) + 1 (+, channel name)
|
||||||
// On some networks, this may be extended with ISUPPORT capabilities,
|
// On some networks, this may be extended with ISUPPORT capabilities,
|
||||||
// however this is extremely uncommon.
|
// however this is extremely uncommon.
|
||||||
if channel[0] == 0x21 {
|
if channel[0] == '!' {
|
||||||
if len(channel) < 7 {
|
if len(channel) < 7 {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// check for valid ID
|
// check for valid ID
|
||||||
for i := 1; i < 6; i++ {
|
for i := 1; i < 6; i++ {
|
||||||
if (channel[i] < 0x30 || channel[i] > 0x39) && (channel[i] < 0x41 || channel[i] > 0x5A) {
|
if (channel[i] < '0' || channel[i] > '9') && (channel[i] < 'A' || channel[i] > 'Z') {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -222,17 +223,15 @@ func IsValidNick(nick string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
nick = ToRFC1459(nick)
|
|
||||||
|
|
||||||
// Check the first index. Some characters aren't allowed for the first
|
// Check the first index. Some characters aren't allowed for the first
|
||||||
// index of an IRC nickname.
|
// index of an IRC nickname.
|
||||||
if nick[0] < 0x41 || nick[0] > 0x7D {
|
if (nick[0] < 'A' || nick[0] > '}') && nick[0] != '?' {
|
||||||
// a-z, A-Z, and _\[]{}^|
|
// a-z, A-Z, '_\[]{}^|', and '?' in the case of znc.
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
for i := 1; i < len(nick); i++ {
|
for i := 1; i < len(nick); i++ {
|
||||||
if (nick[i] < 0x41 || nick[i] > 0x7D) && (nick[i] < 0x30 || nick[i] > 0x39) && nick[i] != 0x2D {
|
if (nick[i] < 'A' || nick[i] > '}') && (nick[i] < '0' || nick[i] > '9') && nick[i] != '-' {
|
||||||
// a-z, A-Z, 0-9, -, and _\[]{}^|
|
// a-z, A-Z, 0-9, -, and _\[]{}^|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
@ -261,10 +260,8 @@ func IsValidUser(name string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
name = ToRFC1459(name)
|
|
||||||
|
|
||||||
// "~" is prepended (commonly) if there was no ident server response.
|
// "~" is prepended (commonly) if there was no ident server response.
|
||||||
if name[0] == 0x7E {
|
if name[0] == '~' {
|
||||||
// Means name only contained "~".
|
// Means name only contained "~".
|
||||||
if len(name) < 2 {
|
if len(name) < 2 {
|
||||||
return false
|
return false
|
||||||
@ -274,12 +271,12 @@ func IsValidUser(name string) bool {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Check to see if the first index is alphanumeric.
|
// Check to see if the first index is alphanumeric.
|
||||||
if (name[0] < 0x41 || name[0] > 0x4A) && (name[0] < 0x61 || name[0] > 0x7A) && (name[0] < 0x30 || name[0] > 0x39) {
|
if (name[0] < 'A' || name[0] > 'J') && (name[0] < 'a' || name[0] > 'z') && (name[0] < '0' || name[0] > '9') {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
for i := 1; i < len(name); i++ {
|
for i := 1; i < len(name); i++ {
|
||||||
if (name[i] < 0x41 || name[i] > 0x7D) && (name[i] < 0x30 || name[i] > 0x39) && name[i] != 0x2D && name[i] != 0x2E {
|
if (name[i] < 'A' || name[i] > '}') && (name[i] < '0' || name[i] > '9') && name[i] != '-' && name[i] != '.' {
|
||||||
// a-z, A-Z, 0-9, -, and _\[]{}^|
|
// a-z, A-Z, 0-9, -, and _\[]{}^|
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
@ -290,8 +287,13 @@ func IsValidUser(name string) bool {
|
|||||||
|
|
||||||
// ToRFC1459 converts a string to the stripped down conversion within RFC
|
// ToRFC1459 converts a string to the stripped down conversion within RFC
|
||||||
// 1459. This will do things like replace an "A" with an "a", "[]" with "{}",
|
// 1459. This will do things like replace an "A" with an "a", "[]" with "{}",
|
||||||
// and so forth. Useful to compare two nicknames or channels.
|
// and so forth. Useful to compare two nicknames or channels. Note that this
|
||||||
func ToRFC1459(input string) (out string) {
|
// should not be used to normalize nicknames or similar, as this may convert
|
||||||
|
// valid input characters to non-rfc-valid characters. As such, it's main use
|
||||||
|
// is for comparing two nicks.
|
||||||
|
func ToRFC1459(input string) string {
|
||||||
|
var out string
|
||||||
|
|
||||||
for i := 0; i < len(input); i++ {
|
for i := 0; i < len(input); i++ {
|
||||||
if input[i] >= 65 && input[i] <= 94 {
|
if input[i] >= 65 && input[i] <= 94 {
|
||||||
out += string(rune(input[i]) + 32)
|
out += string(rune(input[i]) + 32)
|
||||||
|
131
vendor/github.com/lrstanley/girc/handler.go
generated
vendored
131
vendor/github.com/lrstanley/girc/handler.go
generated
vendored
@ -29,11 +29,12 @@ func (c *Client) RunHandlers(event *Event) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Regular wildcard handlers.
|
// Background handlers first.
|
||||||
c.Handlers.exec(ALL_EVENTS, c, event.Copy())
|
c.Handlers.exec(ALL_EVENTS, true, c, event.Copy())
|
||||||
|
c.Handlers.exec(event.Command, true, c, event.Copy())
|
||||||
|
|
||||||
// Then regular handlers.
|
c.Handlers.exec(ALL_EVENTS, false, c, event.Copy())
|
||||||
c.Handlers.exec(event.Command, c, event.Copy())
|
c.Handlers.exec(event.Command, false, c, event.Copy())
|
||||||
|
|
||||||
// Check if it's a CTCP.
|
// Check if it's a CTCP.
|
||||||
if ctcp := decodeCTCP(event.Copy()); ctcp != nil {
|
if ctcp := decodeCTCP(event.Copy()); ctcp != nil {
|
||||||
@ -144,7 +145,7 @@ func (c *Caller) cuid(cmd string, n int) (cuid, uid string) {
|
|||||||
// cuidToID allows easy mapping between a generated cuid and the caller
|
// cuidToID allows easy mapping between a generated cuid and the caller
|
||||||
// external/internal handler maps.
|
// external/internal handler maps.
|
||||||
func (c *Caller) cuidToID(input string) (cmd, uid string) {
|
func (c *Caller) cuidToID(input string) (cmd, uid string) {
|
||||||
i := strings.IndexByte(input, 0x3A)
|
i := strings.IndexByte(input, ':')
|
||||||
if i < 0 {
|
if i < 0 {
|
||||||
return "", ""
|
return "", ""
|
||||||
}
|
}
|
||||||
@ -160,9 +161,9 @@ type execStack struct {
|
|||||||
// exec executes all handlers pertaining to specified event. Internal first,
|
// exec executes all handlers pertaining to specified event. Internal first,
|
||||||
// then external.
|
// then external.
|
||||||
//
|
//
|
||||||
// Please note that there is no specific order/priority for which the
|
// Please note that there is no specific order/priority for which the handlers
|
||||||
// handler types themselves or the handlers are executed.
|
// are executed.
|
||||||
func (c *Caller) exec(command string, client *Client, event *Event) {
|
func (c *Caller) exec(command string, bg bool, client *Client, event *Event) {
|
||||||
// Build a stack of handlers which can be executed concurrently.
|
// Build a stack of handlers which can be executed concurrently.
|
||||||
var stack []execStack
|
var stack []execStack
|
||||||
|
|
||||||
@ -170,13 +171,21 @@ func (c *Caller) exec(command string, client *Client, event *Event) {
|
|||||||
// Get internal handlers first.
|
// Get internal handlers first.
|
||||||
if _, ok := c.internal[command]; ok {
|
if _, ok := c.internal[command]; ok {
|
||||||
for cuid := range c.internal[command] {
|
for cuid := range c.internal[command] {
|
||||||
|
if (strings.HasSuffix(cuid, ":bg") && !bg) || (!strings.HasSuffix(cuid, ":bg") && bg) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
stack = append(stack, execStack{c.internal[command][cuid], cuid})
|
stack = append(stack, execStack{c.internal[command][cuid], cuid})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Aaand then external handlers.
|
// Then external handlers.
|
||||||
if _, ok := c.external[command]; ok {
|
if _, ok := c.external[command]; ok {
|
||||||
for cuid := range c.external[command] {
|
for cuid := range c.external[command] {
|
||||||
|
if (strings.HasSuffix(cuid, ":bg") && !bg) || (!strings.HasSuffix(cuid, ":bg") && bg) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
stack = append(stack, execStack{c.external[command][cuid], cuid})
|
stack = append(stack, execStack{c.external[command][cuid], cuid})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -189,18 +198,29 @@ func (c *Caller) exec(command string, client *Client, event *Event) {
|
|||||||
wg.Add(len(stack))
|
wg.Add(len(stack))
|
||||||
for i := 0; i < len(stack); i++ {
|
for i := 0; i < len(stack); i++ {
|
||||||
go func(index int) {
|
go func(index int) {
|
||||||
c.debug.Printf("executing handler %s for event %s (%d of %d)", stack[index].cuid, command, index+1, len(stack))
|
defer wg.Done()
|
||||||
|
c.debug.Printf("[%d/%d] exec %s => %s", index+1, len(stack), stack[index].cuid, command)
|
||||||
start := time.Now()
|
start := time.Now()
|
||||||
|
|
||||||
// If they want to catch any panics, add to defer stack.
|
if bg {
|
||||||
|
go func() {
|
||||||
|
if client.Config.RecoverFunc != nil {
|
||||||
|
defer recoverHandlerPanic(client, event, stack[index].cuid, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
stack[index].Execute(client, *event)
|
||||||
|
c.debug.Printf("[%d/%d] done %s == %s", index+1, len(stack), stack[index].cuid, time.Since(start))
|
||||||
|
}()
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
if client.Config.RecoverFunc != nil {
|
if client.Config.RecoverFunc != nil {
|
||||||
defer recoverHandlerPanic(client, event, stack[index].cuid, 3)
|
defer recoverHandlerPanic(client, event, stack[index].cuid, 3)
|
||||||
}
|
}
|
||||||
|
|
||||||
stack[index].Execute(client, *event)
|
stack[index].Execute(client, *event)
|
||||||
|
c.debug.Printf("[%d/%d] done %s == %s", index+1, len(stack), stack[index].cuid, time.Since(start))
|
||||||
c.debug.Printf("execution of %s took %s (%d of %d)", stack[index].cuid, time.Since(start), index+1, len(stack))
|
|
||||||
wg.Done()
|
|
||||||
}(i)
|
}(i)
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -281,9 +301,9 @@ func (c *Caller) remove(cuid string) (success bool) {
|
|||||||
|
|
||||||
// sregister is much like Caller.register(), except that it safely locks
|
// sregister is much like Caller.register(), except that it safely locks
|
||||||
// the Caller mutex.
|
// the Caller mutex.
|
||||||
func (c *Caller) sregister(internal bool, cmd string, handler Handler) (cuid string) {
|
func (c *Caller) sregister(internal, bg bool, cmd string, handler Handler) (cuid string) {
|
||||||
c.mu.Lock()
|
c.mu.Lock()
|
||||||
cuid = c.register(internal, cmd, handler)
|
cuid = c.register(internal, bg, cmd, handler)
|
||||||
c.mu.Unlock()
|
c.mu.Unlock()
|
||||||
|
|
||||||
return cuid
|
return cuid
|
||||||
@ -291,30 +311,34 @@ func (c *Caller) sregister(internal bool, cmd string, handler Handler) (cuid str
|
|||||||
|
|
||||||
// register will register a handler in the internal tracker. Unsafe (you
|
// register will register a handler in the internal tracker. Unsafe (you
|
||||||
// must lock c.mu yourself!)
|
// must lock c.mu yourself!)
|
||||||
func (c *Caller) register(internal bool, cmd string, handler Handler) (cuid string) {
|
func (c *Caller) register(internal, bg bool, cmd string, handler Handler) (cuid string) {
|
||||||
var uid string
|
var uid string
|
||||||
|
|
||||||
cmd = strings.ToUpper(cmd)
|
cmd = strings.ToUpper(cmd)
|
||||||
|
|
||||||
|
cuid, uid = c.cuid(cmd, 20)
|
||||||
|
if bg {
|
||||||
|
uid += ":bg"
|
||||||
|
cuid += ":bg"
|
||||||
|
}
|
||||||
|
|
||||||
if internal {
|
if internal {
|
||||||
if _, ok := c.internal[cmd]; !ok {
|
if _, ok := c.internal[cmd]; !ok {
|
||||||
c.internal[cmd] = map[string]Handler{}
|
c.internal[cmd] = map[string]Handler{}
|
||||||
}
|
}
|
||||||
|
|
||||||
cuid, uid = c.cuid(cmd, 20)
|
|
||||||
c.internal[cmd][uid] = handler
|
c.internal[cmd][uid] = handler
|
||||||
} else {
|
} else {
|
||||||
if _, ok := c.external[cmd]; !ok {
|
if _, ok := c.external[cmd]; !ok {
|
||||||
c.external[cmd] = map[string]Handler{}
|
c.external[cmd] = map[string]Handler{}
|
||||||
}
|
}
|
||||||
|
|
||||||
cuid, uid = c.cuid(cmd, 20)
|
|
||||||
c.external[cmd][uid] = handler
|
c.external[cmd][uid] = handler
|
||||||
}
|
}
|
||||||
|
|
||||||
_, file, line, _ := runtime.Caller(3)
|
_, file, line, _ := runtime.Caller(3)
|
||||||
|
|
||||||
c.debug.Printf("registering handler for %q with cuid %q (internal: %t) from: %s:%d", cmd, cuid, internal, file, line)
|
c.debug.Printf("reg %q => %s [int:%t bg:%t] %s:%d", uid, cmd, internal, bg, file, line)
|
||||||
|
|
||||||
return cuid
|
return cuid
|
||||||
}
|
}
|
||||||
@ -323,31 +347,20 @@ func (c *Caller) register(internal bool, cmd string, handler Handler) (cuid stri
|
|||||||
// given event. cuid is the handler uid which can be used to remove the
|
// given event. cuid is the handler uid which can be used to remove the
|
||||||
// handler with Caller.Remove().
|
// handler with Caller.Remove().
|
||||||
func (c *Caller) AddHandler(cmd string, handler Handler) (cuid string) {
|
func (c *Caller) AddHandler(cmd string, handler Handler) (cuid string) {
|
||||||
return c.sregister(false, cmd, handler)
|
return c.sregister(false, false, cmd, handler)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Add registers the handler function for the given event. cuid is the
|
// Add registers the handler function for the given event. cuid is the
|
||||||
// handler uid which can be used to remove the handler with Caller.Remove().
|
// handler uid which can be used to remove the handler with Caller.Remove().
|
||||||
func (c *Caller) Add(cmd string, handler func(client *Client, event Event)) (cuid string) {
|
func (c *Caller) Add(cmd string, handler func(client *Client, event Event)) (cuid string) {
|
||||||
return c.sregister(false, cmd, HandlerFunc(handler))
|
return c.sregister(false, false, cmd, HandlerFunc(handler))
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddBg registers the handler function for the given event and executes it
|
// AddBg registers the handler function for the given event and executes it
|
||||||
// in a go-routine. cuid is the handler uid which can be used to remove the
|
// in a go-routine. cuid is the handler uid which can be used to remove the
|
||||||
// handler with Caller.Remove().
|
// handler with Caller.Remove().
|
||||||
func (c *Caller) AddBg(cmd string, handler func(client *Client, event Event)) (cuid string) {
|
func (c *Caller) AddBg(cmd string, handler func(client *Client, event Event)) (cuid string) {
|
||||||
return c.sregister(false, cmd, HandlerFunc(func(client *Client, event Event) {
|
return c.sregister(false, true, cmd, HandlerFunc(handler))
|
||||||
// Setting up background-based handlers this way allows us to get
|
|
||||||
// clean call stacks for use with panic recovery.
|
|
||||||
go func() {
|
|
||||||
// If they want to catch any panics, add to defer stack.
|
|
||||||
if client.Config.RecoverFunc != nil {
|
|
||||||
defer recoverHandlerPanic(client, &event, "goroutine", 3)
|
|
||||||
}
|
|
||||||
|
|
||||||
handler(client, event)
|
|
||||||
}()
|
|
||||||
}))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// AddTmp adds a "temporary" handler, which is good for one-time or few-time
|
// AddTmp adds a "temporary" handler, which is good for one-time or few-time
|
||||||
@ -361,47 +374,37 @@ func (c *Caller) AddBg(cmd string, handler func(client *Client, event Event)) (c
|
|||||||
//
|
//
|
||||||
// Additionally, AddTmp has a useful option, deadline. When set to greater
|
// Additionally, AddTmp has a useful option, deadline. When set to greater
|
||||||
// than 0, deadline will be the amount of time that passes before the handler
|
// than 0, deadline will be the amount of time that passes before the handler
|
||||||
// is removed from the stack, regardless if the handler returns true or not.
|
// is removed from the stack, regardless of if the handler returns true or not.
|
||||||
// This is useful in that it ensures that the handler is cleaned up if the
|
// This is useful in that it ensures that the handler is cleaned up if the
|
||||||
// server does not respond appropriately, or takes too long to respond.
|
// server does not respond appropriately, or takes too long to respond.
|
||||||
//
|
//
|
||||||
// Note that handlers supplied with AddTmp are executed in a goroutine to
|
// Note that handlers supplied with AddTmp are executed in a goroutine to
|
||||||
// ensure that they are not blocking other handlers. Additionally, use cuid
|
// ensure that they are not blocking other handlers. However, if you are
|
||||||
// with Caller.Remove() to prematurely remove the handler from the stack,
|
// creating a temporary handler from another handler, it should be a
|
||||||
// bypassing the timeout or waiting for the handler to return that it wants
|
// background handler.
|
||||||
// to be removed from the stack.
|
//
|
||||||
|
// Use cuid with Caller.Remove() to prematurely remove the handler from the
|
||||||
|
// stack, bypassing the timeout or waiting for the handler to return that it
|
||||||
|
// wants to be removed from the stack.
|
||||||
func (c *Caller) AddTmp(cmd string, deadline time.Duration, handler func(client *Client, event Event) bool) (cuid string, done chan struct{}) {
|
func (c *Caller) AddTmp(cmd string, deadline time.Duration, handler func(client *Client, event Event) bool) (cuid string, done chan struct{}) {
|
||||||
var uid string
|
|
||||||
cuid, uid = c.cuid(cmd, 20)
|
|
||||||
|
|
||||||
done = make(chan struct{})
|
done = make(chan struct{})
|
||||||
|
|
||||||
c.mu.Lock()
|
cuid = c.sregister(false, true, cmd, HandlerFunc(func(client *Client, event Event) {
|
||||||
if _, ok := c.external[cmd]; !ok {
|
remove := handler(client, event)
|
||||||
c.external[cmd] = map[string]Handler{}
|
if remove {
|
||||||
}
|
if ok := c.Remove(cuid); ok {
|
||||||
c.external[cmd][uid] = HandlerFunc(func(client *Client, event Event) {
|
close(done)
|
||||||
// Setting up background-based handlers this way allows us to get
|
|
||||||
// clean call stacks for use with panic recovery.
|
|
||||||
go func() {
|
|
||||||
// If they want to catch any panics, add to defer stack.
|
|
||||||
if client.Config.RecoverFunc != nil {
|
|
||||||
defer recoverHandlerPanic(client, &event, "tmp-goroutine", 3)
|
|
||||||
}
|
}
|
||||||
|
}
|
||||||
remove := handler(client, event)
|
}))
|
||||||
if remove {
|
|
||||||
if ok := c.Remove(cuid); ok {
|
|
||||||
close(done)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
})
|
|
||||||
c.mu.Unlock()
|
|
||||||
|
|
||||||
if deadline > 0 {
|
if deadline > 0 {
|
||||||
go func() {
|
go func() {
|
||||||
<-time.After(deadline)
|
select {
|
||||||
|
case <-time.After(deadline):
|
||||||
|
case <-done:
|
||||||
|
}
|
||||||
|
|
||||||
if ok := c.Remove(cuid); ok {
|
if ok := c.Remove(cuid); ok {
|
||||||
close(done)
|
close(done)
|
||||||
}
|
}
|
||||||
|
10
vendor/github.com/lrstanley/girc/modes.go
generated
vendored
10
vendor/github.com/lrstanley/girc/modes.go
generated
vendored
@ -206,11 +206,11 @@ func (c *CModes) Parse(flags string, args []string) (out []CMode) {
|
|||||||
var argCount int
|
var argCount int
|
||||||
|
|
||||||
for i := 0; i < len(flags); i++ {
|
for i := 0; i < len(flags); i++ {
|
||||||
if flags[i] == 0x2B {
|
if flags[i] == '+' {
|
||||||
add = true
|
add = true
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if flags[i] == 0x2D {
|
if flags[i] == '-' {
|
||||||
add = false
|
add = false
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@ -265,7 +265,7 @@ func IsValidChannelMode(raw string) bool {
|
|||||||
|
|
||||||
for i := 0; i < len(raw); i++ {
|
for i := 0; i < len(raw); i++ {
|
||||||
// Allowed are: ",", A-Z and a-z.
|
// Allowed are: ",", A-Z and a-z.
|
||||||
if raw[i] != 0x2C && (raw[i] < 0x41 || raw[i] > 0x5A) && (raw[i] < 0x61 || raw[i] > 0x7A) {
|
if raw[i] != ',' && (raw[i] < 'A' || raw[i] > 'Z') && (raw[i] < 'a' || raw[i] > 'z') {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -279,7 +279,7 @@ func isValidUserPrefix(raw string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
if raw[0] != 0x28 { // (.
|
if raw[0] != '(' {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -288,7 +288,7 @@ func isValidUserPrefix(raw string) bool {
|
|||||||
|
|
||||||
// Skip the first one as we know it's (.
|
// Skip the first one as we know it's (.
|
||||||
for i := 1; i < len(raw); i++ {
|
for i := 1; i < len(raw); i++ {
|
||||||
if raw[i] == 0x29 { // ).
|
if raw[i] == ')' {
|
||||||
passedKeys = true
|
passedKeys = true
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
90
vendor/github.com/matrix-org/gomatrix/client.go
generated
vendored
90
vendor/github.com/matrix-org/gomatrix/client.go
generated
vendored
@ -79,7 +79,7 @@ func (cli *Client) BuildBaseURL(urlPath ...string) string {
|
|||||||
return hsURL.String()
|
return hsURL.String()
|
||||||
}
|
}
|
||||||
|
|
||||||
// BuildURLWithQuery builds a URL with query paramters in addition to the Client's homeserver/prefix/access_token set already.
|
// BuildURLWithQuery builds a URL with query parameters in addition to the Client's homeserver/prefix/access_token set already.
|
||||||
func (cli *Client) BuildURLWithQuery(urlPath []string, urlQuery map[string]string) string {
|
func (cli *Client) BuildURLWithQuery(urlPath []string, urlQuery map[string]string) string {
|
||||||
u, _ := url.Parse(cli.BuildURL(urlPath...))
|
u, _ := url.Parse(cli.BuildURL(urlPath...))
|
||||||
q := u.Query()
|
q := u.Query()
|
||||||
@ -387,6 +387,20 @@ func (cli *Client) JoinRoom(roomIDorAlias, serverName string, content interface{
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetDisplayName returns the display name of the user from the specified MXID. See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
func (cli *Client) GetDisplayName(mxid string) (resp *RespUserDisplayName, err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", mxid, "displayname")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetOwnDisplayName returns the user's display name. See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
func (cli *Client) GetOwnDisplayName() (resp *RespUserDisplayName, err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", cli.UserID, "displayname")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// SetDisplayName sets the user's profile display name. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-profile-userid-displayname
|
// SetDisplayName sets the user's profile display name. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-profile-userid-displayname
|
||||||
func (cli *Client) SetDisplayName(displayName string) (err error) {
|
func (cli *Client) SetDisplayName(displayName string) (err error) {
|
||||||
urlPath := cli.BuildURL("profile", cli.UserID, "displayname")
|
urlPath := cli.BuildURL("profile", cli.UserID, "displayname")
|
||||||
@ -450,6 +464,35 @@ func (cli *Client) SendText(roomID, text string) (*RespSendEvent, error) {
|
|||||||
TextMessage{"m.text", text})
|
TextMessage{"m.text", text})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SendImage sends an m.room.message event into the given room with a msgtype of m.image
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#m-image
|
||||||
|
func (cli *Client) SendImage(roomID, body, url string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
ImageMessage{
|
||||||
|
MsgType: "m.image",
|
||||||
|
Body: body,
|
||||||
|
URL: url,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendVideo sends an m.room.message event into the given room with a msgtype of m.video
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
func (cli *Client) SendVideo(roomID, body, url string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
VideoMessage{
|
||||||
|
MsgType: "m.video",
|
||||||
|
Body: body,
|
||||||
|
URL: url,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendNotice sends an m.room.message event into the given room with a msgtype of m.notice
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#m-notice
|
||||||
|
func (cli *Client) SendNotice(roomID, text string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
TextMessage{"m.notice", text})
|
||||||
|
}
|
||||||
|
|
||||||
// RedactEvent redacts the given event. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-redact-eventid-txnid
|
// RedactEvent redacts the given event. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-redact-eventid-txnid
|
||||||
func (cli *Client) RedactEvent(roomID, eventID string, req *ReqRedact) (resp *RespSendEvent, err error) {
|
func (cli *Client) RedactEvent(roomID, eventID string, req *ReqRedact) (resp *RespSendEvent, err error) {
|
||||||
txnID := txnID()
|
txnID := txnID()
|
||||||
@ -518,6 +561,14 @@ func (cli *Client) UnbanUser(roomID string, req *ReqUnbanUser) (resp *RespUnbanU
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// UserTyping sets the typing status of the user. See https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
func (cli *Client) UserTyping(roomID string, typing bool, timeout int64) (resp *RespTyping, err error) {
|
||||||
|
req := ReqTyping{Typing: typing, Timeout: timeout}
|
||||||
|
u := cli.BuildURL("rooms", roomID, "typing", cli.UserID)
|
||||||
|
_, err = cli.MakeRequest("PUT", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// StateEvent gets a single state event in a room. It will attempt to JSON unmarshal into the given "outContent" struct with
|
// StateEvent gets a single state event in a room. It will attempt to JSON unmarshal into the given "outContent" struct with
|
||||||
// the HTTP response body, or return an error.
|
// the HTTP response body, or return an error.
|
||||||
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-state-eventtype-statekey
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-state-eventtype-statekey
|
||||||
@ -556,8 +607,15 @@ func (cli *Client) UploadToContentRepo(content io.Reader, contentType string, co
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if res.StatusCode != 200 {
|
if res.StatusCode != 200 {
|
||||||
|
contents, err := ioutil.ReadAll(res.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, HTTPError{
|
||||||
|
Message: "Upload request failed - Failed to read response body: " + err.Error(),
|
||||||
|
Code: res.StatusCode,
|
||||||
|
}
|
||||||
|
}
|
||||||
return nil, HTTPError{
|
return nil, HTTPError{
|
||||||
Message: "Upload request failed",
|
Message: "Upload request failed: " + string(contents),
|
||||||
Code: res.StatusCode,
|
Code: res.StatusCode,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -588,6 +646,34 @@ func (cli *Client) JoinedRooms() (resp *RespJoinedRooms, err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Messages returns a list of message and state events for a room. It uses
|
||||||
|
// pagination query parameters to paginate history in the room.
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-messages
|
||||||
|
func (cli *Client) Messages(roomID, from, to string, dir rune, limit int) (resp *RespMessages, err error) {
|
||||||
|
query := map[string]string{
|
||||||
|
"from": from,
|
||||||
|
"dir": string(dir),
|
||||||
|
}
|
||||||
|
if to != "" {
|
||||||
|
query["to"] = to
|
||||||
|
}
|
||||||
|
if limit != 0 {
|
||||||
|
query["limit"] = strconv.Itoa(limit)
|
||||||
|
}
|
||||||
|
|
||||||
|
urlPath := cli.BuildURLWithQuery([]string{"rooms", roomID, "messages"}, query)
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TurnServer returns turn server details and credentials for the client to use when initiating calls.
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-voip-turnserver
|
||||||
|
func (cli *Client) TurnServer() (resp *RespTurnServer, err error) {
|
||||||
|
urlPath := cli.BuildURL("voip", "turnServer")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
func txnID() string {
|
func txnID() string {
|
||||||
return "go" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
return "go" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
||||||
}
|
}
|
||||||
|
43
vendor/github.com/matrix-org/gomatrix/events.go
generated
vendored
43
vendor/github.com/matrix-org/gomatrix/events.go
generated
vendored
@ -7,13 +7,13 @@ import (
|
|||||||
|
|
||||||
// Event represents a single Matrix event.
|
// Event represents a single Matrix event.
|
||||||
type Event struct {
|
type Event struct {
|
||||||
StateKey string `json:"state_key"` // The state key for the event. Only present on State Events.
|
StateKey *string `json:"state_key,omitempty"` // The state key for the event. Only present on State Events.
|
||||||
Sender string `json:"sender"` // The user ID of the sender of the event
|
Sender string `json:"sender"` // The user ID of the sender of the event
|
||||||
Type string `json:"type"` // The event type
|
Type string `json:"type"` // The event type
|
||||||
Timestamp int `json:"origin_server_ts"` // The unix timestamp when this message was sent by the origin server
|
Timestamp int64 `json:"origin_server_ts"` // The unix timestamp when this message was sent by the origin server
|
||||||
ID string `json:"event_id"` // The unique ID of this event
|
ID string `json:"event_id"` // The unique ID of this event
|
||||||
RoomID string `json:"room_id"` // The room the event was sent to. May be nil (e.g. for presence)
|
RoomID string `json:"room_id"` // The room the event was sent to. May be nil (e.g. for presence)
|
||||||
Content map[string]interface{} `json:"content"` // The JSON content of the event.
|
Content map[string]interface{} `json:"content"` // The JSON content of the event.
|
||||||
}
|
}
|
||||||
|
|
||||||
// Body returns the value of the "body" key in the event content if it is
|
// Body returns the value of the "body" key in the event content if it is
|
||||||
@ -44,12 +44,31 @@ type TextMessage struct {
|
|||||||
Body string `json:"body"`
|
Body string `json:"body"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// ImageInfo contains info about an image
|
// ImageInfo contains info about an image - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-image
|
||||||
type ImageInfo struct {
|
type ImageInfo struct {
|
||||||
Height uint `json:"h"`
|
Height uint `json:"h,omitempty"`
|
||||||
Width uint `json:"w"`
|
Width uint `json:"w,omitempty"`
|
||||||
Mimetype string `json:"mimetype"`
|
Mimetype string `json:"mimetype,omitempty"`
|
||||||
Size uint `json:"size"`
|
Size uint `json:"size,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// VideoInfo contains info about a video - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
type VideoInfo struct {
|
||||||
|
Mimetype string `json:"mimetype,omitempty"`
|
||||||
|
ThumbnailInfo ImageInfo `json:"thumbnail_info"`
|
||||||
|
ThumbnailURL string `json:"thumbnail_url,omitempty"`
|
||||||
|
Height uint `json:"h,omitempty"`
|
||||||
|
Width uint `json:"w,omitempty"`
|
||||||
|
Duration uint `json:"duration,omitempty"`
|
||||||
|
Size uint `json:"size,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// VideoMessage is an m.video - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
type VideoMessage struct {
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
Body string `json:"body"`
|
||||||
|
URL string `json:"url"`
|
||||||
|
Info VideoInfo `json:"info"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// ImageMessage is an m.image event
|
// ImageMessage is an m.image event
|
||||||
|
43
vendor/github.com/matrix-org/gomatrix/filter.go
generated
vendored
Normal file
43
vendor/github.com/matrix-org/gomatrix/filter.go
generated
vendored
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
// Copyright 2017 Jan Christian Grünhage
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package gomatrix
|
||||||
|
|
||||||
|
//Filter is used by clients to specify how the server should filter responses to e.g. sync requests
|
||||||
|
//Specified by: https://matrix.org/docs/spec/client_server/r0.2.0.html#filtering
|
||||||
|
type Filter struct {
|
||||||
|
AccountData FilterPart `json:"account_data,omitempty"`
|
||||||
|
EventFields []string `json:"event_fields,omitempty"`
|
||||||
|
EventFormat string `json:"event_format,omitempty"`
|
||||||
|
Presence FilterPart `json:"presence,omitempty"`
|
||||||
|
Room struct {
|
||||||
|
AccountData FilterPart `json:"account_data,omitempty"`
|
||||||
|
Ephemeral FilterPart `json:"ephemeral,omitempty"`
|
||||||
|
IncludeLeave bool `json:"include_leave,omitempty"`
|
||||||
|
NotRooms []string `json:"not_rooms,omitempty"`
|
||||||
|
Rooms []string `json:"rooms,omitempty"`
|
||||||
|
State FilterPart `json:"state,omitempty"`
|
||||||
|
Timeline FilterPart `json:"timeline,omitempty"`
|
||||||
|
} `json:"room,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FilterPart struct {
|
||||||
|
NotRooms []string `json:"not_rooms,omitempty"`
|
||||||
|
Rooms []string `json:"rooms,omitempty"`
|
||||||
|
Limit *int `json:"limit,omitempty"`
|
||||||
|
NotSenders []string `json:"not_senders,omitempty"`
|
||||||
|
NotTypes []string `json:"not_types,omitempty"`
|
||||||
|
Senders []string `json:"senders,omitempty"`
|
||||||
|
Types []string `json:"types,omitempty"`
|
||||||
|
}
|
6
vendor/github.com/matrix-org/gomatrix/requests.go
generated
vendored
6
vendor/github.com/matrix-org/gomatrix/requests.go
generated
vendored
@ -70,3 +70,9 @@ type ReqBanUser struct {
|
|||||||
type ReqUnbanUser struct {
|
type ReqUnbanUser struct {
|
||||||
UserID string `json:"user_id"`
|
UserID string `json:"user_id"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ReqTyping is the JSON request for https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
type ReqTyping struct {
|
||||||
|
Typing bool `json:"typing"`
|
||||||
|
Timeout int64 `json:"timeout"`
|
||||||
|
}
|
||||||
|
32
vendor/github.com/matrix-org/gomatrix/responses.go
generated
vendored
32
vendor/github.com/matrix-org/gomatrix/responses.go
generated
vendored
@ -45,6 +45,9 @@ type RespBanUser struct{}
|
|||||||
// RespUnbanUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-unban
|
// RespUnbanUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-unban
|
||||||
type RespUnbanUser struct{}
|
type RespUnbanUser struct{}
|
||||||
|
|
||||||
|
// RespTyping is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
type RespTyping struct{}
|
||||||
|
|
||||||
// RespJoinedRooms is the JSON response for TODO-SPEC https://github.com/matrix-org/synapse/pull/1680
|
// RespJoinedRooms is the JSON response for TODO-SPEC https://github.com/matrix-org/synapse/pull/1680
|
||||||
type RespJoinedRooms struct {
|
type RespJoinedRooms struct {
|
||||||
JoinedRooms []string `json:"joined_rooms"`
|
JoinedRooms []string `json:"joined_rooms"`
|
||||||
@ -58,6 +61,13 @@ type RespJoinedMembers struct {
|
|||||||
} `json:"joined"`
|
} `json:"joined"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RespMessages is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-messages
|
||||||
|
type RespMessages struct {
|
||||||
|
Start string `json:"start"`
|
||||||
|
Chunk []Event `json:"chunk"`
|
||||||
|
End string `json:"end"`
|
||||||
|
}
|
||||||
|
|
||||||
// RespSendEvent is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-send-eventtype-txnid
|
// RespSendEvent is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-send-eventtype-txnid
|
||||||
type RespSendEvent struct {
|
type RespSendEvent struct {
|
||||||
EventID string `json:"event_id"`
|
EventID string `json:"event_id"`
|
||||||
@ -90,6 +100,11 @@ func (r RespUserInteractive) HasSingleStageFlow(stageName string) bool {
|
|||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// RespUserDisplayName is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
type RespUserDisplayName struct {
|
||||||
|
DisplayName string `json:"displayname"`
|
||||||
|
}
|
||||||
|
|
||||||
// RespRegister is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
// RespRegister is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
||||||
type RespRegister struct {
|
type RespRegister struct {
|
||||||
AccessToken string `json:"access_token"`
|
AccessToken string `json:"access_token"`
|
||||||
@ -125,6 +140,16 @@ type RespSync struct {
|
|||||||
Events []Event `json:"events"`
|
Events []Event `json:"events"`
|
||||||
} `json:"presence"`
|
} `json:"presence"`
|
||||||
Rooms struct {
|
Rooms struct {
|
||||||
|
Leave map[string]struct {
|
||||||
|
State struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
} `json:"state"`
|
||||||
|
Timeline struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
Limited bool `json:"limited"`
|
||||||
|
PrevBatch string `json:"prev_batch"`
|
||||||
|
} `json:"timeline"`
|
||||||
|
} `json:"leave"`
|
||||||
Join map[string]struct {
|
Join map[string]struct {
|
||||||
State struct {
|
State struct {
|
||||||
Events []Event `json:"events"`
|
Events []Event `json:"events"`
|
||||||
@ -142,3 +167,10 @@ type RespSync struct {
|
|||||||
} `json:"invite"`
|
} `json:"invite"`
|
||||||
} `json:"rooms"`
|
} `json:"rooms"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type RespTurnServer struct {
|
||||||
|
Username string `json:"username"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
TTL int `json:"ttl"`
|
||||||
|
URIs []string `json:"uris"`
|
||||||
|
}
|
||||||
|
2
vendor/github.com/matrix-org/gomatrix/room.go
generated
vendored
2
vendor/github.com/matrix-org/gomatrix/room.go
generated
vendored
@ -13,7 +13,7 @@ func (room Room) UpdateState(event *Event) {
|
|||||||
if !exists {
|
if !exists {
|
||||||
room.State[event.Type] = make(map[string]*Event)
|
room.State[event.Type] = make(map[string]*Event)
|
||||||
}
|
}
|
||||||
room.State[event.Type][event.StateKey] = event
|
room.State[event.Type][*event.StateKey] = event
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetStateEvent returns the state event for the given type/state_key combo, or nil.
|
// GetStateEvent returns the state event for the given type/state_key combo, or nil.
|
||||||
|
12
vendor/github.com/matrix-org/gomatrix/sync.go
generated
vendored
12
vendor/github.com/matrix-org/gomatrix/sync.go
generated
vendored
@ -73,6 +73,16 @@ func (s *DefaultSyncer) ProcessResponse(res *RespSync, since string) (err error)
|
|||||||
s.notifyListeners(&event)
|
s.notifyListeners(&event)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
for roomID, roomData := range res.Rooms.Leave {
|
||||||
|
room := s.getOrCreateRoom(roomID)
|
||||||
|
for _, event := range roomData.Timeline.Events {
|
||||||
|
if event.StateKey != nil {
|
||||||
|
event.RoomID = roomID
|
||||||
|
room.UpdateState(&event)
|
||||||
|
s.notifyListeners(&event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -102,7 +112,7 @@ func (s *DefaultSyncer) shouldProcessResponse(resp *RespSync, since string) bool
|
|||||||
for roomID, roomData := range resp.Rooms.Join {
|
for roomID, roomData := range resp.Rooms.Join {
|
||||||
for i := len(roomData.Timeline.Events) - 1; i >= 0; i-- {
|
for i := len(roomData.Timeline.Events) - 1; i >= 0; i-- {
|
||||||
e := roomData.Timeline.Events[i]
|
e := roomData.Timeline.Events[i]
|
||||||
if e.Type == "m.room.member" && e.StateKey == s.UserID {
|
if e.Type == "m.room.member" && e.StateKey != nil && *e.StateKey == s.UserID {
|
||||||
m := e.Content["membership"]
|
m := e.Content["membership"]
|
||||||
mship, ok := m.(string)
|
mship, ok := m.(string)
|
||||||
if !ok {
|
if !ok {
|
||||||
|
201
vendor/github.com/matterbridge/gomatrix/LICENSE
generated
vendored
Normal file
201
vendor/github.com/matterbridge/gomatrix/LICENSE
generated
vendored
Normal file
@ -0,0 +1,201 @@
|
|||||||
|
Apache License
|
||||||
|
Version 2.0, January 2004
|
||||||
|
http://www.apache.org/licenses/
|
||||||
|
|
||||||
|
TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
|
||||||
|
|
||||||
|
1. Definitions.
|
||||||
|
|
||||||
|
"License" shall mean the terms and conditions for use, reproduction,
|
||||||
|
and distribution as defined by Sections 1 through 9 of this document.
|
||||||
|
|
||||||
|
"Licensor" shall mean the copyright owner or entity authorized by
|
||||||
|
the copyright owner that is granting the License.
|
||||||
|
|
||||||
|
"Legal Entity" shall mean the union of the acting entity and all
|
||||||
|
other entities that control, are controlled by, or are under common
|
||||||
|
control with that entity. For the purposes of this definition,
|
||||||
|
"control" means (i) the power, direct or indirect, to cause the
|
||||||
|
direction or management of such entity, whether by contract or
|
||||||
|
otherwise, or (ii) ownership of fifty percent (50%) or more of the
|
||||||
|
outstanding shares, or (iii) beneficial ownership of such entity.
|
||||||
|
|
||||||
|
"You" (or "Your") shall mean an individual or Legal Entity
|
||||||
|
exercising permissions granted by this License.
|
||||||
|
|
||||||
|
"Source" form shall mean the preferred form for making modifications,
|
||||||
|
including but not limited to software source code, documentation
|
||||||
|
source, and configuration files.
|
||||||
|
|
||||||
|
"Object" form shall mean any form resulting from mechanical
|
||||||
|
transformation or translation of a Source form, including but
|
||||||
|
not limited to compiled object code, generated documentation,
|
||||||
|
and conversions to other media types.
|
||||||
|
|
||||||
|
"Work" shall mean the work of authorship, whether in Source or
|
||||||
|
Object form, made available under the License, as indicated by a
|
||||||
|
copyright notice that is included in or attached to the work
|
||||||
|
(an example is provided in the Appendix below).
|
||||||
|
|
||||||
|
"Derivative Works" shall mean any work, whether in Source or Object
|
||||||
|
form, that is based on (or derived from) the Work and for which the
|
||||||
|
editorial revisions, annotations, elaborations, or other modifications
|
||||||
|
represent, as a whole, an original work of authorship. For the purposes
|
||||||
|
of this License, Derivative Works shall not include works that remain
|
||||||
|
separable from, or merely link (or bind by name) to the interfaces of,
|
||||||
|
the Work and Derivative Works thereof.
|
||||||
|
|
||||||
|
"Contribution" shall mean any work of authorship, including
|
||||||
|
the original version of the Work and any modifications or additions
|
||||||
|
to that Work or Derivative Works thereof, that is intentionally
|
||||||
|
submitted to Licensor for inclusion in the Work by the copyright owner
|
||||||
|
or by an individual or Legal Entity authorized to submit on behalf of
|
||||||
|
the copyright owner. For the purposes of this definition, "submitted"
|
||||||
|
means any form of electronic, verbal, or written communication sent
|
||||||
|
to the Licensor or its representatives, including but not limited to
|
||||||
|
communication on electronic mailing lists, source code control systems,
|
||||||
|
and issue tracking systems that are managed by, or on behalf of, the
|
||||||
|
Licensor for the purpose of discussing and improving the Work, but
|
||||||
|
excluding communication that is conspicuously marked or otherwise
|
||||||
|
designated in writing by the copyright owner as "Not a Contribution."
|
||||||
|
|
||||||
|
"Contributor" shall mean Licensor and any individual or Legal Entity
|
||||||
|
on behalf of whom a Contribution has been received by Licensor and
|
||||||
|
subsequently incorporated within the Work.
|
||||||
|
|
||||||
|
2. Grant of Copyright License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
copyright license to reproduce, prepare Derivative Works of,
|
||||||
|
publicly display, publicly perform, sublicense, and distribute the
|
||||||
|
Work and such Derivative Works in Source or Object form.
|
||||||
|
|
||||||
|
3. Grant of Patent License. Subject to the terms and conditions of
|
||||||
|
this License, each Contributor hereby grants to You a perpetual,
|
||||||
|
worldwide, non-exclusive, no-charge, royalty-free, irrevocable
|
||||||
|
(except as stated in this section) patent license to make, have made,
|
||||||
|
use, offer to sell, sell, import, and otherwise transfer the Work,
|
||||||
|
where such license applies only to those patent claims licensable
|
||||||
|
by such Contributor that are necessarily infringed by their
|
||||||
|
Contribution(s) alone or by combination of their Contribution(s)
|
||||||
|
with the Work to which such Contribution(s) was submitted. If You
|
||||||
|
institute patent litigation against any entity (including a
|
||||||
|
cross-claim or counterclaim in a lawsuit) alleging that the Work
|
||||||
|
or a Contribution incorporated within the Work constitutes direct
|
||||||
|
or contributory patent infringement, then any patent licenses
|
||||||
|
granted to You under this License for that Work shall terminate
|
||||||
|
as of the date such litigation is filed.
|
||||||
|
|
||||||
|
4. Redistribution. You may reproduce and distribute copies of the
|
||||||
|
Work or Derivative Works thereof in any medium, with or without
|
||||||
|
modifications, and in Source or Object form, provided that You
|
||||||
|
meet the following conditions:
|
||||||
|
|
||||||
|
(a) You must give any other recipients of the Work or
|
||||||
|
Derivative Works a copy of this License; and
|
||||||
|
|
||||||
|
(b) You must cause any modified files to carry prominent notices
|
||||||
|
stating that You changed the files; and
|
||||||
|
|
||||||
|
(c) You must retain, in the Source form of any Derivative Works
|
||||||
|
that You distribute, all copyright, patent, trademark, and
|
||||||
|
attribution notices from the Source form of the Work,
|
||||||
|
excluding those notices that do not pertain to any part of
|
||||||
|
the Derivative Works; and
|
||||||
|
|
||||||
|
(d) If the Work includes a "NOTICE" text file as part of its
|
||||||
|
distribution, then any Derivative Works that You distribute must
|
||||||
|
include a readable copy of the attribution notices contained
|
||||||
|
within such NOTICE file, excluding those notices that do not
|
||||||
|
pertain to any part of the Derivative Works, in at least one
|
||||||
|
of the following places: within a NOTICE text file distributed
|
||||||
|
as part of the Derivative Works; within the Source form or
|
||||||
|
documentation, if provided along with the Derivative Works; or,
|
||||||
|
within a display generated by the Derivative Works, if and
|
||||||
|
wherever such third-party notices normally appear. The contents
|
||||||
|
of the NOTICE file are for informational purposes only and
|
||||||
|
do not modify the License. You may add Your own attribution
|
||||||
|
notices within Derivative Works that You distribute, alongside
|
||||||
|
or as an addendum to the NOTICE text from the Work, provided
|
||||||
|
that such additional attribution notices cannot be construed
|
||||||
|
as modifying the License.
|
||||||
|
|
||||||
|
You may add Your own copyright statement to Your modifications and
|
||||||
|
may provide additional or different license terms and conditions
|
||||||
|
for use, reproduction, or distribution of Your modifications, or
|
||||||
|
for any such Derivative Works as a whole, provided Your use,
|
||||||
|
reproduction, and distribution of the Work otherwise complies with
|
||||||
|
the conditions stated in this License.
|
||||||
|
|
||||||
|
5. Submission of Contributions. Unless You explicitly state otherwise,
|
||||||
|
any Contribution intentionally submitted for inclusion in the Work
|
||||||
|
by You to the Licensor shall be under the terms and conditions of
|
||||||
|
this License, without any additional terms or conditions.
|
||||||
|
Notwithstanding the above, nothing herein shall supersede or modify
|
||||||
|
the terms of any separate license agreement you may have executed
|
||||||
|
with Licensor regarding such Contributions.
|
||||||
|
|
||||||
|
6. Trademarks. This License does not grant permission to use the trade
|
||||||
|
names, trademarks, service marks, or product names of the Licensor,
|
||||||
|
except as required for reasonable and customary use in describing the
|
||||||
|
origin of the Work and reproducing the content of the NOTICE file.
|
||||||
|
|
||||||
|
7. Disclaimer of Warranty. Unless required by applicable law or
|
||||||
|
agreed to in writing, Licensor provides the Work (and each
|
||||||
|
Contributor provides its Contributions) on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
||||||
|
implied, including, without limitation, any warranties or conditions
|
||||||
|
of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
|
||||||
|
PARTICULAR PURPOSE. You are solely responsible for determining the
|
||||||
|
appropriateness of using or redistributing the Work and assume any
|
||||||
|
risks associated with Your exercise of permissions under this License.
|
||||||
|
|
||||||
|
8. Limitation of Liability. In no event and under no legal theory,
|
||||||
|
whether in tort (including negligence), contract, or otherwise,
|
||||||
|
unless required by applicable law (such as deliberate and grossly
|
||||||
|
negligent acts) or agreed to in writing, shall any Contributor be
|
||||||
|
liable to You for damages, including any direct, indirect, special,
|
||||||
|
incidental, or consequential damages of any character arising as a
|
||||||
|
result of this License or out of the use or inability to use the
|
||||||
|
Work (including but not limited to damages for loss of goodwill,
|
||||||
|
work stoppage, computer failure or malfunction, or any and all
|
||||||
|
other commercial damages or losses), even if such Contributor
|
||||||
|
has been advised of the possibility of such damages.
|
||||||
|
|
||||||
|
9. Accepting Warranty or Additional Liability. While redistributing
|
||||||
|
the Work or Derivative Works thereof, You may choose to offer,
|
||||||
|
and charge a fee for, acceptance of support, warranty, indemnity,
|
||||||
|
or other liability obligations and/or rights consistent with this
|
||||||
|
License. However, in accepting such obligations, You may act only
|
||||||
|
on Your own behalf and on Your sole responsibility, not on behalf
|
||||||
|
of any other Contributor, and only if You agree to indemnify,
|
||||||
|
defend, and hold each Contributor harmless for any liability
|
||||||
|
incurred by, or claims asserted against, such Contributor by reason
|
||||||
|
of your accepting any such warranty or additional liability.
|
||||||
|
|
||||||
|
END OF TERMS AND CONDITIONS
|
||||||
|
|
||||||
|
APPENDIX: How to apply the Apache License to your work.
|
||||||
|
|
||||||
|
To apply the Apache License to your work, attach the following
|
||||||
|
boilerplate notice, with the fields enclosed by brackets "{}"
|
||||||
|
replaced with your own identifying information. (Don't include
|
||||||
|
the brackets!) The text should be enclosed in the appropriate
|
||||||
|
comment syntax for the file format. We also recommend that a
|
||||||
|
file or class name and description of purpose be included on the
|
||||||
|
same "printed page" as the copyright notice for easier
|
||||||
|
identification within third-party archives.
|
||||||
|
|
||||||
|
Copyright {yyyy} {name of copyright owner}
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
703
vendor/github.com/matterbridge/gomatrix/client.go
generated
vendored
Normal file
703
vendor/github.com/matterbridge/gomatrix/client.go
generated
vendored
Normal file
@ -0,0 +1,703 @@
|
|||||||
|
// Package gomatrix implements the Matrix Client-Server API.
|
||||||
|
//
|
||||||
|
// Specification can be found at http://matrix.org/docs/spec/client_server/r0.2.0.html
|
||||||
|
package gomatrix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"path"
|
||||||
|
"strconv"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Client represents a Matrix client.
|
||||||
|
type Client struct {
|
||||||
|
HomeserverURL *url.URL // The base homeserver URL
|
||||||
|
Prefix string // The API prefix eg '/_matrix/client/r0'
|
||||||
|
UserID string // The user ID of the client. Used for forming HTTP paths which use the client's user ID.
|
||||||
|
AccessToken string // The access_token for the client.
|
||||||
|
Client *http.Client // The underlying HTTP client which will be used to make HTTP requests.
|
||||||
|
Syncer Syncer // The thing which can process /sync responses
|
||||||
|
Store Storer // The thing which can store rooms/tokens/ids
|
||||||
|
|
||||||
|
// The ?user_id= query parameter for application services. This must be set *prior* to calling a method. If this is empty,
|
||||||
|
// no user_id parameter will be sent.
|
||||||
|
// See http://matrix.org/docs/spec/application_service/unstable.html#identity-assertion
|
||||||
|
AppServiceUserID string
|
||||||
|
|
||||||
|
syncingMutex sync.Mutex // protects syncingID
|
||||||
|
syncingID uint32 // Identifies the current Sync. Only one Sync can be active at any given time.
|
||||||
|
}
|
||||||
|
|
||||||
|
// HTTPError An HTTP Error response, which may wrap an underlying native Go Error.
|
||||||
|
type HTTPError struct {
|
||||||
|
WrappedError error
|
||||||
|
Message string
|
||||||
|
Code int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e HTTPError) Error() string {
|
||||||
|
var wrappedErrMsg string
|
||||||
|
if e.WrappedError != nil {
|
||||||
|
wrappedErrMsg = e.WrappedError.Error()
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("msg=%s code=%d wrapped=%s", e.Message, e.Code, wrappedErrMsg)
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildURL builds a URL with the Client's homserver/prefix/access_token set already.
|
||||||
|
func (cli *Client) BuildURL(urlPath ...string) string {
|
||||||
|
ps := []string{cli.Prefix}
|
||||||
|
for _, p := range urlPath {
|
||||||
|
ps = append(ps, p)
|
||||||
|
}
|
||||||
|
return cli.BuildBaseURL(ps...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildBaseURL builds a URL with the Client's homeserver/access_token set already. You must
|
||||||
|
// supply the prefix in the path.
|
||||||
|
func (cli *Client) BuildBaseURL(urlPath ...string) string {
|
||||||
|
// copy the URL. Purposefully ignore error as the input is from a valid URL already
|
||||||
|
hsURL, _ := url.Parse(cli.HomeserverURL.String())
|
||||||
|
parts := []string{hsURL.Path}
|
||||||
|
parts = append(parts, urlPath...)
|
||||||
|
hsURL.Path = path.Join(parts...)
|
||||||
|
query := hsURL.Query()
|
||||||
|
if cli.AccessToken != "" {
|
||||||
|
query.Set("access_token", cli.AccessToken)
|
||||||
|
}
|
||||||
|
if cli.AppServiceUserID != "" {
|
||||||
|
query.Set("user_id", cli.AppServiceUserID)
|
||||||
|
}
|
||||||
|
hsURL.RawQuery = query.Encode()
|
||||||
|
return hsURL.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// BuildURLWithQuery builds a URL with query parameters in addition to the Client's homeserver/prefix/access_token set already.
|
||||||
|
func (cli *Client) BuildURLWithQuery(urlPath []string, urlQuery map[string]string) string {
|
||||||
|
u, _ := url.Parse(cli.BuildURL(urlPath...))
|
||||||
|
q := u.Query()
|
||||||
|
for k, v := range urlQuery {
|
||||||
|
q.Set(k, v)
|
||||||
|
}
|
||||||
|
u.RawQuery = q.Encode()
|
||||||
|
return u.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetCredentials sets the user ID and access token on this client instance.
|
||||||
|
func (cli *Client) SetCredentials(userID, accessToken string) {
|
||||||
|
cli.AccessToken = accessToken
|
||||||
|
cli.UserID = userID
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClearCredentials removes the user ID and access token on this client instance.
|
||||||
|
func (cli *Client) ClearCredentials() {
|
||||||
|
cli.AccessToken = ""
|
||||||
|
cli.UserID = ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sync starts syncing with the provided Homeserver. If Sync() is called twice then the first sync will be stopped and the
|
||||||
|
// error will be nil.
|
||||||
|
//
|
||||||
|
// This function will block until a fatal /sync error occurs, so it should almost always be started as a new goroutine.
|
||||||
|
// Fatal sync errors can be caused by:
|
||||||
|
// - The failure to create a filter.
|
||||||
|
// - Client.Syncer.OnFailedSync returning an error in response to a failed sync.
|
||||||
|
// - Client.Syncer.ProcessResponse returning an error.
|
||||||
|
// If you wish to continue retrying in spite of these fatal errors, call Sync() again.
|
||||||
|
func (cli *Client) Sync() error {
|
||||||
|
// Mark the client as syncing.
|
||||||
|
// We will keep syncing until the syncing state changes. Either because
|
||||||
|
// Sync is called or StopSync is called.
|
||||||
|
syncingID := cli.incrementSyncingID()
|
||||||
|
nextBatch := cli.Store.LoadNextBatch(cli.UserID)
|
||||||
|
filterID := cli.Store.LoadFilterID(cli.UserID)
|
||||||
|
if filterID == "" {
|
||||||
|
filterJSON := cli.Syncer.GetFilterJSON(cli.UserID)
|
||||||
|
resFilter, err := cli.CreateFilter(filterJSON)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
filterID = resFilter.FilterID
|
||||||
|
cli.Store.SaveFilterID(cli.UserID, filterID)
|
||||||
|
}
|
||||||
|
|
||||||
|
for {
|
||||||
|
resSync, err := cli.SyncRequest(30000, nextBatch, filterID, false, "")
|
||||||
|
if err != nil {
|
||||||
|
duration, err2 := cli.Syncer.OnFailedSync(resSync, err)
|
||||||
|
if err2 != nil {
|
||||||
|
return err2
|
||||||
|
}
|
||||||
|
time.Sleep(duration)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check that the syncing state hasn't changed
|
||||||
|
// Either because we've stopped syncing or another sync has been started.
|
||||||
|
// We discard the response from our sync.
|
||||||
|
if cli.getSyncingID() != syncingID {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Save the token now *before* processing it. This means it's possible
|
||||||
|
// to not process some events, but it means that we won't get constantly stuck processing
|
||||||
|
// a malformed/buggy event which keeps making us panic.
|
||||||
|
cli.Store.SaveNextBatch(cli.UserID, resSync.NextBatch)
|
||||||
|
if err = cli.Syncer.ProcessResponse(resSync, nextBatch); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
nextBatch = resSync.NextBatch
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cli *Client) incrementSyncingID() uint32 {
|
||||||
|
cli.syncingMutex.Lock()
|
||||||
|
defer cli.syncingMutex.Unlock()
|
||||||
|
cli.syncingID++
|
||||||
|
return cli.syncingID
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cli *Client) getSyncingID() uint32 {
|
||||||
|
cli.syncingMutex.Lock()
|
||||||
|
defer cli.syncingMutex.Unlock()
|
||||||
|
return cli.syncingID
|
||||||
|
}
|
||||||
|
|
||||||
|
// StopSync stops the ongoing sync started by Sync.
|
||||||
|
func (cli *Client) StopSync() {
|
||||||
|
// Advance the syncing state so that any running Syncs will terminate.
|
||||||
|
cli.incrementSyncingID()
|
||||||
|
}
|
||||||
|
|
||||||
|
// MakeRequest makes a JSON HTTP request to the given URL.
|
||||||
|
// If "resBody" is not nil, the response body will be json.Unmarshalled into it.
|
||||||
|
//
|
||||||
|
// Returns the HTTP body as bytes on 2xx with a nil error. Returns an error if the response is not 2xx along
|
||||||
|
// with the HTTP body bytes if it got that far. This error is an HTTPError which includes the returned
|
||||||
|
// HTTP status code and possibly a RespError as the WrappedError, if the HTTP body could be decoded as a RespError.
|
||||||
|
func (cli *Client) MakeRequest(method string, httpURL string, reqBody interface{}, resBody interface{}) ([]byte, error) {
|
||||||
|
var req *http.Request
|
||||||
|
var err error
|
||||||
|
if reqBody != nil {
|
||||||
|
var jsonStr []byte
|
||||||
|
jsonStr, err = json.Marshal(reqBody)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
req, err = http.NewRequest(method, httpURL, bytes.NewBuffer(jsonStr))
|
||||||
|
} else {
|
||||||
|
req, err = http.NewRequest(method, httpURL, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
res, err := cli.Client.Do(req)
|
||||||
|
if res != nil {
|
||||||
|
defer res.Body.Close()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
contents, err := ioutil.ReadAll(res.Body)
|
||||||
|
if res.StatusCode/100 != 2 { // not 2xx
|
||||||
|
var wrap error
|
||||||
|
var respErr RespError
|
||||||
|
if _ = json.Unmarshal(contents, &respErr); respErr.ErrCode != "" {
|
||||||
|
wrap = respErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// If we failed to decode as RespError, don't just drop the HTTP body, include it in the
|
||||||
|
// HTTP error instead (e.g proxy errors which return HTML).
|
||||||
|
msg := "Failed to " + method + " JSON to " + req.URL.Path
|
||||||
|
if wrap == nil {
|
||||||
|
msg = msg + ": " + string(contents)
|
||||||
|
}
|
||||||
|
|
||||||
|
return contents, HTTPError{
|
||||||
|
Code: res.StatusCode,
|
||||||
|
Message: msg,
|
||||||
|
WrappedError: wrap,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if resBody != nil {
|
||||||
|
if err = json.Unmarshal(contents, &resBody); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return contents, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateFilter makes an HTTP request according to http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-user-userid-filter
|
||||||
|
func (cli *Client) CreateFilter(filter json.RawMessage) (resp *RespCreateFilter, err error) {
|
||||||
|
urlPath := cli.BuildURL("user", cli.UserID, "filter")
|
||||||
|
_, err = cli.MakeRequest("POST", urlPath, &filter, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// SyncRequest makes an HTTP request according to http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-sync
|
||||||
|
func (cli *Client) SyncRequest(timeout int, since, filterID string, fullState bool, setPresence string) (resp *RespSync, err error) {
|
||||||
|
query := map[string]string{
|
||||||
|
"timeout": strconv.Itoa(timeout),
|
||||||
|
}
|
||||||
|
if since != "" {
|
||||||
|
query["since"] = since
|
||||||
|
}
|
||||||
|
if filterID != "" {
|
||||||
|
query["filter"] = filterID
|
||||||
|
}
|
||||||
|
if setPresence != "" {
|
||||||
|
query["set_presence"] = setPresence
|
||||||
|
}
|
||||||
|
if fullState {
|
||||||
|
query["full_state"] = "true"
|
||||||
|
}
|
||||||
|
urlPath := cli.BuildURLWithQuery([]string{"sync"}, query)
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cli *Client) register(u string, req *ReqRegister) (resp *RespRegister, uiaResp *RespUserInteractive, err error) {
|
||||||
|
var bodyBytes []byte
|
||||||
|
bodyBytes, err = cli.MakeRequest("POST", u, req, nil)
|
||||||
|
if err != nil {
|
||||||
|
httpErr, ok := err.(HTTPError)
|
||||||
|
if !ok { // network error
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if httpErr.Code == 401 {
|
||||||
|
// body should be RespUserInteractive, if it isn't, fail with the error
|
||||||
|
err = json.Unmarshal(bodyBytes, &uiaResp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// body should be RespRegister
|
||||||
|
err = json.Unmarshal(bodyBytes, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register makes an HTTP request according to http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
||||||
|
//
|
||||||
|
// Registers with kind=user. For kind=guest, see RegisterGuest.
|
||||||
|
func (cli *Client) Register(req *ReqRegister) (*RespRegister, *RespUserInteractive, error) {
|
||||||
|
u := cli.BuildURL("register")
|
||||||
|
return cli.register(u, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RegisterGuest makes an HTTP request according to http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
||||||
|
// with kind=guest.
|
||||||
|
//
|
||||||
|
// For kind=user, see Register.
|
||||||
|
func (cli *Client) RegisterGuest(req *ReqRegister) (*RespRegister, *RespUserInteractive, error) {
|
||||||
|
query := map[string]string{
|
||||||
|
"kind": "guest",
|
||||||
|
}
|
||||||
|
u := cli.BuildURLWithQuery([]string{"register"}, query)
|
||||||
|
return cli.register(u, req)
|
||||||
|
}
|
||||||
|
|
||||||
|
// RegisterDummy performs m.login.dummy registration according to https://matrix.org/docs/spec/client_server/r0.2.0.html#dummy-auth
|
||||||
|
//
|
||||||
|
// Only a username and password need to be provided on the ReqRegister struct. Most local/developer homeservers will allow registration
|
||||||
|
// this way. If the homeserver does not, an error is returned.
|
||||||
|
//
|
||||||
|
// This does not set credentials on the client instance. See SetCredentials() instead.
|
||||||
|
//
|
||||||
|
// res, err := cli.RegisterDummy(&gomatrix.ReqRegister{
|
||||||
|
// Username: "alice",
|
||||||
|
// Password: "wonderland",
|
||||||
|
// })
|
||||||
|
// if err != nil {
|
||||||
|
// panic(err)
|
||||||
|
// }
|
||||||
|
// token := res.AccessToken
|
||||||
|
func (cli *Client) RegisterDummy(req *ReqRegister) (*RespRegister, error) {
|
||||||
|
res, uia, err := cli.Register(req)
|
||||||
|
if err != nil && uia == nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if uia != nil && uia.HasSingleStageFlow("m.login.dummy") {
|
||||||
|
req.Auth = struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Session string `json:"session,omitempty"`
|
||||||
|
}{"m.login.dummy", uia.Session}
|
||||||
|
res, _, err = cli.Register(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if res == nil {
|
||||||
|
return nil, fmt.Errorf("registration failed: does this server support m.login.dummy?")
|
||||||
|
}
|
||||||
|
return res, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Login a user to the homeserver according to http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-login
|
||||||
|
// This does not set credentials on this client instance. See SetCredentials() instead.
|
||||||
|
func (cli *Client) Login(req *ReqLogin) (resp *RespLogin, err error) {
|
||||||
|
urlPath := cli.BuildURL("login")
|
||||||
|
_, err = cli.MakeRequest("POST", urlPath, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Logout the current user. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-logout
|
||||||
|
// This does not clear the credentials from the client instance. See ClearCredentials() instead.
|
||||||
|
func (cli *Client) Logout() (resp *RespLogout, err error) {
|
||||||
|
urlPath := cli.BuildURL("logout")
|
||||||
|
_, err = cli.MakeRequest("POST", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Versions returns the list of supported Matrix versions on this homeserver. See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-versions
|
||||||
|
func (cli *Client) Versions() (resp *RespVersions, err error) {
|
||||||
|
urlPath := cli.BuildBaseURL("_matrix", "client", "versions")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// JoinRoom joins the client to a room ID or alias. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-join-roomidoralias
|
||||||
|
//
|
||||||
|
// If serverName is specified, this will be added as a query param to instruct the homeserver to join via that server. If content is specified, it will
|
||||||
|
// be JSON encoded and used as the request body.
|
||||||
|
func (cli *Client) JoinRoom(roomIDorAlias, serverName string, content interface{}) (resp *RespJoinRoom, err error) {
|
||||||
|
var urlPath string
|
||||||
|
if serverName != "" {
|
||||||
|
urlPath = cli.BuildURLWithQuery([]string{"join", roomIDorAlias}, map[string]string{
|
||||||
|
"server_name": serverName,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
urlPath = cli.BuildURL("join", roomIDorAlias)
|
||||||
|
}
|
||||||
|
_, err = cli.MakeRequest("POST", urlPath, content, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetDisplayName returns the display name of the user from the specified MXID. See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
func (cli *Client) GetDisplayName(mxid string) (resp *RespUserDisplayName, err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", mxid, "displayname")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetOwnDisplayName returns the user's display name. See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
func (cli *Client) GetOwnDisplayName() (resp *RespUserDisplayName, err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", cli.UserID, "displayname")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetDisplayName sets the user's profile display name. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-profile-userid-displayname
|
||||||
|
func (cli *Client) SetDisplayName(displayName string) (err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", cli.UserID, "displayname")
|
||||||
|
s := struct {
|
||||||
|
DisplayName string `json:"displayname"`
|
||||||
|
}{displayName}
|
||||||
|
_, err = cli.MakeRequest("PUT", urlPath, &s, nil)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetAvatarURL gets the user's avatar URL. See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-avatar-url
|
||||||
|
func (cli *Client) GetAvatarURL() (url string, err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", cli.UserID, "avatar_url")
|
||||||
|
s := struct {
|
||||||
|
AvatarURL string `json:"avatar_url"`
|
||||||
|
}{}
|
||||||
|
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &s)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return s.AvatarURL, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetAvatarURL sets the user's avatar URL. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-profile-userid-avatar-url
|
||||||
|
func (cli *Client) SetAvatarURL(url string) (err error) {
|
||||||
|
urlPath := cli.BuildURL("profile", cli.UserID, "avatar_url")
|
||||||
|
s := struct {
|
||||||
|
AvatarURL string `json:"avatar_url"`
|
||||||
|
}{url}
|
||||||
|
_, err = cli.MakeRequest("PUT", urlPath, &s, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendMessageEvent sends a message event into a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-send-eventtype-txnid
|
||||||
|
// contentJSON should be a pointer to something that can be encoded as JSON using json.Marshal.
|
||||||
|
func (cli *Client) SendMessageEvent(roomID string, eventType string, contentJSON interface{}) (resp *RespSendEvent, err error) {
|
||||||
|
txnID := txnID()
|
||||||
|
urlPath := cli.BuildURL("rooms", roomID, "send", eventType, txnID)
|
||||||
|
_, err = cli.MakeRequest("PUT", urlPath, contentJSON, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendStateEvent sends a state event into a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-state-eventtype-statekey
|
||||||
|
// contentJSON should be a pointer to something that can be encoded as JSON using json.Marshal.
|
||||||
|
func (cli *Client) SendStateEvent(roomID, eventType, stateKey string, contentJSON interface{}) (resp *RespSendEvent, err error) {
|
||||||
|
urlPath := cli.BuildURL("rooms", roomID, "state", eventType, stateKey)
|
||||||
|
_, err = cli.MakeRequest("PUT", urlPath, contentJSON, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendText sends an m.room.message event into the given room with a msgtype of m.text
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#m-text
|
||||||
|
func (cli *Client) SendText(roomID, text string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
TextMessage{"m.text", text})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendImage sends an m.room.message event into the given room with a msgtype of m.image
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#m-image
|
||||||
|
func (cli *Client) SendImage(roomID, body, url string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
ImageMessage{
|
||||||
|
MsgType: "m.image",
|
||||||
|
Body: body,
|
||||||
|
URL: url,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendVideo sends an m.room.message event into the given room with a msgtype of m.video
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
func (cli *Client) SendVideo(roomID, body, url string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
VideoMessage{
|
||||||
|
MsgType: "m.video",
|
||||||
|
Body: body,
|
||||||
|
URL: url,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendNotice sends an m.room.message event into the given room with a msgtype of m.notice
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#m-notice
|
||||||
|
func (cli *Client) SendNotice(roomID, text string) (*RespSendEvent, error) {
|
||||||
|
return cli.SendMessageEvent(roomID, "m.room.message",
|
||||||
|
TextMessage{"m.notice", text})
|
||||||
|
}
|
||||||
|
|
||||||
|
// RedactEvent redacts the given event. See http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-redact-eventid-txnid
|
||||||
|
func (cli *Client) RedactEvent(roomID, eventID string, req *ReqRedact) (resp *RespSendEvent, err error) {
|
||||||
|
txnID := txnID()
|
||||||
|
urlPath := cli.BuildURL("rooms", roomID, "redact", eventID, txnID)
|
||||||
|
_, err = cli.MakeRequest("PUT", urlPath, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// CreateRoom creates a new Matrix room. See https://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-createroom
|
||||||
|
// resp, err := cli.CreateRoom(&gomatrix.ReqCreateRoom{
|
||||||
|
// Preset: "public_chat",
|
||||||
|
// })
|
||||||
|
// fmt.Println("Room:", resp.RoomID)
|
||||||
|
func (cli *Client) CreateRoom(req *ReqCreateRoom) (resp *RespCreateRoom, err error) {
|
||||||
|
urlPath := cli.BuildURL("createRoom")
|
||||||
|
_, err = cli.MakeRequest("POST", urlPath, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// LeaveRoom leaves the given room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-leave
|
||||||
|
func (cli *Client) LeaveRoom(roomID string) (resp *RespLeaveRoom, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "leave")
|
||||||
|
_, err = cli.MakeRequest("POST", u, struct{}{}, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// ForgetRoom forgets a room entirely. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-forget
|
||||||
|
func (cli *Client) ForgetRoom(roomID string) (resp *RespForgetRoom, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "forget")
|
||||||
|
_, err = cli.MakeRequest("POST", u, struct{}{}, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// InviteUser invites a user to a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-invite
|
||||||
|
func (cli *Client) InviteUser(roomID string, req *ReqInviteUser) (resp *RespInviteUser, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "invite")
|
||||||
|
_, err = cli.MakeRequest("POST", u, struct{}{}, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// InviteUserByThirdParty invites a third-party identifier to a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#invite-by-third-party-id-endpoint
|
||||||
|
func (cli *Client) InviteUserByThirdParty(roomID string, req *ReqInvite3PID) (resp *RespInviteUser, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "invite")
|
||||||
|
_, err = cli.MakeRequest("POST", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// KickUser kicks a user from a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-kick
|
||||||
|
func (cli *Client) KickUser(roomID string, req *ReqKickUser) (resp *RespKickUser, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "kick")
|
||||||
|
_, err = cli.MakeRequest("POST", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// BanUser bans a user from a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-ban
|
||||||
|
func (cli *Client) BanUser(roomID string, req *ReqBanUser) (resp *RespBanUser, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "ban")
|
||||||
|
_, err = cli.MakeRequest("POST", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// UnbanUser unbans a user from a room. See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-unban
|
||||||
|
func (cli *Client) UnbanUser(roomID string, req *ReqUnbanUser) (resp *RespUnbanUser, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "unban")
|
||||||
|
_, err = cli.MakeRequest("POST", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// UserTyping sets the typing status of the user. See https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
func (cli *Client) UserTyping(roomID string, typing bool, timeout int64) (resp *RespTyping, err error) {
|
||||||
|
req := ReqTyping{Typing: typing, Timeout: timeout}
|
||||||
|
u := cli.BuildURL("rooms", roomID, "typing", cli.UserID)
|
||||||
|
_, err = cli.MakeRequest("PUT", u, req, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// StateEvent gets a single state event in a room. It will attempt to JSON unmarshal into the given "outContent" struct with
|
||||||
|
// the HTTP response body, or return an error.
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-state-eventtype-statekey
|
||||||
|
func (cli *Client) StateEvent(roomID, eventType, stateKey string, outContent interface{}) (err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "state", eventType, stateKey)
|
||||||
|
_, err = cli.MakeRequest("GET", u, nil, outContent)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadLink uploads an HTTP URL and then returns an MXC URI.
|
||||||
|
func (cli *Client) UploadLink(link string) (*RespMediaUpload, error) {
|
||||||
|
res, err := cli.Client.Get(link)
|
||||||
|
if res != nil {
|
||||||
|
defer res.Body.Close()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return cli.UploadToContentRepo(res.Body, res.Header.Get("Content-Type"), res.ContentLength)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UploadToContentRepo uploads the given bytes to the content repository and returns an MXC URI.
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-media-r0-upload
|
||||||
|
func (cli *Client) UploadToContentRepo(content io.Reader, contentType string, contentLength int64) (*RespMediaUpload, error) {
|
||||||
|
req, err := http.NewRequest("POST", cli.BuildBaseURL("_matrix/media/r0/upload"), content)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", contentType)
|
||||||
|
req.ContentLength = contentLength
|
||||||
|
res, err := cli.Client.Do(req)
|
||||||
|
if res != nil {
|
||||||
|
defer res.Body.Close()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if res.StatusCode != 200 {
|
||||||
|
contents, err := ioutil.ReadAll(res.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, HTTPError{
|
||||||
|
Message: "Upload request failed - Failed to read response body: " + err.Error(),
|
||||||
|
Code: res.StatusCode,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil, HTTPError{
|
||||||
|
Message: "Upload request failed: " + string(contents),
|
||||||
|
Code: res.StatusCode,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
var m RespMediaUpload
|
||||||
|
if err := json.NewDecoder(res.Body).Decode(&m); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &m, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// JoinedMembers returns a map of joined room members. See TODO-SPEC. https://github.com/matrix-org/synapse/pull/1680
|
||||||
|
//
|
||||||
|
// In general, usage of this API is discouraged in favour of /sync, as calling this API can race with incoming membership changes.
|
||||||
|
// This API is primarily designed for application services which may want to efficiently look up joined members in a room.
|
||||||
|
func (cli *Client) JoinedMembers(roomID string) (resp *RespJoinedMembers, err error) {
|
||||||
|
u := cli.BuildURL("rooms", roomID, "joined_members")
|
||||||
|
_, err = cli.MakeRequest("GET", u, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// JoinedRooms returns a list of rooms which the client is joined to. See TODO-SPEC. https://github.com/matrix-org/synapse/pull/1680
|
||||||
|
//
|
||||||
|
// In general, usage of this API is discouraged in favour of /sync, as calling this API can race with incoming membership changes.
|
||||||
|
// This API is primarily designed for application services which may want to efficiently look up joined rooms.
|
||||||
|
func (cli *Client) JoinedRooms() (resp *RespJoinedRooms, err error) {
|
||||||
|
u := cli.BuildURL("joined_rooms")
|
||||||
|
_, err = cli.MakeRequest("GET", u, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Messages returns a list of message and state events for a room. It uses
|
||||||
|
// pagination query parameters to paginate history in the room.
|
||||||
|
// See https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-messages
|
||||||
|
func (cli *Client) Messages(roomID, from, to string, dir rune, limit int) (resp *RespMessages, err error) {
|
||||||
|
query := map[string]string{
|
||||||
|
"from": from,
|
||||||
|
"dir": string(dir),
|
||||||
|
}
|
||||||
|
if to != "" {
|
||||||
|
query["to"] = to
|
||||||
|
}
|
||||||
|
if limit != 0 {
|
||||||
|
query["limit"] = strconv.Itoa(limit)
|
||||||
|
}
|
||||||
|
|
||||||
|
urlPath := cli.BuildURLWithQuery([]string{"rooms", roomID, "messages"}, query)
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TurnServer returns turn server details and credentials for the client to use when initiating calls.
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-voip-turnserver
|
||||||
|
func (cli *Client) TurnServer() (resp *RespTurnServer, err error) {
|
||||||
|
urlPath := cli.BuildURL("voip", "turnServer")
|
||||||
|
_, err = cli.MakeRequest("GET", urlPath, nil, &resp)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func txnID() string {
|
||||||
|
return "go" + strconv.FormatInt(time.Now().UnixNano(), 10)
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewClient creates a new Matrix Client ready for syncing
|
||||||
|
func NewClient(homeserverURL, userID, accessToken string) (*Client, error) {
|
||||||
|
hsURL, err := url.Parse(homeserverURL)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
// By default, use an in-memory store which will never save filter ids / next batch tokens to disk.
|
||||||
|
// The client will work with this storer: it just won't remember across restarts.
|
||||||
|
// In practice, a database backend should be used.
|
||||||
|
store := NewInMemoryStore()
|
||||||
|
cli := Client{
|
||||||
|
AccessToken: accessToken,
|
||||||
|
HomeserverURL: hsURL,
|
||||||
|
UserID: userID,
|
||||||
|
Prefix: "/_matrix/client/r0",
|
||||||
|
Syncer: NewDefaultSyncer(userID, store),
|
||||||
|
Store: store,
|
||||||
|
}
|
||||||
|
// By default, use the default HTTP client.
|
||||||
|
cli.Client = http.DefaultClient
|
||||||
|
|
||||||
|
return &cli, nil
|
||||||
|
}
|
102
vendor/github.com/matterbridge/gomatrix/events.go
generated
vendored
Normal file
102
vendor/github.com/matterbridge/gomatrix/events.go
generated
vendored
Normal file
@ -0,0 +1,102 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"html"
|
||||||
|
"regexp"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Event represents a single Matrix event.
|
||||||
|
type Event struct {
|
||||||
|
StateKey *string `json:"state_key,omitempty"` // The state key for the event. Only present on State Events.
|
||||||
|
Sender string `json:"sender"` // The user ID of the sender of the event
|
||||||
|
Type string `json:"type"` // The event type
|
||||||
|
Timestamp int64 `json:"origin_server_ts"` // The unix timestamp when this message was sent by the origin server
|
||||||
|
ID string `json:"event_id"` // The unique ID of this event
|
||||||
|
RoomID string `json:"room_id"` // The room the event was sent to. May be nil (e.g. for presence)
|
||||||
|
Content map[string]interface{} `json:"content"` // The JSON content of the event.
|
||||||
|
Redacts string `json:"redacts,omitempty"` // The event ID that was redacted if a m.room.redaction event
|
||||||
|
}
|
||||||
|
|
||||||
|
// Body returns the value of the "body" key in the event content if it is
|
||||||
|
// present and is a string.
|
||||||
|
func (event *Event) Body() (body string, ok bool) {
|
||||||
|
value, exists := event.Content["body"]
|
||||||
|
if !exists {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
body, ok = value.(string)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// MessageType returns the value of the "msgtype" key in the event content if
|
||||||
|
// it is present and is a string.
|
||||||
|
func (event *Event) MessageType() (msgtype string, ok bool) {
|
||||||
|
value, exists := event.Content["msgtype"]
|
||||||
|
if !exists {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
msgtype, ok = value.(string)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// TextMessage is the contents of a Matrix formated message event.
|
||||||
|
type TextMessage struct {
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
Body string `json:"body"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ImageInfo contains info about an image - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-image
|
||||||
|
type ImageInfo struct {
|
||||||
|
Height uint `json:"h,omitempty"`
|
||||||
|
Width uint `json:"w,omitempty"`
|
||||||
|
Mimetype string `json:"mimetype,omitempty"`
|
||||||
|
Size uint `json:"size,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// VideoInfo contains info about a video - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
type VideoInfo struct {
|
||||||
|
Mimetype string `json:"mimetype,omitempty"`
|
||||||
|
ThumbnailInfo ImageInfo `json:"thumbnail_info"`
|
||||||
|
ThumbnailURL string `json:"thumbnail_url,omitempty"`
|
||||||
|
Height uint `json:"h,omitempty"`
|
||||||
|
Width uint `json:"w,omitempty"`
|
||||||
|
Duration uint `json:"duration,omitempty"`
|
||||||
|
Size uint `json:"size,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// VideoMessage is an m.video - http://matrix.org/docs/spec/client_server/r0.2.0.html#m-video
|
||||||
|
type VideoMessage struct {
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
Body string `json:"body"`
|
||||||
|
URL string `json:"url"`
|
||||||
|
Info VideoInfo `json:"info"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ImageMessage is an m.image event
|
||||||
|
type ImageMessage struct {
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
Body string `json:"body"`
|
||||||
|
URL string `json:"url"`
|
||||||
|
Info ImageInfo `json:"info"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// An HTMLMessage is the contents of a Matrix HTML formated message event.
|
||||||
|
type HTMLMessage struct {
|
||||||
|
Body string `json:"body"`
|
||||||
|
MsgType string `json:"msgtype"`
|
||||||
|
Format string `json:"format"`
|
||||||
|
FormattedBody string `json:"formatted_body"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var htmlRegex = regexp.MustCompile("<[^<]+?>")
|
||||||
|
|
||||||
|
// GetHTMLMessage returns an HTMLMessage with the body set to a stripped version of the provided HTML, in addition
|
||||||
|
// to the provided HTML.
|
||||||
|
func GetHTMLMessage(msgtype, htmlText string) HTMLMessage {
|
||||||
|
return HTMLMessage{
|
||||||
|
Body: html.UnescapeString(htmlRegex.ReplaceAllLiteralString(htmlText, "")),
|
||||||
|
MsgType: msgtype,
|
||||||
|
Format: "org.matrix.custom.html",
|
||||||
|
FormattedBody: htmlText,
|
||||||
|
}
|
||||||
|
}
|
43
vendor/github.com/matterbridge/gomatrix/filter.go
generated
vendored
Normal file
43
vendor/github.com/matterbridge/gomatrix/filter.go
generated
vendored
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
// Copyright 2017 Jan Christian Grünhage
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
//
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package gomatrix
|
||||||
|
|
||||||
|
//Filter is used by clients to specify how the server should filter responses to e.g. sync requests
|
||||||
|
//Specified by: https://matrix.org/docs/spec/client_server/r0.2.0.html#filtering
|
||||||
|
type Filter struct {
|
||||||
|
AccountData FilterPart `json:"account_data,omitempty"`
|
||||||
|
EventFields []string `json:"event_fields,omitempty"`
|
||||||
|
EventFormat string `json:"event_format,omitempty"`
|
||||||
|
Presence FilterPart `json:"presence,omitempty"`
|
||||||
|
Room struct {
|
||||||
|
AccountData FilterPart `json:"account_data,omitempty"`
|
||||||
|
Ephemeral FilterPart `json:"ephemeral,omitempty"`
|
||||||
|
IncludeLeave bool `json:"include_leave,omitempty"`
|
||||||
|
NotRooms []string `json:"not_rooms,omitempty"`
|
||||||
|
Rooms []string `json:"rooms,omitempty"`
|
||||||
|
State FilterPart `json:"state,omitempty"`
|
||||||
|
Timeline FilterPart `json:"timeline,omitempty"`
|
||||||
|
} `json:"room,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type FilterPart struct {
|
||||||
|
NotRooms []string `json:"not_rooms,omitempty"`
|
||||||
|
Rooms []string `json:"rooms,omitempty"`
|
||||||
|
Limit *int `json:"limit,omitempty"`
|
||||||
|
NotSenders []string `json:"not_senders,omitempty"`
|
||||||
|
NotTypes []string `json:"not_types,omitempty"`
|
||||||
|
Senders []string `json:"senders,omitempty"`
|
||||||
|
Types []string `json:"types,omitempty"`
|
||||||
|
}
|
78
vendor/github.com/matterbridge/gomatrix/requests.go
generated
vendored
Normal file
78
vendor/github.com/matterbridge/gomatrix/requests.go
generated
vendored
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
// ReqRegister is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
||||||
|
type ReqRegister struct {
|
||||||
|
Username string `json:"username,omitempty"`
|
||||||
|
BindEmail bool `json:"bind_email,omitempty"`
|
||||||
|
Password string `json:"password,omitempty"`
|
||||||
|
DeviceID string `json:"device_id,omitempty"`
|
||||||
|
InitialDeviceDisplayName string `json:"initial_device_display_name"`
|
||||||
|
Auth interface{} `json:"auth,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqLogin is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-login
|
||||||
|
type ReqLogin struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Password string `json:"password,omitempty"`
|
||||||
|
Medium string `json:"medium,omitempty"`
|
||||||
|
User string `json:"user,omitempty"`
|
||||||
|
Address string `json:"address,omitempty"`
|
||||||
|
Token string `json:"token,omitempty"`
|
||||||
|
DeviceID string `json:"device_id,omitempty"`
|
||||||
|
InitialDeviceDisplayName string `json:"initial_device_display_name,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqCreateRoom is the JSON request for https://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-createroom
|
||||||
|
type ReqCreateRoom struct {
|
||||||
|
Visibility string `json:"visibility,omitempty"`
|
||||||
|
RoomAliasName string `json:"room_alias_name,omitempty"`
|
||||||
|
Name string `json:"name,omitempty"`
|
||||||
|
Topic string `json:"topic,omitempty"`
|
||||||
|
Invite []string `json:"invite,omitempty"`
|
||||||
|
Invite3PID []ReqInvite3PID `json:"invite_3pid,omitempty"`
|
||||||
|
CreationContent map[string]interface{} `json:"creation_content,omitempty"`
|
||||||
|
InitialState []Event `json:"initial_state,omitempty"`
|
||||||
|
Preset string `json:"preset,omitempty"`
|
||||||
|
IsDirect bool `json:"is_direct,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqRedact is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-redact-eventid-txnid
|
||||||
|
type ReqRedact struct {
|
||||||
|
Reason string `json:"reason,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqInvite3PID is the JSON request for https://matrix.org/docs/spec/client_server/r0.2.0.html#id57
|
||||||
|
// It is also a JSON object used in https://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-createroom
|
||||||
|
type ReqInvite3PID struct {
|
||||||
|
IDServer string `json:"id_server"`
|
||||||
|
Medium string `json:"medium"`
|
||||||
|
Address string `json:"address"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqInviteUser is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-invite
|
||||||
|
type ReqInviteUser struct {
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqKickUser is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-kick
|
||||||
|
type ReqKickUser struct {
|
||||||
|
Reason string `json:"reason,omitempty"`
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqBanUser is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-ban
|
||||||
|
type ReqBanUser struct {
|
||||||
|
Reason string `json:"reason,omitempty"`
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqUnbanUser is the JSON request for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-unban
|
||||||
|
type ReqUnbanUser struct {
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReqTyping is the JSON request for https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
type ReqTyping struct {
|
||||||
|
Typing bool `json:"typing"`
|
||||||
|
Timeout int64 `json:"timeout"`
|
||||||
|
}
|
176
vendor/github.com/matterbridge/gomatrix/responses.go
generated
vendored
Normal file
176
vendor/github.com/matterbridge/gomatrix/responses.go
generated
vendored
Normal file
@ -0,0 +1,176 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
// RespError is the standard JSON error response from Homeservers. It also implements the Golang "error" interface.
|
||||||
|
// See http://matrix.org/docs/spec/client_server/r0.2.0.html#api-standards
|
||||||
|
type RespError struct {
|
||||||
|
ErrCode string `json:"errcode"`
|
||||||
|
Err string `json:"error"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error returns the errcode and error message.
|
||||||
|
func (e RespError) Error() string {
|
||||||
|
return e.ErrCode + ": " + e.Err
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespCreateFilter is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-user-userid-filter
|
||||||
|
type RespCreateFilter struct {
|
||||||
|
FilterID string `json:"filter_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespVersions is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-versions
|
||||||
|
type RespVersions struct {
|
||||||
|
Versions []string `json:"versions"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespJoinRoom is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-join
|
||||||
|
type RespJoinRoom struct {
|
||||||
|
RoomID string `json:"room_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespLeaveRoom is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-leave
|
||||||
|
type RespLeaveRoom struct{}
|
||||||
|
|
||||||
|
// RespForgetRoom is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-forget
|
||||||
|
type RespForgetRoom struct{}
|
||||||
|
|
||||||
|
// RespInviteUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-invite
|
||||||
|
type RespInviteUser struct{}
|
||||||
|
|
||||||
|
// RespKickUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-kick
|
||||||
|
type RespKickUser struct{}
|
||||||
|
|
||||||
|
// RespBanUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-ban
|
||||||
|
type RespBanUser struct{}
|
||||||
|
|
||||||
|
// RespUnbanUser is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-rooms-roomid-unban
|
||||||
|
type RespUnbanUser struct{}
|
||||||
|
|
||||||
|
// RespTyping is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-typing-userid
|
||||||
|
type RespTyping struct{}
|
||||||
|
|
||||||
|
// RespJoinedRooms is the JSON response for TODO-SPEC https://github.com/matrix-org/synapse/pull/1680
|
||||||
|
type RespJoinedRooms struct {
|
||||||
|
JoinedRooms []string `json:"joined_rooms"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespJoinedMembers is the JSON response for TODO-SPEC https://github.com/matrix-org/synapse/pull/1680
|
||||||
|
type RespJoinedMembers struct {
|
||||||
|
Joined map[string]struct {
|
||||||
|
DisplayName *string `json:"display_name"`
|
||||||
|
AvatarURL *string `json:"avatar_url"`
|
||||||
|
} `json:"joined"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespMessages is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-rooms-roomid-messages
|
||||||
|
type RespMessages struct {
|
||||||
|
Start string `json:"start"`
|
||||||
|
Chunk []Event `json:"chunk"`
|
||||||
|
End string `json:"end"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespSendEvent is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#put-matrix-client-r0-rooms-roomid-send-eventtype-txnid
|
||||||
|
type RespSendEvent struct {
|
||||||
|
EventID string `json:"event_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespMediaUpload is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-media-r0-upload
|
||||||
|
type RespMediaUpload struct {
|
||||||
|
ContentURI string `json:"content_uri"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespUserInteractive is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#user-interactive-authentication-api
|
||||||
|
type RespUserInteractive struct {
|
||||||
|
Flows []struct {
|
||||||
|
Stages []string `json:"stages"`
|
||||||
|
} `json:"flows"`
|
||||||
|
Params map[string]interface{} `json:"params"`
|
||||||
|
Session string `json:"string"`
|
||||||
|
Completed []string `json:"completed"`
|
||||||
|
ErrCode string `json:"errcode"`
|
||||||
|
Error string `json:"error"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// HasSingleStageFlow returns true if there exists at least 1 Flow with a single stage of stageName.
|
||||||
|
func (r RespUserInteractive) HasSingleStageFlow(stageName string) bool {
|
||||||
|
for _, f := range r.Flows {
|
||||||
|
if len(f.Stages) == 1 && f.Stages[0] == stageName {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespUserDisplayName is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-profile-userid-displayname
|
||||||
|
type RespUserDisplayName struct {
|
||||||
|
DisplayName string `json:"displayname"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespRegister is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-register
|
||||||
|
type RespRegister struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
DeviceID string `json:"device_id"`
|
||||||
|
HomeServer string `json:"home_server"`
|
||||||
|
RefreshToken string `json:"refresh_token"`
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespLogin is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-login
|
||||||
|
type RespLogin struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
DeviceID string `json:"device_id"`
|
||||||
|
HomeServer string `json:"home_server"`
|
||||||
|
UserID string `json:"user_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespLogout is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-logout
|
||||||
|
type RespLogout struct{}
|
||||||
|
|
||||||
|
// RespCreateRoom is the JSON response for https://matrix.org/docs/spec/client_server/r0.2.0.html#post-matrix-client-r0-createroom
|
||||||
|
type RespCreateRoom struct {
|
||||||
|
RoomID string `json:"room_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// RespSync is the JSON response for http://matrix.org/docs/spec/client_server/r0.2.0.html#get-matrix-client-r0-sync
|
||||||
|
type RespSync struct {
|
||||||
|
NextBatch string `json:"next_batch"`
|
||||||
|
AccountData struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
} `json:"account_data"`
|
||||||
|
Presence struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
} `json:"presence"`
|
||||||
|
Rooms struct {
|
||||||
|
Leave map[string]struct {
|
||||||
|
State struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
} `json:"state"`
|
||||||
|
Timeline struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
Limited bool `json:"limited"`
|
||||||
|
PrevBatch string `json:"prev_batch"`
|
||||||
|
} `json:"timeline"`
|
||||||
|
} `json:"leave"`
|
||||||
|
Join map[string]struct {
|
||||||
|
State struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
} `json:"state"`
|
||||||
|
Timeline struct {
|
||||||
|
Events []Event `json:"events"`
|
||||||
|
Limited bool `json:"limited"`
|
||||||
|
PrevBatch string `json:"prev_batch"`
|
||||||
|
} `json:"timeline"`
|
||||||
|
} `json:"join"`
|
||||||
|
Invite map[string]struct {
|
||||||
|
State struct {
|
||||||
|
Events []Event
|
||||||
|
} `json:"invite_state"`
|
||||||
|
} `json:"invite"`
|
||||||
|
} `json:"rooms"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type RespTurnServer struct {
|
||||||
|
Username string `json:"username"`
|
||||||
|
Password string `json:"password"`
|
||||||
|
TTL int `json:"ttl"`
|
||||||
|
URIs []string `json:"uris"`
|
||||||
|
}
|
50
vendor/github.com/matterbridge/gomatrix/room.go
generated
vendored
Normal file
50
vendor/github.com/matterbridge/gomatrix/room.go
generated
vendored
Normal file
@ -0,0 +1,50 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
// Room represents a single Matrix room.
|
||||||
|
type Room struct {
|
||||||
|
ID string
|
||||||
|
State map[string]map[string]*Event
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateState updates the room's current state with the given Event. This will clobber events based
|
||||||
|
// on the type/state_key combination.
|
||||||
|
func (room Room) UpdateState(event *Event) {
|
||||||
|
_, exists := room.State[event.Type]
|
||||||
|
if !exists {
|
||||||
|
room.State[event.Type] = make(map[string]*Event)
|
||||||
|
}
|
||||||
|
room.State[event.Type][*event.StateKey] = event
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetStateEvent returns the state event for the given type/state_key combo, or nil.
|
||||||
|
func (room Room) GetStateEvent(eventType string, stateKey string) *Event {
|
||||||
|
stateEventMap, _ := room.State[eventType]
|
||||||
|
event, _ := stateEventMap[stateKey]
|
||||||
|
return event
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetMembershipState returns the membership state of the given user ID in this room. If there is
|
||||||
|
// no entry for this member, 'leave' is returned for consistency with left users.
|
||||||
|
func (room Room) GetMembershipState(userID string) string {
|
||||||
|
state := "leave"
|
||||||
|
event := room.GetStateEvent("m.room.member", userID)
|
||||||
|
if event != nil {
|
||||||
|
membershipState, found := event.Content["membership"]
|
||||||
|
if found {
|
||||||
|
mState, isString := membershipState.(string)
|
||||||
|
if isString {
|
||||||
|
state = mState
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return state
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewRoom creates a new Room with the given ID
|
||||||
|
func NewRoom(roomID string) *Room {
|
||||||
|
// Init the State map and return a pointer to the Room
|
||||||
|
return &Room{
|
||||||
|
ID: roomID,
|
||||||
|
State: make(map[string]map[string]*Event),
|
||||||
|
}
|
||||||
|
}
|
65
vendor/github.com/matterbridge/gomatrix/store.go
generated
vendored
Normal file
65
vendor/github.com/matterbridge/gomatrix/store.go
generated
vendored
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
// Storer is an interface which must be satisfied to store client data.
|
||||||
|
//
|
||||||
|
// You can either write a struct which persists this data to disk, or you can use the
|
||||||
|
// provided "InMemoryStore" which just keeps data around in-memory which is lost on
|
||||||
|
// restarts.
|
||||||
|
type Storer interface {
|
||||||
|
SaveFilterID(userID, filterID string)
|
||||||
|
LoadFilterID(userID string) string
|
||||||
|
SaveNextBatch(userID, nextBatchToken string)
|
||||||
|
LoadNextBatch(userID string) string
|
||||||
|
SaveRoom(room *Room)
|
||||||
|
LoadRoom(roomID string) *Room
|
||||||
|
}
|
||||||
|
|
||||||
|
// InMemoryStore implements the Storer interface.
|
||||||
|
//
|
||||||
|
// Everything is persisted in-memory as maps. It is not safe to load/save filter IDs
|
||||||
|
// or next batch tokens on any goroutine other than the syncing goroutine: the one
|
||||||
|
// which called Client.Sync().
|
||||||
|
type InMemoryStore struct {
|
||||||
|
Filters map[string]string
|
||||||
|
NextBatch map[string]string
|
||||||
|
Rooms map[string]*Room
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveFilterID to memory.
|
||||||
|
func (s *InMemoryStore) SaveFilterID(userID, filterID string) {
|
||||||
|
s.Filters[userID] = filterID
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadFilterID from memory.
|
||||||
|
func (s *InMemoryStore) LoadFilterID(userID string) string {
|
||||||
|
return s.Filters[userID]
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveNextBatch to memory.
|
||||||
|
func (s *InMemoryStore) SaveNextBatch(userID, nextBatchToken string) {
|
||||||
|
s.NextBatch[userID] = nextBatchToken
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadNextBatch from memory.
|
||||||
|
func (s *InMemoryStore) LoadNextBatch(userID string) string {
|
||||||
|
return s.NextBatch[userID]
|
||||||
|
}
|
||||||
|
|
||||||
|
// SaveRoom to memory.
|
||||||
|
func (s *InMemoryStore) SaveRoom(room *Room) {
|
||||||
|
s.Rooms[room.ID] = room
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoadRoom from memory.
|
||||||
|
func (s *InMemoryStore) LoadRoom(roomID string) *Room {
|
||||||
|
return s.Rooms[roomID]
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewInMemoryStore constructs a new InMemoryStore.
|
||||||
|
func NewInMemoryStore() *InMemoryStore {
|
||||||
|
return &InMemoryStore{
|
||||||
|
Filters: make(map[string]string),
|
||||||
|
NextBatch: make(map[string]string),
|
||||||
|
Rooms: make(map[string]*Room),
|
||||||
|
}
|
||||||
|
}
|
164
vendor/github.com/matterbridge/gomatrix/sync.go
generated
vendored
Normal file
164
vendor/github.com/matterbridge/gomatrix/sync.go
generated
vendored
Normal file
@ -0,0 +1,164 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"runtime/debug"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Syncer represents an interface that must be satisfied in order to do /sync requests on a client.
|
||||||
|
type Syncer interface {
|
||||||
|
// Process the /sync response. The since parameter is the since= value that was used to produce the response.
|
||||||
|
// This is useful for detecting the very first sync (since=""). If an error is return, Syncing will be stopped
|
||||||
|
// permanently.
|
||||||
|
ProcessResponse(resp *RespSync, since string) error
|
||||||
|
// OnFailedSync returns either the time to wait before retrying or an error to stop syncing permanently.
|
||||||
|
OnFailedSync(res *RespSync, err error) (time.Duration, error)
|
||||||
|
// GetFilterJSON for the given user ID. NOT the filter ID.
|
||||||
|
GetFilterJSON(userID string) json.RawMessage
|
||||||
|
}
|
||||||
|
|
||||||
|
// DefaultSyncer is the default syncing implementation. You can either write your own syncer, or selectively
|
||||||
|
// replace parts of this default syncer (e.g. the ProcessResponse method). The default syncer uses the observer
|
||||||
|
// pattern to notify callers about incoming events. See DefaultSyncer.OnEventType for more information.
|
||||||
|
type DefaultSyncer struct {
|
||||||
|
UserID string
|
||||||
|
Store Storer
|
||||||
|
listeners map[string][]OnEventListener // event type to listeners array
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnEventListener can be used with DefaultSyncer.OnEventType to be informed of incoming events.
|
||||||
|
type OnEventListener func(*Event)
|
||||||
|
|
||||||
|
// NewDefaultSyncer returns an instantiated DefaultSyncer
|
||||||
|
func NewDefaultSyncer(userID string, store Storer) *DefaultSyncer {
|
||||||
|
return &DefaultSyncer{
|
||||||
|
UserID: userID,
|
||||||
|
Store: store,
|
||||||
|
listeners: make(map[string][]OnEventListener),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProcessResponse processes the /sync response in a way suitable for bots. "Suitable for bots" means a stream of
|
||||||
|
// unrepeating events. Returns a fatal error if a listener panics.
|
||||||
|
func (s *DefaultSyncer) ProcessResponse(res *RespSync, since string) (err error) {
|
||||||
|
if !s.shouldProcessResponse(res, since) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
if r := recover(); r != nil {
|
||||||
|
err = fmt.Errorf("ProcessResponse panicked! userID=%s since=%s panic=%s\n%s", s.UserID, since, r, debug.Stack())
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
for roomID, roomData := range res.Rooms.Join {
|
||||||
|
room := s.getOrCreateRoom(roomID)
|
||||||
|
for _, event := range roomData.State.Events {
|
||||||
|
event.RoomID = roomID
|
||||||
|
room.UpdateState(&event)
|
||||||
|
s.notifyListeners(&event)
|
||||||
|
}
|
||||||
|
for _, event := range roomData.Timeline.Events {
|
||||||
|
event.RoomID = roomID
|
||||||
|
s.notifyListeners(&event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for roomID, roomData := range res.Rooms.Invite {
|
||||||
|
room := s.getOrCreateRoom(roomID)
|
||||||
|
for _, event := range roomData.State.Events {
|
||||||
|
event.RoomID = roomID
|
||||||
|
room.UpdateState(&event)
|
||||||
|
s.notifyListeners(&event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for roomID, roomData := range res.Rooms.Leave {
|
||||||
|
room := s.getOrCreateRoom(roomID)
|
||||||
|
for _, event := range roomData.Timeline.Events {
|
||||||
|
if event.StateKey != nil {
|
||||||
|
event.RoomID = roomID
|
||||||
|
room.UpdateState(&event)
|
||||||
|
s.notifyListeners(&event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnEventType allows callers to be notified when there are new events for the given event type.
|
||||||
|
// There are no duplicate checks.
|
||||||
|
func (s *DefaultSyncer) OnEventType(eventType string, callback OnEventListener) {
|
||||||
|
_, exists := s.listeners[eventType]
|
||||||
|
if !exists {
|
||||||
|
s.listeners[eventType] = []OnEventListener{}
|
||||||
|
}
|
||||||
|
s.listeners[eventType] = append(s.listeners[eventType], callback)
|
||||||
|
}
|
||||||
|
|
||||||
|
// shouldProcessResponse returns true if the response should be processed. May modify the response to remove
|
||||||
|
// stuff that shouldn't be processed.
|
||||||
|
func (s *DefaultSyncer) shouldProcessResponse(resp *RespSync, since string) bool {
|
||||||
|
if since == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
// This is a horrible hack because /sync will return the most recent messages for a room
|
||||||
|
// as soon as you /join it. We do NOT want to process those events in that particular room
|
||||||
|
// because they may have already been processed (if you toggle the bot in/out of the room).
|
||||||
|
//
|
||||||
|
// Work around this by inspecting each room's timeline and seeing if an m.room.member event for us
|
||||||
|
// exists and is "join" and then discard processing that room entirely if so.
|
||||||
|
// TODO: We probably want to process messages from after the last join event in the timeline.
|
||||||
|
for roomID, roomData := range resp.Rooms.Join {
|
||||||
|
for i := len(roomData.Timeline.Events) - 1; i >= 0; i-- {
|
||||||
|
e := roomData.Timeline.Events[i]
|
||||||
|
if e.Type == "m.room.member" && e.StateKey != nil && *e.StateKey == s.UserID {
|
||||||
|
m := e.Content["membership"]
|
||||||
|
mship, ok := m.(string)
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if mship == "join" {
|
||||||
|
_, ok := resp.Rooms.Join[roomID]
|
||||||
|
if !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
delete(resp.Rooms.Join, roomID) // don't re-process messages
|
||||||
|
delete(resp.Rooms.Invite, roomID) // don't re-process invites
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// getOrCreateRoom must only be called by the Sync() goroutine which calls ProcessResponse()
|
||||||
|
func (s *DefaultSyncer) getOrCreateRoom(roomID string) *Room {
|
||||||
|
room := s.Store.LoadRoom(roomID)
|
||||||
|
if room == nil { // create a new Room
|
||||||
|
room = NewRoom(roomID)
|
||||||
|
s.Store.SaveRoom(room)
|
||||||
|
}
|
||||||
|
return room
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *DefaultSyncer) notifyListeners(event *Event) {
|
||||||
|
listeners, exists := s.listeners[event.Type]
|
||||||
|
if !exists {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, fn := range listeners {
|
||||||
|
fn(event)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// OnFailedSync always returns a 10 second wait period between failed /syncs, never a fatal error.
|
||||||
|
func (s *DefaultSyncer) OnFailedSync(res *RespSync, err error) (time.Duration, error) {
|
||||||
|
return 10 * time.Second, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFilterJSON returns a filter with a timeline limit of 50.
|
||||||
|
func (s *DefaultSyncer) GetFilterJSON(userID string) json.RawMessage {
|
||||||
|
return json.RawMessage(`{"room":{"timeline":{"limit":50}}}`)
|
||||||
|
}
|
130
vendor/github.com/matterbridge/gomatrix/userids.go
generated
vendored
Normal file
130
vendor/github.com/matterbridge/gomatrix/userids.go
generated
vendored
Normal file
@ -0,0 +1,130 @@
|
|||||||
|
package gomatrix
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const lowerhex = "0123456789abcdef"
|
||||||
|
|
||||||
|
// encode the given byte using quoted-printable encoding (e.g "=2f")
|
||||||
|
// and writes it to the buffer
|
||||||
|
// See https://golang.org/src/mime/quotedprintable/writer.go
|
||||||
|
func encode(buf *bytes.Buffer, b byte) {
|
||||||
|
buf.WriteByte('=')
|
||||||
|
buf.WriteByte(lowerhex[b>>4])
|
||||||
|
buf.WriteByte(lowerhex[b&0x0f])
|
||||||
|
}
|
||||||
|
|
||||||
|
// escape the given alpha character and writes it to the buffer
|
||||||
|
func escape(buf *bytes.Buffer, b byte) {
|
||||||
|
buf.WriteByte('_')
|
||||||
|
if b == '_' {
|
||||||
|
buf.WriteByte('_') // another _
|
||||||
|
} else {
|
||||||
|
buf.WriteByte(b + 0x20) // ASCII shift A-Z to a-z
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func shouldEncode(b byte) bool {
|
||||||
|
return b != '-' && b != '.' && b != '_' && !(b >= '0' && b <= '9') && !(b >= 'a' && b <= 'z') && !(b >= 'A' && b <= 'Z')
|
||||||
|
}
|
||||||
|
|
||||||
|
func shouldEscape(b byte) bool {
|
||||||
|
return (b >= 'A' && b <= 'Z') || b == '_'
|
||||||
|
}
|
||||||
|
|
||||||
|
func isValidByte(b byte) bool {
|
||||||
|
return isValidEscapedChar(b) || (b >= '0' && b <= '9') || b == '.' || b == '=' || b == '-'
|
||||||
|
}
|
||||||
|
|
||||||
|
func isValidEscapedChar(b byte) bool {
|
||||||
|
return b == '_' || (b >= 'a' && b <= 'z')
|
||||||
|
}
|
||||||
|
|
||||||
|
// EncodeUserLocalpart encodes the given string into Matrix-compliant user ID localpart form.
|
||||||
|
// See http://matrix.org/docs/spec/intro.html#mapping-from-other-character-sets
|
||||||
|
//
|
||||||
|
// This returns a string with only the characters "a-z0-9._=-". The uppercase range A-Z
|
||||||
|
// are encoded using leading underscores ("_"). Characters outside the aforementioned ranges
|
||||||
|
// (including literal underscores ("_") and equals ("=")) are encoded as UTF8 code points (NOT NCRs)
|
||||||
|
// and converted to lower-case hex with a leading "=". For example:
|
||||||
|
// Alph@Bet_50up => _alph=40_bet=5f50up
|
||||||
|
func EncodeUserLocalpart(str string) string {
|
||||||
|
strBytes := []byte(str)
|
||||||
|
var outputBuffer bytes.Buffer
|
||||||
|
for _, b := range strBytes {
|
||||||
|
if shouldEncode(b) {
|
||||||
|
encode(&outputBuffer, b)
|
||||||
|
} else if shouldEscape(b) {
|
||||||
|
escape(&outputBuffer, b)
|
||||||
|
} else {
|
||||||
|
outputBuffer.WriteByte(b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return outputBuffer.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// DecodeUserLocalpart decodes the given string back into the original input string.
|
||||||
|
// Returns an error if the given string is not a valid user ID localpart encoding.
|
||||||
|
// See http://matrix.org/docs/spec/intro.html#mapping-from-other-character-sets
|
||||||
|
//
|
||||||
|
// This decodes quoted-printable bytes back into UTF8, and unescapes casing. For
|
||||||
|
// example:
|
||||||
|
// _alph=40_bet=5f50up => Alph@Bet_50up
|
||||||
|
// Returns an error if the input string contains characters outside the
|
||||||
|
// range "a-z0-9._=-", has an invalid quote-printable byte (e.g. not hex), or has
|
||||||
|
// an invalid _ escaped byte (e.g. "_5").
|
||||||
|
func DecodeUserLocalpart(str string) (string, error) {
|
||||||
|
strBytes := []byte(str)
|
||||||
|
var outputBuffer bytes.Buffer
|
||||||
|
for i := 0; i < len(strBytes); i++ {
|
||||||
|
b := strBytes[i]
|
||||||
|
if !isValidByte(b) {
|
||||||
|
return "", fmt.Errorf("Byte pos %d: Invalid byte", i)
|
||||||
|
}
|
||||||
|
|
||||||
|
if b == '_' { // next byte is a-z and should be upper-case or is another _ and should be a literal _
|
||||||
|
if i+1 >= len(strBytes) {
|
||||||
|
return "", fmt.Errorf("Byte pos %d: expected _[a-z_] encoding but ran out of string", i)
|
||||||
|
}
|
||||||
|
if !isValidEscapedChar(strBytes[i+1]) { // invalid escaping
|
||||||
|
return "", fmt.Errorf("Byte pos %d: expected _[a-z_] encoding", i)
|
||||||
|
}
|
||||||
|
if strBytes[i+1] == '_' {
|
||||||
|
outputBuffer.WriteByte('_')
|
||||||
|
} else {
|
||||||
|
outputBuffer.WriteByte(strBytes[i+1] - 0x20) // ASCII shift a-z to A-Z
|
||||||
|
}
|
||||||
|
i++ // skip next byte since we just handled it
|
||||||
|
} else if b == '=' { // next 2 bytes are hex and should be buffered ready to be read as utf8
|
||||||
|
if i+2 >= len(strBytes) {
|
||||||
|
return "", fmt.Errorf("Byte pos: %d: expected quote-printable encoding but ran out of string", i)
|
||||||
|
}
|
||||||
|
dst := make([]byte, 1)
|
||||||
|
_, err := hex.Decode(dst, strBytes[i+1:i+3])
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
outputBuffer.WriteByte(dst[0])
|
||||||
|
i += 2 // skip next 2 bytes since we just handled it
|
||||||
|
} else { // pass through
|
||||||
|
outputBuffer.WriteByte(b)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return outputBuffer.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExtractUserLocalpart extracts the localpart portion of a user ID.
|
||||||
|
// See http://matrix.org/docs/spec/intro.html#user-identifiers
|
||||||
|
func ExtractUserLocalpart(userID string) (string, error) {
|
||||||
|
if len(userID) == 0 || userID[0] != '@' {
|
||||||
|
return "", fmt.Errorf("%s is not a valid user id", userID)
|
||||||
|
}
|
||||||
|
return strings.TrimPrefix(
|
||||||
|
strings.SplitN(userID, ":", 2)[0], // @foo:bar:8448 => [ "@foo", "bar:8448" ]
|
||||||
|
"@", // remove "@" prefix
|
||||||
|
), nil
|
||||||
|
}
|
20
vendor/github.com/matterbridge/slack/websocket_utils.go
generated
vendored
20
vendor/github.com/matterbridge/slack/websocket_utils.go
generated
vendored
@ -1,20 +0,0 @@
|
|||||||
package slack
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net"
|
|
||||||
"net/url"
|
|
||||||
)
|
|
||||||
|
|
||||||
var portMapping = map[string]string{"ws": "80", "wss": "443"}
|
|
||||||
|
|
||||||
func websocketizeURLPort(orig string) (string, error) {
|
|
||||||
urlObj, err := url.ParseRequestURI(orig)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
_, _, err = net.SplitHostPort(urlObj.Host)
|
|
||||||
if err != nil {
|
|
||||||
return urlObj.Scheme + "://" + urlObj.Host + ":" + portMapping[urlObj.Scheme] + urlObj.Path, nil
|
|
||||||
}
|
|
||||||
return orig, nil
|
|
||||||
}
|
|
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user