mirror of https://github.com/velour/catbase.git
slackapp: most things working
* User lists are definitely not working yet
This commit is contained in:
parent
980b079bf3
commit
c72dc7b2c8
|
@ -3,13 +3,10 @@ package slackapp
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"html"
|
"html"
|
||||||
"io/ioutil"
|
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
|
||||||
"regexp"
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
@ -29,7 +26,8 @@ type SlackApp struct {
|
||||||
config *config.Config
|
config *config.Config
|
||||||
api *slack.Client
|
api *slack.Client
|
||||||
|
|
||||||
token string
|
botToken string
|
||||||
|
userToken string
|
||||||
verification string
|
verification string
|
||||||
id string
|
id string
|
||||||
|
|
||||||
|
@ -55,8 +53,10 @@ func New(c *config.Config) *SlackApp {
|
||||||
return &SlackApp{
|
return &SlackApp{
|
||||||
api: api,
|
api: api,
|
||||||
config: c,
|
config: c,
|
||||||
token: token,
|
botToken: token,
|
||||||
|
userToken: c.Get("slack.usertoken", "NONE"),
|
||||||
verification: c.Get("slack.verification", "NONE"),
|
verification: c.Get("slack.verification", "NONE"),
|
||||||
|
myBotID: c.Get("slack.botid", ""),
|
||||||
lastRecieved: time.Now(),
|
lastRecieved: time.Now(),
|
||||||
users: make(map[string]string),
|
users: make(map[string]string),
|
||||||
emoji: make(map[string]string),
|
emoji: make(map[string]string),
|
||||||
|
@ -69,6 +69,7 @@ func (s *SlackApp) RegisterEvent(f bot.Callback) {
|
||||||
|
|
||||||
func (s *SlackApp) Serve() error {
|
func (s *SlackApp) Serve() error {
|
||||||
s.populateEmojiList()
|
s.populateEmojiList()
|
||||||
|
|
||||||
http.HandleFunc("/evt", func(w http.ResponseWriter, r *http.Request) {
|
http.HandleFunc("/evt", func(w http.ResponseWriter, r *http.Request) {
|
||||||
buf := new(bytes.Buffer)
|
buf := new(bytes.Buffer)
|
||||||
buf.ReadFrom(r.Body)
|
buf.ReadFrom(r.Body)
|
||||||
|
@ -100,7 +101,6 @@ func (s *SlackApp) Serve() error {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
log.Println("[INFO] Server listening")
|
|
||||||
log.Fatal(http.ListenAndServe("0.0.0.0:1337", nil))
|
log.Fatal(http.ListenAndServe("0.0.0.0:1337", nil))
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -117,7 +117,7 @@ func (s *SlackApp) msgReceivd(msg *slackevents.MessageEvent) {
|
||||||
}
|
}
|
||||||
} else if msg.ThreadTimeStamp != "" {
|
} else if msg.ThreadTimeStamp != "" {
|
||||||
//we're throwing away some information here by not parsing the correct reply object type, but that's okay
|
//we're throwing away some information here by not parsing the correct reply object type, but that's okay
|
||||||
s.event(bot.Reply, s.buildLightReplyMessage(msg), msg.ThreadTimeStamp)
|
s.event(bot.Reply, s.buildMessage(msg), msg.ThreadTimeStamp)
|
||||||
} else {
|
} else {
|
||||||
log.Printf("THAT MESSAGE WAS HIDDEN: %+v", msg.Text)
|
log.Printf("THAT MESSAGE WAS HIDDEN: %+v", msg.Text)
|
||||||
}
|
}
|
||||||
|
@ -149,55 +149,26 @@ func (s *SlackApp) Send(kind bot.Kind, args ...interface{}) (string, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) sendMessageType(channel, message string, meMessage bool) (string, error) {
|
func (s *SlackApp) sendMessageType(channel, message string, meMessage bool) (string, error) {
|
||||||
postUrl := "https://slack.com/api/chat.postMessage"
|
ts, err := "", fmt.Errorf("")
|
||||||
if meMessage {
|
|
||||||
postUrl = "https://slack.com/api/chat.meMessage"
|
|
||||||
}
|
|
||||||
|
|
||||||
nick := s.config.Get("Nick", "bot")
|
nick := s.config.Get("Nick", "bot")
|
||||||
icon := s.config.Get("IconURL", "https://placekitten.com/128/128")
|
|
||||||
|
|
||||||
resp, err := http.PostForm(postUrl,
|
if meMessage {
|
||||||
url.Values{"token": {s.token},
|
_, ts, err = s.api.PostMessage(channel,
|
||||||
"username": {nick},
|
slack.MsgOptionUsername(nick),
|
||||||
"icon_url": {icon},
|
slack.MsgOptionText(message, false),
|
||||||
"channel": {channel},
|
slack.MsgOptionMeMessage())
|
||||||
"text": {message},
|
} else {
|
||||||
})
|
_, ts, err = s.api.PostMessage(channel,
|
||||||
|
slack.MsgOptionUsername(nick),
|
||||||
|
slack.MsgOptionText(message, false))
|
||||||
|
}
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error sending Slack message: %s", err)
|
log.Printf("Error sending message: %+v", err)
|
||||||
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
return ts, nil
|
||||||
resp.Body.Close()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error reading Slack API body: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println(string(body))
|
|
||||||
|
|
||||||
type MessageResponse struct {
|
|
||||||
OK bool `json:"ok"`
|
|
||||||
Timestamp string `json:"ts"`
|
|
||||||
Message struct {
|
|
||||||
BotID string `json:"bot_id"`
|
|
||||||
} `json:"message"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var mr MessageResponse
|
|
||||||
err = json.Unmarshal(body, &mr)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error parsing message response: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
if !mr.OK {
|
|
||||||
return "", errors.New("failure response received")
|
|
||||||
}
|
|
||||||
|
|
||||||
s.myBotID = mr.Message.BotID
|
|
||||||
|
|
||||||
return mr.Timestamp, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) sendMessage(channel, message string) (string, error) {
|
func (s *SlackApp) sendMessage(channel, message string) (string, error) {
|
||||||
|
@ -214,48 +185,12 @@ func (s *SlackApp) sendAction(channel, message string) (string, error) {
|
||||||
|
|
||||||
func (s *SlackApp) replyToMessageIdentifier(channel, message, identifier string) (string, error) {
|
func (s *SlackApp) replyToMessageIdentifier(channel, message, identifier string) (string, error) {
|
||||||
nick := s.config.Get("Nick", "bot")
|
nick := s.config.Get("Nick", "bot")
|
||||||
icon := s.config.Get("IconURL", "https://placekitten.com/128/128")
|
_, ts, err := s.api.PostMessage(channel,
|
||||||
|
slack.MsgOptionUsername(nick),
|
||||||
resp, err := http.PostForm("https://slack.com/api/chat.postMessage",
|
slack.MsgOptionText(message, false),
|
||||||
url.Values{"token": {s.token},
|
slack.MsgOptionMeMessage(),
|
||||||
"username": {nick},
|
slack.MsgOptionTS(identifier))
|
||||||
"icon_url": {icon},
|
return ts, err
|
||||||
"channel": {channel},
|
|
||||||
"text": {message},
|
|
||||||
"thread_ts": {identifier},
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("Error sending Slack reply: %s", err)
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
|
||||||
resp.Body.Close()
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("Error reading Slack API body: %s", err)
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Println(string(body))
|
|
||||||
|
|
||||||
type MessageResponse struct {
|
|
||||||
OK bool `json:"ok"`
|
|
||||||
Timestamp string `json:"ts"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var mr MessageResponse
|
|
||||||
err = json.Unmarshal(body, &mr)
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("Error parsing message response: %s", err)
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
|
|
||||||
if !mr.OK {
|
|
||||||
return "", fmt.Errorf("Got !OK from slack message response")
|
|
||||||
}
|
|
||||||
|
|
||||||
return mr.Timestamp, err
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) replyToMessage(channel, message string, replyTo msg.Message) (string, error) {
|
func (s *SlackApp) replyToMessage(channel, message string, replyTo msg.Message) (string, error) {
|
||||||
|
@ -264,30 +199,23 @@ func (s *SlackApp) replyToMessage(channel, message string, replyTo msg.Message)
|
||||||
|
|
||||||
func (s *SlackApp) react(channel, reaction string, message msg.Message) (string, error) {
|
func (s *SlackApp) react(channel, reaction string, message msg.Message) (string, error) {
|
||||||
log.Printf("Reacting in %s: %s", channel, reaction)
|
log.Printf("Reacting in %s: %s", channel, reaction)
|
||||||
resp, err := http.PostForm("https://slack.com/api/reactions.add",
|
ref := slack.ItemRef{
|
||||||
url.Values{"token": {s.token},
|
Channel: channel,
|
||||||
"name": {reaction},
|
Timestamp: message.AdditionalData["RAW_SLACK_TIMESTAMP"],
|
||||||
"channel": {channel},
|
|
||||||
"timestamp": {message.AdditionalData["RAW_SLACK_TIMESTAMP"]}})
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("reaction failed: %s", err)
|
|
||||||
return "", err
|
|
||||||
}
|
}
|
||||||
return "", checkReturnStatus(resp)
|
err := s.api.AddReaction(reaction, ref)
|
||||||
|
return "", err
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) edit(channel, newMessage, identifier string) (string, error) {
|
func (s *SlackApp) edit(channel, newMessage, identifier string) (string, error) {
|
||||||
log.Printf("Editing in (%s) %s: %s", identifier, channel, newMessage)
|
log.Printf("Editing in (%s) %s: %s", identifier, channel, newMessage)
|
||||||
resp, err := http.PostForm("https://slack.com/api/chat.update",
|
nick := s.config.Get("Nick", "bot")
|
||||||
url.Values{"token": {s.token},
|
_, ts, err := s.api.PostMessage(channel,
|
||||||
"channel": {channel},
|
slack.MsgOptionUsername(nick),
|
||||||
"text": {newMessage},
|
slack.MsgOptionText(newMessage, false),
|
||||||
"ts": {identifier}})
|
slack.MsgOptionMeMessage(),
|
||||||
if err != nil {
|
slack.MsgOptionUpdate(identifier))
|
||||||
err := fmt.Errorf("edit failed: %s", err)
|
return ts, err
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
return "", checkReturnStatus(resp)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) GetEmojiList() map[string]string {
|
func (s *SlackApp) GetEmojiList() map[string]string {
|
||||||
|
@ -295,30 +223,21 @@ func (s *SlackApp) GetEmojiList() map[string]string {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) populateEmojiList() {
|
func (s *SlackApp) populateEmojiList() {
|
||||||
resp, err := http.PostForm("https://slack.com/api/emoji.list",
|
if s.userToken == "NONE" {
|
||||||
url.Values{"token": {s.token}})
|
log.Println("Cannot get emoji list without slack.usertoken")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
dbg := slack.OptionDebug(true)
|
||||||
|
api := slack.New(s.userToken)
|
||||||
|
dbg(api)
|
||||||
|
|
||||||
|
em, err := api.GetEmoji()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error retrieving emoji list from Slack: %s", err)
|
log.Printf("Error retrieving emoji list from Slack: %s", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
s.emoji = em
|
||||||
resp.Body.Close()
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error reading Slack API body: %s", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
type EmojiListResponse struct {
|
|
||||||
OK bool `json:"ok"`
|
|
||||||
Emoji map[string]string `json:"emoji"`
|
|
||||||
}
|
|
||||||
|
|
||||||
var list EmojiListResponse
|
|
||||||
err = json.Unmarshal(body, &list)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatalf("Error parsing emoji list: %s", err)
|
|
||||||
}
|
|
||||||
s.emoji = list.Emoji
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// I think it's horseshit that I have to do this
|
// I think it's horseshit that I have to do this
|
||||||
|
@ -329,27 +248,6 @@ func slackTStoTime(t string) time.Time {
|
||||||
return time.Unix(sec, nsec)
|
return time.Unix(sec, nsec)
|
||||||
}
|
}
|
||||||
|
|
||||||
func checkReturnStatus(response *http.Response) error {
|
|
||||||
type Response struct {
|
|
||||||
OK bool `json:"ok"`
|
|
||||||
}
|
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(response.Body)
|
|
||||||
response.Body.Close()
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("Error reading Slack API body: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
var resp Response
|
|
||||||
err = json.Unmarshal(body, &resp)
|
|
||||||
if err != nil {
|
|
||||||
err := fmt.Errorf("Error parsing message response: %s", err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
var urlDetector = regexp.MustCompile(`<(.+)://([^|^>]+).*>`)
|
var urlDetector = regexp.MustCompile(`<(.+)://([^|^>]+).*>`)
|
||||||
|
|
||||||
// Convert a slackMessage to a msg.Message
|
// Convert a slackMessage to a msg.Message
|
||||||
|
@ -386,39 +284,6 @@ func (s *SlackApp) buildMessage(m *slackevents.MessageEvent) msg.Message {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SlackApp) buildLightReplyMessage(m *slackevents.MessageEvent) msg.Message {
|
|
||||||
text := html.UnescapeString(m.Text)
|
|
||||||
|
|
||||||
text = fixText(s.getUser, text)
|
|
||||||
|
|
||||||
isCmd, text := bot.IsCmd(s.config, text)
|
|
||||||
|
|
||||||
isAction := m.SubType == "me_message"
|
|
||||||
|
|
||||||
u, _ := s.getUser(m.User)
|
|
||||||
if m.Username != "" {
|
|
||||||
u = m.Username
|
|
||||||
}
|
|
||||||
|
|
||||||
tstamp := slackTStoTime(m.TimeStamp)
|
|
||||||
|
|
||||||
return msg.Message{
|
|
||||||
User: &user.User{
|
|
||||||
ID: m.User,
|
|
||||||
Name: u,
|
|
||||||
},
|
|
||||||
Body: text,
|
|
||||||
Raw: m.Text,
|
|
||||||
Channel: m.Channel,
|
|
||||||
Command: isCmd,
|
|
||||||
Action: isAction,
|
|
||||||
Time: tstamp,
|
|
||||||
AdditionalData: map[string]string{
|
|
||||||
"RAW_SLACK_TIMESTAMP": m.TimeStamp,
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get username for Slack user ID
|
// Get username for Slack user ID
|
||||||
func (s *SlackApp) getUser(id string) (string, bool) {
|
func (s *SlackApp) getUser(id string) (string, bool) {
|
||||||
if name, ok := s.users[id]; ok {
|
if name, ok := s.users[id]; ok {
|
||||||
|
|
Loading…
Reference in New Issue