add support for a command to be run after connect, e.g. for identification

Author: raidancampbell

Diff for: conf/config.go

@@ -9,10 +9,11 @@ type Cfg struct {
 }
 
 type Network struct {
-	Host     string   `mapstructure:"host"`
-	Port     int      `mapstructure:"port"`
-	SSL      bool     `mapstructure:"ssl"`
-	Channels []string `mapstructure:"channels"`
+	Host      string   `mapstructure:"host"`
+	Port      int      `mapstructure:"port"`
+	SSL       bool     `mapstructure:"ssl"`
+	Channels  []string `mapstructure:"channels"`
+	OnConnect []string `mapstructure:"onConnect"`
 }
 
 type Logging struct {

Diff for: gossip/bot.go

@@ -27,7 +27,7 @@ func New(cfg *conf.Cfg) *Bot {
 		addr:         fmt.Sprintf("%s:%d", cfg.Network.Host, cfg.Network.Port),
 		nick:         cfg.Nick,
 		msgChan:      make(chan *irc.Message),
-		triggers:     []Trigger{pingPong, NewJoin(cfg.Network.Channels), invite, NewPush(cfg), userPingPong, htmlTitle, die, part, rename, karma, toggle, source},
+		triggers:     []Trigger{pingPong, onConnect, NewJoin(cfg.Network.Channels), invite, NewPush(cfg), userPingPong, htmlTitle, die, part, rename, karma, toggle, source},
 		cfg:          cfg,
 	}
 	/* Feature todo:

Diff for: gossip/join.go

@@ -2,8 +2,10 @@ package gossip
 
 import (
 	"github.com/raidancampbell/gossip/data"
+	"github.com/sirupsen/logrus"
 	"gopkg.in/sorcix/irc.v2"
 	"sync"
+	"time"
 )
 
 // join desired channels on startup
@@ -31,12 +33,19 @@ func (j joinChannels) Condition(_ *Bot, msg *irc.Message) (shouldApply bool) {
 
 func (j joinChannels) Action(g *Bot, _ *irc.Message) (shouldContinue bool) {
 	j.o.Do(func() {
-		for _, chn := range j.channels {
-			g.msgChan <- &irc.Message{
-				Command: irc.JOIN,
-				Params:  []string{chn},
+		go func() {
+			if len(g.cfg.Network.OnConnect) > 0 {
+				waitTime := 10 * time.Second
+				logrus.Infof("waiting %s before joining channels to allow onConnect command to run...", waitTime.String())
+				time.Sleep(waitTime)
 			}
-		}
+			for _, chn := range j.channels {
+				g.msgChan <- &irc.Message{
+					Command: irc.JOIN,
+					Params:  []string{chn},
+				}
+			}
+		}()
 	})
 	return true
 }

Diff for: gossip/onConnect.go

@@ -0,0 +1,25 @@
+package gossip
+
+import (
+	"github.com/raidancampbell/gossip/data"
+	"gopkg.in/sorcix/irc.v2"
+	"time"
+)
+
+var onConnect = &SyncTrigger{
+	Cond: func(g *Bot, msg *irc.Message) bool {
+		return msg.Command == irc.RPL_WELCOME
+	},
+	Act: func(g *Bot, msg *irc.Message) bool {
+		time.Sleep(1 * time.Second)
+		for _, rawCMD := range g.cfg.Network.OnConnect {
+			g.msgChan <- irc.ParseMessage(rawCMD)
+		}
+		return true
+	},
+	meta: &data.TriggerMeta{
+		Disabled: false,
+		Priority: 0,
+		Name:     "onConnect",
+	},
+}