Archived
165 lines
3.5 KiB
Go
165 lines
3.5 KiB
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
|
|
)
|
|
|
|
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) {
|
|
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, ":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
|
|
}
|