From 600bafe08de5dd2bdef484d2e42d343f9bf19278 Mon Sep 17 00:00:00 2001
From: Simon Marsh <simon@burble.com>
Date: Tue, 12 Jan 2021 10:15:54 +0000
Subject: [PATCH] Add utility functions for filtering results and rename
 templates

---
 frontend/bindata/templates/{bgpmap => bgpmap.tpl}   |  0
 frontend/bindata/templates/{bird => bird.tpl}       |  0
 frontend/bindata/templates/{page => page.tpl}       |  0
 frontend/bindata/templates/{summary => summary.tpl} |  0
 frontend/bindata/templates/{whois => whois.tpl}     |  0
 frontend/template.go                                | 13 ++++++++++++-
 6 files changed, 12 insertions(+), 1 deletion(-)
 rename frontend/bindata/templates/{bgpmap => bgpmap.tpl} (100%)
 rename frontend/bindata/templates/{bird => bird.tpl} (100%)
 rename frontend/bindata/templates/{page => page.tpl} (100%)
 rename frontend/bindata/templates/{summary => summary.tpl} (100%)
 rename frontend/bindata/templates/{whois => whois.tpl} (100%)

diff --git a/frontend/bindata/templates/bgpmap b/frontend/bindata/templates/bgpmap.tpl
similarity index 100%
rename from frontend/bindata/templates/bgpmap
rename to frontend/bindata/templates/bgpmap.tpl
diff --git a/frontend/bindata/templates/bird b/frontend/bindata/templates/bird.tpl
similarity index 100%
rename from frontend/bindata/templates/bird
rename to frontend/bindata/templates/bird.tpl
diff --git a/frontend/bindata/templates/page b/frontend/bindata/templates/page.tpl
similarity index 100%
rename from frontend/bindata/templates/page
rename to frontend/bindata/templates/page.tpl
diff --git a/frontend/bindata/templates/summary b/frontend/bindata/templates/summary.tpl
similarity index 100%
rename from frontend/bindata/templates/summary
rename to frontend/bindata/templates/summary.tpl
diff --git a/frontend/bindata/templates/whois b/frontend/bindata/templates/whois.tpl
similarity index 100%
rename from frontend/bindata/templates/whois
rename to frontend/bindata/templates/whois.tpl
diff --git a/frontend/template.go b/frontend/template.go
index 0c041dc..89b8ce9 100644
--- a/frontend/template.go
+++ b/frontend/template.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"strings"
 	"text/template"
 )
 
@@ -42,6 +43,16 @@ type SummaryRowData struct {
 	Info        string
 }
 
+// utility functions to allow filtering of results in the template
+
+func (r SummaryRowData) NameHasPrefix(prefix string) bool {
+	return strings.HasPrefix(r.Name, prefix)
+}
+
+func (r SummaryRowData) NameContains(prefix string) bool {
+	return strings.Contains(r.Name, prefix)
+}
+
 type TemplateSummary struct {
 	ServerName string
 	Raw        string
@@ -94,7 +105,7 @@ func ImportTemplates() {
 	for _, tmpl := range requiredTemplates {
 
 		// extract the template definition from the bindata
-		def := MustAssetString("templates/" + tmpl)
+		def := MustAssetString("templates/" + tmpl + ".tpl")
 
 		// and add it to the template library
 		template, err := template.New(tmpl).Parse(def)