From 23df075253741ef6aa4606dcf8e07f33442a2e1d Mon Sep 17 00:00:00 2001
From: Dane Everitt <dane@daneeveritt.com>
Date: Sat, 6 Apr 2019 15:24:25 -0700
Subject: [PATCH] Add support for checking/creating the Pterodactyl system user

---
 config/config.go | 53 ++++++++++++++++++++++++++++++++++++++++++++++--
 wings.go         |  8 ++++++++
 2 files changed, 59 insertions(+), 2 deletions(-)

diff --git a/config/config.go b/config/config.go
index e3f03ae..4ef90f3 100644
--- a/config/config.go
+++ b/config/config.go
@@ -1,9 +1,13 @@
 package config
 
 import (
+	"fmt"
 	"gopkg.in/yaml.v2"
 	"io/ioutil"
 	"os"
+	"os/exec"
+	"os/user"
+	"strings"
 )
 
 type Configuration struct {
@@ -188,8 +192,42 @@ func ReadConfiguration(path string) (*Configuration, error) {
 //
 // If files are not owned by this user there will be issues with permissions on Docker
 // mount points.
-func (c *Configuration) EnsurePterodactylUser() error {
-	return nil
+func (c *Configuration) EnsurePterodactylUser() (*user.User, error) {
+	u, err := user.Lookup(c.System.User)
+
+	// If an error is returned but it isn't the unknown user error just abort
+	// the process entirely. If we did find a user, return it immediately.
+	if err == nil {
+		return u, nil
+	} else if _, ok := err.(user.UnknownUserError); !ok {
+		return nil, err
+	}
+
+	sysName, err := getSystemName()
+	if err != nil {
+		return nil, err
+	}
+
+	var command = fmt.Sprintf("useradd --system --no-create-home --shell /bin/false %s", c.System.User)
+
+	// Alpine Linux is the only OS we currently support that doesn't work with the useradd command, so
+	// in those cases we just modify the command a bit to work as expected.
+	if strings.HasPrefix(sysName, "Alpine") {
+		command = fmt.Sprintf("adduser -S -D -H -G %[1]s -s /bin/false %[1]s", c.System.User)
+
+		// We have to create the group first on Alpine, so do that here before continuing on
+		// to the user creation process.
+		if _, err := exec.Command("addgroup", "-s", c.System.User).Output(); err != nil {
+			return nil, err
+		}
+	}
+
+	split := strings.Split(command, " ")
+	if _, err := exec.Command(split[0], split[1:]...).Output(); err != nil {
+		return nil, err
+	}
+
+	return user.Lookup(c.System.User)
 }
 
 // Ensures that the configured data directory has the correct permissions assigned to
@@ -197,3 +235,14 @@ func (c *Configuration) EnsurePterodactylUser() error {
 func (c *Configuration) EnsureFilePermissions() error {
 	return nil
 }
+
+func getSystemName() (string, error) {
+	cmd := exec.Command("lsb_release", "-is")
+
+	b, err := cmd.Output()
+	if err != nil {
+		return "", err
+	}
+
+	return string(b), nil
+}
\ No newline at end of file
diff --git a/wings.go b/wings.go
index aeb9748..2a7ed58 100644
--- a/wings.go
+++ b/wings.go
@@ -38,6 +38,14 @@ func main() {
 		zap.S().Debugw("running in debug mode")
 	}
 
+	zap.S().Infof("checking for pterodactyl system user \"%s\"", c.System.User)
+	if su, err := c.EnsurePterodactylUser(); err != nil {
+		zap.S().Panicw("failed to create pterodactyl system user", zap.Error(err))
+		return
+	} else {
+		zap.S().Infow("configured system user", zap.String("username", su.Username), zap.String("uid", su.Uid), zap.String("gid", su.Gid))
+	}
+
 	servers, err := server.LoadDirectory("data/servers", c.System)
 	if err != nil {
 		zap.S().Fatalw("failed to load server configurations", zap.Error(err))