dm
/
vain
forked from sm/vain
1
0
Fork 0
vain/server.go

59 lines
1.5 KiB
Go
Raw Normal View History

2016-02-11 11:57:16 -08:00
package vain
2016-02-08 00:15:22 -08:00
2016-02-11 12:12:13 -08:00
import (
"encoding/json"
2016-02-11 12:12:13 -08:00
"fmt"
"net/http"
"strings"
2016-02-11 12:12:13 -08:00
)
2016-02-08 00:15:22 -08:00
2016-02-15 01:10:14 -08:00
// NewServer populates a server, adds the routes, and returns it for use.
func NewServer(sm *http.ServeMux, store Storage) *Server {
2016-02-15 01:10:14 -08:00
s := &Server{
storage: store,
2016-02-15 01:10:14 -08:00
}
sm.Handle("/", s)
return s
}
// Server serves up the http.
2016-02-08 00:15:22 -08:00
type Server struct {
storage Storage
2016-02-08 00:15:22 -08:00
}
2016-02-11 12:12:13 -08:00
func (s *Server) ServeHTTP(w http.ResponseWriter, req *http.Request) {
switch req.Method {
case "GET":
fmt.Fprintf(w, "<!DOCTYPE html>\n<html><head>\n")
for _, p := range s.storage.All() {
fmt.Fprintf(w, "%s\n", p)
}
fmt.Fprintf(w, "</head>\n</html>\n")
2016-02-11 12:12:13 -08:00
case "POST":
if req.URL.Path == "/" {
http.Error(w, fmt.Sprintf("invalid path %q", req.URL.Path), http.StatusBadRequest)
return
}
p := Package{}
if err := json.NewDecoder(req.Body).Decode(&p); err != nil {
2016-02-15 01:10:14 -08:00
http.Error(w, fmt.Sprintf("unable to parse json from body: %v", err), http.StatusBadRequest)
return
}
if p.Repo == "" {
http.Error(w, fmt.Sprintf("invalid repository %q", req.URL.Path), http.StatusBadRequest)
return
}
p.path = fmt.Sprintf("%s/%s", req.Host, strings.Trim(req.URL.Path, "/"))
2016-02-15 01:10:14 -08:00
if !Valid(p.path, s.storage.All()) {
http.Error(w, fmt.Sprintf("invalid path; prefix already taken %q", req.URL.Path), http.StatusConflict)
return
}
if err := s.storage.Add(p); err != nil {
http.Error(w, fmt.Sprintf("unable to add package: %v", err), http.StatusInternalServerError)
return
}
2016-02-11 12:12:13 -08:00
default:
2016-02-14 21:46:12 -08:00
http.Error(w, fmt.Sprintf("unsupported method %q; accepted: POST, GET", req.Method), http.StatusMethodNotAllowed)
2016-02-11 12:12:13 -08:00
}
}