package handler import ( "encoding/json" "net/http" "github.com/drone/drone/server/worker" "github.com/drone/drone/server/worker/director" "github.com/drone/drone/server/worker/docker" "github.com/drone/drone/server/worker/pool" "github.com/goji/context" "github.com/zenazn/goji/web" ) // GetWorkers accepts a request to retrieve the list // of registered workers and return the results // in JSON format. // // GET /api/workers // func GetWorkers(c web.C, w http.ResponseWriter, r *http.Request) { ctx := context.FromC(c) workers := pool.FromContext(ctx).List() json.NewEncoder(w).Encode(workers) } // PostWorker accepts a request to allocate a new // worker to the pool. // // POST /api/workers // func PostWorker(c web.C, w http.ResponseWriter, r *http.Request) { ctx := context.FromC(c) workers := pool.FromContext(ctx) node := r.FormValue("node") workers.Allocate(docker.NewHost(node)) w.WriteHeader(http.StatusOK) } // GetWorkPending accepts a request to retrieve the list // of pending work and returns in JSON format. // // GET /api/work/pending // func GetWorkPending(c web.C, w http.ResponseWriter, r *http.Request) { ctx := context.FromC(c) d := worker.FromContext(ctx).(*director.Director) json.NewEncoder(w).Encode(d.GetPending()) } // GetWorkStarted accepts a request to retrieve the list // of started work and returns in JSON format. // // GET /api/work/started // func GetWorkStarted(c web.C, w http.ResponseWriter, r *http.Request) { ctx := context.FromC(c) d := worker.FromContext(ctx).(*director.Director) json.NewEncoder(w).Encode(d.GetStarted()) } // GetWorkAssigned accepts a request to retrieve the list // of started work and returns in JSON format. // // GET /api/work/assignments // func GetWorkAssigned(c web.C, w http.ResponseWriter, r *http.Request) { ctx := context.FromC(c) d := worker.FromContext(ctx).(*director.Director) json.NewEncoder(w).Encode(d.GetAssignemnts()) }