New RSSFeed doesn't auto update, server callback adds to feed pool, monitor callback updates feed
This commit is contained in:
@@ -1,10 +1,13 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path"
|
||||
"regexp"
|
||||
"strings"
|
||||
"syscall"
|
||||
"time"
|
||||
@@ -12,10 +15,10 @@ import (
|
||||
|
||||
type Server struct {
|
||||
addr string
|
||||
newItemHandler func(string, time.Duration)
|
||||
newItemHandler func(string, string, string, time.Duration)
|
||||
}
|
||||
|
||||
func New(addr string, newItemHandler func(string, time.Duration)) (*Server, error) {
|
||||
func New(addr string, newItemHandler func(string, string, string, time.Duration)) (*Server, error) {
|
||||
return &Server{
|
||||
addr: addr,
|
||||
newItemHandler: newItemHandler,
|
||||
@@ -47,20 +50,28 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
case "api":
|
||||
s.api(w, r)
|
||||
default:
|
||||
s.bad(w, r)
|
||||
s.notFound(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) bad(w http.ResponseWriter, r *http.Request) {
|
||||
func (s *Server) notFound(w http.ResponseWriter, r *http.Request) {
|
||||
http.NotFound(w, r)
|
||||
}
|
||||
|
||||
func (s *Server) bad(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusBadRequest)
|
||||
}
|
||||
|
||||
func (s *Server) mybad(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
}
|
||||
|
||||
func (s *Server) api(w http.ResponseWriter, r *http.Request) {
|
||||
switch advance(r) {
|
||||
case "feed":
|
||||
s.feed(w, r)
|
||||
default:
|
||||
s.bad(w, r)
|
||||
s.notFound(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -68,12 +79,51 @@ func (s *Server) feed(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case "GET":
|
||||
case "POST":
|
||||
s.newItem(w, r)
|
||||
case "PUT":
|
||||
s.newItem(w, r)
|
||||
default:
|
||||
s.bad(w, r)
|
||||
s.notFound(w, r)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) newItem(w http.ResponseWriter, r *http.Request) {
|
||||
defer r.Body.Close()
|
||||
newItemBody := struct {
|
||||
URL string `json:"url"`
|
||||
Refresh string `json:"refresh"`
|
||||
ItemFilter string `json:"items"`
|
||||
ContentFilter string `json:"content"`
|
||||
}{}
|
||||
if err := json.NewDecoder(r.Body).Decode(&newItemBody); err != nil {
|
||||
s.bad(w, r)
|
||||
return
|
||||
}
|
||||
interval, err := time.ParseDuration(newItemBody.Refresh)
|
||||
if err != nil {
|
||||
s.bad(w, r)
|
||||
return
|
||||
}
|
||||
if !validURL(newItemBody.URL) {
|
||||
s.bad(w, r)
|
||||
return
|
||||
}
|
||||
if _, err := regexp.Compile(newItemBody.ItemFilter); err != nil {
|
||||
s.bad(w, r)
|
||||
return
|
||||
}
|
||||
if _, err := regexp.Compile(newItemBody.ContentFilter); err != nil {
|
||||
s.bad(w, r)
|
||||
return
|
||||
}
|
||||
s.newItemHandler(newItemBody.URL, newItemBody.ItemFilter, newItemBody.ContentFilter, interval)
|
||||
}
|
||||
|
||||
func validURL(loc string) bool {
|
||||
_, err := url.ParseRequestURI(loc)
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func advance(r *http.Request) string {
|
||||
p := path.Clean("/" + r.URL.Path)
|
||||
i := strings.Index(p[1:], "/") + 1
|
||||
|
||||
Reference in New Issue
Block a user