very first iteration

2016-04-24 03:42:50 +02:00
parent 971bd1d6d4
commit 21eba2d77d
6 changed files with 600 additions and 0 deletions
@@ -0,0 +1,162 @@
+// ts3.go
+package modules
+
+import (
+	"bufio"
+	"log"
+	"net"
+	"regexp"
+	"strings"
+	"time"
+)
+
+var (
+	users   = map[string]bool{}
+	host    string
+	port    string
+	user    string
+	pass    string
+	botname string
+	channel string
+)
+
+func init() {
+	MsgHandlers["ts3"] = ts3HandleMessage
+	log.Println("Initializing ts3 module")
+	go setConfig()
+}
+func setConfig() {
+	time.Sleep(5 * time.Second)
+	host = ModParams["tshost"]
+	port = ModParams["tsport"]
+	user = ModParams["tsuser"]
+	pass = ModParams["tspass"]
+	botname = ModParams["tsbotname"]
+	channel = ModParams["tschan"]
+	go loop()
+}
+func loop() {
+	for {
+		var err error
+		users, err = getUsers()
+		if err != nil {
+			log.Println(err)
+			log.Println("Trying again in 60 seconds.")
+			time.Sleep(60 * time.Second)
+		} else {
+			break
+		}
+	}
+
+	for {
+		if channel == "" { // dont spam if no channel is set
+			break
+		}
+		neu, err := getUsers()
+		if err != nil {
+			log.Println(err)
+			time.Sleep(50 * time.Second)
+		} else {
+			for i, _ := range neu {
+				if users[i] != neu[i] {
+					SayCh <- GeneratePayload(channel, ":poop:", i+" joined TS3.", botname)
+				}
+			}
+			for i, _ := range users {
+				if users[i] != neu[i] {
+					SayCh <- GeneratePayload(channel, ":poop:", i+" left TS3.", botname)
+				}
+			}
+			users = neu
+		}
+
+		time.Sleep(10 * time.Second)
+	}
+}
+func ts3HandleMessage(payload *WebhookPayload) {
+	log.Println("TS3 users requested by " + payload.UserName)
+	users, _ := getUsers()
+	var s string
+	i := 0
+	s += "Current users on TS3 Server:\n"
+	for u, _ := range users {
+		i++
+		s += "- " + u
+		if i != len(users) {
+			s += "\n"
+		}
+	}
+	SayCh <- GeneratePayload("@"+payload.UserName, ":poop:", s, botname)
+}
+
+func getUsers() (u map[string]bool, e error) {
+	conn, err := net.Dial("tcp", host+":"+port)
+
+	if err != nil {
+		return nil, err
+	}
+
+	reader := bufio.NewReader(conn)
+	writer := bufio.NewWriter(conn)
+
+	for i := 0; i < 2; i++ { // get rid of banner
+		_, _, err := reader.ReadLine()
+		if err != nil {
+			if err.Error() == "EOF" {
+				log.Println("Connection closed by foreign host.")
+			}
+			return nil, err
+		}
+	}
+	writer.WriteString("login " + user + " " + pass + "\n")
+	writer.Flush()
+	_, _, err = reader.ReadLine() // get rid of "OK" message
+	if err != nil {
+		if err.Error() == "EOF" {
+			log.Println("Connection closed by foreign host.")
+		}
+		return nil, err
+	}
+
+	writer.WriteString("use sid=1\n")
+	writer.Flush()
+	_, _, err = reader.ReadLine() // get rid of "OK" message
+	if err != nil {
+		if err.Error() == "EOF" {
+			log.Println("Connection closed by foreign host.")
+		}
+		return nil, err
+	}
+
+	writer.WriteString("clientlist\n")
+	writer.Flush()
+	line, _, err := reader.ReadLine() // get rid of "OK" message
+	if err != nil {
+		if err.Error() == "EOF" {
+			log.Println("Connection closed by foreign host.")
+		}
+		return nil, err
+	}
+	users_raw := string(line)
+
+	re := regexp.MustCompile("client_nickname=(.*?) ")
+	users_still_raw := re.FindAllString(users_raw, -1)
+
+	for i := range users_still_raw {
+		users_still_raw[i] = strings.Trim(users_still_raw[i], " ")
+		users_still_raw[i] = strings.Replace(users_still_raw[i], "client_nickname=", "", 1)
+		users_still_raw[i] = strings.Replace(users_still_raw[i], "\\s", " ", -1)
+	}
+	users := make([]string, 0)
+	for i := range users_still_raw {
+		if !strings.HasPrefix(users_still_raw[i], "serveradmin from") {
+			users = append(users, users_still_raw[i])
+		}
+	}
+	result := make(map[string]bool)
+	for i := range users {
+		result[users[i]] = true
+	}
+	conn.Close()
+	return result, nil
+}