1 files changed, 49 insertions, 40 deletions
diff --git a/pkg/html/pages/stats.go b/pkg/html/pages/stats.go
index 2682d28e9..3bc60cc5d 100644
--- a/pkg/html/pages/stats.go
+++ b/pkg/html/pages/stats.go
@@ -3,48 +3,57 @@
 
 package pages
 
-var StatsTemplate = Create(`
-<!doctype html>
-<html>
-<head>
-	<title>syzkaller stats</title>
-	<script type="text/javascript" src="https://www.google.com/jsapi"></script>
-	<script type="text/javascript">
-		google.load("visualization", "1", {packages:["corechart"]});
-		google.setOnLoadCallback(function() {
-			{{range $g := .}}
-			new google.visualization. {{if $g.Stacked}} AreaChart {{else}} LineChart {{end}} (
-				document.getElementById('div_{{$g.ID}}')).
-				draw(google.visualization.arrayToDataTable([
-					["-" {{range $line := $g.Lines}} , '{{$line}}' {{end}}],
-					{{range $p := $g.Points}} [ {{$p.X}} {{range $y := $p.Y}} , {{$y}} {{end}} ], {{end}}
-				]), {
-					title: '{{$g.Title}}',
-					titlePosition: 'in',
-					width: "95%",
-					height: "400",
-					chartArea: {width: '95%', height: '85%'},
-					legend: {position: 'in'},
-					lineWidth: 2,
-					focusTarget: "category",
-					{{if $g.Stacked}} isStacked: true, {{end}}
-					vAxis: {minValue: 1, textPosition: 'in', gridlines: {multiple: 1}, minorGridlines: {multiple: 1}},
-					hAxis: {minValue: 1, textPosition: 'out', maxAlternation: 1, gridlines: {multiple: 1},
-						minorGridlines: {multiple: 1}},
-				})
-			{{end}}
+import (
+	"bytes"
+	"fmt"
+	"html/template"
+
+	"github.com/google/syzkaller/pkg/stat"
+)
+
+func StatsHTML() (template.HTML, error) {
+	buf := new(bytes.Buffer)
+	data := stat.RenderGraphs()
+	if err := statsTemplate.Execute(buf, data); err != nil {
+		return "", fmt.Errorf("failed to execute stats template: %w", err)
+	}
+	return template.HTML(buf.String()), nil
+}
+
+var statsTemplate = Create(`
+<script type="text/javascript" src="https://www.google.com/jsapi"></script>
+<script type="text/javascript">
+	google.load("visualization", "1", {packages:["corechart"]});
+	google.setOnLoadCallback(function() {
+		{{range $g := .}}
+		new google.visualization. {{if $g.Stacked}} AreaChart {{else}} LineChart {{end}} (
+			document.getElementById('div_{{$g.ID}}')).
+			draw(google.visualization.arrayToDataTable([
+				["-" {{range $line := $g.Lines}} , '{{$line}}' {{end}}],
+				{{range $p := $g.Points}} [ {{$p.X}} {{range $y := $p.Y}} , {{$y}} {{end}} ], {{end}}
+			]), {
+				title: '{{$g.Title}}',
+				titlePosition: 'in',
+				width: "95%",
+				height: "400",
+				chartArea: {width: '95%', height: '85%'},
+				legend: {position: 'in'},
+				lineWidth: 2,
+				focusTarget: "category",
+				{{if $g.Stacked}} isStacked: true, {{end}}
+				vAxis: {minValue: 1, textPosition: 'in', gridlines: {multiple: 1}, minorGridlines: {multiple: 1}},
+				hAxis: {minValue: 1, textPosition: 'out', maxAlternation: 1, gridlines: {multiple: 1},
+					minorGridlines: {multiple: 1}},
+			})
+		{{end}}
+
+		{{/* Preserve vertical scroll position after page reloads. Otherwise it's random. */}}
+		window.scroll(0, window.location.hash.substring(1));
+		document.onscroll = function(e) { window.location.hash = Math.round(window.scrollY); };
+	});
+</script>
 
-			{{/* Preserve vertical scroll position after page reloads. Otherwise it's random. */}}
-			window.scroll(0, window.location.hash.substring(1));
-			document.onscroll = function(e) { window.location.hash = Math.round(window.scrollY); };
-		});
-	</script>
-	{{HEAD}}
-</head>
-<body>
 {{range $g := .}}
 	<div id="div_{{$g.ID}}"></div>
 {{end}}
-</body>
-</html>
 `)