From 4326f83ae57f7ca478778724753d1fc6b7303e1c Mon Sep 17 00:00:00 2001
From: Benjamin Bannier <bbannier@gmail.com>
Date: Wed, 7 Feb 2024 21:56:10 +0100
Subject: [PATCH] Flag full `print` statement for `print-in-prod`

---
 rule-tests/__snapshots__/print-in-prod-snapshot.yml | 12 ++++++------
 rules/print-in-prod.yml                             |  2 +-
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/rule-tests/__snapshots__/print-in-prod-snapshot.yml b/rule-tests/__snapshots__/print-in-prod-snapshot.yml
index 2d7609a..8b0059b 100644
--- a/rule-tests/__snapshots__/print-in-prod-snapshot.yml
+++ b/rule-tests/__snapshots__/print-in-prod-snapshot.yml
@@ -2,30 +2,30 @@ id: print-in-prod
 snapshots:
   on A::%done { if (True) print 1; }:
     labels:
-    - source: print
+    - source: print 1;
       style: primary
       start: 24
-      end: 29
+      end: 32
     - source: on A::%done { if (True) print 1; }
       style: secondary
       start: 0
       end: 34
   on A::%done { print 1; }:
     labels:
-    - source: print
+    - source: print 1;
       style: primary
       start: 14
-      end: 19
+      end: 22
     - source: on A::%done { print 1; }
       style: secondary
       start: 0
       end: 24
   'type X = unit { x: uint8 { print $$; }};':
     labels:
-    - source: print
+    - source: print $$;
       style: primary
       start: 27
-      end: 32
+      end: 36
     - source: 'x: uint8 { print $$; }'
       style: secondary
       start: 16
diff --git a/rules/print-in-prod.yml b/rules/print-in-prod.yml
index 3a15b34..f24c81f 100644
--- a/rules/print-in-prod.yml
+++ b/rules/print-in-prod.yml
@@ -4,7 +4,7 @@ severity: warning
 language: spicy
 
 rule:
-  pattern: "print"
+  kind: "print"
   inside:
     any:
       - kind: "hook_decl"