164 lines
3.2 KiB
Go
164 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
log "github.com/sirupsen/logrus"
|
|
"meow.tf/residentsleeper/commands"
|
|
"meow.tf/residentsleeper/events"
|
|
"meow.tf/residentsleeper/rcon"
|
|
"meow.tf/residentsleeper/scripting/config"
|
|
"meow.tf/residentsleeper/scripting/minecraft"
|
|
"os"
|
|
"path"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
serverPath string
|
|
scriptPath string
|
|
rconPort int
|
|
rconPassword string
|
|
debug bool
|
|
|
|
client *rcon.Session
|
|
|
|
configRegexp = regexp.MustCompile("^(.*?)=(.*)$")
|
|
|
|
flagDebug = flag.Bool("debug", false, "debug messages")
|
|
)
|
|
|
|
func init() {
|
|
flag.StringVar(&serverPath, "dir", "", "server directory")
|
|
flag.StringVar(&scriptPath, "scriptPath", "", "script path override")
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
|
|
debug = *flagDebug
|
|
|
|
if debug {
|
|
log.SetLevel(log.DebugLevel)
|
|
}
|
|
|
|
// Load properties
|
|
configPath := path.Join(serverPath, "server.properties")
|
|
|
|
cfg, err := loadServerConfig(configPath)
|
|
|
|
if err != nil {
|
|
log.WithError(err).Fatalln("Unable to load config")
|
|
}
|
|
|
|
if rconEnabled, ok := cfg["enable-rcon"]; !ok || rconEnabled != "true" {
|
|
log.Fatalln("RCON not enabled.")
|
|
}
|
|
|
|
var rconPortStr string
|
|
var ok bool
|
|
|
|
if rconPortStr, ok = cfg["rcon.port"]; !ok {
|
|
log.Fatalln("RCON is not enabled: No port set")
|
|
}
|
|
|
|
rconPort, _ = strconv.Atoi(rconPortStr)
|
|
|
|
if rconPassword, ok = cfg["rcon.password"]; !ok {
|
|
log.Fatalln("RCON is not enabled: No password set")
|
|
}
|
|
|
|
events.On(events.Message, handleCommands)
|
|
|
|
rconHost := "localhost"
|
|
|
|
if envHost := os.Getenv("RCON_HOST"); envHost != "" {
|
|
rconHost = envHost
|
|
}
|
|
|
|
client = rcon.NewSession(rconHost, rconPort, rconPassword)
|
|
|
|
client.Debug = debug
|
|
|
|
if scriptPath == "" {
|
|
scriptPath = path.Join(serverPath, "scripts")
|
|
}
|
|
|
|
if _, err := os.Stat(scriptPath); !os.IsNotExist(err) {
|
|
config.BasePath = path.Join(scriptPath, "config")
|
|
|
|
minecraft.BasePath = serverPath
|
|
minecraft.WorldPath = path.Join(serverPath, cfg["level-name"])
|
|
|
|
if _, err = os.Stat(config.BasePath); os.IsNotExist(err) {
|
|
os.Mkdir(config.BasePath, 0755)
|
|
}
|
|
|
|
log.WithField("path", scriptPath).Info("Loading scripts")
|
|
err = loadScripts(scriptPath)
|
|
|
|
if err != nil {
|
|
log.WithError(err).Warn("Unable to load scripts")
|
|
}
|
|
}
|
|
|
|
logPath := path.Join(serverPath, "logs/latest.log")
|
|
|
|
logParser(logPath)
|
|
}
|
|
|
|
func handleCommands(eventArgs ...interface{}) {
|
|
user := eventArgs[0].(string)
|
|
str := eventArgs[1].(string)
|
|
|
|
log.WithFields(log.Fields{
|
|
"user": user,
|
|
"message": str,
|
|
}).Debug("Message received")
|
|
|
|
args := commands.ParseCommandArguments(str)
|
|
|
|
idx := strings.Index(str, " ")
|
|
|
|
var argString string
|
|
|
|
if idx == -1 {
|
|
argString = ""
|
|
} else {
|
|
argString = strings.TrimSpace(str[idx+1:])
|
|
}
|
|
|
|
var command string
|
|
|
|
if len(args) > 1 {
|
|
command, args = args[0], args[1:]
|
|
} else {
|
|
command = str
|
|
args = []string{}
|
|
}
|
|
|
|
// Find the channel that the message came from. Override and use State if enabled.
|
|
|
|
match := commands.Find("!", command, str)
|
|
|
|
if match == nil {
|
|
return
|
|
}
|
|
|
|
ctx := &commands.CommandContext{
|
|
User: user,
|
|
Prefix: "!",
|
|
Command: command,
|
|
ArgumentString: argString,
|
|
Arguments: args,
|
|
ArgumentCount: len(args),
|
|
|
|
Reply: func(text string) {
|
|
client.SendMessage(user, text)
|
|
},
|
|
}
|
|
|
|
go match.Call(ctx)
|
|
}
|