@flisk/analyze-tracking 0.8.7 → 0.9.0

package/README.md

@@ -95,25 +95,25 @@ Use this to understand where your events live in the code and how they're being
 Your LLM of choice is used for generating descriptions of events, properties, and implementations.
 
 See [schema.json](schema.json) for a JSON Schema of the output.
- 
+
 
 ## Supported tracking libraries & languages
 
-| Library | JavaScript/TypeScript | Python | Ruby | Go |
-|---------|:---------------------:|:------:|:----:|:--:|
-| Google Analytics   | ✅ | ❌ | ❌ | ❌ |
-| Google Tag Manager | ✅ | ❌ | ❌ | ❌ |
-| Segment            | ✅ | ✅ | ✅ | ✅ |
-| Mixpanel           | ✅ | ✅ | ✅ | ✅ |
-| Amplitude          | ✅ | ✅ | ❌ | ✅ |
-| Rudderstack        | ✅ | ✅ | ✳️ | ✳️ |
-| mParticle          | ✅ | ❌ | ❌ | ❌ |
-| PostHog            | ✅ | ✅ | ✅ | ✅ |
-| Pendo              | ✅ | ❌ | ❌ | ❌ |
-| Heap               | ✅ | ❌ | ❌ | ❌ |
-| Snowplow           | ✅ | ✅ | ✅ | ✅ |
-| Datadog RUM        | ✅ | ❌ | ❌ | ❌ |
-| Custom Function    | ✅ | ✅ | ✅ | ✅ |
+| Library | JavaScript/TypeScript | Python | Ruby | Go | Swift |
+|---------|:---------------------:|:------:|:----:|:--:|:--:|
+| Google Analytics   | ✅ | ❌ | ❌ | ❌ | ✅ |
+| Google Tag Manager | ✅ | ❌ | ❌ | ❌ | ✅ |
+| Segment            | ✅ | ✅ | ✅ | ✅ | ✅ |
+| Mixpanel           | ✅ | ✅ | ✅ | ✅ | ✅ |
+| Amplitude          | ✅ | ✅ | ❌ | ✅ | ✅ |
+| Rudderstack        | ✅ | ✅ | ✳️ | ✳️ | ✅ |
+| mParticle          | ✅ | ❌ | ❌ | ❌ | ✅ |
+| PostHog            | ✅ | ✅ | ✅ | ✅ | ✅ |
+| Pendo              | ✅ | ❌ | ❌ | ❌ | ✅ |
+| Heap               | ✅ | ❌ | ❌ | ❌ | ✅ |
+| Snowplow           | ✅ | ✅ | ✅ | ✅ | ❌ |
+| Datadog RUM        | ✅ | ❌ | ❌ | ❌ | ❌ |
+| Custom Function    | ✅ | ✅ | ✅ | ✅ | ✅ |
 
 ✳️ Rudderstack's SDKs often use the same format as Segment, so Rudderstack events may be detected as Segment events.
 
@@ -129,6 +129,13 @@ See [schema.json](schema.json) for a JSON Schema of the output.
     '<property_name>': '<property_value>'
   });
   ```
+
+  **Swift**
+  ```swift
+  Analytics.logEvent("<event_name>", parameters: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -147,6 +154,11 @@ See [schema.json](schema.json) for a JSON Schema of the output.
     '<property_name>': '<property_value>'
   });
   ```
+
+  **Swift**
+  ```swift
+  dataLayer.push(["event": "<event_name>", "<property_name>": "<property_value>"])
+  ```
 </details>
 
 <details>
@@ -185,6 +197,11 @@ See [schema.json](schema.json) for a JSON Schema of the output.
       Set("<property_name>", "<property_value>"),
   })
   ```
+
+  **Swift**
+  ```swift
+  analytics.track(name: "<event_name>", properties: TrackProperties("<property_name>": "<property_value>"))
+  ```
 </details>
 
 <details>
