server/bserv/main.go

package main

import (
	"flag"
	"log"
	"math/rand"
	"net/http"
	_ "net/http/pprof"
	"os"
	"runtime/pprof"
	"time"

	"bitbucket.org/hackerbots/botserv"
	"code.google.com/p/go.net/websocket"
)

var addr = flag.String("addr", ":8666", "http service address")
var profile = flag.String("pprof", "", "if specified will run with pprof")
var mprofile = flag.String("mprof", "", "if specified will dump a memory profile")
var netprofile = flag.Bool("netprof", false, "if specified will run with net/http/pprof on :8667")
var verbose = flag.Bool("verbose", false, "")
var config = flag.String("config", "~/.config/hackerbots/config.json", "location of config file")

func main() {
	log.SetFlags(log.Ldate | log.Ltime | log.Lshortfile)
	log.Printf("Starting Server...")
	var err error

	rand.Seed(time.Now().UnixNano())
	flag.Parse()
	if *profile != "" {
		f, err := os.Create(*profile)
		if err != nil {
			log.Fatal(err)
		}
		pprof.StartCPUProfile(f)
	}
	if *netprofile {
		go func() {
			log.Println(http.ListenAndServe("localhost:8667", nil))
		}()
		log.Println("serving profile info at http://localhost:8667/debug/pprof/")
	}

	conf, err := botserv.LoadConfig(*config)
	if err != nil {
		log.Fatal(err)
	}

	sm := http.NewServeMux()

	c := botserv.Controller{
		Idg:  botserv.NewIdGenerator(),
		Conf: conf,
		Games: botserv.MapLock{
			M: make(map[string]*botserv.Game),
		},
		Memprofile: *mprofile,
		Profile:    *profile,
	}

	sm.Handle("/", botserv.JsonHandler(c.Index))
	sm.Handle("/ws/", websocket.Handler(c.AddPlayer))
	sm.Handle("/game/start/", botserv.JsonHandler(c.StartGame))
	sm.Handle("/game/list/", botserv.JsonHandler(c.ListGames))
	sm.Handle("/game/stats/", botserv.JsonHandler(c.GameStats))
	sm.Handle("/game/bw/", botserv.JsonHandler(c.BW))
	sm.Handle("/game/stop/", botserv.JsonHandler(c.StopGame))
	sm.HandleFunc("/fuck/shit/up/", c.KillServer)

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