package server import ( "context" "fmt" "log" "net/http" "sync" "time" "github.com/gorilla/mux" "github.com/mdiluz/rove/pkg/accounts" "github.com/mdiluz/rove/pkg/game" ) const ( // PersistentData will allow the server to load and save it's state PersistentData = iota // EphemeralData will let the server neither load or save out any of it's data EphemeralData ) // Server contains the relevant data to run a game server type Server struct { port int accountant *accounts.Accountant world *game.World server *http.Server router *mux.Router persistence int sync sync.WaitGroup } // ServerOption defines a server creation option type ServerOption func(s *Server) // OptionPort sets the server port for hosting func OptionPort(port int) ServerOption { return func(s *Server) { s.port = port } } // OptionPersistentData sets the server data to be persistent func OptionPersistentData() ServerOption { return func(s *Server) { s.persistence = PersistentData } } // NewServer sets up a new server func NewServer(opts ...ServerOption) *Server { router := mux.NewRouter().StrictSlash(true) // Set up the default server s := &Server{ port: 8080, accountant: accounts.NewAccountant(), world: game.NewWorld(), persistence: EphemeralData, router: router, server: &http.Server{Addr: ":8080", Handler: router}, } // Apply all options for _, o := range opts { o(s) } return s } // Initialise sets up internal state ready to serve func (s *Server) Initialise() error { // Set up the world s.world = game.NewWorld() fmt.Printf("World created\n\t%+v\n", s.world) // Load the accounts if requested if s.persistence == PersistentData { if err := s.accountant.Load(); err != nil { return err } } // Create a new router s.SetUpRouter() fmt.Printf("Routes Created\n") // Add to our sync s.sync.Add(1) return nil } // Run executes the server func (s *Server) Run() { defer s.sync.Done() // Listen and serve the http requests fmt.Println("Serving HTTP") if err := s.server.ListenAndServe(); err != nil && err != http.ErrServerClosed { log.Fatal(err) } } // Close closes up the server func (s *Server) Close() error { // Try and shut down the http server ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) defer cancel() if err := s.server.Shutdown(ctx); err != nil { return err } // Wait until the server is shut down s.sync.Wait() // Save the accounts if requested if s.persistence == PersistentData { if err := s.accountant.Save(); err != nil { return err } } return nil }