ADDING THE FILES FOR THAT PROGRESS!
This commit is contained in:
parent
7f93845e9e
commit
12ae393998
10
README.md
Normal file
10
README.md
Normal file
@ -0,0 +1,10 @@
|
||||
|
||||
# mybot
|
||||
|
||||
`mybot` is an working Slack bot written in Go. Fork it and use it to build
|
||||
your very own cool Slack bot!
|
||||
|
||||
Check the [blog post](https://www.opsdash.com/blog/slack-bot-in-golang.html)
|
||||
for a description of mybot internals.
|
||||
|
||||
Follow us on Twitter today! [@therapidloop](https://twitter.com/therapidloop)
|
||||
188
slack.go
Normal file
188
slack.go
Normal file
@ -0,0 +1,188 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strconv"
|
||||
"strings"
|
||||
"sync/atomic"
|
||||
"time"
|
||||
|
||||
"golang.org/x/net/websocket"
|
||||
)
|
||||
|
||||
// Slack manages the slack connection and implements API calls
|
||||
type Slack struct {
|
||||
apiToken string
|
||||
id string
|
||||
socket *websocket.Conn
|
||||
alive bool
|
||||
}
|
||||
|
||||
// CreateSlack creates the slack object, opening a websocket
|
||||
func CreateSlack(token string) (*Slack, error) {
|
||||
s := Slack{apiToken: token}
|
||||
url := fmt.Sprintf("https://slack.com/api/rtm.start?token=%s", token)
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.StatusCode != 200 {
|
||||
err = fmt.Errorf("API request failed with code %d", resp.StatusCode)
|
||||
return nil, err
|
||||
}
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var respObj responseRtmStart
|
||||
err = json.Unmarshal(body, &respObj)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if !respObj.Ok {
|
||||
err = fmt.Errorf("Slack error: %s", respObj.Error)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
wsurl := respObj.URL
|
||||
s.id = respObj.Self.ID
|
||||
|
||||
ws, err := websocket.Dial(wsurl, "", "https://api.slack.com/")
|
||||
s.socket = ws
|
||||
s.keepAlive()
|
||||
|
||||
return &s, nil
|
||||
}
|
||||
|
||||
func (s *Slack) keepAlive() {
|
||||
s.alive = true
|
||||
go func() {
|
||||
for {
|
||||
if s.alive {
|
||||
s.ping()
|
||||
}
|
||||
time.Sleep(time.Second * 30)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
var counter uint64
|
||||
|
||||
func (s *Slack) ping() error {
|
||||
return s.postMessage(Message{Type: "ping"})
|
||||
}
|
||||
|
||||
func (s *Slack) postMessage(m Message) error {
|
||||
m.ID = atomic.AddUint64(&counter, 1)
|
||||
return websocket.JSON.Send(s.socket, m)
|
||||
}
|
||||
|
||||
func (s *Slack) getMessage() (Message, error) {
|
||||
var m Message
|
||||
err := websocket.JSON.Receive(s.socket, &m)
|
||||
|
||||
txtArr := strings.Split(m.Ts, ".")
|
||||
if t, err := strconv.Atoi(txtArr[0]); err == nil {
|
||||
rawts := int64(t)
|
||||
m.Time = time.Unix(0, rawts*1000000000)
|
||||
}
|
||||
|
||||
return m, err
|
||||
}
|
||||
|
||||
func (s *Slack) sendMessageToUser(uid string, msg string) error {
|
||||
// TODO: Actually send a direct message to uid
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Slack) getUserInfo(uid string) (*User, error) {
|
||||
url := fmt.Sprintf("https://slack.com/api/users.info?token=%s&user=%s", s.apiToken, uid)
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.StatusCode != 200 {
|
||||
err = fmt.Errorf("API request failed with code %d", resp.StatusCode)
|
||||
return nil, err
|
||||
}
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
writeToLog("User Lookup: \n")
|
||||
writeToLog(string(body))
|
||||
writeToLog("\n\n")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var respObj responseUserLookup
|
||||
err = json.Unmarshal(body, &respObj)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if !respObj.Ok {
|
||||
err = fmt.Errorf("Slack error: %s", respObj.Error)
|
||||
return nil, err
|
||||
}
|
||||
return respObj.User, nil
|
||||
}
|
||||
|
||||
func (s *Slack) getChannelInfo(cid string) (*Channel, error) {
|
||||
url := fmt.Sprintf("https://slack.com/api/channels.info?token=%s&channel=%s", s.apiToken, cid)
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if resp.StatusCode != 200 {
|
||||
err = fmt.Errorf("API request failed with code %d", resp.StatusCode)
|
||||
return nil, err
|
||||
}
|
||||
body, err := ioutil.ReadAll(resp.Body)
|
||||
resp.Body.Close()
|
||||
writeToLog("Channel Lookup: \n")
|
||||
writeToLog(string(body))
|
||||
writeToLog("\n\n")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var respObj responseChannelLookup
|
||||
err = json.Unmarshal(body, &respObj)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if !respObj.Ok {
|
||||
err = fmt.Errorf("Slack error: %s", respObj.Error)
|
||||
return nil, err
|
||||
}
|
||||
return respObj.Channel, nil
|
||||
}
|
||||
|
||||
// These structures represent the response of the Slack API events
|
||||
// Only some fields are included. The rest are ignored by json.Unmarshal.
|
||||
type responseRtmStart struct {
|
||||
Ok bool `json:"ok"`
|
||||
Error string `json:"error"`
|
||||
URL string `json:"url"`
|
||||
Self responseSelf `json:"self"`
|
||||
}
|
||||
|
||||
type responseSelf struct {
|
||||
ID string `json:"id"`
|
||||
}
|
||||
|
||||
type responseUserLookup struct {
|
||||
Ok bool `json:"ok"`
|
||||
Error string `json:"error"`
|
||||
User *User `json:"user"`
|
||||
}
|
||||
|
||||
type responseChannelLookup struct {
|
||||
Ok bool `json:"ok"`
|
||||
Error string `json:"error"`
|
||||
Channel *Channel `json:"channel"`
|
||||
}
|
||||
107
statbot.go
Normal file
107
statbot.go
Normal file
@ -0,0 +1,107 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
const programName = "statbot"
|
||||
|
||||
func main() {
|
||||
if len(os.Args) != 2 {
|
||||
fmt.Fprintf(os.Stderr, "usage: statbot slack-bot-token\n")
|
||||
os.Exit(1)
|
||||
}
|
||||
writeToLog("\n\n\n== " + time.Now().Format(time.RFC3339) + " ==\n")
|
||||
|
||||
// start a websocket-based Real Time API session
|
||||
var slack *Slack
|
||||
var err error
|
||||
|
||||
if err = initDatabase(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if slack, err = CreateSlack(os.Args[1]); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
fmt.Println("statbot ready, ^C exits")
|
||||
|
||||
for {
|
||||
// read each incoming message
|
||||
m, err := slack.getMessage()
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
processMessage(slack, &m)
|
||||
}
|
||||
}
|
||||
|
||||
func processMessage(slack *Slack, m *Message) {
|
||||
if mb, me := json.Marshal(m); me == nil {
|
||||
// Write the JSON representation to the log
|
||||
writeToLog(string(mb) + "\n")
|
||||
}
|
||||
// TODO: Handle reaction_added messages
|
||||
// TODO: Handle reaction_removed messages
|
||||
|
||||
if m.Type == "message" {
|
||||
// Check if we know who the user is
|
||||
usr, err := getUserInfo(m.User)
|
||||
// If the user information hasn't been updated in the past day, update it.
|
||||
if err != nil || usr.LastUpdated.IsZero() || time.Since(usr.LastUpdated) > (time.Hour*24) {
|
||||
if u, ue := slack.getUserInfo(m.User); ue == nil {
|
||||
saveUserInfo(u)
|
||||
}
|
||||
}
|
||||
// If 'channel' is defined, save the message to the 'channels' bucket
|
||||
if m.Channel != "" {
|
||||
// Check if we know what the channel is
|
||||
chnl, err := getChannelInfo(m.Channel)
|
||||
// If the channel information hasn't been updated in the past day, update it.
|
||||
if err != nil || chnl.LastUpdated.IsZero() || time.Since(chnl.LastUpdated) > (time.Hour*24) {
|
||||
// Either we don't have this channel, or it's a direct message
|
||||
if c, ce := slack.getChannelInfo(m.Channel); ce != nil {
|
||||
// Invalid channel, save as a direct message
|
||||
saveUserMessage(m.User, m)
|
||||
} else {
|
||||
// Save channel info
|
||||
saveChannelInfo(c)
|
||||
// And save the channel message
|
||||
saveChannelMessage(m.Channel, m)
|
||||
}
|
||||
}
|
||||
}
|
||||
// check if we're mentioned
|
||||
/*
|
||||
if m.Type == "message" && strings.HasPrefix(m.Text, "<@"+slack.id+">") {
|
||||
parts := strings.Fields(m.Text)
|
||||
if len(parts) == 3 && parts[1] == "stock" {
|
||||
// looks good, get the quote and reply with the result
|
||||
go func(m Message) {
|
||||
m.Text = getQuote(parts[2])
|
||||
postMessage(ws, m)
|
||||
}(m)
|
||||
// NOTE: the Message object is copied, this is intentional
|
||||
} else {
|
||||
// huh?
|
||||
m.Text = fmt.Sprintf("Beep boop beep, that does not compute\n")
|
||||
postMessage(ws, m)
|
||||
}
|
||||
}
|
||||
*/
|
||||
}
|
||||
}
|
||||
|
||||
func writeToLog(d string) {
|
||||
f, err := os.OpenFile("statbot.log", os.O_CREATE|os.O_APPEND|os.O_WRONLY, 0664)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
f.WriteString(d)
|
||||
f.Close()
|
||||
}
|
||||
680
statbot_model.go
Normal file
680
statbot_model.go
Normal file
@ -0,0 +1,680 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/boltdb/bolt"
|
||||
)
|
||||
|
||||
const databaseFile = programName + ".db"
|
||||
|
||||
var db *bolt.DB
|
||||
|
||||
func openDatabase() error {
|
||||
var err error
|
||||
db, err = bolt.Open(databaseFile, 0600, nil)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func closeDatabase() error {
|
||||
return db.Close()
|
||||
}
|
||||
|
||||
func initDatabase() error {
|
||||
openDatabase()
|
||||
|
||||
// Make sure the needed buckets exists
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
|
||||
// The config bucket holds config info for statbot
|
||||
_, err := tx.CreateBucketIfNotExists([]byte("config"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// The 'users' bucket holds user info, including direct messages
|
||||
_, err = tx.CreateBucketIfNotExists([]byte("users"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// The 'channels' bucket holds channel info, including channel messages
|
||||
_, err = tx.CreateBucketIfNotExists([]byte("channels"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
// Channel object
|
||||
type Channel struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
IsChannel bool `json:"is_channel"`
|
||||
CreatedRaw int `json:"created"`
|
||||
Created time.Time
|
||||
Creator string `json:"creator"`
|
||||
IsArchived bool `json:"is_archived"`
|
||||
IsGeneral bool `json:"is_general"`
|
||||
IsMember bool `json:"is_member"`
|
||||
LastReadRaw int `json:"last_read"`
|
||||
LastRead time.Time
|
||||
Latest *Message `json:"latest"`
|
||||
UnreadCount int `json:"unread_count"`
|
||||
UnreadCountDisplay int `json:"unread_count_display"`
|
||||
Members []string `json:"members"`
|
||||
Topic *ChannelTopic `json:"topic"`
|
||||
Purpose *ChannelTopic `json:"purpose"`
|
||||
|
||||
// LastUpdated is the last time we updated the info in the DB
|
||||
LastUpdated time.Time
|
||||
}
|
||||
|
||||
// ChannelTopic A simple 'Channel Topic' object
|
||||
// used for several things in the slack api (channel topic/purpose, etc.
|
||||
type ChannelTopic struct {
|
||||
Value string `json:"value"`
|
||||
Creator string `json:"creator"`
|
||||
LastSetRaw int `json:"last_set"`
|
||||
LastSet time.Time
|
||||
}
|
||||
|
||||
func getChannelInfo(chnl string) (*Channel, error) {
|
||||
openDatabase()
|
||||
ret := Channel{ID: chnl}
|
||||
err := db.View(func(tx *bolt.Tx) error {
|
||||
var b, chB, chIB *bolt.Bucket
|
||||
var err error
|
||||
|
||||
b = tx.Bucket([]byte("channels"))
|
||||
if b == nil {
|
||||
return fmt.Errorf("Error opening 'channels' bucket")
|
||||
}
|
||||
if chB = b.Bucket([]byte(chnl)); chB == nil {
|
||||
return fmt.Errorf("Error opening channel bucket (%s)", chnl)
|
||||
}
|
||||
if chIB = chB.Bucket([]byte("info")); chIB == nil {
|
||||
return fmt.Errorf("Error opening channel info bucket (%s/info)", chnl)
|
||||
}
|
||||
if ret.Name, err = bktGetString(chIB, "name"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsChannel, err = bktGetBool(chIB, "is_channel"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Created, err = bktGetTime(chIB, "created"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Creator, err = bktGetString(chIB, "creator"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsArchived, err = bktGetBool(chIB, "is_archived"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsGeneral, err = bktGetBool(chIB, "is_general"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsMember, err = bktGetBool(chIB, "is_member"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.LastRead, err = bktGetTime(chIB, "last_read"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.UnreadCount, err = bktGetInt(chIB, "unread_count"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.UnreadCountDisplay, err = bktGetInt(chIB, "unread_count_display"); err != nil {
|
||||
return err
|
||||
}
|
||||
// Get all of the members into a []string
|
||||
chMembersB := chIB.Bucket([]byte("members"))
|
||||
if chMembersB != nil {
|
||||
chMembersB.ForEach(func(k, v []byte) error {
|
||||
ret.Members = append(ret.Members, string(v))
|
||||
return nil
|
||||
})
|
||||
}
|
||||
ret.Topic = new(ChannelTopic)
|
||||
if ret.Topic.Value, err = bktGetString(chIB, "topic_value"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Topic.Creator, err = bktGetString(chIB, "topic_creator"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Topic.LastSet, err = bktGetTime(chIB, "topic_last_set"); err != nil {
|
||||
return err
|
||||
}
|
||||
ret.Purpose = new(ChannelTopic)
|
||||
if ret.Purpose.Value, err = bktGetString(chIB, "purpose_value"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Purpose.Creator, err = bktGetString(chIB, "purpose_creator"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Purpose.LastSet, err = bktGetTime(chIB, "purpose_last_set"); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
closeDatabase()
|
||||
return &ret, err
|
||||
}
|
||||
|
||||
func saveChannelInfo(chnl *Channel) error {
|
||||
openDatabase()
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("channel"))
|
||||
var err error
|
||||
var chB, chIB *bolt.Bucket
|
||||
if chB, err = b.CreateBucketIfNotExists([]byte(chnl.ID)); err != nil {
|
||||
return err
|
||||
}
|
||||
if chIB, err = chB.CreateBucketIfNotExists([]byte("info")); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(chIB, "id", chnl.ID); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(chIB, "name", chnl.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(chIB, "is_channel", chnl.IsChannel); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(chIB, "created", chnl.Created); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(chIB, "creator", chnl.Creator); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(chIB, "is_archived", chnl.IsArchived); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(chIB, "is_general", chnl.IsGeneral); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(chIB, "is_member", chnl.IsMember); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(chIB, "last_read", chnl.LastRead); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutInt(chIB, "unread_count", chnl.UnreadCount); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutInt(chIB, "unread_count_display", chnl.UnreadCountDisplay); err != nil {
|
||||
return err
|
||||
}
|
||||
chMembersB := chIB.Bucket([]byte("members"))
|
||||
if chMembersB != nil {
|
||||
// Recreate the Members bucket
|
||||
chIB.DeleteBucket([]byte("members"))
|
||||
}
|
||||
if chMembersB, err = chIB.CreateBucket([]byte("members")); err != nil {
|
||||
return err
|
||||
}
|
||||
mbIdx := 0
|
||||
for _, mm := range chnl.Members {
|
||||
idxKey := strconv.Itoa(mbIdx)
|
||||
if err = bktPutString(chMembersB, idxKey, mm); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if chnl.Topic != nil {
|
||||
if err = bktPutString(chIB, "topic_value", chnl.Topic.Value); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(chIB, "topic_creator", chnl.Topic.Creator); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(chIB, "topic_last_set", chnl.Topic.LastSet); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
if chnl.Purpose != nil {
|
||||
if err = bktPutString(chIB, "purpose_value", chnl.Topic.Value); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(chIB, "purpose_creator", chnl.Topic.Creator); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(chIB, "purpose_last_set", chnl.Topic.LastSet); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
err = bktPutTime(chIB, "last_updated", chnl.LastUpdated)
|
||||
return err
|
||||
})
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func saveChannelStat(channel string, key string, val string) error {
|
||||
openDatabase()
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("channels"))
|
||||
if chB, err := b.CreateBucketIfNotExists([]byte(channel)); err == nil {
|
||||
chB.Put([]byte(key), []byte(val))
|
||||
}
|
||||
return nil
|
||||
})
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func saveChannelMessage(channel string, message *Message) error {
|
||||
openDatabase()
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("channels"))
|
||||
var err error
|
||||
var chB, chMB, msgBkt *bolt.Bucket
|
||||
if chB, err = b.CreateBucketIfNotExists([]byte(channel)); err != nil {
|
||||
return err
|
||||
}
|
||||
if chMB, err = chB.CreateBucketIfNotExists([]byte("messages")); err != nil {
|
||||
return err
|
||||
}
|
||||
idx, _ := chMB.NextSequence()
|
||||
idxKey := []byte(strconv.FormatUint(idx, 10))
|
||||
if msgBkt, err = chMB.CreateBucketIfNotExists(idxKey); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "type", message.Type); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "user", message.User); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "text", message.Text); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "name", message.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(msgBkt, "time", message.Time); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func incrementChannelStat(channel string, key string) error {
|
||||
openDatabase()
|
||||
strRet, err := getChannelStat(channel, key)
|
||||
if err == nil {
|
||||
if i, err := strconv.Atoi(strRet); err == nil {
|
||||
i++
|
||||
return saveChannelStat(channel, key, fmt.Sprintf("%d", i))
|
||||
}
|
||||
return err
|
||||
}
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func decrementChannelStat(channel string, key string) error {
|
||||
openDatabase()
|
||||
strRet, err := getChannelStat(channel, key)
|
||||
if err == nil {
|
||||
if i, err := strconv.Atoi(strRet); err == nil {
|
||||
i--
|
||||
return saveChannelStat(channel, key, fmt.Sprintf("%d", i))
|
||||
}
|
||||
return err
|
||||
}
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func getChannelStat(channel string, key string) (string, error) {
|
||||
openDatabase()
|
||||
var ret string
|
||||
err := db.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("channels"))
|
||||
if chB := b.Bucket([]byte(channel)); chB != nil {
|
||||
ret = string(chB.Get([]byte(key)))
|
||||
}
|
||||
return nil
|
||||
})
|
||||
closeDatabase()
|
||||
return ret, err
|
||||
}
|
||||
|
||||
// Message We save the message struct into the db,
|
||||
// it's also what we send/receive from slack
|
||||
type Message struct {
|
||||
ID uint64 `json:"id"`
|
||||
Type string `json:"type"`
|
||||
Channel string `json:"channel"`
|
||||
User string `json:"user"`
|
||||
Name string `json:"name"`
|
||||
Text string `json:"text"`
|
||||
Ts string `json:"ts"`
|
||||
Time time.Time
|
||||
}
|
||||
|
||||
// User object
|
||||
type User struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Deleted bool `json:"deleted"`
|
||||
Status string `json:"status"`
|
||||
Color string `json:"color"`
|
||||
RealName string `json:"real_name"`
|
||||
TZ string `json:"tz"`
|
||||
TZLabel string `json:"tz_label"`
|
||||
TZOffset int `json:"tz_offset"`
|
||||
IsAdmin bool `json:"is_admin"`
|
||||
IsOwner bool `json:"is_owner"`
|
||||
IsPrimaryOwner bool `json:"is_primary_owner"`
|
||||
IsRestricted bool `json:"is_restricted"`
|
||||
IsUltraRestricted bool `json:"is_ultra_restricted"`
|
||||
IsBot bool `json:"is_bot"`
|
||||
HasFiles bool `json:"has_files"`
|
||||
|
||||
// LastUpdated is the last time we updated the info in the DB
|
||||
LastUpdated time.Time
|
||||
}
|
||||
|
||||
func getUserInfo(usr string) (*User, error) {
|
||||
openDatabase()
|
||||
ret := User{ID: usr}
|
||||
err := db.View(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("users"))
|
||||
if uB := b.Bucket([]byte(usr)); uB != nil {
|
||||
var err error
|
||||
if ret.Name, err = bktGetString(uB, "name"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Deleted, err = bktGetBool(uB, "deleted"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Status, err = bktGetString(uB, "status"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.Color, err = bktGetString(uB, "color"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.RealName, err = bktGetString(uB, "real_name"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.TZ, err = bktGetString(uB, "tz"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.TZLabel, err = bktGetString(uB, "tz_label"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.TZOffset, err = bktGetInt(uB, "tz_offset"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsAdmin, err = bktGetBool(uB, "is_admin"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsOwner, err = bktGetBool(uB, "is_owner"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsPrimaryOwner, err = bktGetBool(uB, "is_primary_owner"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsRestricted, err = bktGetBool(uB, "is_restricted"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsUltraRestricted, err = bktGetBool(uB, "is_ultra_restricted"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.IsBot, err = bktGetBool(uB, "is_bot"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.HasFiles, err = bktGetBool(uB, "has_files"); err != nil {
|
||||
return err
|
||||
}
|
||||
if ret.LastUpdated, err = bktGetTime(uB, "last_updated"); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
return fmt.Errorf("User (%s) bucket not found.", usr)
|
||||
})
|
||||
closeDatabase()
|
||||
return &ret, err
|
||||
}
|
||||
|
||||
func saveUserInfo(usr *User) error {
|
||||
openDatabase()
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("users"))
|
||||
var err error
|
||||
var uB, uIB *bolt.Bucket
|
||||
if uB, err = b.CreateBucketIfNotExists([]byte(usr.ID)); err != nil {
|
||||
return err
|
||||
}
|
||||
if uIB, err = uB.CreateBucketIfNotExists([]byte("info")); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "id", usr.ID); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "name", usr.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "deleted", usr.Deleted); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "status", usr.Status); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "color", usr.Color); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "real_name", usr.RealName); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "tz", usr.TZ); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(uIB, "tz_label", usr.TZLabel); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutInt(uIB, "tz_offset", usr.TZOffset); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_admin", usr.IsAdmin); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_owner", usr.IsOwner); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_primary_owner", usr.IsPrimaryOwner); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_restricted", usr.IsRestricted); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_ultra_restricted", usr.IsUltraRestricted); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "is_bot", usr.IsBot); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutBool(uIB, "has_files", usr.HasFiles); err != nil {
|
||||
return err
|
||||
}
|
||||
err = bktPutTime(uIB, "last_udpated", usr.LastUpdated)
|
||||
return err
|
||||
})
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func saveUserMessage(user string, message *Message) error {
|
||||
openDatabase()
|
||||
err := db.Update(func(tx *bolt.Tx) error {
|
||||
b := tx.Bucket([]byte("users"))
|
||||
var err error
|
||||
var uB, uMB, msgBkt *bolt.Bucket
|
||||
if uB, err = b.CreateBucketIfNotExists([]byte(user)); err != nil {
|
||||
return err
|
||||
}
|
||||
if uMB, err = uB.CreateBucketIfNotExists([]byte("messages")); err != nil {
|
||||
return err
|
||||
}
|
||||
idx, _ := uMB.NextSequence()
|
||||
idxKey := []byte(strconv.FormatUint(idx, 10))
|
||||
if msgBkt, err = uMB.CreateBucketIfNotExists(idxKey); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "type", message.Type); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "text", message.Text); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutString(msgBkt, "name", message.Name); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = bktPutTime(msgBkt, "time", message.Time); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
closeDatabase()
|
||||
return err
|
||||
}
|
||||
|
||||
func bktGetBucket(b *bolt.Bucket, key string) (*bolt.Bucket, error) {
|
||||
bkt := b.Bucket([]byte(key))
|
||||
if bkt != nil {
|
||||
return nil, fmt.Errorf("Bucket (%s) not found.", key)
|
||||
}
|
||||
return bkt, nil
|
||||
}
|
||||
|
||||
func bktGetBytes(b *bolt.Bucket, key string) ([]byte, error) {
|
||||
bb := b.Get([]byte(key))
|
||||
if bb == nil {
|
||||
return nil, fmt.Errorf("Error reading key (%s)", key)
|
||||
}
|
||||
return bb, nil
|
||||
}
|
||||
|
||||
func bktGetString(b *bolt.Bucket, key string) (string, error) {
|
||||
bb, err := bktGetBytes(b, key)
|
||||
return string(bb), err
|
||||
}
|
||||
|
||||
func bktPutString(b *bolt.Bucket, key string, val string) error {
|
||||
return b.Put([]byte(key), []byte(val))
|
||||
}
|
||||
|
||||
func bktGetBool(b *bolt.Bucket, key string) (bool, error) {
|
||||
bb, err := bktGetBytes(b, key)
|
||||
return (string(bb) == "true"), err
|
||||
}
|
||||
|
||||
func bktPutBool(b *bolt.Bucket, key string, val bool) error {
|
||||
if val {
|
||||
return b.Put([]byte(key), []byte("true"))
|
||||
}
|
||||
return b.Put([]byte(key), []byte("false"))
|
||||
}
|
||||
|
||||
func bktGetInt(b *bolt.Bucket, key string) (int, error) {
|
||||
bb, err := bktGetBytes(b, key)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return strconv.Atoi(string(bb))
|
||||
}
|
||||
|
||||
func bktPutInt(b *bolt.Bucket, key string, val int) error {
|
||||
return b.Put([]byte(key), []byte(fmt.Sprintf("%d", val)))
|
||||
}
|
||||
|
||||
func bktGetTime(b *bolt.Bucket, key string) (time.Time, error) {
|
||||
bs, err := bktGetString(b, key)
|
||||
if err != nil {
|
||||
return time.Time{}, err
|
||||
}
|
||||
return time.Parse(time.RFC3339, bs)
|
||||
}
|
||||
|
||||
func bktPutTime(b *bolt.Bucket, key string, val time.Time) error {
|
||||
bs := val.Format(time.RFC3339)
|
||||
return b.Put([]byte(key), []byte(bs))
|
||||
}
|
||||
|
||||
// All user stats are saved in the db like so:
|
||||
// users (bucket)
|
||||
// |- username1 (bucket)
|
||||
// | |-info (bucket)
|
||||
// | | |-userinfo1 (pair)
|
||||
// | | \-userinfo2 (pair)
|
||||
// | |
|
||||
// | |-messages (bucket)
|
||||
// | | |-message1 (message bucket)
|
||||
// | | \-message2 (message bucket)
|
||||
// | |
|
||||
// | \-stats (bucket)
|
||||
// | |-stat1 (pair)
|
||||
// | \-stat2 (pair)
|
||||
// |
|
||||
// \- username2 (bucket)
|
||||
// |-info (bucket)
|
||||
// | |-userinfo1 (pair)
|
||||
// | \-userinfo2 (pair)
|
||||
// |
|
||||
// |-messages (bucket)
|
||||
// | |-message1 (message bucket)
|
||||
// | \-message2 (message bucket)
|
||||
// |
|
||||
// \-stats (bucket)
|
||||
// |-stat1 (pair)
|
||||
// \-stat2 (pair)
|
||||
//
|
||||
// And channel stats are saved in the db like so:
|
||||
// channels (bucket)
|
||||
// |- channel1 (bucket)
|
||||
// | |-info (bucket)
|
||||
// | | |-chaninfo1 (pair)
|
||||
// | | \-chaninfo2 (pair)
|
||||
// | |
|
||||
// | |-messages (bucket)
|
||||
// | | |-message1 (message bucket)
|
||||
// | | \-message2 (message bucket)
|
||||
// | |
|
||||
// | \-stats (bucket)
|
||||
// | |-stat1 (pair)
|
||||
// | \-stat2 (pair)
|
||||
// |
|
||||
// \- channel2 (bucket)
|
||||
// |-info (bucket)
|
||||
// | |-chaninfo1 (pair)
|
||||
// | \-chaninfo2 (pair)
|
||||
// |
|
||||
// |-messages (bucket)
|
||||
// | |-message1 (message bucket)
|
||||
// | \-message2 (message bucket)
|
||||
// |
|
||||
// \-stats (bucket)
|
||||
// |-stat1 (pair)
|
||||
// \-stat2 (pair)
|
||||
//
|
||||
// A (message bucket) looks like this:
|
||||
// messageid (unique to parent bucket)
|
||||
// |-Type (slack event type)
|
||||
// |-User (slack user ID)
|
||||
// |-Text (Text of the message)
|
||||
// \-Time RFC3339 format timestamp
|
||||
Loading…
Reference in New Issue
Block a user