forked from johansundell/cocbot
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlastdonations.go
More file actions
122 lines (113 loc) · 2.6 KB
/
lastdonations.go
File metadata and controls
122 lines (113 loc) · 2.6 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
package main
import (
"context"
"fmt"
"strings"
"time"
)
type donations struct {
name string
min int64
amount int64
}
func (v donations) String() string {
return fmt.Sprintf("%s donated %d troops %s ago\n", v.name, v.amount, (time.Duration(v.min) * time.Second).String())
}
func init() {
key := commandFunc{"!last donations", "To list the last 10 donations", "", categoryStats}
lockMap.Lock()
defer lockMap.Unlock()
botFuncs[key] = func(ctx context.Context, command string) (string, error) {
if key.command == command {
don, err := getDonations(10)
if err != nil {
return "", err
}
msg := ""
for _, v := range don {
msg += v.String()
}
rec, err := getReceive(10)
if err != nil {
return "", err
}
msg += "--------\n"
for _, v := range rec {
msg += v.String()
}
return msg, nil
}
return "", nil
}
key2 := commandFunc{"!last donations for [name]", "To see the last donations by that member", "", categoryStats}
botFuncs[key2] = func(ctx context.Context, command string) (string, error) {
cmd := strings.Replace(key2.command, " [name]", "", -1)
if strings.HasPrefix(command, cmd) {
name := strings.TrimSpace(command[len(cmd):])
don, err := getUserDonations(name, 10)
if err != nil {
return "", err
}
msg := ""
if len(don) != 0 {
msg += "These are the last donations by " + name + "\n"
for _, v := range don {
msg += v.String()
}
}
return msg, nil
}
return "", nil
}
}
var queryDonations = `
SELECT
(d.current_donations - d.prev_donations) AS diff,
m.name,
TIME_TO_SEC(TIMEDIFF(NOW(), d.ts)) AS since
FROM
donations d
JOIN
members m ON m.member_id = d.member_id
ORDER BY d.donate_id DESC
LIMIT 0 , ?
`
func getDonations(numToFetch int) ([]donations, error) {
rows, err := db.Query(queryDonations, numToFetch)
if err != nil {
return nil, err
}
don := []donations{}
for rows.Next() {
d := donations{}
rows.Scan(&d.amount, &d.name, &d.min)
don = append(don, d)
}
return don, nil
}
var queryUserDonations = `
SELECT
(d.current_donations - d.prev_donations) AS diff,
m.name,
TIME_TO_SEC(TIMEDIFF(NOW(), d.ts)) AS since
FROM
donations d
JOIN
members m ON m.member_id = d.member_id
WHERE m.name LIKE ?
ORDER BY d.donate_id DESC
LIMIT 0 , ?
`
func getUserDonations(name string, numToFetch int) ([]donations, error) {
rows, err := db.Query(queryUserDonations, name, numToFetch)
if err != nil {
return nil, err
}
don := []donations{}
for rows.Next() {
d := donations{}
rows.Scan(&d.amount, &d.name, &d.min)
don = append(don, d)
}
return don, nil
}