Method handleSVG

src/server/server.go:112–165

(w http.ResponseWriter, r *http.Request, b *rod.Browser)

}

func (s *Server) handleSVG(w http.ResponseWriter, r *http.Request, b *rod.Browser) {
	if svgContent, found := s.c.Get("cachedSvgContent:" + r.URL.Query().Encode()); found {
		log.Println("SVG found in cache.")

		s.sendSVG(svgContent.(string), w)

		return
	}

	if svgContent, found := s.c.Get("cachedSvgContent:old:" + r.URL.Query().Encode()); found {
		log.Println("SVG found in old cache.")

		s.c.Delete("cachedSvgContent:old:" + r.URL.Query().Encode())
		s.sendSVG(svgContent.(string), w)

		go s.handleSVGInBackground(r, b)

		return
	}

	log.Println("New SVG request.")

	page := b.MustPage(fmt.Sprintf("http://localhost:8687/?%s", r.URL.Query().Encode()))
	defer page.Close()
	page.MustWaitLoad()

	if page.MustHas("pre") {
		pre := page.MustElement("pre")

		w.Header().Set("Content-Type", "text/plain")
		w.Header().Set("Cache-Control", "no-store, max-age=0")
		w.WriteHeader(http.StatusInternalServerError)

		fmt.Fprintln(w, pre.MustText())

		return
	} else if page.MustHas("#svg-container") {
		svg := page.MustElement("#svg-container")
		svgContent := svg.MustHTML()

		s.sendSVG(svgContent, w)

		s.c.Set("cachedSvgContent:"+r.URL.Query().Encode(), svgContent, cache.DefaultExpiration)
		s.c.Set("cachedSvgContent:old:"+r.URL.Query().Encode(), svgContent, 15778463000000000)

		return
	}

	w.Header().Set("Content-Type", "text/plain")
	w.Header().Set("Cache-Control", "no-store, max-age=0")
	w.WriteHeader(http.StatusInternalServerError)

	fmt.Fprintln(w, "Error while parsing SVG.")
}

func (s *Server) Start() {
	s.c = cache.New(30*time.Minute, 1*time.Minute)