mirror of https://github.com/velour/catbase.git
counter: add leaderboards
This commit is contained in:
parent
22bca5bf4d
commit
e19d982fe5
|
@ -45,6 +45,32 @@ func GetItems(db *sqlx.DB, nick string) ([]Item, error) {
|
||||||
return items, nil
|
return items, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func LeaderAll(db *sqlx.DB) ([]Item, error) {
|
||||||
|
s := `select id,item,nick,max(count) as count from counter group by item having count(nick) > 1 and max(count) > 1 order by count desc`
|
||||||
|
var items []Item
|
||||||
|
err := db.Select(&items, s)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for i := range items {
|
||||||
|
items[i].DB = db
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func Leader(db *sqlx.DB, itemName string) ([]Item, error) {
|
||||||
|
s := `select * from counter where item=? order by count desc`
|
||||||
|
var items []Item
|
||||||
|
err := db.Select(&items, s, itemName)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for i := range items {
|
||||||
|
items[i].DB = db
|
||||||
|
}
|
||||||
|
return items, nil
|
||||||
|
}
|
||||||
|
|
||||||
// GetItem returns a specific counter for a subject
|
// GetItem returns a specific counter for a subject
|
||||||
func GetItem(db *sqlx.DB, nick, itemName string) (Item, error) {
|
func GetItem(db *sqlx.DB, nick, itemName string) (Item, error) {
|
||||||
var item Item
|
var item Item
|
||||||
|
@ -136,7 +162,36 @@ func (p *CounterPlugin) Message(message msg.Message) bool {
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
if tea, _ := regexp.MatchString("(?i)^tea\\. [^.]*\\. ((hot)|(iced))\\.?$", message.Body); tea {
|
if parts[0] == "leaderboard" {
|
||||||
|
var cmd func() ([]Item, error)
|
||||||
|
itNameTxt := ""
|
||||||
|
|
||||||
|
if len(parts) == 1 {
|
||||||
|
cmd = func() ([]Item, error) { return LeaderAll(p.DB) }
|
||||||
|
} else {
|
||||||
|
itNameTxt = fmt.Sprintf(" for %s", parts[1])
|
||||||
|
cmd = func() ([]Item, error) { return Leader(p.DB, parts[1]) }
|
||||||
|
}
|
||||||
|
|
||||||
|
its, err := cmd()
|
||||||
|
if err != nil {
|
||||||
|
log.Println(err)
|
||||||
|
return false
|
||||||
|
} else if len(its) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
out := fmt.Sprintf("Leaderboard%s:\n", itNameTxt)
|
||||||
|
for _, it := range its {
|
||||||
|
out += fmt.Sprintf("%s with %d %s\n",
|
||||||
|
it.Nick,
|
||||||
|
it.Count,
|
||||||
|
it.Item,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
p.Bot.SendMessage(channel, out)
|
||||||
|
return true
|
||||||
|
} else if tea, _ := regexp.MatchString("(?i)^tea\\. [^.]*\\. ((hot)|(iced))\\.?$", message.Body); tea {
|
||||||
item, err := GetItem(p.DB, nick, ":tea:")
|
item, err := GetItem(p.DB, nick, ":tea:")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("Error finding item %s.%s: %s.", nick, ":tea:", err)
|
log.Printf("Error finding item %s.%s: %s.", nick, ":tea:", err)
|
||||||
|
|
Loading…
Reference in New Issue