107 lines
2.4 KiB
Go
107 lines
2.4 KiB
Go
package plugins
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"kurumibot/database/mdb"
|
|
"kurumibot/utils"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.nix13.pw/scuroneko/extypes"
|
|
"git.nix13.pw/scuroneko/laniakea"
|
|
"git.nix13.pw/scuroneko/slog"
|
|
)
|
|
|
|
func RegisterLogs(bot *laniakea.Bot) {
|
|
p := laniakea.NewPlugin("Logs")
|
|
p.Command(getLogs, "logs")
|
|
p.Command(getMsgLogs, "msglogs")
|
|
p.AddMiddleware(AdminMiddleware())
|
|
bot.AddPlugins(p.Build())
|
|
}
|
|
func InitLogMiddleware() laniakea.Middleware {
|
|
return laniakea.NewMiddleware("LogMiddleware", logMiddleware).SetAsync(true).Build()
|
|
}
|
|
|
|
func getLogs(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
|
logs, err := mdb.GetConsoleLogs(db)
|
|
if err != nil {
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
out := encodeLogs(logs)
|
|
ctx.Answer(strings.Join(out, ""))
|
|
}
|
|
func getMsgLogs(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
|
logs, err := mdb.GetMessageLogs(db)
|
|
if err != nil {
|
|
ctx.Error(err)
|
|
return
|
|
}
|
|
out := encodeLogs(logs)
|
|
ctx.Answer(strings.Join(out, ""))
|
|
}
|
|
|
|
// Utils
|
|
|
|
func encodeLogs[T comparable](logs extypes.Slice[T]) extypes.Slice[string] {
|
|
out := make(extypes.Slice[string], len(logs))
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
enc := json.NewEncoder(buf)
|
|
enc.SetEscapeHTML(false)
|
|
enc.SetIndent("", " ")
|
|
for i, log := range logs {
|
|
_ = enc.Encode(log)
|
|
out[i] = fmt.Sprintf("`%s`", buf.String())
|
|
buf.Reset()
|
|
}
|
|
return out
|
|
}
|
|
|
|
func logMiddleware(ctx *laniakea.MsgContext, db *laniakea.DatabaseContext) {
|
|
if ctx.Msg == nil {
|
|
return
|
|
}
|
|
entry := &mdb.MessageLogEntry{
|
|
MessageID: ctx.Msg.MessageID,
|
|
SenderID: ctx.FromID,
|
|
ChatID: ctx.Msg.Chat.ID,
|
|
Text: ctx.Msg.Text,
|
|
TimeStamp: time.Now().Unix(),
|
|
}
|
|
err := mdb.WriteMessageLog(db, entry)
|
|
if err != nil {
|
|
ctx.Bot.Logger().Errorln(err)
|
|
}
|
|
}
|
|
|
|
type DatabaseWriter struct {
|
|
slog.LoggerWriter
|
|
db *laniakea.DatabaseContext
|
|
}
|
|
|
|
func (w *DatabaseWriter) Print(level slog.LogLevel, prefix string, traceback []*slog.MethodTraceback, messages ...any) error {
|
|
t := time.Now()
|
|
if messages[len(messages)-1] == "\n" {
|
|
messages = messages[:len(messages)-1]
|
|
}
|
|
entry := &mdb.ConsoleLogEntry{
|
|
Level: level.GetName(),
|
|
Prefix: prefix,
|
|
Traceback: slog.FormatFullTraceback(traceback),
|
|
Message: strings.Join(utils.Map(messages, func(el any) string {
|
|
return fmt.Sprintf("%v", el)
|
|
}), "\n"),
|
|
Time: t,
|
|
TimeStamp: t.Unix(),
|
|
}
|
|
err := mdb.WriteConsoleLog(w.db, entry)
|
|
return err
|
|
}
|
|
func DatabaseLogger(db *laniakea.DatabaseContext) slog.LoggerWriter {
|
|
return &DatabaseWriter{db: db}
|
|
}
|