Ejemplo n.º 1
0
func watchUntappd(irc *goirc.Connection) {
	var (
		userEvents []string
		err        error
	)
	lastCheckins := make(map[string]float64)
	for {
		for _, user := range cfg.Untappd.Users {
			// store the last checkin ID, to avoid printing the same checkin twice
			var lastcheckin float64
			if _, ok := lastCheckins[user]; ok {
				lastcheckin = lastCheckins[user]
			}
			if cfg.Untappd.Debug {
				fmt.Println("query untappd activity for", user, "with last checkin set to", lastcheckin)
			}
			userEvents, lastCheckins[user], err = getUntappdActivityFor(user, lastcheckin)
			if err != nil {
				log.Println("Failed to get", user, "'s Untappd activity:", err)
			} else {
				for _, ev := range userEvents {
					irc.Privmsgf(cfg.Irc.Channel, "%s", ev)
				}
			}
			time.Sleep(60 * time.Second)
		}
	}
}
Ejemplo n.º 2
0
func watchGithub(irc *goirc.Connection) {
	var err error
	// start the github watcher
	evchan := make(chan string)
	githubCli := makeGithubClient(cfg.Github.Token)
	for _, repo := range cfg.Github.Repos {
		splitted := strings.Split(repo, "/")
		if len(splitted) != 2 {
			irc.Privmsgf(cfg.Irc.Channel, "Invalid repository syntax '%s'. Must be <owner>/<reponame>", repo)
			continue
		}
		// don't run everything at once, we've got time...
		time.Sleep(time.Second)
		go func() {
			for {
				err = followRepoEvents(githubCli, splitted[0], splitted[1], evchan)
				if err != nil {
					log.Println("github follower crashed with error", err)
				}
				time.Sleep(60 * time.Second)
			}
		}()
	}
	go func() {
		for ev := range evchan {
			// no more than one post per second
			time.Sleep(time.Second)
			irc.Privmsgf(cfg.Irc.Channel, "%s", ev)
		}
	}()

}
Ejemplo n.º 3
0
// send strings to IRC
func sendIRC(s []string, i *irc.Connection, e *irc.Event, ready chan bool) {
	i.SendRaw("PRIVMSG " + e.Arguments[0] + " :\x031,9d-(^_^)z \x039,1 check your PMs for /r/" + e.Message[8:])
	for _, str := range s {
		i.SendRaw("PRIVMSG " + e.Nick + " :" + str)
		time.Sleep(time.Second * 1)
	}
	ready <- true
}
Ejemplo n.º 4
0
// JoinCallback triggers on JOIN event which happens whenever *anyone* joins
func JoinCallback(con *irc.Connection) {
	con.AddCallback("JOIN", func(e *irc.Event) {
		if e.Nick != config.IRCNick { // Exclude bot joins
			log.Printf("%s joined %s", e.Nick, e.Arguments[0])
		} else {
			log.Printf(e.Message())
		}
	})
}
Ejemplo n.º 5
0
func handleAuth(irc *goirc.Connection) {
	// place a callback on nickserv identification and wait until it is done
	if cfg.Irc.Nickpass != "" {
		identwaiter := make(chan bool)
		irc.AddCallback("NOTICE", func(e *goirc.Event) {
			re := regexp.MustCompile("NickServ IDENTIFY")
			if e.Nick == "NickServ" && re.MatchString(e.Message()) {
				irc.Privmsgf("NickServ", "IDENTIFY %s", cfg.Irc.Nickpass)
			}
			reaccepted := regexp.MustCompile("(?i)Password accepted")
			if e.Nick == "NickServ" && reaccepted.MatchString(e.Message()) {
				identwaiter <- true
			}
		})
		for {
			select {
			case <-identwaiter:
				goto identified
			case <-time.After(5 * time.Second):
				irc.Privmsgf("NickServ", "IDENTIFY %s", cfg.Irc.Nickpass)
			}
		}
	identified:
		irc.ClearCallback("NOTICE")
		close(identwaiter)
	}
	return
}
Ejemplo n.º 6
0
func githubPrintReposList(irc *goirc.Connection) {
	list := "list of followed github repositories: "
	for _, repo := range cfg.Github.Repos {
		list += repo + ", "
		if len(list) > 300 {
			irc.Privmsgf(cfg.Irc.Channel, "%s", list)
			list = ""
		}
	}
	if len(list) > 0 {
		irc.Privmsgf(cfg.Irc.Channel, "%s", list)
	}
	return
}
Ejemplo n.º 7
0
// PrivMsgCallback (PRIVMSG) is really any message in IRC that the bot sees
func PrivMsgCallback(con *irc.Connection) {
	con.AddCallback("PRIVMSG", func(e *irc.Event) {
		message := e.Arguments[1]
		channel := e.Arguments[0]
		log.Printf("%s said: %s\n", e.Nick, message)

		// Look for urls in messages
		urls := urlRegex.FindAllString(message, -1)
		if len(urls) > 0 {
			for _, url := range urls {
				messageCh := make(chan string, 1)
				go scrapePage(url, messageCh)
				con.Privmsg(channel, <-messageCh)
			}
		}
	})
}
Ejemplo n.º 8
0
func fetchPageTitles(irc *goirc.Connection) {
	irc.AddCallback("PRIVMSG", func(e *goirc.Event) {
		rehttp := regexp.MustCompile("(https?://.+)")
		if rehttp.MatchString(e.Message()) {
			url := rehttp.FindStringSubmatch(e.Message())
			if len(url) < 2 {
				return
			}
			title := fetchTitle(url[1])
			log.Printf("Retrieved tile '%s' from url %s\n", title, url[1])
			if title != "" {
				irc.Privmsgf(cfg.Irc.Channel, "Title: %s", title)
			}
		}
	})
	return
}
Ejemplo n.º 9
0
// WelcomeCallback is a PrivMsgCallback for the welcome event 001, join channels when we're logged in
func WelcomeCallback(con *irc.Connection) {
	con.AddCallback("001", func(e *irc.Event) {
		con.Privmsgf("NickServ", "identify %s", config.IRCNickPass)
		time.Sleep(time.Second * 15) // Need to wait a little for the registration to take effect

		for _, room := range config.IRCChannels {
			con.Join(room)
			log.Printf("Connected to channel %s\n", room)
		}
	})
}
Ejemplo n.º 10
0
// ErrorCallback should log anytime an error happens
func ErrorCallback(con *irc.Connection) {
	con.AddCallback("ERROR", func(e *irc.Event) {
		log.Printf("Error: %#v\n", e)
	})
}
Ejemplo n.º 11
0
func main() {
	var (
		irc *goirc.Connection
		err error
	)
	var configFile = flag.String("c", "r2d2.cfg", "Load configuration from file")
	flag.Parse()
	_, err = os.Stat(*configFile)
	if err != nil {
		log.Fatal("%v", err)
		os.Exit(1)
	}
	err = gcfg.ReadFileInto(&cfg, *configFile)
	if err != nil {
		log.Fatal("Error in configuration file: %v", err)
		os.Exit(1)
	}
	irc = goirc.IRC(cfg.Irc.Nick, cfg.Irc.Nick)
	irc.UseTLS = cfg.Irc.TLS
	irc.VerboseCallbackHandler = cfg.Irc.Debug
	irc.Debug = cfg.Irc.Debug
	err = irc.Connect(cfg.Irc.Server)
	if err != nil {
		log.Fatal("Connection to IRC server failed: %v", err)
		os.Exit(1)
	}

	// block while performing authentication
	handleAuth(irc)

	// we are identified, let's continue
	if cfg.Irc.ChannelPass != "" {
		// if a channel pass is used, craft a join command
		// of the form "&<channel>; <key>"
		irc.Join(cfg.Irc.Channel + " " + cfg.Irc.ChannelPass)
	} else {
		irc.Join(cfg.Irc.Channel)
	}
	if cfg.Irc.Debug {
		irc.Privmsg(cfg.Irc.Channel, "beep beedibeep dibeep")
	}
	go watchGithub(irc)
	go watchUntappd(irc)
	go fetchPageTitles(irc)
	initMaxmind()

	// add callback that captures messages sent to bot
	terminate := make(chan bool)
	irc.AddCallback("PRIVMSG", func(e *goirc.Event) {
		re := regexp.MustCompile("^" + cfg.Irc.Nick + ":(.+)$")
		if re.MatchString(e.Message()) {
			parsed := re.FindStringSubmatch(e.Message())
			if len(parsed) != 2 {
				return
			}
			req := strings.Trim(parsed[1], " ")
			resp := handleRequest(e.Nick, req, irc)
			if resp != "" {
				irc.Privmsgf(cfg.Irc.Channel, "%s: %s", e.Nick, resp)
			}
		}
	})
	<-terminate
	irc.Loop()
	irc.Disconnect()
}