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
|
// Copyright 2015 Robert S. Gerus. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package bot
import (
"encoding/json"
"fmt"
"log"
"strings"
"time"
"github.com/cloudflare/gokabinet/kt"
"github.com/arachnist/gorepost/irc"
)
var k *kt.Conn
type seenRecord struct {
Network string
Target string
Action string
Time time.Time
Text string
}
func seenrecord(output func(irc.Message), msg irc.Message) {
var target string
if msg.Params == nil || msg.Command == "QUIT" {
target = ""
} else {
target = msg.Params[0]
}
v := seenRecord{
Network: msg.Context["Network"],
Target: target,
Action: msg.Command,
Time: time.Now(),
Text: msg.Trailing,
}
b, _ := json.Marshal(v)
err := k.Set("seen/"+msg.Prefix.Name, b)
if err != nil {
log.Println("Context:", msg.Context, "error recording seen record:", err)
}
}
func seen(output func(irc.Message), msg irc.Message) {
var v seenRecord
var r string
args := strings.Split(msg.Trailing, " ")
if args[0] != ":seen" {
return
}
if len(args) < 2 {
return
}
b, err := k.GetBytes("seen/" + args[1])
if err == kt.ErrNotFound {
output(reply(msg, cfg.LookupString(msg.Context, "NotSeenMessage")))
return
} else if err != nil {
output(reply(msg, fmt.Sprint("error getting record for", args[1], err)))
return
}
_ = json.Unmarshal(b, &v)
r = fmt.Sprintf("Last seen %s on %s/%s at %v ", args[1], v.Network, v.Target, v.Time.Round(time.Second))
switch v.Action {
case "JOIN":
r += "joining"
case "PART":
r += fmt.Sprint("leaving: ", v.Text)
case "QUIT":
r += fmt.Sprint("quitting with reasson: ", v.Text)
case "PRIVMSG":
r += fmt.Sprint("saying: ", v.Text)
case "NOTICE":
r += fmt.Sprint("noticing: ", v.Text)
}
output(reply(msg, r))
}
func seenInit() {
var err error
var ktHost = cfg.LookupString(nil, "KTHost")
var ktPort = cfg.LookupInt(nil, "KTPort")
log.Println("seen: connecting to KT")
k, err = kt.NewConn(ktHost, ktPort, 4, 2*time.Second)
if err != nil {
log.Println("error connecting to kyoto tycoon", err)
return
}
log.Println("Registering callbacks")
addCallback("PRIVMSG", "seen", seen)
addCallback("PRIVMSG", "seenrecord", seenrecord)
addCallback("JOIN", "seenrecord", seenrecord)
addCallback("PART", "seenrecord", seenrecord)
addCallback("QUIT", "seenrecord", seenrecord)
addCallback("NOTICE", "seenrecord", seenrecord)
}
func init() {
log.Println("Defering \"papiez\" initialization")
addInit(seenInit)
}
|
