harness-drone/server/handler/server.go

package handler

import (
	"encoding/json"
	"net/http"

	"github.com/drone/drone/server/database"
	"github.com/drone/drone/server/session"
	"github.com/drone/drone/shared/model"
	"github.com/gorilla/pat"
)

type ServerHandler struct {
	servers database.ServerManager
	sess    session.Session
}

func NewServerHandler(servers database.ServerManager, sess session.Session) *ServerHandler {
	return &ServerHandler{servers, sess}
}

// GetServers gets all servers.
// GET /api/servers
func (h *ServerHandler) GetServers(w http.ResponseWriter, r *http.Request) error {
	// get the user form the session
	user := h.sess.User(r)
	switch {
	case user == nil:
		return notAuthorized{}
	case user.Admin == false:
		return forbidden{}
	}
	// get all servers
	servers, err := h.servers.List()
	if err != nil {
		return internalServerError{err}
	}

	return json.NewEncoder(w).Encode(servers)
}

// PostServer creates a new server.
// POST /api/servers
func (h *ServerHandler) PostServer(w http.ResponseWriter, r *http.Request) error {
	// get the user form the session
	user := h.sess.User(r)
	switch {
	case user == nil:
		return notAuthorized{}
	case user.Admin == false:
		return forbidden{}
	}
	// unmarshal the server from the payload
	defer r.Body.Close()
	in := model.Server{}
	if err := json.NewDecoder(r.Body).Decode(&in); err != nil {
		return badRequest{err}
	}
	// insert the server in the database
	if err := h.servers.Insert(&in); err != nil {
		return internalServerError{err}
	}

	return json.NewEncoder(w).Encode(&in)
}

// DeleteServers deletes the named server.
// GET /api/servers/:name
func (h *ServerHandler) DeleteServer(w http.ResponseWriter, r *http.Request) error {
	name := r.FormValue(":name")

	// get the user form the session
	user := h.sess.User(r)
	switch {
	case user == nil:
		return notAuthorized{}
	case user.Admin == false:
		return forbidden{}
	}
	// get the server
	server, err := h.servers.FindName(name)
	if err != nil {
		return notFound{err}
	}
	if err := h.servers.Delete(server); err != nil {
		return internalServerError{err}
	}

	w.WriteHeader(http.StatusNoContent)
	return nil
}

func (h *ServerHandler) Register(r *pat.Router) {
	r.Delete("/v1/servers/:name", errorHandler(h.DeleteServer))
	r.Post("/v1/servers", errorHandler(h.PostServer))
	r.Get("/v1/servers", errorHandler(h.GetServers))
}