improve logging

Author: lachlan-eagling

.gitignore

@@ -4,6 +4,7 @@
 *.dll
 *.so
 *.dylib
+RSSBot
 
 # Test binary, built with `go test -c`
 *.test
@@ -18,3 +19,4 @@
 */config-prod.yml
 */seen.txt
 .DS_Store
+*.log

app/bot.go

@@ -35,6 +35,13 @@ func FormatAuthor(a string) string {
 }
 
 func (b *Bot) Run() error {
+	if b.config.TestRun {
+		if err := b.twitterClient.PostTweet("Testing 1,2,3"); err != nil {
+			log.Debugf("Error posting test tweet. (%s)", err.Error())
+			return err
+		}
+		return nil
+	}
 	defer b.store.Close()
 
 	pendingTweets := make([]string, 0)
@@ -72,7 +79,13 @@ func (b *Bot) Run() error {
 		return nil
 	}
 
-	log.Infof("Sending %s tweets to Twitter API", len(pendingTweets))
+	if len(pendingTweets) > 0 {
+		log.Infof("Sending %d tweets to Twitter API", len(pendingTweets))
+	} else {
+		log.Info("No tweets to send. Exiting.")
+		return nil
+	}
+
 	for _, tweet := range pendingTweets {
 		log.Infof("Posting tweet (%s)", tweet)
 		if err := b.twitterClient.PostTweet(tweet); err != nil {
@@ -81,7 +94,6 @@ func (b *Bot) Run() error {
 		}
 	}
 	log.Info("Done sending tweets")
-
 	return nil
 }
 

app/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"flag"
+	"os"
 
 	log "github.com/sirupsen/logrus"
 )
@@ -17,6 +18,15 @@ func main() {
 	flag.StringVar(&seenPath, "seen", "seen.txt", "Path to text file containing seen URLs.")
 	flag.Parse()
 
+	// Setup logging to file
+	f, err := os.OpenFile("rssbot.log", os.O_APPEND|os.O_CREATE|os.O_RDWR, 0666)
+	if err != nil {
+		log.Errorf("error opening file: %v", err)
+	}
+	defer f.Close()
+	log.SetOutput(f)
+	log.SetLevel(log.InfoLevel)
+
 	cfg, err := LoadConfig(configPath)
 	if err != nil {
 		log.Fatalf("Error loading config. Shutting Down. (%s)", err.Error())