Teach root path HTML

Depending on whats in the accept request header, the root path either response with a plain text or some nice HTML.

Teach root path HTML
a42a3ae8 · gjahn · bf2ab576 · a42a3ae8 · a42a3ae8 · a42a3ae8
Commit a42a3ae8 authored 1 year ago by gjahn
--- a/routing/routes.go
+++ b/routing/routes.go
@@ -4,7 +4,11 @@ import (
    "encoding/json"
    "os"
    "fmt"
+    "strings"
    "net/http"
+    "html/template"
+    "log"
+    "bytes"
    "webservice/configuration"
@@ -14,8 +18,32 @@ import (
 func SetRoutes( router *f.App, config *configuration.Config, healthiness *bool ){
+    indexHtmlTemplate, err := template.New( "index" ).Parse( indexHtml )
+    if err != nil {
+        log.Fatal( err )
+    }
    router.Get( "/", func( c *f.Ctx ) error {
-        return c.SendString( "Hello, World!" )
+        headers := c.GetReqHeaders()
+        if ! strings.Contains( headers[ "Accept" ], "html" ) {
+            c.Set( "Content-Type", "text/plain; charset=utf-8" )
+            return c.SendString( "Hello, World!" )
+        }
+        data := indexHtmlData{
+            Version: "",
+            Color: "",
+        }
+        buffer := &bytes.Buffer{}
+        err := indexHtmlTemplate.Execute( buffer, data )
+        if err != nil {
+            return err
+        }
+        c.Set( "Content-Type", "text/html; charset=utf-8" )
+        return c.Send( buffer.Bytes() )
    })

--- a/routing/routes_test.go
+++ b/routing/routes_test.go
@@ -71,10 +71,17 @@ func TestIndexRoute( t *testing.T ){
    router, _, _ := setup()
    req := ht.NewRequest( "GET", "/", nil )
-    res, err := router.Test( req, -1 )
+    req.Header.Add( "Accept", "text/html" )
+    res, _ := router.Test( req, -1 )
    bodyContent, err := bodyToString( &res.Body )
    assert.Nil( t, err )
+    assert.Contains( t, bodyContent, "</html>" )
+    assert.Contains( t, bodyContent, "<head>" )
+    req = ht.NewRequest( "GET", "/", nil )
+    res, _ = router.Test( req, -1 )
+    bodyContent, err = bodyToString( &res.Body )
+    assert.Nil( t, err )
    assert.Equal( t, "Hello, World!", bodyContent )
 }

--- a/routing/templates.go
+++ b/routing/templates.go
+package routing
+const indexHtml = `
+    <!DOCTYPE html>
+    <html>
+        <head>
+            <meta charset="UTF-8">
+            <title>Webservice</title>
+        </head>
+        <body>
+            <h1 style="color: {{ .Color }}">Hello World, again!</h1>
+            <p>Version: {{ .Version }}</p>
+        </body>
+    </html>
+`
+type indexHtmlData struct {
+    Version string
+    Color   string
+}