@@ -221,6 +238,13 @@ See [schema.json](schema.json) for a JSON Schema of the output.
     }),
   })
   ```
+
+  **Swift**
+  ```swift
+  Mixpanel.mainInstance().track(event: "<event_name>", properties: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -256,6 +280,14 @@ See [schema.json](schema.json) for a JSON Schema of the output.
     },
   })
   ```
+
+  **Swift**
+  ```swift
+  amplitude.track(
+    eventType: "<event_name>",
+    eventProperties: ["<property_name>": "<property_value>"]
+  )
+  ```
 </details>
 
 <details>
@@ -295,6 +327,13 @@ See [schema.json](schema.json) for a JSON Schema of the output.
       Set("<property_name>", "<property_value>"),
   })
   ```
+
+  **Swift**
+  ```swift
+  RSClient.sharedInstance()?.track("<event_name>", properties: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -306,6 +345,15 @@ See [schema.json](schema.json) for a JSON Schema of the output.
     '<property_name>': '<property_value>'
   });
   ```
+
+  **Swift**
+  ```swift
+  let event = MPEvent(name: "<event_name>", type: .other)
+  event.customAttributes = [
+    "<property_name>": "<property_value>"
+  ]
+  MParticle.sharedInstance().logEvent(event)
+  ```
 </details>
 
 <details>
