WGoSlackBot/modules/ts3.go

// 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
	tsboticon 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"]
	tsboticon = ModParams["tsboticon"]
	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, tsboticon, i+" joined TS3.", botname)
				}
			}
			for i, _ := range users {
				if users[i] != neu[i] {
					SayCh <- GeneratePayload(channel, tsboticon, i+" left TS3.", botname)
				}
			}
			users = neu
		}

		time.Sleep(10 * time.Second)
	}
}
func ts3HandleMessage(payload *WebhookPayload) {
	if payload.Command == "/ts" || payload.TriggerWord == "!ts" {
		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, tsboticon, 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
}