tux21b · September 30, 2012 13:26
diff --git a/weierbot.go b/weierbot.go
 package main

 import (
 	"fmt"
 	irc "github.com/fluffle/goirc/client"
 	"html/template"
 	"io/ioutil"
 	"log"
 	"net/http"
 	"net/url"
 	"os"
 	"path/filepath"
 	"regexp"
 	"sort"
 	"strings"
 	"time"
 )

 var (
 	timezone   = time.UTC
 	wolframURL = "http://www.wolframalpha.com/input/?i=%s"
 )

 func init() {
 	tz, err := time.LoadLocation("Europe/Vienna")
 	if err != nil {
 		log.Fatal(err)
 	}
 	timezone = tz
 }

 var tmpl = template.Must(template.New("weierbot").Parse(`<!doctype html>
 <html>
 <head>
 <title>{{.Title}} - Weierbot</title>
 <style>a { color: #111; }</style>
 </head>
 <body>
 <div style="float:right;">
 {{if .Prev}}<a style="color:#FF1A00;" href="/{{.Prev}}">Previous</a>{{end}}
 {{if .Next}}<a style="color:#FF1A00;" href="/{{.Next}}">Next</a>{{end}}
 </div>
 <h1 style="color:#111; font-size: 1.2em; margin:0;">Weierbot</h1>
 <h2 style="margin: .1em 0 .5em; font-size: 1.8em;">
 <a style="color:#FF1A00; text-decoration: none;" href="/{{.Title}}">{{.Title}}</a>
 </h2>
 <pre>{{.Chatlog}}</pre>
 </body>
 </html>`))

 var reURL = regexp.MustCompile(`https?://[-A-Za-z0-9+&@#/%?=~_|!:,.;]*[-A-Za-z0-9+&@#/%=~_|]`)
 var reTime = regexp.MustCompile(`(?m)^\[\S+\]`)

 type WeierBot struct {
 	server     string
 	client     *irc.Conn
 	log        chan Message
 	disconnect chan bool
 }

 func NewWeierBot(server, nick, password string, channels []string) *WeierBot {
 	w := &WeierBot{
 		server:     server,
 		client:     irc.SimpleClient(nick),
 		log:        make(chan Message, 512),
 		disconnect: make(chan bool),
 	}
 	w.client.Me.Name = nick
 	w.client.Me.Ident = nick

 	w.client.AddHandler("connected", func(conn *irc.Conn, line *irc.Line) {
 		conn.Privmsg("NickServ", "identify "+password)
 		for _, ch := range channels {
 			log.Printf("Joining %s\n", ch)
 			conn.Join(ch)
 		}
 	})
 	w.client.AddHandler("disconnected", func(conn *irc.Conn, line *irc.Line) {
 		w.disconnect <- true
 	})
 	w.client.AddHandler("join", func(conn *irc.Conn, line *irc.Line) {
 		if len(line.Args) >= 1 && strings.HasPrefix(line.Args[0], "#") {
 			conn.Privmsg(line.Nick, fmt.Sprintf("Hallo %s, willkommen auf %s. "+
 				"Dieser Channel wird unter http://weierbot.tux21b.org/ mitgespeichert.",
 				line.Nick, line.Args[0]))
 		}
 	})
 	w.client.AddHandler("privmsg", func(conn *irc.Conn, line *irc.Line) {
 		if len(line.Args) < 2 {
 			return
 		}
 		w.handleMessage(line.Args[0], Message{
 			Nick:    line.Nick,
 			Time:    line.Time,
 			Message: line.Args[1],
 		})
 	})

 	go w.writeLog()

 	return w
 }

 func (bot *WeierBot) handleMessage(target string, msg Message) {
 	if strings.HasPrefix(target, "#") {
 		bot.log <- msg
 	} else {
 		target = msg.Nick
 	}

 	switch {
 	case strings.HasPrefix(msg.Message, "!wolfram "):
 		bot.send(target, fmt.Sprintf(wolframURL, url.QueryEscape(msg.Message[9:])))
 	case msg.Message == "!log":
 		bot.send(target, "http://weierbot.tux21b.org/")
 	}
 }

 func (bot *WeierBot) send(target, message string) {
 	bot.client.Privmsg(target, message)
 	if strings.HasPrefix(target, "#") {
 		bot.log <- Message{Nick: bot.client.Me.Nick, Time: time.Now(),
 			Message: message}
 	}
 }

 func (bot *WeierBot) writeLog() {
 	var (
 		file     *os.File
 		format   = "2006-01-02.log"
 		filename string
 	)
 	defer func() {
 		if file != nil {
 			file.Close()
 		}
 	}()
 	for msg := range bot.log {
 		msg.Time = msg.Time.In(timezone)
 		current := msg.Time.Format(format)
 		if file == nil || current != filename {
 			if file != nil {
 				file.Close()
 			}
 			f, err := os.OpenFile(current,
 				os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0666)
 			if err != nil {
 				log.Println(err)
 			} else {
 				file, filename = f, current
 			}
 		}
 		fmt.Fprintf(file, "%s\n", msg)
 	}
 }

 func (bot *WeierBot) ServeIRC() {
 	tries := 0
 	for {
 		tries++
 		if err := bot.client.Connect(bot.server); err != nil {
 			log.Println(err)
 			if tries >= 5 {
 				log.Fatal("maximum number of tries exceeded")
 			}
 			<-time.After(5 * time.Second)
 			continue
 		}
 		tries = 0
 		log.Printf("Connected to %s\n", bot.server)
 		<-bot.disconnect
 	}
 }

 func (bot *WeierBot) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	files, err := filepath.Glob("*.log")
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}
 	sort.Strings(files)

 	if r.URL.Path == "/" && len(files) > 0 {
 		http.Redirect(w, r, "/"+filepath.Base(files[len(files)-1]),
 			http.StatusTemporaryRedirect)
 	}
 	index := sort.SearchStrings(files, r.URL.Path[1:])
 	if index < 0 || index >= len(files) {
 		http.NotFound(w, r)
 		return
 	}

 	chatlog, err := ioutil.ReadFile(files[index])
 	if err != nil {
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}

 	chatlog = []byte(template.HTMLEscapeString(string(chatlog)))
 	chatlog = reURL.ReplaceAllFunc(chatlog, func(m []byte) []byte {
 		return []byte(fmt.Sprintf("<a href=\"%s\">%s</a>", m, m))
 	})
 	chatlog = reTime.ReplaceAllFunc(chatlog, func(m []byte) []byte {
 		return []byte(fmt.Sprintf("<span style=\"color: #888;\">%s</span>", m))
 	})

 	data := struct {
 		Title      string
 		Prev, Next string
 		Chatlog    template.HTML
 	}{
 		Title:   filepath.Base(files[index]),
 		Chatlog: template.HTML(chatlog),
 	}
 	if index > 0 {
 		data.Prev = files[index-1]
 	}
 	if index < len(files)-1 {
 		data.Next = files[index+1]
 	}
 	if err := tmpl.Execute(w, data); err != nil {
 		log.Println(err.Error())
 	}
 }

 type Message struct {
 	Time    time.Time
 	Nick    string
 	Message string
 }

 func (m Message) String() string {
 	return fmt.Sprintf("[%s] %s: %s", m.Time.Format(time.Kitchen), m.Nick, m.Message)
 }

 func main() {
 	bot := NewWeierBot("irc.euirc.net", "weierbot", "secret", []string{"#tm"})

 	go func() {
 		if err := http.ListenAndServe(":8005", bot); err != nil {
 			log.Fatal(err)
 		}
 	}()

 	bot.ServeIRC()
 }
	package main

	import (
	"fmt"
	irc "github.com/fluffle/goirc/client"
	"html/template"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"regexp"
	"sort"
	"strings"
	"time"
	)

	var (
	timezone = time.UTC
	wolframURL = "http://www.wolframalpha.com/input/?i=%s"
	)

	func init() {
	tz, err := time.LoadLocation("Europe/Vienna")
	if err != nil {
	log.Fatal(err)
	}
	timezone = tz
	}

	var tmpl = template.Must(template.New("weierbot").Parse(`<!doctype html>
	<html>
	<head>
	<title>{{.Title}} - Weierbot</title>
	<style>a { color: #111; }</style>
	</head>
	<body>
	<div style="float:right;">
	{{if .Prev}}<a style="color:#FF1A00;" href="/{{.Prev}}">Previous</a>{{end}}
	{{if .Next}}<a style="color:#FF1A00;" href="/{{.Next}}">Next</a>{{end}}
	</div>
	<h1 style="color:#111; font-size: 1.2em; margin:0;">Weierbot</h1>
	<h2 style="margin: .1em 0 .5em; font-size: 1.8em;">
	<a style="color:#FF1A00; text-decoration: none;" href="/{{.Title}}">{{.Title}}</a>
	</h2>
	<pre>{{.Chatlog}}</pre>
	</body>
	</html>`))

	var reURL = regexp.MustCompile(`https?://[-A-Za-z0-9+&@#/%?=~_\|!:,.;]*[-A-Za-z0-9+&@#/%=~_\|]`)
	var reTime = regexp.MustCompile(`(?m)^\[\S+\]`)

	type WeierBot struct {
	server string
	client *irc.Conn
	log chan Message
	disconnect chan bool
	}

	func NewWeierBot(server, nick, password string, channels []string) *WeierBot {
	w := &WeierBot{
	server: server,
	client: irc.SimpleClient(nick),
	log: make(chan Message, 512),
	disconnect: make(chan bool),
	}
	w.client.Me.Name = nick
	w.client.Me.Ident = nick

	w.client.AddHandler("connected", func(conn irc.Conn, line irc.Line) {
	conn.Privmsg("NickServ", "identify "+password)
	for _, ch := range channels {
	log.Printf("Joining %s\n", ch)
	conn.Join(ch)
	}
	})
	w.client.AddHandler("disconnected", func(conn irc.Conn, line irc.Line) {
	w.disconnect <- true
	})
	w.client.AddHandler("join", func(conn irc.Conn, line irc.Line) {
	if len(line.Args) >= 1 && strings.HasPrefix(line.Args[0], "#") {
	conn.Privmsg(line.Nick, fmt.Sprintf("Hallo %s, willkommen auf %s. "+
	"Dieser Channel wird unter http://weierbot.tux21b.org/ mitgespeichert.",
	line.Nick, line.Args[0]))
	}
	})
	w.client.AddHandler("privmsg", func(conn irc.Conn, line irc.Line) {
	if len(line.Args) < 2 {
	return
	}
	w.handleMessage(line.Args[0], Message{
	Nick: line.Nick,
	Time: line.Time,
	Message: line.Args[1],
	})
	})

	go w.writeLog()

	return w
	}

	func (bot *WeierBot) handleMessage(target string, msg Message) {
	if strings.HasPrefix(target, "#") {
	bot.log <- msg
	} else {
	target = msg.Nick
	}

	switch {
	case strings.HasPrefix(msg.Message, "!wolfram "):
	bot.send(target, fmt.Sprintf(wolframURL, url.QueryEscape(msg.Message[9:])))
	case msg.Message == "!log":
	bot.send(target, "http://weierbot.tux21b.org/")
	}
	}

	func (bot *WeierBot) send(target, message string) {
	bot.client.Privmsg(target, message)
	if strings.HasPrefix(target, "#") {
	bot.log <- Message{Nick: bot.client.Me.Nick, Time: time.Now(),
	Message: message}
	}
	}

	func (bot *WeierBot) writeLog() {
	var (
	file *os.File
	format = "2006-01-02.log"
	filename string
	)
	defer func() {
	if file != nil {
	file.Close()
	}
	}()
	for msg := range bot.log {
	msg.Time = msg.Time.In(timezone)
	current := msg.Time.Format(format)
	if file == nil \|\| current != filename {
	if file != nil {
	file.Close()
	}
	f, err := os.OpenFile(current,
	os.O_WRONLY\|os.O_APPEND\|os.O_CREATE, 0666)
	if err != nil {
	log.Println(err)
	} else {
	file, filename = f, current
	}
	}
	fmt.Fprintf(file, "%s\n", msg)
	}
	}

	func (bot *WeierBot) ServeIRC() {
	tries := 0
	for {
	tries++
	if err := bot.client.Connect(bot.server); err != nil {
	log.Println(err)
	if tries >= 5 {
	log.Fatal("maximum number of tries exceeded")
	}
	<-time.After(5 * time.Second)
	continue
	}
	tries = 0
	log.Printf("Connected to %s\n", bot.server)
	<-bot.disconnect
	}
	}

	func (bot WeierBot) ServeHTTP(w http.ResponseWriter, r http.Request) {
	files, err := filepath.Glob("*.log")
	if err != nil {
	http.Error(w, err.Error(), http.StatusInternalServerError)
	return
	}
	sort.Strings(files)

	if r.URL.Path == "/" && len(files) > 0 {
	http.Redirect(w, r, "/"+filepath.Base(files[len(files)-1]),
	http.StatusTemporaryRedirect)
	}
	index := sort.SearchStrings(files, r.URL.Path[1:])
	if index < 0 \|\| index >= len(files) {
	http.NotFound(w, r)
	return
	}

	chatlog, err := ioutil.ReadFile(files[index])
	if err != nil {
	http.Error(w, err.Error(), http.StatusInternalServerError)
	return
	}

	chatlog = []byte(template.HTMLEscapeString(string(chatlog)))
	chatlog = reURL.ReplaceAllFunc(chatlog, func(m []byte) []byte {
	return []byte(fmt.Sprintf("<a href=\"%s\">%s</a>", m, m))
	})
	chatlog = reTime.ReplaceAllFunc(chatlog, func(m []byte) []byte {
	return []byte(fmt.Sprintf("<span style=\"color: #888;\">%s</span>", m))
	})

	data := struct {
	Title string
	Prev, Next string
	Chatlog template.HTML
	}{
	Title: filepath.Base(files[index]),
	Chatlog: template.HTML(chatlog),
	}
	if index > 0 {
	data.Prev = files[index-1]
	}
	if index < len(files)-1 {
	data.Next = files[index+1]
	}
	if err := tmpl.Execute(w, data); err != nil {
	log.Println(err.Error())
	}
	}

	type Message struct {
	Time time.Time
	Nick string
	Message string
	}

	func (m Message) String() string {
	return fmt.Sprintf("[%s] %s: %s", m.Time.Format(time.Kitchen), m.Nick, m.Message)
	}

	func main() {
	bot := NewWeierBot("irc.euirc.net", "weierbot", "secret", []string{"#tm"})

	go func() {
	if err := http.ListenAndServe(":8005", bot); err != nil {
	log.Fatal(err)
	}
	}()

	bot.ServeIRC()
	}