kevin/matrix-veles
1
0
Fork 0
mirror of https://github.com/Unkn0wnCat/matrix-veles.git synced 2025-06-03 19:22:00 +02:00
Code Issues Projects Releases Wiki Activity

Add user API endpoints and metrics endpoint

Browse source
This commit is contained in:
Kevin Kandlbinder 2022-03-01 14:08:53 +01:00
parent c572bbce85
commit ff7339d88f
10 changed files with 299 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

22
internal/bot/handleHashing.go
View file

@ -10,6 +10,8 @@ import (
"github.com/Unkn0wnCat/matrix-veles/internal/config"
"github.com/Unkn0wnCat/matrix-veles/internal/db"
"github.com/Unkn0wnCat/matrix-veles/internal/db/model"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto"
"go.mongodb.org/mongo-driver/mongo"
"io"
"log"
@ -17,6 +19,21 @@ import (
"maunium.net/go/mautrix/event"
)
var (
filesFlagged = promauto.NewCounter(prometheus.CounterOpts{
Namespace: "bot",
Subsystem: "hash_handler",
Name: "flagged_files_total",
Help: "The total number of files found to be matching a hash",
})
filesCleared = promauto.NewCounter(prometheus.CounterOpts{
Namespace: "bot",
Subsystem: "hash_handler",
Name: "cleared_files_total",
Help: "The total number of files found to not be matching a hash",
})
)
// handleHashing hashes and checks a message, taking configured actions on match
func handleHashing(content *event.MessageEventContent, evt *event.Event, matrixClient *mautrix.Client) {
url, err := content.URL.Parse()
@ -44,9 +61,12 @@ func handleHashing(content *event.MessageEventContent, evt *event.Event, matrixC
// Fetch room configuration for adjusting behaviour
roomConfig := config.GetRoomConfig(evt.RoomID.String())
defer filesProcessed.Inc()
hashObj, err := db.GetEntryByHash(sum)
if err != nil {
if errors.Is(err, mongo.ErrNoDocuments) {
filesCleared.Inc()
if roomConfig.Debug {
matrixClient.SendNotice(evt.RoomID, fmt.Sprintf("DEBUG - This file is not on the hashlist: %s", sum))
}
@ -116,6 +136,8 @@ func checkSubscription(roomConfig *config.RoomConfig, hashObj *model.DBEntry) bo
// handleIllegalContent is called when a hash-match is found to take configured actions
func handleIllegalContent(evt *event.Event, matrixClient *mautrix.Client, hashObj *model.DBEntry, roomConfig config.RoomConfig) {
filesFlagged.Inc()
switch roomConfig.HashChecker.HashCheckMode {
case 0:
postNotice(evt, matrixClient, hashObj, roomConfig)