quake-kube/cmd/q3/app/server/server.go

package server

import (
	sdk "agones.dev/agones/sdks/go"
	"context"
	"fmt"
	quakenet "github.com/criticalstack/quake-kube/internal/quake/net"
	"log"
	"net/url"
	"time"

	"github.com/pkg/errors"
	"github.com/spf13/cobra"

	quakeclient "github.com/criticalstack/quake-kube/internal/quake/client"
	"github.com/criticalstack/quake-kube/internal/quake/content"
	quakeserver "github.com/criticalstack/quake-kube/internal/quake/server"
	httputil "github.com/criticalstack/quake-kube/internal/util/net/http"
	"github.com/criticalstack/quake-kube/public"
)

var opts struct {
	ClientAddr    string
	ServerAddr    string
	ContentServer string
	AcceptEula    bool
	AssetsDir     string
	ConfigFile    string
	WatchInterval time.Duration
	WithAgones    bool
}

func NewCommand() *cobra.Command {
	cmd := &cobra.Command{
		Use:          "server",
		Short:        "q3 server",
		SilenceUsage: true,
		RunE: func(cmd *cobra.Command, args []string) error {
			ctx, cancel := context.WithCancel(context.Background())
			defer cancel()

			csurl, err := url.Parse(opts.ContentServer)
			if err != nil {
				return err
			}
			if !opts.AcceptEula {
				fmt.Println(quakeserver.Q3DemoEULA)
				return errors.New("You must agree to the EULA to continue")
			}
			if err := httputil.GetUntil(opts.ContentServer, ctx.Done()); err != nil {
				return err
			}

			// TODO(chrism): only download what is in map config
			if err := content.CopyAssets(csurl, opts.AssetsDir); err != nil {
				return err
			}

			if opts.WithAgones {
				if err := startAgones(ctx, opts.ServerAddr); err != nil {
					log.Printf("[Agones] Error: %v", err)
				}
			}

			go func() {
				s := quakeserver.Server{
					Dir:           opts.AssetsDir,
					WatchInterval: opts.WatchInterval,
					ConfigFile:    opts.ConfigFile,
					Addr:          opts.ServerAddr,
				}
				if err := s.Start(ctx); err != nil {
					panic(err)
				}
			}()

			e, err := quakeclient.NewRouter(&quakeclient.Config{
				ContentServerURL: opts.ContentServer,
				ServerAddr:       opts.ServerAddr,
				Files:            public.Files,
			})
			if err != nil {
				return err
			}
			s := &quakeclient.Server{
				Addr:       opts.ClientAddr,
				Handler:    e,
				ServerAddr: opts.ServerAddr,
			}
			fmt.Printf("Starting server %s\n", opts.ClientAddr)
			return s.ListenAndServe()
		},
	}
	cmd.Flags().StringVarP(&opts.ConfigFile, "config", "c", "", "server configuration file")
	cmd.Flags().StringVar(&opts.ContentServer, "content-server", "http://content.quakejs.com", "content server url")
	cmd.Flags().BoolVar(&opts.AcceptEula, "agree-eula", false, "agree to the Quake 3 demo EULA")
	cmd.Flags().StringVar(&opts.AssetsDir, "assets-dir", "assets", "location for game files")
	cmd.Flags().StringVar(&opts.ClientAddr, "client-addr", "0.0.0.0:8080", "client address <host>:<port>")
	cmd.Flags().StringVar(&opts.ServerAddr, "server-addr", "0.0.0.0:27960", "dedicated server <host>:<port>")
	cmd.Flags().DurationVar(&opts.WatchInterval, "watch-interval", 15*time.Second, "dedicated server <host>:<port>")
	cmd.Flags().BoolVar(&opts.WithAgones, "with-agones", false, "use Agones SDK integration")
	return cmd
}

func startAgones(ctx context.Context, addr string) error {
	log.Println("[Agones] starting Agones SDK client")
	s, err := sdk.NewSDK()
	if err != nil {
		return err
	}

	if err := s.Ready(); err != nil {
		log.Println("[Agones] failed to make the server Ready")
	}

	go func() {
		tick := time.Tick(2 * time.Second)
		maxAttempts := 0
		for {
			if err := s.Health(); err != nil {
				if maxAttempts > 5 {
					log.Fatalf("[Agones] could not send health ping: %v", err)
				}
				maxAttempts++
			} else {
				maxAttempts = 0
			}

			select {
			case <-ctx.Done():
				log.Print("[Agones] stopped health pings")
				return
			case <-tick:
			}
		}
	}()

	go func() {
		tick := time.Tick(5 * time.Second)
		for {
			if status, err := quakenet.GetStatus(addr); err == nil {
				if err != nil {
					log.Printf("[Agones] failed to get status from server: %s", err.Error())
					continue
				}
				for _, p := range status.Players {
					log.Printf("[Agones] player: %s", p.Name)

					if _, err := s.Alpha().PlayerConnect(p.Name); err != nil {
						log.Printf("[Agones] failed to register player: %s", err.Error())
					}
				}
				log.Println("[Agones] status checked")
			}

			select {
			case <-ctx.Done():
				log.Print("[Agones] stopped status checks")
				return
			case <-tick:
			}
		}
	}()

	return nil
}