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 "" } }