f2037c5d52
There's no reason why messgeLogger needs access to the whole connection, the network is enough.
126 lines
3.1 KiB
Go
126 lines
3.1 KiB
Go
package soju
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"gopkg.in/irc.v3"
|
|
)
|
|
|
|
type messageLogger struct {
|
|
network *network
|
|
entity string
|
|
|
|
path string
|
|
file *os.File
|
|
}
|
|
|
|
func newMessageLogger(network *network, entity string) *messageLogger {
|
|
return &messageLogger{
|
|
network: network,
|
|
entity: entity,
|
|
}
|
|
}
|
|
|
|
func logPath(network *network, entity string, t time.Time) string {
|
|
user := network.user
|
|
srv := user.srv
|
|
|
|
// TODO: handle/forbid network/entity names with illegal path characters
|
|
year, month, day := t.Date()
|
|
filename := fmt.Sprintf("%04d-%02d-%02d.log", year, month, day)
|
|
return filepath.Join(srv.LogPath, user.Username, network.GetName(), entity, filename)
|
|
}
|
|
|
|
func (ml *messageLogger) Append(msg *irc.Message) error {
|
|
s := formatMessage(msg)
|
|
if s == "" {
|
|
return nil
|
|
}
|
|
|
|
// TODO: parse time from msg.Tags["time"], if available
|
|
|
|
// TODO: enforce maximum open file handles (LRU cache of file handles)
|
|
// TODO: handle non-monotonic clock behaviour
|
|
now := time.Now()
|
|
path := logPath(ml.network, ml.entity, now)
|
|
if ml.path != path {
|
|
if ml.file != nil {
|
|
ml.file.Close()
|
|
}
|
|
|
|
dir := filepath.Dir(path)
|
|
if err := os.MkdirAll(dir, 0700); err != nil {
|
|
return fmt.Errorf("failed to create logs directory %q: %v", dir, err)
|
|
}
|
|
|
|
f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0600)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to open log file %q: %v", path, err)
|
|
}
|
|
|
|
ml.path = path
|
|
ml.file = f
|
|
}
|
|
|
|
_, err := fmt.Fprintf(ml.file, "[%02d:%02d:%02d] %s\n", now.Hour(), now.Minute(), now.Second(), s)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to log message to %q: %v", ml.path, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (ml *messageLogger) Close() error {
|
|
if ml.file == nil {
|
|
return nil
|
|
}
|
|
return ml.file.Close()
|
|
}
|
|
|
|
// formatMessage formats a message log line. It assumes a well-formed IRC
|
|
// message.
|
|
func formatMessage(msg *irc.Message) string {
|
|
switch strings.ToUpper(msg.Command) {
|
|
case "NICK":
|
|
return fmt.Sprintf("*** %s is now known as %s", msg.Prefix.Name, msg.Params[0])
|
|
case "JOIN":
|
|
return fmt.Sprintf("*** Joins: %s (%s@%s)", msg.Prefix.Name, msg.Prefix.User, msg.Prefix.Host)
|
|
case "PART":
|
|
var reason string
|
|
if len(msg.Params) > 1 {
|
|
reason = msg.Params[1]
|
|
}
|
|
return fmt.Sprintf("*** Parts: %s (%s@%s) (%s)", msg.Prefix.Name, msg.Prefix.User, msg.Prefix.Host, reason)
|
|
case "KICK":
|
|
nick := msg.Params[1]
|
|
var reason string
|
|
if len(msg.Params) > 2 {
|
|
reason = msg.Params[2]
|
|
}
|
|
return fmt.Sprintf("*** %s was kicked by %s (%s)", nick, msg.Prefix.Name, reason)
|
|
case "QUIT":
|
|
var reason string
|
|
if len(msg.Params) > 0 {
|
|
reason = msg.Params[0]
|
|
}
|
|
return fmt.Sprintf("*** Quits: %s (%s@%s) (%s)", msg.Prefix.Name, msg.Prefix.User, msg.Prefix.Host, reason)
|
|
case "TOPIC":
|
|
var topic string
|
|
if len(msg.Params) > 1 {
|
|
topic = msg.Params[1]
|
|
}
|
|
return fmt.Sprintf("*** %s changes topic to '%s'", msg.Prefix.Name, topic)
|
|
case "MODE":
|
|
return fmt.Sprintf("*** %s sets mode: %s", msg.Prefix.Name, strings.Join(msg.Params[1:], " "))
|
|
case "NOTICE":
|
|
return fmt.Sprintf("-%s- %s", msg.Prefix.Name, msg.Params[1])
|
|
case "PRIVMSG":
|
|
return fmt.Sprintf("<%s> %s", msg.Prefix.Name, msg.Params[1])
|
|
default:
|
|
return ""
|
|
}
|
|
}
|