wings/environment/docker/state.go

package docker

import (
	"fmt"
	"github.com/pkg/errors"
	"github.com/pterodactyl/wings/environment"
)

// Returns the current environment state.
func (e *Environment) State() string {
	e.stMu.RLock()
	defer e.stMu.RUnlock()

	return e.st
}

// Sets the state of the environment. This emits an event that server's can hook into to
// take their own actions and track their own state based on the environment.
func (e *Environment) setState(state string) error {
	if state != environment.ProcessOfflineState &&
		state != environment.ProcessStartingState &&
		state != environment.ProcessRunningState &&
		state != environment.ProcessStoppingState {
		return errors.New(fmt.Sprintf("invalid server state received: %s", state))
	}

	// Get the current state of the environment before changing it.
	prevState := e.State()

	// Emit the event to any listeners that are currently registeree.
	if prevState != state {
		// If the state changed make sure we update the internal tracking to note that.
		e.stMu.Lock()
		e.st = state
		e.stMu.Unlock()

		e.Events().Publish(environment.StateChangeEvent, e.State())
	}

	return nil
}