nmdc-log-service/main.go

147 lines
3.5 KiB
Go
Raw Normal View History

2016-04-02 01:04:58 +00:00
package main
import (
"flag"
"fmt"
"libnmdc"
"os"
"path/filepath"
"regexp"
"strings"
"time"
)
var BaseDir string = "."
var PMResponse string = ""
var LogConnectionState bool
var DebugMode bool
var CharacterMatcher *regexp.Regexp
func init() {
CharacterMatcher = regexp.MustCompile("[^a-zA-Z0-9]")
}
func GetDirectoryNameForHub(hub string) string {
return filepath.Join(BaseDir, CharacterMatcher.ReplaceAllString(hub, "_"))
}
func LogMessage(hub, message string) {
instant := time.Now() // using system timezone
// Log file path
monthStamp := instant.Format("2006-01")
logFilePath := filepath.Join(GetDirectoryNameForHub(hub), monthStamp+".log")
// Produce a timestamp
timeStamp := instant.Format("2006-01-02 15:04:05")
fh, err := os.OpenFile(logFilePath, os.O_APPEND|os.O_CREATE, 0644)
if err != nil {
fmt.Fprintf(os.Stderr, "Couldn't open file '%s': %s\n", logFilePath, err.Error())
return
}
defer fh.Close()
_, err = fh.WriteString("[" + timeStamp + "] " + message + "\n")
if err != nil {
fmt.Fprintf(os.Stderr, "Error writing to file '%s': %s\n", logFilePath, err.Error())
return
}
}
func HubWorker(addr, nick, password string) {
opts := libnmdc.HubConnectionOptions{
Address: libnmdc.HubAddress(addr),
Self: libnmdc.UserInfo{Nick: nick},
NickPassword: password,
}
hub := opts.Connect()
for {
event := <-hub.OnEvent
if DebugMode {
fmt.Printf("DEBUG: %s %v\n", addr, event)
}
switch event.EventType {
case libnmdc.EVENT_CONNECTION_STATE_CHANGED:
if LogConnectionState {
if len(event.Message) > 0 {
LogMessage(addr, "* "+event.StateChange.Format()+" ("+event.Message+")")
} else {
LogMessage(addr, "* "+event.StateChange.Format())
}
}
case libnmdc.EVENT_PUBLIC:
LogMessage(addr, "<"+event.Nick+"> "+event.Message)
case libnmdc.EVENT_PRIVATE:
fmt.Printf("Got PM %v\n", event)
hub.SayPrivate(event.Nick, PMResponse)
case libnmdc.EVENT_SYSTEM_MESSAGE_FROM_CONN, libnmdc.EVENT_SYSTEM_MESSAGE_FROM_HUB:
if strings.HasPrefix(event.Message, "* ") {
LogMessage(addr, event.Message)
} else {
LogMessage(addr, "* "+event.Message)
}
}
}
}
func main() {
// Parse arguments
hubs := flag.String("Server", "", "Addresses to connect to (comma-separated)")
nick := flag.String("Nick", "nmdc-log-service", "Nick")
password := flag.String("Password", "", "Registered nick password")
flag.StringVar(&BaseDir, "Dir", ".", "Output directory")
2016-04-02 01:04:58 +00:00
flag.BoolVar(&LogConnectionState, "LogConnectionState", true, "Include connection state changes in log")
flag.StringVar(&PMResponse, "PMResponse", "This is an automated service. For enquiries, please contact an administrator.", "Message to respond with on PM")
flag.BoolVar(&DebugMode, "Debug", false, "Print additional information on stdout")
flag.Parse()
// Assert dir exists
dinfo, err := os.Stat(BaseDir)
if err != nil {
fmt.Fprintf(os.Stderr, "FATAL: %s", err.Error())
os.Exit(1)
}
if !dinfo.IsDir() {
fmt.Fprintf(os.Stderr, "FATAL: Path '%s' is not a directory\n", BaseDir)
}
// Launch loggers
all_hubs := strings.Split(*hubs, ",")
launch_ct := 0
for _, hubaddr := range all_hubs {
if len(hubaddr) == 0 {
continue
}
// Assert target directory exists
os.MkdirAll(GetDirectoryNameForHub(hubaddr), 0755)
// Launch logger
go HubWorker(hubaddr, *nick, *password)
launch_ct++
}
if launch_ct == 0 {
fmt.Fprintln(os.Stderr, "FATAL: No hubs specified")
os.Exit(1)
}
// Wait forever
var forever chan bool = nil
select {
case <-forever:
}
}