package vain import ( "encoding/json" "fmt" "net/http" "strings" ) // NewServer populates a server, adds the routes, and returns it for use. func NewServer(sm *http.ServeMux, store Storage) *Server { s := &Server{ storage: store, } addRoutes(sm, s) return s } // Server serves up the http. type Server struct { storage Storage } func (s *Server) ServeHTTP(w http.ResponseWriter, req *http.Request) { switch req.Method { case "GET": fmt.Fprintf(w, "\n\n") for _, p := range s.storage.All() { fmt.Fprintf(w, "%s\n", p) } fmt.Fprintf(w, "\n\n") 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 { 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 } if p.Vcs == "" { p.Vcs = "git" } if !valid(p.Vcs) { http.Error(w, fmt.Sprintf("invalid vcs %q", p.Vcs), http.StatusBadRequest) return } p.path = fmt.Sprintf("%s/%s", req.Host, strings.Trim(req.URL.Path, "/")) 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 } case "DELETE": p := fmt.Sprintf("%s/%s", req.Host, strings.Trim(req.URL.Path, "/")) if !s.storage.Contains(p) { http.Error(w, fmt.Sprintf("package %q not found", p), http.StatusNotFound) return } if err := s.storage.Remove(p); err != nil { http.Error(w, fmt.Sprintf("unable to delete package: %v", err), http.StatusInternalServerError) return } default: http.Error(w, fmt.Sprintf("unsupported method %q; accepted: POST, GET, DELETE", req.Method), http.StatusMethodNotAllowed) } } func (s *Server) db(w http.ResponseWriter, req *http.Request) { all := s.storage.All() w.Header().Set("Content-type", "application/json") json.NewEncoder(w).Encode(&all) } func addRoutes(sm *http.ServeMux, s *Server) { sm.Handle("/", s) sm.HandleFunc("/db/", s.db) }