94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
|
package slack
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"errors"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"golang.org/x/net/websocket"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
// MaxMessageTextLength is the current maximum message length in number of characters as defined here
|
||
|
|
// https://api.slack.com/rtm#limits
|
||
|
|
MaxMessageTextLength = 4000
|
||
|
|
)
|
||
|
|
|
||
|
|
// RTM represents a managed websocket connection. It also supports
|
||
|
|
// all the methods of the `Client` type.
|
||
|
|
//
|
||
|
|
// Create this element with Client's NewRTM().
|
||
|
|
type RTM struct {
|
||
|
|
idGen IDGenerator
|
||
|
|
pings map[int]time.Time
|
||
|
|
|
||
|
|
// Connection life-cycle
|
||
|
|
conn *websocket.Conn
|
||
|
|
IncomingEvents chan RTMEvent
|
||
|
|
outgoingMessages chan OutgoingMessage
|
||
|
|
killChannel chan bool
|
||
|
|
forcePing chan bool
|
||
|
|
rawEvents chan json.RawMessage
|
||
|
|
wasIntentional bool
|
||
|
|
isConnected bool
|
||
|
|
|
||
|
|
// Client is the main API, embedded
|
||
|
|
Client
|
||
|
|
websocketURL string
|
||
|
|
|
||
|
|
// UserDetails upon connection
|
||
|
|
info *Info
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewRTM returns a RTM, which provides a fully managed connection to
|
||
|
|
// Slack's websocket-based Real-Time Messaging protocol.
|
||
|
|
func newRTM(api *Client) *RTM {
|
||
|
|
return &RTM{
|
||
|
|
Client: *api,
|
||
|
|
IncomingEvents: make(chan RTMEvent, 50),
|
||
|
|
outgoingMessages: make(chan OutgoingMessage, 20),
|
||
|
|
pings: make(map[int]time.Time),
|
||
|
|
isConnected: false,
|
||
|
|
wasIntentional: true,
|
||
|
|
killChannel: make(chan bool),
|
||
|
|
forcePing: make(chan bool),
|
||
|
|
rawEvents: make(chan json.RawMessage),
|
||
|
|
idGen: NewSafeID(1),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Disconnect and wait, blocking until a successful disconnection.
|
||
|
|
func (rtm *RTM) Disconnect() error {
|
||
|
|
if !rtm.isConnected {
|
||
|
|
return errors.New("Invalid call to Disconnect - Slack API is already disconnected")
|
||
|
|
}
|
||
|
|
rtm.killChannel <- true
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// Reconnect only makes sense if you've successfully disconnectd with Disconnect().
|
||
|
|
func (rtm *RTM) Reconnect() error {
|
||
|
|
logger.Println("RTM::Reconnect not implemented!")
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetInfo returns the info structure received when calling
|
||
|
|
// "startrtm", holding all channels, groups and other metadata needed
|
||
|
|
// to implement a full chat client. It will be non-nil after a call to
|
||
|
|
// StartRTM().
|
||
|
|
func (rtm *RTM) GetInfo() *Info {
|
||
|
|
return rtm.info
|
||
|
|
}
|
||
|
|
|
||
|
|
// SendMessage submits a simple message through the websocket. For
|
||
|
|
// more complicated messages, use `rtm.PostMessage` with a complete
|
||
|
|
// struct describing your attachments and all.
|
||
|
|
func (rtm *RTM) SendMessage(msg *OutgoingMessage) {
|
||
|
|
if msg == nil {
|
||
|
|
rtm.Debugln("Error: Attempted to SendMessage(nil)")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
rtm.outgoingMessages <- *msg
|
||
|
|
}
|