2013-07-18 02:26:55 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
2013-07-18 11:39:49 +00:00
|
|
|
"flag"
|
2013-07-18 02:26:55 +00:00
|
|
|
"log"
|
2014-11-25 16:42:54 +00:00
|
|
|
|
|
|
|
"github.com/vbatts/flaming-happiness/common"
|
2013-07-18 02:26:55 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2013-07-18 11:39:49 +00:00
|
|
|
flag.Parse()
|
|
|
|
|
2014-11-25 16:42:54 +00:00
|
|
|
if len(ignoreChannels) > 0 {
|
|
|
|
common.SetIgnores(ignoreChannels)
|
|
|
|
}
|
2013-07-18 17:28:44 +00:00
|
|
|
|
2014-11-25 16:42:54 +00:00
|
|
|
if len(flag.Args()) != 1 {
|
2013-07-18 11:39:49 +00:00
|
|
|
log.Fatalf("provide the zmq_notify publisher! like tcp://example.com:2428")
|
2013-07-18 02:26:55 +00:00
|
|
|
}
|
2014-11-25 16:42:54 +00:00
|
|
|
|
|
|
|
subscriber, err := NewSubscriber(flag.Args()[0])
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("ERROR: %s", err)
|
|
|
|
}
|
|
|
|
if err = subscriber.Connect(); err != nil {
|
|
|
|
log.Fatalf("ERROR: %s", err)
|
|
|
|
}
|
|
|
|
if !quiet {
|
|
|
|
log.Printf("Connected to [%s]", flag.Args()[0])
|
|
|
|
}
|
|
|
|
if err = subscriber.SetSubscribe(""); err != nil {
|
|
|
|
log.Fatalf("ERROR: %s", err)
|
|
|
|
}
|
2013-07-18 11:39:49 +00:00
|
|
|
|
2013-07-18 02:26:55 +00:00
|
|
|
for {
|
2014-11-25 16:42:54 +00:00
|
|
|
msg, err := subscriber.RecvMessageBytes(0)
|
2013-07-18 02:26:55 +00:00
|
|
|
if err != nil {
|
2014-11-25 16:42:54 +00:00
|
|
|
log.Fatalf("ERROR: %s", err)
|
2013-07-18 02:26:55 +00:00
|
|
|
break
|
|
|
|
}
|
2013-07-18 17:28:44 +00:00
|
|
|
noti_msg := common.IrcNotify{}
|
2014-11-25 16:42:54 +00:00
|
|
|
json.Unmarshal(msg[0], ¬i_msg)
|
2013-07-18 17:28:44 +00:00
|
|
|
go common.Display(noti_msg, linger, quiet)
|
2013-07-18 02:26:55 +00:00
|
|
|
}
|
|
|
|
}
|
2013-07-18 11:39:49 +00:00
|
|
|
|
2014-11-25 16:42:54 +00:00
|
|
|
type Subscriber interface {
|
|
|
|
Connect() error
|
|
|
|
SetSubscribe(string) error
|
|
|
|
RecvMessage(flag int) (msg []string, err error)
|
|
|
|
RecvMessageBytes(flag int) (msg [][]byte, err error)
|
|
|
|
}
|
|
|
|
|
2013-07-18 11:39:49 +00:00
|
|
|
var (
|
2013-07-18 17:28:44 +00:00
|
|
|
linger int64 = 5
|
2013-07-18 11:39:49 +00:00
|
|
|
quiet bool = false
|
|
|
|
ignoreChannels string
|
|
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
2013-07-18 17:28:44 +00:00
|
|
|
flag.Int64Var(&linger, "linger",
|
|
|
|
linger, "time to let the notification linger")
|
2013-07-18 11:39:49 +00:00
|
|
|
flag.BoolVar(&quiet, "quiet",
|
|
|
|
false, "less output")
|
|
|
|
flag.StringVar(&ignoreChannels, "ignore",
|
|
|
|
"", "comma seperated list of pattern of channels to ignore")
|
|
|
|
}
|