xevion/r2park
1
0
Fork 0
mirror of https://github.com/Xevion/r2park.git synced 2025-12-05 23:15:56 -06:00
Code Issues Packages Projects Releases Wiki Activity

Use regex to filter which levels can match

Browse Source
This commit is contained in:
Xevion
2024-03-09 17:52:53 -06:00
parent a72de046e1
commit 800bc54fbf
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
main.go
View File

@@ -1,10 +1,10 @@
package main package main
import ( import (
"bytes"
"flag" "flag"
"os" "os"
"os/signal" "os/signal"
"regexp"
"strings" "strings"
"github.com/bwmarrin/discordgo" "github.com/bwmarrin/discordgo"
@@ -23,8 +23,9 @@ var (
modalHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){ modalHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
"register": RegisterModalHandler, "register": RegisterModalHandler,
} }
db *redis.Client db *redis.Client
debugFlag = flag.Bool("debug", false, "Enable debug logging") debugFlag = flag.Bool("debug", false, "Enable debug logging")
levelPattern = regexp.MustCompile(`level=(error|fatal|panic|warning)`)
) )
func init() { func init() {
@@ -152,7 +153,8 @@ func Bot() {
type OutputSplitter struct{} type OutputSplitter struct{}
func (splitter *OutputSplitter) Write(p []byte) (n int, err error) { func (splitter *OutputSplitter) Write(p []byte) (n int, err error) {
if bytes.Contains(p, []byte("level=error")) { // Levels matched are contained within the OR'd group (error|fatal|panic|warning)
if levelPattern.Match(p) {
return os.Stderr.Write(p) return os.Stderr.Write(p)
} }
return os.Stdout.Write(p) return os.Stdout.Write(p)