@@ -353,6 +401,13 @@ See [schema.json](schema.json) for a JSON Schema of the output.
       Set("<property_name>", "<property_value>"),
   })
   ```
+
+  **Swift**
+  ```swift
+  PostHogSDK.shared.capture("<event_name>", properties: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -372,7 +427,12 @@ See [schema.json](schema.json) for a JSON Schema of the output.
   })
   ```
 
-
+  **Swift**
+  ```swift
+  PendoManager.shared().track("<event_name>", properties: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -392,7 +452,12 @@ See [schema.json](schema.json) for a JSON Schema of the output.
   })
   ```
 
-
+  **Swift**
+  ```swift
+  Heap.shared.track("<event_name>", properties: [
+    "<property_name>": "<property_value>"
+  ])
+  ```
 </details>
 
 <details>
@@ -403,7 +468,7 @@ See [schema.json](schema.json) for a JSON Schema of the output.
   datadogRum.addAction('<event_name>', {
     '<property_name>': '<property_value>'
   });
-  
+
   // Or via window
   window.DD_RUM.addAction('<event_name>', {
     '<property_name>': '<property_value>'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@flisk/analyze-tracking",
-  "version": "0.8.7",
+  "version": "0.9.0",
   "description": "Analyzes tracking code in a project and generates data schemas",
   "main": "src/index.js",
   "bin": {
@@ -13,6 +13,7 @@
     "test:python": "node --experimental-vm-modules --test tests/analyzePython.test.js",
     "test:ruby": "node --experimental-vm-modules --test tests/analyzeRuby.test.js",
     "test:go": "node --test tests/analyzeGo.test.js",
+    "test:swift": "node --test tests/analyzeSwift.test.js",
     "test:cli": "node --test tests/cli.test.js",
     "test:schema": "node --test tests/schema.test.js",
     "test:generateDescriptions": "node --test tests/generateDescriptions.test.js",
@@ -34,6 +35,7 @@
   },
   "homepage": "https://github.com/fliskdata/analyze-tracking#readme",
   "dependencies": {
+    "@flisk/swift-ast": "^0.1.3",
     "@langchain/core": "^0.3.56",
     "@langchain/google-vertexai": "^0.2.9",
     "@langchain/openai": "^0.5.10",
@@ -52,9 +54,9 @@
     "zod": "^3.24.4"
   },
   "devDependencies": {
+    "@types/react": "^19.1.6",
     "ajv": "^8.17.1",
     "lodash": "^4.17.21",
-    "react": "^19.1.0",
-    "@types/react": "^19.1.6"
+    "react": "^19.1.0"
   }
 }

package/src/analyze/index.js

@@ -13,12 +13,13 @@ const { analyzeTsFiles } = require('./typescript');
 const { analyzePythonFile } = require('./python');
 const { analyzeRubyFile, prebuildConstantMaps } = require('./ruby');
 const { analyzeGoFile } = require('./go');
+const { analyzeSwiftFile } = require('./swift');
 
 /**
  * Analyzes a single file for analytics tracking calls
- * 
+ *
  * Note: typescript files are handled separately by analyzeTsFiles, which is a batch processor
- * 
+ *
  * @param {string} file - Path to the file to analyze
  * @param {Array<string>} customFunctionSignatures - Custom function signatures to detect
  * @returns {Promise<Array<Object>>} Array of events found in the file
@@ -28,19 +29,20 @@ async function analyzeFile(file, customFunctionSignatures) {
   if (/\.py$/.test(file))   return analyzePythonFile(file, customFunctionSignatures)
   if (/\.rb$/.test(file))   return analyzeRubyFile(file, customFunctionSignatures)
   if (/\.go$/.test(file))   return analyzeGoFile(file, customFunctionSignatures)
+  if (/\.swift$/.test(file)) return analyzeSwiftFile(file, customFunctionSignatures)
   return []
 }
 
 /**
  * Adds an event to the events collection, merging properties if event already exists
- * 
+ *
  * @param {Object} allEvents - Collection of all events
  * @param {Object} event - Event to add
  * @param {string} baseDir - Base directory for relative path calculation
  */
 function addEventToCollection(allEvents, event, baseDir) {
   const relativeFilePath = path.relative(baseDir, event.filePath);
-  
+
   const implementation = {
     path: relativeFilePath,
     line: event.line,
@@ -64,12 +66,12 @@ function addEventToCollection(allEvents, event, baseDir) {
 
 /**
  * Processes all files that are not TypeScript files in parallel
- * 
+ *
  * Checks the system's file descriptor limit and uses 80% of it to avoid running out of file descriptors
  * Creates a promise pool and launches one analysis for each file in parallel
  * When a slot frees up, the next file is launched
  * Waits for the remaining work to complete
- * 
+ *
  * @param {Array<string>} files - Array of file paths
  * @param {Object} allEvents - Collection to add events to
  * @param {string} baseDir - Base directory for relative paths
@@ -121,10 +123,10 @@ async function processFiles(files, allEvents, baseDir, customFunctionSignatures)
 
 /**
  * Analyze a directory recursively for analytics tracking calls
- * 
+ *
  * This function scans all supported files in a directory tree and identifies analytics tracking calls,
  * handling different file types appropriately.
- * 
+ *
  * @param {string} dirPath - Path to the directory to analyze
  * @param {Array<string>} [customFunctions=null] - Array of custom tracking function signatures to detect
  * @returns {Promise<Object>} Object mapping event names to their tracking implementations
@@ -132,17 +134,22 @@ async function processFiles(files, allEvents, baseDir, customFunctionSignatures)
 async function analyzeDirectory(dirPath, customFunctions) {
   const allEvents = {};
 
-  const customFunctionSignatures = (customFunctions?.length > 0) 
-    ? customFunctions.map(parseCustomFunctionSignature) 
-    : null;
+  let customFunctionSignatures = null;
+  if (Array.isArray(customFunctions) && customFunctions.length > 0) {
+    customFunctionSignatures = customFunctions.map(cf => {
+      if (cf && typeof cf === 'object' && cf.functionName) return cf;
+      if (typeof cf === 'string') return parseCustomFunctionSignature(cf);
+      return null;
+    }).filter(Boolean);
+  }
 
   const files = getAllFiles(dirPath);
-  
+
   // Separate TypeScript files from others for optimized processing
   const tsFiles = [];
   const nonTsFiles = [];
   const rubyFiles = [];
-  
+
   for (const file of files) {
     const isTsFile = /\.(tsx?)$/.test(file);
     if (isTsFile) {

package/src/analyze/javascript/detectors/analytics-source.js

@@ -45,20 +45,23 @@ function detectAnalyticsSource(node, customFunction) {
 function isCustomFunction(node, customFunction) {
   if (!customFunction) return false;
 
-  // Support dot-separated names like "CustomModule.track"
-  const parts = customFunction.split('.');
+  // Support dot-separated names like "CustomModule.track" and chained calls like "getTrackingService().track"
+  // Normalize each segment by stripping trailing parentheses
+  const parts = customFunction.split('.').map(p => p.replace(/\(\s*\)$/, ''));
 
   // Simple identifier (no dot)
   if (parts.length === 1) {
-    return node.callee.type === NODE_TYPES.IDENTIFIER && node.callee.name === customFunction;
+    return node.callee.type === NODE_TYPES.IDENTIFIER && node.callee.name === parts[0];
   }
 
-  // For dot-separated names, the callee should be a MemberExpression chain.
-  if (node.callee.type !== NODE_TYPES.MEMBER_EXPRESSION) {
+  // For dot-separated names, the callee should be a MemberExpression chain,
+  // but we also allow CallExpression in the chain (e.g., getService().track)
+  const callee = node.callee;
+  if (callee.type !== NODE_TYPES.MEMBER_EXPRESSION && callee.type !== NODE_TYPES.CALL_EXPRESSION) {
     return false;
   }
 
-  return matchesMemberChain(node.callee, parts);
+  return matchesMemberChain(callee, parts);
 }
 
 /**
@@ -75,9 +78,8 @@ function matchesMemberChain(memberExpr, parts) {
   while (currentNode && idx >= 0) {
     const expectedPart = parts[idx];
 
-    // property should match current expectedPart
     if (currentNode.type === NODE_TYPES.MEMBER_EXPRESSION) {
-      // Ensure property is Identifier and matches
+      // Ensure property is Identifier and matches the expected part
       if (
         currentNode.property.type !== NODE_TYPES.IDENTIFIER ||
         currentNode.property.name !== expectedPart
@@ -85,16 +87,25 @@ function matchesMemberChain(memberExpr, parts) {
         return false;
       }
 
-      // Move to the object of the MemberExpression
+      // Move to the object (which could itself be a MemberExpression, Identifier, or CallExpression)
       currentNode = currentNode.object;
       idx -= 1;
-    } else if (currentNode.type === NODE_TYPES.IDENTIFIER) {
-      // We reached the leftmost Identifier; it should match the first part
+      continue;
+    }
+
+    // If we encounter a CallExpression in the chain (e.g., getService().track),
+    // step into its callee without consuming an expected part.
+    if (currentNode.type === NODE_TYPES.CALL_EXPRESSION) {
+      currentNode = currentNode.callee;
+      continue;
+    }
+
+    if (currentNode.type === NODE_TYPES.IDENTIFIER) {
       return idx === 0 && currentNode.name === expectedPart;
-    } else {
-      // Unexpected node type (e.g., ThisExpression, CallExpression, etc.)
-      return false;
     }
+
+    // Unexpected node type (e.g., ThisExpression, Literal, etc.)
+    return false;
   }
 
   return false;

package/src/analyze/javascript/extractors/event-extractor.js

@@ -257,7 +257,11 @@ function getStringValue(node, constantMap = {}) {
     return node.value;
   }
   if (node.type === NODE_TYPES.MEMBER_EXPRESSION) {
-    return resolveMemberExpressionToString(node, constantMap);
+    const resolved = resolveMemberExpressionToString(node, constantMap);
+    if (resolved) return resolved;
+    // Fallback: return a dotted path for member expressions when we cannot
+    // resolve to a literal (e.g., imported constants like TELEMETRY_EVENTS.X)
+    return memberExpressionToPath(node);
   }
   return null;
 }
@@ -315,6 +319,25 @@ function resolveMemberExpressionToString(node, constantMap) {
   return null;
 }
 
+// Build a dotted path string for a MemberExpression (e.g., OBJ.KEY.SUBKEY)
+function memberExpressionToPath(node) {
+  if (!node || node.type !== NODE_TYPES.MEMBER_EXPRESSION) return null;
+  const parts = [];
+  let current = node;
+  while (current && current.type === NODE_TYPES.MEMBER_EXPRESSION && !current.computed) {
+    if (current.property && current.property.type === NODE_TYPES.IDENTIFIER) {
+      parts.unshift(current.property.name);
+    } else if (current.property && current.property.type === NODE_TYPES.LITERAL) {
+      parts.unshift(String(current.property.value));
+    }
+    current = current.object;
+  }
+  if (current && current.type === NODE_TYPES.IDENTIFIER) {
+    parts.unshift(current.name);
+  }
+  return parts.length ? parts.join('.') : null;
+}
+
 module.exports = {
   extractEventData,
   processEventData

package/src/analyze/javascript/parser.js

@@ -12,6 +12,7 @@ const { PARSER_OPTIONS, NODE_TYPES } = require('./constants');
 const { detectAnalyticsSource } = require('./detectors');
 const { extractEventData, processEventData } = require('./extractors');
 const { findWrappingFunction } = require('./utils/function-finder');
+const { collectImportedConstantStringMap } = require('./utils/import-resolver');
 
 // Extend walker to support JSX
 extend(walk.base);
@@ -82,19 +83,15 @@ function parseFile(filePath) {
 function nodeMatchesCustomFunction(node, fnName) {
   if (!fnName || !node.callee) return false;
 
-  const parts = fnName.split('.');
+  // Support chained calls in function name by stripping trailing parens from each segment
+  const parts = fnName.split('.').map(p => p.replace(/\(\s*\)$/, ''));
 
   // Simple identifier case
   if (parts.length === 1) {
-    return node.callee.type === NODE_TYPES.IDENTIFIER && node.callee.name === fnName;
+    return node.callee.type === NODE_TYPES.IDENTIFIER && node.callee.name === parts[0];
   }
 
-  // Member expression chain case
-  if (node.callee.type !== NODE_TYPES.MEMBER_EXPRESSION) {
-    return false;
-  }
-
-  // Walk the chain from the right-most property to the leftmost object
+  // Allow MemberExpression and CallExpression within the chain (e.g., getService().track)
   let currentNode = node.callee;
   let idx = parts.length - 1;
 
@@ -108,13 +105,23 @@ function nodeMatchesCustomFunction(node, fnName) {
       ) {
         return false;
       }
+      // step to the object; do not decrement idx for call expressions yet
       currentNode = currentNode.object;
       idx -= 1;
-    } else if (currentNode.type === NODE_TYPES.IDENTIFIER) {
+      continue;
+    }
+
+    if (currentNode.type === NODE_TYPES.CALL_EXPRESSION) {
+      // descend into the callee of the call without consuming a part
+      currentNode = currentNode.callee;
+      continue;
+    }
+
+    if (currentNode.type === NODE_TYPES.IDENTIFIER) {
       return idx === 0 && currentNode.name === expected;
-    } else {
-      return false;
     }
+
+    return false;
   }
 
   return false;
@@ -183,8 +190,11 @@ function collectConstantStringMap(ast) {
 function findTrackingEvents(ast, filePath, customConfigs = []) {
   const events = [];
 
-  // Collect constant mappings once per file
-  const constantMap = collectConstantStringMap(ast);
+  // Collect constant mappings once per file (locals + imported)
+  const constantMap = {
+    ...collectConstantStringMap(ast),
+    ...collectImportedConstantStringMap(filePath, ast)
+  };
 
   walk.ancestor(ast, {
     [NODE_TYPES.CALL_EXPRESSION]: (node, ancestors) => {