You've already forked speedtest-go
Migrate code from Go branch to new repo
This commit is contained in:
Maddie Zhan
@@ -0,0 +1,178 @@
|
||||
package results
|
||||
|
||||
import (
|
||||
"html/template"
|
||||
"net/http"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
|
||||
"github.com/librespeed/speedtest/config"
|
||||
"github.com/librespeed/speedtest/database"
|
||||
"github.com/librespeed/speedtest/database/schema"
|
||||
)
|
||||
|
||||
type StatsData struct {
|
||||
NoPassword bool
|
||||
LoggedIn bool
|
||||
Data []schema.TelemetryData
|
||||
}
|
||||
|
||||
func Stats(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||
t, err := template.New("template").Parse(htmlTemplate)
|
||||
if err != nil {
|
||||
log.Errorf("Failed to parse template: %s", err)
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
conf := config.LoadedConfig()
|
||||
var data StatsData
|
||||
|
||||
if conf.StatsPassword == "PASSWORD" {
|
||||
data.NoPassword = true
|
||||
}
|
||||
|
||||
if !data.NoPassword {
|
||||
op := r.FormValue("op")
|
||||
c, _ := r.Cookie("logged")
|
||||
|
||||
if c != nil && c.Value == "true" {
|
||||
if op == "logout" {
|
||||
cookie := &http.Cookie{
|
||||
Name: "logged",
|
||||
Value: "false",
|
||||
}
|
||||
http.SetCookie(w, cookie)
|
||||
http.Redirect(w, r, "/stats", http.StatusTemporaryRedirect)
|
||||
} else {
|
||||
data.LoggedIn = true
|
||||
|
||||
id := r.FormValue("id")
|
||||
switch id {
|
||||
case "L100":
|
||||
stats, err := database.DB.FetchLast100()
|
||||
if err != nil {
|
||||
log.Errorf("Error fetching data from database: %s", err)
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
data.Data = stats
|
||||
case "":
|
||||
default:
|
||||
stat, err := database.DB.FetchByUUID(id)
|
||||
if err != nil {
|
||||
log.Errorf("Error fetching data from database: %s", err)
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
data.Data = append(data.Data, *stat)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if op == "login" {
|
||||
password := r.FormValue("password")
|
||||
if password == conf.StatsPassword {
|
||||
cookie := &http.Cookie{
|
||||
Name: "logged",
|
||||
Value: "true",
|
||||
}
|
||||
http.SetCookie(w, cookie)
|
||||
http.Redirect(w, r, "/stats", http.StatusTemporaryRedirect)
|
||||
} else {
|
||||
w.WriteHeader(http.StatusForbidden)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if err := t.Execute(w, data); err != nil {
|
||||
log.Errorf("Error executing template: %s", err)
|
||||
w.WriteHeader(http.StatusInternalServerError)
|
||||
}
|
||||
}
|
||||
|
||||
const htmlTemplate = `<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<title>LibreSpeed - Stats</title>
|
||||
<style type="text/css">
|
||||
html,body{
|
||||
margin:0;
|
||||
padding:0;
|
||||
border:none;
|
||||
width:100%; min-height:100%;
|
||||
}
|
||||
html{
|
||||
background-color: hsl(198,72%,35%);
|
||||
font-family: "Segoe UI","Roboto",sans-serif;
|
||||
}
|
||||
body{
|
||||
background-color:#FFFFFF;
|
||||
box-sizing:border-box;
|
||||
width:100%;
|
||||
max-width:70em;
|
||||
margin:4em auto;
|
||||
box-shadow:0 1em 6em #00000080;
|
||||
padding:1em 1em 4em 1em;
|
||||
border-radius:0.4em;
|
||||
}
|
||||
h1,h2,h3,h4,h5,h6{
|
||||
font-weight:300;
|
||||
margin-bottom: 0.1em;
|
||||
}
|
||||
h1{
|
||||
text-align:center;
|
||||
}
|
||||
table{
|
||||
margin:2em 0;
|
||||
width:100%;
|
||||
}
|
||||
table, tr, th, td {
|
||||
border: 1px solid #AAAAAA;
|
||||
}
|
||||
th {
|
||||
width: 6em;
|
||||
}
|
||||
td {
|
||||
word-break: break-all;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<h1>LibreSpeed - Stats</h1>
|
||||
{{ if .NoPassword }}
|
||||
Please set statistics_password in settings.toml to enable access.
|
||||
{{ else if .LoggedIn }}
|
||||
<form action="stats" method="GET"><input type="hidden" name="op" value="logout" /><input type="submit" value="Logout" /></form>
|
||||
<form action="stats" method="GET">
|
||||
<h3>Search test results</h6>
|
||||
<input type="hidden" name="op" value="id" />
|
||||
<input type="text" name="id" id="id" placeholder="Test ID" value=""/>
|
||||
<input type="submit" value="Find" />
|
||||
<input type="submit" onclick="document.getElementById('id').value='L100'" value="Show last 100 tests" />
|
||||
</form>
|
||||
|
||||
{{ range $i, $v := .Data }}
|
||||
<table>
|
||||
<tr><th>Test ID</th><td>{{ $v.UUID }}</td></tr>
|
||||
<tr><th>Date and time</th><td>{{ $v.Timestamp }}</td></tr>
|
||||
<tr><th>IP and ISP Info</th><td>{{ $v.IPAddress }}<br/>{{ $v.ISPInfo }}</td></tr>
|
||||
<tr><th>User agent and locale</th><td>{{ $v.UserAgent }}<br/>{{ $v.Language }}</td></tr>
|
||||
<tr><th>Download speed</th><td>{{ $v.Download }}</td></tr>
|
||||
<tr><th>Upload speed</th><td>{{ $v.Upload }}</td></tr>
|
||||
<tr><th>Ping</th><td>{{ $v.Ping }}</td></tr>
|
||||
<tr><th>Jitter</th><td>{{ $v.Jitter }}</td></tr>
|
||||
<tr><th>Log</th><td>{{ $v.Log }}</td></tr>
|
||||
<tr><th>Extra info</th><td>{{ $v.Extra }}</td></tr>
|
||||
</table>
|
||||
{{ end }}
|
||||
{{ else }}
|
||||
<form action="stats?op=login" method="POST">
|
||||
<h3>Login</h3>
|
||||
<input type="password" name="password" placeholder="Password" value=""/>
|
||||
<input type="submit" value="Login" />
|
||||
</form>
|
||||
{{ end }}
|
||||
</body>
|
||||
</html>`
|
||||
Reference in New Issue
Block a user