rssc.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"
	"time"

	_ "embed"

	"github.com/larrasket/rssc/filter"
	"github.com/patrickmn/go-cache"
)

//go:embed README.html
var README string

func main() {
	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte(README))
	})

	c := cache.New(5*time.Second, 20*time.Second)
	http.HandleFunc("/rss", func(w http.ResponseWriter, r *http.Request) {
		q := r.URL.Query()

		if req, found := c.Get(q.Encode()); found {
			res := req.(string)
			w.Write([]byte(res))
			return
		}

		prams := filter.FilterPrams{AuthorRegex: q.Get("authorf"),
			ContentRegex:     q.Get("contentf"),
			TitleRegex:       q.Get("titlef"),
			DescriptionRegex: q.Get("descriptionf"),
			LinkRegex:        q.Get("linkf"),
			URL:              q.Get("src"),
			DotNet: func(b string) bool {
				if b == "1" {
					return true
				}
				return false
			}(q.Get("net")),
		}

		entries, err := filter.FilterFeeds(&prams)
		if err != nil {
			w.Write([]byte(err.Error()))
		}

		ftype := q.Get("t")
		feeds, err := filter.GenerateFeeds(entries, ftype)
		if err != nil {
			w.Write([]byte(err.Error()))
		}

		if ftype == "json" {
			w.Header().Set("Content-Type", "application/json")
		} else {
			w.Header().Set("Content-Type", "application/xml")
		}

		c.Set(q.Encode(), feeds, cache.DefaultExpiration)
		w.Write([]byte(feeds))
	})

	port, ok := os.LookupEnv("PORT")
	if ok == false {
		port = "8080"
	}

	fmt.Printf("Listening on :%s...\n", port)
	log.Fatal(http.ListenAndServe(":"+port, nil))

}