Add get endpoint for url via query param

master
Bel LaPointe 2018-10-09 18:29:17 -06:00
parent d216949d91
commit 682f69746e
2 changed files with 45 additions and 14 deletions

View File

@ -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 {

View File

@ -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)
} }