Add rules statistics

1 files changed, 18 insertions, 10 deletions

diff --git a/rules_manager.go b/rules_manager.go
index 636fc74..0e6c3d1 100644
--- a/rules_manager.go
+++ b/rules_manager.go
@@ -121,16 +121,24 @@ func LoadRulesManager(storage Storage, flagRegex string) (RulesManager, error) {
 
 	// if there are no rules in database (e.g. first run), set flagRegex as first rule
 	if len(rulesManager.rules) == 0 {
-		if _, err := rulesManager.AddRule(context.Background(), Rule{
-			Name:  "flag",
-			Color: "#E53935",
-			Notes: "Mark connections where the flag is stolen",
-			Patterns: []Pattern{
-				{Regex: flagRegex, Direction: DirectionToClient},
-			},
-		}); err != nil {
-			return nil, err
-		}
+		go func() {
+			_, _ = rulesManager.AddRule(context.Background(), Rule{
+				Name:  "flag_out",
+				Color: "#e53935",
+				Notes: "Mark connections where the flags are stolen",
+				Patterns: []Pattern{
+					{Regex: flagRegex, Direction: DirectionToClient, Flags: RegexFlags{Utf8Mode: true}},
+				},
+			})
+			_, _ = rulesManager.AddRule(context.Background(), Rule{
+				Name:  "flag_in",
+				Color: "#43A047",
+				Notes: "Mark connections where the flags are placed",
+				Patterns: []Pattern{
+					{Regex: flagRegex, Direction: DirectionToServer, Flags: RegexFlags{Utf8Mode: true}},
+				},
+			})
+		}()
 	} else {
 		if err := rulesManager.generateDatabase(rules[len(rules)-1].ID); err != nil {
 			return nil, err