mirror of https://github.com/velour/catbase.git
downtime: remove dead plugin
This commit is contained in:
parent
c72dc7b2c8
commit
104ff85a0d
|
|
@ -1,233 +0,0 @@
|
||||||
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
|
|
||||||
|
|
||||||
package downtime
|
|
||||||
|
|
||||||
import (
|
|
||||||
"database/sql"
|
|
||||||
|
|
||||||
"github.com/jmoiron/sqlx"
|
|
||||||
"github.com/velour/catbase/bot"
|
|
||||||
"github.com/velour/catbase/bot/msg"
|
|
||||||
)
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"sort"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// This is a downtime plugin to monitor how much our users suck
|
|
||||||
|
|
||||||
type DowntimePlugin struct {
|
|
||||||
Bot bot.Bot
|
|
||||||
db *sqlx.DB
|
|
||||||
}
|
|
||||||
|
|
||||||
type idleEntry struct {
|
|
||||||
id sql.NullInt64
|
|
||||||
nick string
|
|
||||||
lastSeen time.Time
|
|
||||||
}
|
|
||||||
|
|
||||||
func (entry idleEntry) saveIdleEntry(db *sqlx.DB) error {
|
|
||||||
var err error
|
|
||||||
if entry.id.Valid {
|
|
||||||
log.Println("Updating downtime for: ", entry)
|
|
||||||
_, err = db.Exec(`update downtime set
|
|
||||||
nick=?, lastSeen=?
|
|
||||||
where id=?;`, entry.nick, entry.lastSeen.Unix(), entry.id.Int64)
|
|
||||||
} else {
|
|
||||||
log.Println("Inserting downtime for: ", entry)
|
|
||||||
_, err = db.Exec(`insert into downtime (nick, lastSeen)
|
|
||||||
values (?, ?)`, entry.nick, entry.lastSeen.Unix())
|
|
||||||
}
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func getIdleEntryByNick(db *sqlx.DB, nick string) (idleEntry, error) {
|
|
||||||
var id sql.NullInt64
|
|
||||||
var lastSeen sql.NullInt64
|
|
||||||
err := db.QueryRow(`select id, max(lastSeen) from downtime
|
|
||||||
where nick = ?`, nick).Scan(&id, &lastSeen)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("Error selecting downtime: ", err)
|
|
||||||
return idleEntry{}, err
|
|
||||||
}
|
|
||||||
if !id.Valid {
|
|
||||||
return idleEntry{
|
|
||||||
nick: nick,
|
|
||||||
lastSeen: time.Now(),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
return idleEntry{
|
|
||||||
id: id,
|
|
||||||
nick: nick,
|
|
||||||
lastSeen: time.Unix(lastSeen.Int64, 0),
|
|
||||||
}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func getAllIdleEntries(db *sqlx.DB) (idleEntries, error) {
|
|
||||||
rows, err := db.Query(`select id, nick, max(lastSeen) from downtime
|
|
||||||
group by nick`)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
entries := idleEntries{}
|
|
||||||
for rows.Next() {
|
|
||||||
var e idleEntry
|
|
||||||
err := rows.Scan(&e.id, &e.nick, &e.lastSeen)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
entries = append(entries, &e)
|
|
||||||
}
|
|
||||||
return entries, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type idleEntries []*idleEntry
|
|
||||||
|
|
||||||
func (ie idleEntries) Len() int {
|
|
||||||
return len(ie)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ie idleEntries) Less(i, j int) bool {
|
|
||||||
return ie[i].lastSeen.Before(ie[j].lastSeen)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (ie idleEntries) Swap(i, j int) {
|
|
||||||
ie[i], ie[j] = ie[j], ie[i]
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewDowntimePlugin creates a new DowntimePlugin with the Plugin interface
|
|
||||||
func New(bot bot.Bot) *DowntimePlugin {
|
|
||||||
p := DowntimePlugin{
|
|
||||||
Bot: bot,
|
|
||||||
db: bot.DB(),
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err := p.db.Exec(`create table if not exists downtime (
|
|
||||||
id integer primary key,
|
|
||||||
nick string,
|
|
||||||
lastSeen integer
|
|
||||||
);`)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal("Error creating downtime table: ", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &p
|
|
||||||
}
|
|
||||||
|
|
||||||
// Message responds to the bot hook on recieving messages.
|
|
||||||
// This function returns true if the plugin responds in a meaningful way to the users message.
|
|
||||||
// Otherwise, the function returns false and the bot continues execution of other plugins.
|
|
||||||
func (p *DowntimePlugin) Message(message msg.Message) bool {
|
|
||||||
// If it's a command and the payload is idle <nick>, give it. Log everything.
|
|
||||||
|
|
||||||
parts := strings.Fields(strings.ToLower(message.Body))
|
|
||||||
channel := message.Channel
|
|
||||||
ret := false
|
|
||||||
|
|
||||||
if len(parts) == 0 {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
if parts[0] == "idle" && len(parts) == 2 {
|
|
||||||
nick := parts[1]
|
|
||||||
// parts[1] must be the userid, or we don't know them
|
|
||||||
entry, err := getIdleEntryByNick(p.db, nick)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("Error getting idle entry: ", err)
|
|
||||||
}
|
|
||||||
if !entry.id.Valid {
|
|
||||||
// couldn't find em
|
|
||||||
p.Bot.Send(bot.Message, channel, fmt.Sprintf("Sorry, I don't know %s.", nick))
|
|
||||||
} else {
|
|
||||||
p.Bot.Send(bot.Message, channel, fmt.Sprintf("%s has been idle for: %s",
|
|
||||||
nick, time.Now().Sub(entry.lastSeen)))
|
|
||||||
}
|
|
||||||
ret = true
|
|
||||||
} else if parts[0] == "idle" && len(parts) == 1 {
|
|
||||||
// Find all idle times, report them.
|
|
||||||
entries, err := getAllIdleEntries(p.db)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("Error retrieving idle entries: ", err)
|
|
||||||
}
|
|
||||||
sort.Sort(entries)
|
|
||||||
tops := "The top entries are: "
|
|
||||||
for _, e := range entries {
|
|
||||||
|
|
||||||
// filter out ZNC entries and ourself
|
|
||||||
if strings.HasPrefix(e.nick, "*") || strings.ToLower(p.Bot.Config().Get("Nick", "bot")) == e.nick {
|
|
||||||
p.remove(e.nick)
|
|
||||||
} else {
|
|
||||||
tops = fmt.Sprintf("%s%s: %s ", tops, e.nick, time.Now().Sub(e.lastSeen))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
p.Bot.Send(bot.Message, channel, tops)
|
|
||||||
ret = true
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
p.record(strings.ToLower(message.User.Name))
|
|
||||||
|
|
||||||
return ret
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DowntimePlugin) record(user string) {
|
|
||||||
entry, err := getIdleEntryByNick(p.db, user)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("Error recording downtime: ", err)
|
|
||||||
}
|
|
||||||
entry.lastSeen = time.Now()
|
|
||||||
entry.saveIdleEntry(p.db)
|
|
||||||
log.Println("Inserted downtime for:", user)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DowntimePlugin) remove(user string) error {
|
|
||||||
_, err := p.db.Exec(`delete from downtime where nick = ?`, user)
|
|
||||||
if err != nil {
|
|
||||||
log.Println("Error removing downtime for user: ", user, err)
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
log.Println("Removed downtime for:", user)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Help responds to help requests. Every plugin must implement a help function.
|
|
||||||
func (p *DowntimePlugin) Help(channel string, parts []string) {
|
|
||||||
p.Bot.Send(bot.Message, channel, "Ask me how long one of your friends has been idele with, \"idle <nick>\"")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Empty event handler because this plugin does not do anything on event recv
|
|
||||||
func (p *DowntimePlugin) Event(kind string, message msg.Message) bool {
|
|
||||||
log.Println(kind, "\t", message)
|
|
||||||
if kind != "PART" && message.User.Name != p.Bot.Config().Get("Nick", "bot") {
|
|
||||||
// user joined, let's nail them for it
|
|
||||||
if kind == "NICK" {
|
|
||||||
p.record(strings.ToLower(message.Channel))
|
|
||||||
p.remove(strings.ToLower(message.User.Name))
|
|
||||||
} else {
|
|
||||||
p.record(strings.ToLower(message.User.Name))
|
|
||||||
}
|
|
||||||
} else if kind == "PART" || kind == "QUIT" {
|
|
||||||
p.remove(strings.ToLower(message.User.Name))
|
|
||||||
} else {
|
|
||||||
log.Println("Unknown event: ", kind, message.User, message)
|
|
||||||
p.record(strings.ToLower(message.User.Name))
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// Handler for bot's own messages
|
|
||||||
func (p *DowntimePlugin) BotMessage(message msg.Message) bool {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// Register any web URLs desired
|
|
||||||
func (p *DowntimePlugin) RegisterWeb() *string {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *DowntimePlugin) ReplyMessage(message msg.Message, identifier string) bool { return false }
|
|
||||||
Loading…
Reference in New Issue