package web
import (
"adammathes.com/neko/config"
"adammathes.com/neko/crawler"
"adammathes.com/neko/exporter"
"adammathes.com/neko/models/feed"
"adammathes.com/neko/models/item"
"encoding/base64"
"encoding/json"
"fmt"
"github.com/GeertJohan/go.rice"
"golang.org/x/crypto/bcrypt"
"io/ioutil"
"log"
"net/http"
"strconv"
"time"
)
func indexHandler(w http.ResponseWriter, r *http.Request) {
serveBoxedFile(w, r, "ui.html")
}
func streamHandler(w http.ResponseWriter, r *http.Request) {
max_id := 0
if r.FormValue("max_id") != "" {
max_id, _ = strconv.Atoi(r.FormValue("max_id"))
}
feed_id := int64(0)
if r.FormValue("feed_url") != "" {
feed_url := r.FormValue("feed_url")
var f feed.Feed
f.ByUrl(feed_url)
feed_id = f.Id
}
category := ""
if r.FormValue("tag") != "" {
category = r.FormValue("tag")
}
unread_only := true
if r.FormValue("read_filter") != "" {
unread_only = false
}
starred_only := false
if r.FormValue("starred") != "" {
starred_only = true
}
search_query := ""
if r.FormValue("q") != "" {
search_query = r.FormValue("q")
unread_only = false
}
var items []*item.Item
items, err := item.Filter(int64(max_id), feed_id, category, unread_only, starred_only, 0, search_query)
if err != nil {
log.Println(err)
}
w.Header().Set("Content-Type", "application/json")
js, _ := json.Marshal(items)
w.Write(js)
}
func itemHandler(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "PUT":
var i item.Item
err := json.NewDecoder(r.Body).Decode(&i)
if err != nil {
log.Println(err)
} else {
i.Save()
}
case "GET":
fullTextHandler(w, r)
}
defer r.Body.Close()
}
func feedHandler(w http.ResponseWriter, r *http.Request) {
if r.Method == "GET" {
feeds, err := feed.All()
if err != nil {
log.Println(err)
}
js, err := json.Marshal(feeds)
if err != nil {
log.Println(err)
}
w.Write(js)
return
}
var f feed.Feed
err := json.NewDecoder(r.Body).Decode(&f)
if err != nil {
log.Println(err)
}
switch r.Method {
case "POST":
feed.NewFeed(f.Url)
f.ByUrl(f.Url)
ch := make(chan string)
// log.Println("crawling")
crawler.CrawlFeed(&f, ch)
log.Println(<-ch)
case "PUT":
f.Update()
case "DELETE":
feed_id, err := strconv.Atoi(r.URL.Path[len("/feed/"):])
if err != nil {
log.Println(err)
return
}
f.Id = int64(feed_id)
f.Delete()
}
}
func categoryHandler(w http.ResponseWriter, r *http.Request) {
if r.Method == "GET" {
categories, err := feed.Categories()
if err != nil {
log.Println(err)
}
js, err := json.Marshal(categories)
if err != nil {
log.Println(err)
}
w.Write(js)
return
}
}
func imageProxyHandler(w http.ResponseWriter, r *http.Request) {
imgURL := r.URL.String()
decodedURL, err := base64.URLEncoding.DecodeString(imgURL)
// pseudo-caching
if r.Header.Get("If-None-Match") == string(decodedURL) {
w.WriteHeader(http.StatusNotModified)
return
}
if r.Header.Get("Etag") == string(decodedURL) {
w.WriteHeader(http.StatusNotModified)
return
}
// set headers
// grab the img
c := &http.Client{
// give up after 5 seconds
Timeout: 5 * time.Second,
}
request, err := http.NewRequest("GET", string(decodedURL), nil)
if err != nil {
http.Error(w, "failed to proxy image", 404)
return
}
userAgent := "neko RSS Reader Image Proxy +https://github.com/adammathes/neko"
request.Header.Set("User-Agent", userAgent)
resp, err := c.Do(request)
if err != nil {
http.Error(w, "failed to proxy image", 404)
return
}
bts, err := ioutil.ReadAll(resp.Body)
if err != nil {
http.Error(w, "failed to read proxy image", 404)
return
}
w.Header().Set("ETag", string(decodedURL))
w.Header().Set("Cache-Control", "public")
w.Header().Set("Expires", time.Now().Add(48*time.Hour).Format(time.RFC1123))
w.Write(bts)
return
}
func exportHandler(w http.ResponseWriter, r *http.Request) {
format := r.URL.String()
w.Header().Set("content-type", "text/plain")
w.Write([]byte(exporter.ExportFeeds(format)))
return
}
func crawlHandler(w http.ResponseWriter, r *http.Request) {
fmt.Fprintf(w, "crawling...\n\n")
if f, ok := w.(http.Flusher); ok {
f.Flush()
} else {
log.Println("Damn, no flush")
}
crawler.Crawl()
fmt.Fprintf(w, "done...\n\n")
return
}
func fullTextHandler(w http.ResponseWriter, r *http.Request) {
fmt.Printf("request: %v\n\n", r)
fmt.Printf("url string: %s\n\n", r.URL.String())
itemID, _ := strconv.Atoi(r.URL.String())
// fmt.Printf("item id: %v\n\n", itemID)
if itemID == 0 {
fmt.Printf("wah wah wah\n")
return
}
i := item.ItemById(int64(itemID))
// fmt.Println("item fetched: %v\n\n", i)
if i.FullContent == "" {
i.GetFullContent()
}
w.Header().Set("Content-Type", "application/json")
js, _ := json.Marshal(i)
w.Write(js)
}
var AuthCookie = "auth"
var SecondsInAYear = 60 * 60 * 24 * 365
func loginHandler(w http.ResponseWriter, r *http.Request) {
switch r.Method {
case "GET":
serveBoxedFile(w, r, "login.html")
case "POST":
password := r.FormValue("password")
if password == config.Config.DigestPassword {
v, _ := bcrypt.GenerateFromPassword([]byte(password), 0)
c := http.Cookie{Name: AuthCookie, Value: string(v), Path: "/", MaxAge: SecondsInAYear, HttpOnly: false}
http.SetCookie(w, &c)
http.Redirect(w, r, "/", 307)
} else {
http.Error(w, "bad login", 401)
}
default:
http.Error(w, "nope", 500)
}
}
func logoutHandler(w http.ResponseWriter, r *http.Request) {
c := http.Cookie{Name: AuthCookie, MaxAge: 0, Path: "/", HttpOnly: false}
http.SetCookie(w, &c)
fmt.Fprintf(w, "you are logged out")
}
func Authenticated(r *http.Request) bool {
pc, err := r.Cookie("auth")
if err != nil {
return false
}
err = bcrypt.CompareHashAndPassword([]byte(pc.Value), []byte(config.Config.DigestPassword))
if err == nil {
return true
}
return false
}
func AuthWrap(wrapped http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
if Authenticated(r) {
wrapped(w, r)
} else {
http.Redirect(w, r, "/login/", 307)
}
}
}
func serveBoxedFile(w http.ResponseWriter, r *http.Request, filename string) {
box := rice.MustFindBox("../static")
ui, err := box.Open(filename)
if err != nil {
panic(err)
}
fi, _ := ui.Stat()
http.ServeContent(w, r, filename, fi.ModTime(), ui)
}
func Serve() {
box := rice.MustFindBox("../static")
staticFileServer := http.StripPrefix("/static/", http.FileServer(box.HTTPBox()))
http.Handle("/static/", staticFileServer)
http.HandleFunc("/stream/", AuthWrap(streamHandler))
http.Handle("/item/", http.StripPrefix("/item/", AuthWrap(itemHandler)))
http.HandleFunc("/feed/", AuthWrap(feedHandler))
http.HandleFunc("/tag/", AuthWrap(categoryHandler))
http.Handle("/image/", http.StripPrefix("/image/", AuthWrap(imageProxyHandler)))
http.Handle("/export/", http.StripPrefix("/export/", AuthWrap(exportHandler)))
http.HandleFunc("/crawl/", AuthWrap(crawlHandler))
http.HandleFunc("/login/", loginHandler)
http.HandleFunc("/logout/", logoutHandler)
http.HandleFunc("/", AuthWrap(indexHandler))
log.Fatal(http.ListenAndServe(":"+strconv.Itoa(config.Config.Port), nil))
}
