webweav.ing/cmd/web/handlers.go

package main

import (
	"fmt"
	"html"
	"net/http"
	"strings"

	"git.32bit.cafe/yequari/rss-gen/feed"
)

func (app *application) home(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path != "/" {
        http.NotFound(w, r)
        return
    }

    app.render(w, http.StatusOK, "home.tmpl.html", nil)
}

func (app *application) generateRss(w http.ResponseWriter, r *http.Request) {
    q := r.URL.Query()
    siteName := q.Get("site-name")
    siteUrl := q.Get("site-url")
    siteDesc := q.Get("site-description")
    pageUrls := q.Get("page-urls")
    pages := strings.Split(pageUrls, "\n")
    for i := range pages {
        pages[i] = strings.TrimSpace(pages[i])
    }

    feed, err := feed.GenerateRss(siteUrl, siteName, siteDesc, pages...)
    if err != nil {
        w.Write([]byte(fmt.Sprintf("<p class=\"error\">Error generating feed: %s</p>", err.Error())))
        app.infoLog.Printf("Error generating feed: %s\n", err.Error())
    }
    for _, line := range strings.Split(feed, "\n") {
        w.Write([]byte(html.EscapeString(line) + "\n"))
    }
}