From f22f247d3b1ccd0bd789adf10ee2dfc7518f7914 Mon Sep 17 00:00:00 2001
From: Will Charczuk <will@clotheshor.se>
Date: Sat, 9 Jul 2016 09:11:47 -0700
Subject: [PATCH] testserver tweaks.

---
 testserver/main.go | 99 +++++++++++++++++++++++++++-------------------
 1 file changed, 58 insertions(+), 41 deletions(-)

diff --git a/testserver/main.go b/testserver/main.go
index a428135..037f792 100644
--- a/testserver/main.go
+++ b/testserver/main.go
@@ -8,50 +8,67 @@ import (
 	"github.com/wcharczuk/go-web"
 )
 
+func chartHandler(rc *web.RequestContext) web.ControllerResult {
+	format, err := rc.RouteParameter("format")
+	if err != nil {
+		format = "png"
+	}
+
+	if format == "png" {
+		rc.Response.Header().Set("Content-Type", "image/png")
+	} else {
+		rc.Response.Header().Set("Content-Type", "image/svg+xml")
+	}
+
+	c := chart.Chart{
+		Title: "A Test Chart",
+		TitleStyle: chart.Style{
+			Show: true,
+		},
+		Width:  640,
+		Height: 480,
+		Axes: chart.Style{
+			Show:        true,
+			StrokeWidth: 1.0,
+		},
+		YRange: chart.Range{
+			Min: 0.0,
+			Max: 7.0,
+		},
+		FinalValueLabel: chart.Style{
+			Show: true,
+		},
+		Series: []chart.Series{
+			chart.ContinuousSeries{
+				Name:    "a",
+				XValues: []float64{1.0, 2.0, 3.0, 4.0},
+				YValues: []float64{2.5, 5.0, 2.0, 3.3},
+			},
+			chart.ContinuousSeries{
+				Name:    "b",
+				XValues: []float64{3.0, 4.0, 5.0, 6.0},
+				YValues: []float64{6.0, 5.0, 4.0, 1.0},
+			},
+		},
+	}
+
+	if format == "png" {
+		err = c.Render(chart.PNG, rc.Response)
+	} else {
+		err = c.Render(chart.SVG, rc.Response)
+	}
+	if err != nil {
+		return rc.API().InternalError(err)
+	}
+	rc.Response.WriteHeader(http.StatusOK)
+	return nil
+}
+
 func main() {
 	app := web.New()
 	app.SetName("Chart Test Server")
 	app.SetLogger(web.NewStandardOutputLogger())
-	app.GET("/", func(rc *web.RequestContext) web.ControllerResult {
-		rc.Response.Header().Set("Content-Type", "image/png")
-		c := chart.Chart{
-			Title: "A Test Chart",
-			TitleStyle: chart.Style{
-				Show: true,
-			},
-			Width:  640,
-			Height: 480,
-			Axes: chart.Style{
-				Show:        true,
-				StrokeWidth: 1.0,
-			},
-			YRange: chart.Range{
-				Min: 0.0,
-				Max: 7.0,
-			},
-			FinalValueLabel: chart.Style{
-				Show: true,
-			},
-			Series: []chart.Series{
-				chart.ContinuousSeries{
-					Name:    "a",
-					XValues: []float64{1.0, 2.0, 3.0, 4.0},
-					YValues: []float64{2.5, 5.0, 2.0, 3.3},
-				},
-				chart.ContinuousSeries{
-					Name:    "b",
-					XValues: []float64{3.0, 4.0, 5.0, 6.0},
-					YValues: []float64{6.0, 5.0, 4.0, 1.0},
-				},
-			},
-		}
-
-		err := c.Render(chart.PNG, rc.Response)
-		if err != nil {
-			return rc.API().InternalError(err)
-		}
-		rc.Response.WriteHeader(http.StatusOK)
-		return nil
-	})
+	app.GET("/", chartHandler)
+	app.GET("/:format", chartHandler)
 	log.Fatal(app.Start())
 }