feed finder

server/handlers.go

@@ -1,12 +1,15 @@
 package server
 
 import (
+	"github.com/nkanaev/yarr/worker"
 	"net/http"
+	"encoding/json"
 	"os"
 	"log"
 	"io"
 	"fmt"
 	"mime"
+	"strings"
 )
 
 func IndexHandler(rw http.ResponseWriter, req *http.Request) {
@@ -40,7 +43,54 @@ func FolderListHandler(rw http.ResponseWriter, req *http.Request) {
 func FolderHandler(rw http.ResponseWriter, req *http.Request) {
 }
 
+type NewFeed struct {
+	Url string	   `json:"url"`
+	FolderID int64 `json:"folder_id,omitempty"`
+}
+
 func FeedListHandler(rw http.ResponseWriter, req *http.Request) {
+	if req.Method == "POST" {
+		var feed NewFeed
+		if err := json.NewDecoder(req.Body).Decode(&feed); err != nil {
+			log.Print(err)
+			rw.WriteHeader(http.StatusBadRequest)
+			return
+		}
+		feedUrl := feed.Url
+		res, err := http.Get(feedUrl)	
+		if err != nil {
+			log.Print(err)
+			rw.WriteHeader(http.StatusBadRequest)
+			return
+		} else if res.StatusCode != 200 {
+			rw.WriteHeader(http.StatusBadRequest)
+			return
+		}
+
+		contentType := res.Header.Get("Content-Type")
+		if strings.HasPrefix(contentType, "text/html") {
+			sources, err := worker.FindFeeds(res)
+			if err != nil {
+				log.Print(err)
+			}
+			if len(sources) == 0 {
+				writeJSON(rw, map[string]string{"status": "notfound"})
+			} else if len(sources) > 1 {
+				writeJSON(rw, map[string]interface{}{
+					"status": "multiple",
+					"choice": sources,
+				})
+			} else if len(sources) == 1 {
+				feedUrl = sources[0].Url
+				fmt.Println("feedUrl:", feedUrl)
+				writeJSON(rw, map[string]string{"status": "success"})
+			}
+			fmt.Println("got html url", sources, feedUrl)
+		} else if strings.HasPrefix(contentType, "text/xml") {
+			log.Print("got rss feed")
+		}
+		log.Print(res.Header.Get("Content-Type"))
+	}
 }
 
 func FeedHandler(rw http.ResponseWriter, req *http.Request) {

server/server.go

@@ -1,9 +1,11 @@
 package server
 
 import (
+	"encoding/json"
 	"context"
 	"regexp"
 	"net/http"
+	"log"
 )
 
 type Route struct {
@@ -38,6 +40,7 @@ var routes []Route = []Route{
 	p("/api/folders/:id", FolderHandler),
 	p("/api/feeds", FeedListHandler),
 	p("/api/feeds/:id", FeedHandler),
+	p("/api/feeds/find", FeedHandler),
 }
 
 func Vars(req *http.Request) map[string]string {
@@ -66,6 +69,16 @@ func (h Handler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 	rw.WriteHeader(http.StatusNotFound)
 }
 
+func writeJSON(rw http.ResponseWriter, data interface{}) {
+	rw.Header().Set("Content-Type", "application/json; charset=utf-8")
+	reply, err := json.Marshal(data)
+	if err != nil {
+		log.Fatal(err)
+	}
+	rw.Write(reply)
+	rw.Write([]byte("\n"))
+}
+
 func New() *http.Server {
 	h := Handler{}
 	s := &http.Server{Addr: "127.0.0.1:8000", Handler: h}