handle ruby hash with compact

Author: skarim

src/analyze/ruby/index.js

@@ -244,6 +244,7 @@ async function collectVariableAssignments(node, variableMap) {
   const prism = await import('@ruby/prism');
   const LocalVariableWriteNode = prism.LocalVariableWriteNode;
   const HashNode = prism.HashNode;
+  const CallNode = prism.CallNode;
 
   if (LocalVariableWriteNode && node instanceof LocalVariableWriteNode) {
     if (node.value instanceof HashNode) {
@@ -252,6 +253,18 @@ async function collectVariableAssignments(node, variableMap) {
       const { extractHashProperties } = require('./extractors');
       const props = await extractHashProperties(node.value);
       variableMap[varName] = props;
+    } else if (node.value instanceof CallNode) {
+      // Handle patterns like { ... }.compact or { ... }.compact!
+      const callNode = node.value;
+      const methodName = callNode.name;
+
+      // Check if the call is a compact/compact! call with a Hash receiver
+      if ((methodName === 'compact' || methodName === 'compact!') && callNode.receiver instanceof HashNode) {
+        const varName = node.name;
+        const { extractHashProperties } = require('./extractors');
+        const props = await extractHashProperties(callNode.receiver);
+        variableMap[varName] = props;
+      }
     }
   }
 

tests/analyzeRuby.test.js

@@ -392,4 +392,20 @@ test.describe('analyzeRubyFile', () => {
     // If we get here, it means the analysis didn't hang
     assert.ok(true, 'Analysis completed without infinite recursion');
   });
+
+  test('should extract properties from hash assigned via .compact', async () => {
+    const compactFile = path.join(fixturesDir, 'ruby', 'compact_event.rb');
+    const sig = parseCustomFunctionSignature('TrackingModule.track(userId, EVENT_NAME, PROPERTIES)');
+    const events = await analyzeRubyFile(compactFile, [sig]);
+    assert.strictEqual(events.length, 1);
+
+    const evt = events[0];
+    assert.strictEqual(evt.eventName, 'RegisteredUser');
+    assert.deepStrictEqual(evt.properties, {
+      userId: { type: 'number' },
+      createdByAdminEmail: { type: 'string' },
+      _email: { type: 'string' },
+      emailOptIn: { type: 'boolean' }
+    });
+  });
 });

tests/fixtures/ruby/compact_event.rb

@@ -0,0 +1,13 @@
+def user_registration
+  actor_data = {
+    createdByAdminEmail: '[email protected]',
+    _email: '[email protected]',
+    emailOptIn: true
+  }.compact
+
+  TrackingModule.track(
+    123,
+    'RegisteredUser',
+    actor_data
+  )
+end