Add and adjust triggerwords, improve word seperation

OpenShock · Sep 13, 2024 · beb58a4 · beb58a4
1 parent b463bab
commit beb58a4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/DiscordBot/Utils/ProfanityDetector.cs b/DiscordBot/Utils/ProfanityDetector.cs
@@ -27,10 +27,11 @@ public static class ProfanityDetector
  { "cock", 0.5f },
  { "crap", 0.3f },
  { "cum", 0.4f },
- { "damn", 0.3f },
+ { "damn", 0.15f },
  { "micropython", 0.5f },
  { "piss", 0.3f },
  { "ass", 0.2f },
+ { "wtf", 0.15f },
  };
 
  private static float CalculateWeight(float accumulated, float weight) => accumulated == 0 ? weight : MathF.Max(accumulated, weight) + 0.1f;
@@ -54,7 +55,7 @@ private static void GetContainedWordsCountAndWeight(ReadOnlySpan<char> lowerCase
  }
 
  private readonly record struct WordRange(int Start, int End);
- private static readonly SearchValues<char> _whiteSpaceSearchValues = SearchValues.Create([' ', '\t', '\r', '\n']);
+ private static readonly SearchValues<char> _seperationValues = SearchValues.Create([' ', '\t', '\r', '\n', '?', '!', ',', '.']);
  private static List<WordRange> GetWordRanges(ReadOnlySpan<char> span)
  {
  List<WordRange> wordRanges = [];
@@ -63,7 +64,7 @@ private static List<WordRange> GetWordRanges(ReadOnlySpan<char> span)
  while (true)
  {
  // Find the next white space character
- int index = span.IndexOfAny(_whiteSpaceSearchValues);
+ int index = span.IndexOfAny(_seperationValues);
  if (index < 0)
  {
  // Add the remaining word range if the word is not empty