2019-05-26 20:32:02 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
|
|
|
"html/template"
|
|
|
|
"net/http"
|
2019-05-27 13:03:30 +00:00
|
|
|
"net/url"
|
2019-05-26 20:32:02 +00:00
|
|
|
"strconv"
|
|
|
|
"strings"
|
2019-10-13 11:13:11 +00:00
|
|
|
|
2021-01-04 09:56:42 +00:00
|
|
|
"git.jlel.se/jlelse/kis3/helpers"
|
2019-05-26 20:32:02 +00:00
|
|
|
)
|
|
|
|
|
2019-05-27 13:03:30 +00:00
|
|
|
func initStatsRouter() {
|
2021-01-04 09:56:42 +00:00
|
|
|
app.router.HandleFunc("/stats", statsHandler)
|
2019-05-26 20:32:02 +00:00
|
|
|
}
|
|
|
|
|
2021-01-04 09:56:42 +00:00
|
|
|
func statsHandler(w http.ResponseWriter, r *http.Request) {
|
2019-05-26 20:32:02 +00:00
|
|
|
// Require authentication
|
|
|
|
if appConfig.statsAuth() {
|
|
|
|
if !helpers.CheckAuth(w, r, appConfig.StatsUsername, appConfig.StatsPassword) {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Do request
|
2019-05-27 13:03:30 +00:00
|
|
|
queryValues := r.URL.Query()
|
|
|
|
result, e := doRequest(queryValues)
|
|
|
|
if e != nil {
|
|
|
|
fmt.Println("Database request failed:", e)
|
|
|
|
w.WriteHeader(500)
|
|
|
|
} else if result != nil {
|
|
|
|
w.Header().Set("Cache-Control", "max-age=0")
|
|
|
|
switch queryValues.Get("format") {
|
|
|
|
case "json":
|
2021-01-04 09:56:42 +00:00
|
|
|
sendJSONResponse(result, w)
|
2019-05-27 13:03:30 +00:00
|
|
|
case "chart":
|
|
|
|
sendChartResponse(result, w)
|
|
|
|
default: // "plain"
|
|
|
|
sendPlainResponse(result, w)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func doRequest(queries url.Values) (result []*RequestResultRow, e error) {
|
2019-05-26 20:32:02 +00:00
|
|
|
view := PAGES
|
|
|
|
switch strings.ToLower(queries.Get("view")) {
|
|
|
|
case "pages":
|
|
|
|
view = PAGES
|
|
|
|
case "referrers":
|
|
|
|
view = REFERRERS
|
|
|
|
case "useragents":
|
|
|
|
view = USERAGENTS
|
|
|
|
case "useragentnames":
|
|
|
|
view = USERAGENTNAMES
|
2019-10-19 10:13:02 +00:00
|
|
|
case "os":
|
|
|
|
view = OS
|
2019-05-26 20:32:02 +00:00
|
|
|
case "hours":
|
|
|
|
view = HOURS
|
|
|
|
case "days":
|
|
|
|
view = DAYS
|
|
|
|
case "weeks":
|
|
|
|
view = WEEKS
|
|
|
|
case "months":
|
|
|
|
view = MONTHS
|
|
|
|
case "allhours":
|
|
|
|
view = ALLHOURS
|
|
|
|
case "alldays":
|
|
|
|
view = ALLDAYS
|
|
|
|
case "count":
|
|
|
|
view = COUNT
|
|
|
|
}
|
2019-05-27 13:03:30 +00:00
|
|
|
result, e = request(&ViewsRequest{
|
2019-05-26 20:32:02 +00:00
|
|
|
view: view,
|
|
|
|
from: queries.Get("from"),
|
|
|
|
fromRel: queries.Get("fromrel"),
|
|
|
|
to: queries.Get("to"),
|
|
|
|
toRel: queries.Get("torel"),
|
|
|
|
url: queries.Get("url"),
|
|
|
|
ref: queries.Get("ref"),
|
|
|
|
ua: queries.Get("ua"),
|
|
|
|
ordercol: strings.ToLower(queries.Get("ordercol")),
|
|
|
|
order: strings.ToUpper(queries.Get("order")),
|
|
|
|
limit: queries.Get("limit"),
|
2019-08-02 17:32:34 +00:00
|
|
|
bots: queries.Get("bots"),
|
2019-10-19 10:13:02 +00:00
|
|
|
os: queries.Get("os"),
|
2019-05-26 20:32:02 +00:00
|
|
|
})
|
2019-05-27 13:03:30 +00:00
|
|
|
return
|
2019-05-26 20:32:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func sendPlainResponse(result []*RequestResultRow, w http.ResponseWriter) {
|
|
|
|
w.Header().Set("Content-Type", "text/plain")
|
|
|
|
for _, row := range result {
|
|
|
|
_, _ = fmt.Fprintln(w, (*row).First+": "+strconv.Itoa((*row).Second))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-01-04 09:56:42 +00:00
|
|
|
func sendJSONResponse(result []*RequestResultRow, w http.ResponseWriter) {
|
2019-05-26 20:32:02 +00:00
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
|
|
jsonBytes, _ := json.Marshal(result)
|
|
|
|
_, _ = fmt.Fprintln(w, string(jsonBytes))
|
|
|
|
}
|
|
|
|
|
|
|
|
func sendChartResponse(result []*RequestResultRow, w http.ResponseWriter) {
|
|
|
|
labels := make([]string, len(result))
|
|
|
|
values := make([]int, len(result))
|
|
|
|
for i, row := range result {
|
|
|
|
labels[i] = row.First
|
|
|
|
values[i] = row.Second
|
|
|
|
}
|
|
|
|
chartJSString, e := app.staticBox.FindString("Chart.min.js")
|
|
|
|
if e != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
data := struct {
|
|
|
|
Labels []string
|
|
|
|
Values []int
|
|
|
|
ChartJS template.JS
|
|
|
|
}{
|
|
|
|
Labels: labels,
|
|
|
|
Values: values,
|
|
|
|
ChartJS: template.JS(chartJSString),
|
|
|
|
}
|
|
|
|
chartTemplateString, e := app.staticBox.FindString("chart.html")
|
|
|
|
if e != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
t, e := template.New("chart").Parse(chartTemplateString)
|
|
|
|
if e != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
_ = t.Execute(w, data)
|
|
|
|
}
|
2019-05-27 13:03:30 +00:00
|
|
|
|
2020-04-22 19:40:28 +00:00
|
|
|
func respondToTelegramUpdate(u *telegramUpdate) {
|
|
|
|
if app.telegram != nil && strings.HasPrefix(u.Message.Text, "/stats") {
|
|
|
|
response := ""
|
2021-01-04 09:56:42 +00:00
|
|
|
fakeURL, e := url.Parse("/stats?" + strings.TrimSpace(strings.TrimPrefix(u.Message.Text, "/stats")))
|
2020-04-22 19:40:28 +00:00
|
|
|
if e != nil {
|
|
|
|
response = "Request failed"
|
|
|
|
} else {
|
2021-01-04 09:56:42 +00:00
|
|
|
if appConfig.statsAuth() && (fakeURL.Query().Get("username") != appConfig.StatsUsername || fakeURL.Query().Get("password") != appConfig.StatsPassword) {
|
2020-04-22 19:40:28 +00:00
|
|
|
response = "Not authorized. Add username=yourusername&password=yourpassword to the query."
|
2019-05-27 13:03:30 +00:00
|
|
|
} else {
|
2021-01-04 09:56:42 +00:00
|
|
|
result, e := doRequest(fakeURL.Query())
|
2020-04-22 19:40:28 +00:00
|
|
|
if e != nil {
|
|
|
|
response = "Request failed"
|
2019-05-27 13:03:30 +00:00
|
|
|
} else {
|
2020-04-22 19:40:28 +00:00
|
|
|
rowStrings := make([]string, len(result))
|
|
|
|
for i, row := range result {
|
|
|
|
rowStrings[i] = (*row).First + ": " + strconv.Itoa((*row).Second)
|
2019-05-27 13:03:30 +00:00
|
|
|
}
|
2020-04-22 19:40:28 +00:00
|
|
|
response = strings.Join(rowStrings, "\n")
|
2019-05-27 13:03:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2020-04-22 19:40:28 +00:00
|
|
|
e = app.telegram.replyToMessage(u.Message.Chat.Id, response, u.Message.Id)
|
|
|
|
if e != nil {
|
|
|
|
fmt.Println("Failed to send message:", e)
|
2019-05-27 13:03:30 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|