jlelse
/
kis3
Archived
1
Fork 0
This repository has been archived on 2021-01-04. You can view files and clone it, but cannot push or open issues or pull requests.
kis3/stats.go

185 lines
4.5 KiB
Go

package main
import (
"encoding/json"
"fmt"
"html/template"
"net/http"
"net/url"
"strconv"
"strings"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
"github.com/kis3/kis3/helpers"
"github.com/whiteshtef/clockwork"
)
func initStatsRouter() {
app.router.HandleFunc("/stats", StatsHandler)
}
func StatsHandler(w http.ResponseWriter, r *http.Request) {
// Require authentication
if appConfig.statsAuth() {
if !helpers.CheckAuth(w, r, appConfig.StatsUsername, appConfig.StatsPassword) {
return
}
}
// Do request
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":
sendJsonResponse(result, w)
case "chart":
sendChartResponse(result, w)
default: // "plain"
sendPlainResponse(result, w)
}
}
}
func doRequest(queries url.Values) (result []*RequestResultRow, e error) {
view := PAGES
switch strings.ToLower(queries.Get("view")) {
case "pages":
view = PAGES
case "referrers":
view = REFERRERS
case "useragents":
view = USERAGENTS
case "useragentnames":
view = USERAGENTNAMES
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
}
result, e = request(&ViewsRequest{
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"),
bots: queries.Get("bots"),
})
return
}
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))
}
}
func sendJsonResponse(result []*RequestResultRow, w http.ResponseWriter) {
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)
}
func startStatsTelegram() {
if app.tgBot == nil {
return
}
u := tgbotapi.NewUpdate(0)
scheduler := clockwork.NewScheduler()
scheduler.Schedule().Every(5).Seconds().Do(func() {
checkForTelegramUpdates(&u)
})
scheduler.Run()
}
func checkForTelegramUpdates(u *tgbotapi.UpdateConfig) {
updates, e := app.tgBot.GetUpdates(*u)
if e != nil {
return
}
for _, update := range updates {
if update.Message != nil && update.Message.Command() == "stats" {
response := ""
fakeUrl, e := url.Parse("/stats?" + update.Message.CommandArguments())
if e != nil {
response = "Request failed"
} else {
if appConfig.statsAuth() && (fakeUrl.Query().Get("username") != appConfig.StatsUsername || fakeUrl.Query().Get("password") != appConfig.StatsPassword) {
response = "Not authorized. Add username=yourusername&password=yourpassword to the query."
} else {
result, e := doRequest(fakeUrl.Query())
if e != nil {
response = "Request failed"
} else {
rowStrings := make([]string, len(result))
for i, row := range result {
rowStrings[i] = (*row).First + ": " + strconv.Itoa((*row).Second)
}
response = strings.Join(rowStrings, "\n")
}
}
}
msg := tgbotapi.NewMessage(update.Message.Chat.ID, response)
msg.ReplyToMessageID = update.Message.MessageID
_, e = app.tgBot.Send(msg)
if e != nil {
fmt.Println("Failed to send message:", e)
}
}
if update.UpdateID >= u.Offset {
u.Offset = update.UpdateID + 1
}
}
}