server/main.go

package main

import (
	"bitbucket.org/hackerbots/botserv/protocol"
	"code.google.com/p/go.net/websocket"
	"flag"
	"fmt"
	"log"
	"math/rand"
	"net/http"
	"time"
)

var addr = flag.String("addr", ":8666", "http service address")
var velocity = flag.Float64("velocity", 30, "")
var tick = flag.Int("tick", 33, "")
var weapon_radius = flag.Int("weapon_radius", 35, "")
var verbose = flag.Bool("verbose", false, "")
var width = flag.Float64("width", 800, "width of field")
var height = flag.Float64("height", 550, "height of field")

var delta float64
var g = game{
	register:    make(chan *player),
	unregister:  make(chan *player),
	projectiles: make(map[*projectile]bool),
	splosions:   make(map[*splosion]bool),
	players:     make(map[*player]bool),
	turn:        0,
	spectators:  make(map[*Spectator]bool),
	sregister:   make(chan *Spectator),
	sunregister: make(chan *Spectator),
}

func main() {
	rand.Seed(time.Now().UnixNano())
	flag.Parse()

	delta = float64(*tick) / 1000

	http.Handle("/ws/", websocket.Handler(addPlayer))

	go g.run()

	err := http.ListenAndServe(*addr, nil)
	if err != nil {
		log.Fatal("unable to start server")
	}
}

func addPlayer(ws *websocket.Conn) {
	var err error

	id := fmt.Sprintf("robot%d", <-g.id)
	log.Printf("sending robot id: %s", id)

	err = websocket.JSON.Send(ws, protocol.NewIdRequest(id))
	if err != nil {
		log.Printf("%s: problem sending initial identification", id)
		websocket.JSON.Send(ws, protocol.NewFailure("generic server error"))
		return
	}

	var clientid protocol.ClientID
	err = websocket.JSON.Receive(ws, &clientid)
	if err != nil {
		log.Printf("%s: problem parsing clientID", id)
		websocket.JSON.Send(ws, protocol.NewFailure("could not parse id"))
		return
	}
	if v, msg := clientid.Valid(); !v {
		log.Printf("%s: invalid clientid: %+v", id, clientid)
		websocket.JSON.Send(
			ws,
			protocol.NewFailure(msg),
		)
		return
	}

	gameParam := protocol.NewGameParam(*width, *height)
	err = websocket.JSON.Send(ws, gameParam)
	if err != nil {
		log.Println("%s: problem sending game info: %+v", id, gameParam)
		websocket.JSON.Send(ws, protocol.NewFailure("generic server error"))
		return
	}

	switch clientid.Type {
	case "robot":
		var conf Config
		for {
			err = websocket.JSON.Receive(ws, &conf)
			if err != nil {
				log.Print(err)
				return
			}
			if conf.Stats.valid() {
				_ = websocket.JSON.Send(ws, protocol.NewHandshake(id, true))
				break
			} else {
				_ = websocket.JSON.Send(ws, protocol.NewHandshake(id, false))
				log.Printf("%s invalid config", id)
			}
		}
		log.Printf("%s eventually sent valid config: %+v", id, conf)

		p := &player{
			Robot: robot{
				Stats:    conf.Stats,
				Id:       id,
				Health:   conf.Stats.Hp,
				Scanners: make([]scanner, 0)},
			send: make(chan *boardstate),
			ws:   ws,
		}
		p.reset()
		g.register <- p
		defer func() {
			g.unregister <- p
		}()
		go p.sender()
		p.recv()
		log.Printf("%v has been disconnect from this game\n", p.Robot.Id)
	case "spectator":
		s := &Spectator{
			send: make(chan *boardstate),
			ws:   ws,
		}
		g.sregister <- s
		defer func() {
			g.sunregister <- s
		}()
		s.sender()
		log.Printf("%+v has been disconnect from this game\n", s)
	}
}