bugfixes and finetuning

modules/twitch.go

@@ -129,8 +129,13 @@ var (
 
 func init() {
 	MsgHandlers["twitch"] = twitchHandleMessage
-	announceChannel = ModParams["twitchchannel"]
 	log.Println("Initializing twitch module")
+	go configAndPoll()
+}
+
+func configAndPoll() {
+	time.Sleep(5 * time.Second)
+	announceChannel = ModParams["twitchchan"]
 	go pollStreamData()
 }
 
@@ -140,38 +145,43 @@ func twitchHandleMessage(payload *WebhookPayload) {
 		return
 	}
 	switch tok[0] {
-	case "!twitch" || "/twitch":
+	case "!twitch":
-		switch len(tok) {
+		handleCommand(tok, "#"+payload.ChannelName)
-		case 1:
+	case "/twitch":
-			onlinestreams := 0
+		handleCommand(tok, "@"+payload.UserName)
-			for streamname, _ := range twitch {
+	default:
-				var so TwitchStreamObject
+	}
-				var co TwitchChannelObject
+}
-				so = getTwitchStreamObject(streamname)
+func handleCommand(tok []string, target string) {
-				if so.Stream.Game != "" {
+	switch len(tok) {
-					onlinestreams++
+	case 1:
-					co = getTwitchChannelObject(streamname)
+		onlinestreams := 0
-					twitchSendMsg(co, so)
+		for streamname, _ := range twitch {
-					twitch[streamname] = true
-				} else {
-					twitch[streamname] = false
-				}
-			}
-			if onlinestreams == 0 {
-				SayCh <- GeneratePayload(payload.ChannelName, "", "All streams offline", "Twitch_Bot")
-			}
-		case 2:
-			streamname := tok[1]
 			var so TwitchStreamObject
 			var co TwitchChannelObject
 			so = getTwitchStreamObject(streamname)
 			if so.Stream.Game != "" {
+				onlinestreams++
 				co = getTwitchChannelObject(streamname)
-				twitchSendMsg(co, so)
+				twitchSendMsg(co, so, target)
+				twitch[streamname] = true
 			} else {
-				SayCh <- GeneratePayload("@"+payload.UserName, "", streamname+" not found or offline", "Twitch_Bot")
+				twitch[streamname] = false
 			}
-		default:
+		}
+		if onlinestreams == 0 {
+			SayCh <- GeneratePayload(target, "", "All streams offline", "Twitch_Bot")
+		}
+	case 2:
+		streamname := tok[1]
+		var so TwitchStreamObject
+		var co TwitchChannelObject
+		so = getTwitchStreamObject(streamname)
+		if so.Stream.Game != "" {
+			co = getTwitchChannelObject(streamname)
+			twitchSendMsg(co, so, target)
+		} else {
+			SayCh <- GeneratePayload(target, "", streamname+" not found or offline", "Twitch_Bot")
 		}
 	default:
 	}
@@ -187,7 +197,7 @@ func pollStreamData() {
 				so = getTwitchStreamObject(streamname)
 				if so.Stream.Game != "" && !twitch[streamname] {
 					co = getTwitchChannelObject(streamname)
-					twitchSendMsg(co, so)
+					twitchSendMsg(co, so, "")
 					twitch[streamname] = true
 				} else if so.Stream.Game == "" && twitch[streamname] {
 					twitch[streamname] = false
@@ -236,8 +246,11 @@ func getTwitchChannelObject(streamname string) TwitchChannelObject {
 	return tcobj
 }
 
-func twitchSendMsg(tcobj TwitchChannelObject, tso TwitchStreamObject) {
+func twitchSendMsg(tcobj TwitchChannelObject, tso TwitchStreamObject, target string) {
-	SayCh <- GeneratePayload(announceChannel,
+	if target == "" {
+		target = announceChannel
+	}
+	SayCh <- GeneratePayload(target,
 		"",
 		"*"+tso.Stream.Channel.DisplayName+
 			"*\n*Title:* "+tcobj.Status+