89 lines
2.1 KiB
Go
89 lines
2.1 KiB
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)
|
||
|
|
if user == nil || !user.Admin {
|
||
|
|
return notAuthorized{}
|
||
|
|
}
|
||
|
|
// 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)
|
||
|
|
if user == nil || !user.Admin {
|
||
|
|
return notAuthorized{}
|
||
|
|
}
|
||
|
|
// 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)
|
||
|
|
if user == nil || !user.Admin {
|
||
|
|
return notAuthorized{}
|
||
|
|
}
|
||
|
|
// 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))
|
||
|
|
}
|