Add get endpoint for url via query param
parent
d216949d91
commit
682f69746e
|
|
@ -2,12 +2,15 @@ package server
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"local1/logger"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"os"
|
"os"
|
||||||
"os/signal"
|
"os/signal"
|
||||||
"path"
|
"path"
|
||||||
"regexp"
|
"regexp"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
@ -15,13 +18,15 @@ import (
|
||||||
|
|
||||||
type Server struct {
|
type Server struct {
|
||||||
addr string
|
addr string
|
||||||
newItemHandler func(string, string, string, time.Duration)
|
newFeedHandler func(string, string, string, time.Duration)
|
||||||
|
getFeedHandler func(string, int) (string, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(addr string, newItemHandler func(string, string, string, time.Duration)) (*Server, error) {
|
func New(addr string, newFeedHandler func(string, string, string, time.Duration), getFeedHandler func(string, int) (string, error)) (*Server, error) {
|
||||||
return &Server{
|
return &Server{
|
||||||
addr: addr,
|
addr: addr,
|
||||||
newItemHandler: newItemHandler,
|
newFeedHandler: newFeedHandler,
|
||||||
|
getFeedHandler: getFeedHandler,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -80,18 +85,19 @@ func (s *Server) api(w http.ResponseWriter, r *http.Request) {
|
||||||
func (s *Server) feed(w http.ResponseWriter, r *http.Request) {
|
func (s *Server) feed(w http.ResponseWriter, r *http.Request) {
|
||||||
switch r.Method {
|
switch r.Method {
|
||||||
case "GET":
|
case "GET":
|
||||||
|
s.getFeed(w, r)
|
||||||
case "POST":
|
case "POST":
|
||||||
s.newItem(w, r)
|
s.newFeed(w, r)
|
||||||
case "PUT":
|
case "PUT":
|
||||||
s.newItem(w, r)
|
s.newFeed(w, r)
|
||||||
default:
|
default:
|
||||||
s.notFound(w, r)
|
s.notFound(w, r)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) newItem(w http.ResponseWriter, r *http.Request) {
|
func (s *Server) newFeed(w http.ResponseWriter, r *http.Request) {
|
||||||
defer r.Body.Close()
|
defer r.Body.Close()
|
||||||
newItemBody := struct {
|
newFeedBody := struct {
|
||||||
URL string `json:"url"`
|
URL string `json:"url"`
|
||||||
Refresh string `json:"refresh"`
|
Refresh string `json:"refresh"`
|
||||||
ItemFilter string `json:"items"`
|
ItemFilter string `json:"items"`
|
||||||
|
|
@ -99,28 +105,53 @@ func (s *Server) newItem(w http.ResponseWriter, r *http.Request) {
|
||||||
}{
|
}{
|
||||||
Refresh: "3h",
|
Refresh: "3h",
|
||||||
}
|
}
|
||||||
if err := json.NewDecoder(r.Body).Decode(&newItemBody); err != nil {
|
if err := json.NewDecoder(r.Body).Decode(&newFeedBody); err != nil {
|
||||||
s.bad(w, r)
|
s.bad(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
interval, err := time.ParseDuration(newItemBody.Refresh)
|
interval, err := time.ParseDuration(newFeedBody.Refresh)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
s.bad(w, r)
|
s.bad(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !validURL(newItemBody.URL) {
|
if !validURL(newFeedBody.URL) {
|
||||||
s.bad(w, r)
|
s.bad(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if _, err := regexp.Compile(newItemBody.ItemFilter); err != nil {
|
if _, err := regexp.Compile(newFeedBody.ItemFilter); err != nil {
|
||||||
s.bad(w, r)
|
s.bad(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if _, err := regexp.Compile(newItemBody.ContentFilter); err != nil {
|
if _, err := regexp.Compile(newFeedBody.ContentFilter); err != nil {
|
||||||
s.bad(w, r)
|
s.bad(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
s.newItemHandler(newItemBody.URL, newItemBody.ItemFilter, newItemBody.ContentFilter, interval)
|
s.newFeedHandler(newFeedBody.URL, newFeedBody.ItemFilter, newFeedBody.ContentFilter, interval)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Server) getFeed(w http.ResponseWriter, r *http.Request) {
|
||||||
|
url, err := url.ParseQuery(r.URL.RawQuery)
|
||||||
|
if err != nil {
|
||||||
|
logger.Logf("cannot get feed to read: %v", err)
|
||||||
|
s.mybad(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer r.Body.Close()
|
||||||
|
limit := 20
|
||||||
|
if url.Get("limit") != "" {
|
||||||
|
limit, err = strconv.Atoi(url.Get("limit"))
|
||||||
|
if err != nil {
|
||||||
|
s.bad(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
feedBody, err := s.getFeedHandler(url.Get("url"), limit)
|
||||||
|
if err != nil {
|
||||||
|
logger.Logf("cannot get feed %s: %v", url.Get("url"), err)
|
||||||
|
s.mybad(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprintln(w, feedBody)
|
||||||
}
|
}
|
||||||
|
|
||||||
func validURL(loc string) bool {
|
func validURL(loc string) bool {
|
||||||
|
|
|
||||||
|
|
@ -19,7 +19,7 @@ func Test_Server(t *testing.T) {
|
||||||
|
|
||||||
for _, _ = range cases {
|
for _, _ = range cases {
|
||||||
var err error
|
var err error
|
||||||
s, err := New(testPort, func(string, string, string, time.Duration) {})
|
s, err := New(testPort, func(string, string, string, time.Duration) {}, func(string, int) (string, error) { return "", nil })
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Errorf("failed to create server: %v", err)
|
t.Errorf("failed to create server: %v", err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue