feat: add DISABLE_ANALYTICS check

amirabbas-gh · amirabbas-gh · commit e6824fe53cbb · 2025-07-14T17:27:13.000+03:30
diff --git a/crates/cli/src/commands/publish.rs b/crates/cli/src/commands/publish.rs
@@ -200,17 +200,21 @@ pub async fn handler(args: &Command, telemetry: &PostHogSender) -> Result<()> {
 
     let cli_version = env!("CARGO_PKG_VERSION");
 
-    let _ = telemetry.send_event(
-        BaseEvent {
-            kind: "codemodPublished".to_string(),
-            properties: HashMap::from([
-                ("codemodName".to_string(), manifest.name.clone()),
-                ("version".to_string(), manifest.version.clone()),
-                ("cliVersion".to_string(), cli_version.to_string()),
-            ]),
-        },
-        None,
-    );
+    if std::env::var("DISABLE_ANALYTICS") == Ok("false".to_string())
+        || std::env::var("DISABLE_ANALYTICS").is_err()
+    {
+        let _ = telemetry.send_event(
+            BaseEvent {
+                kind: "codemodPublished".to_string(),
+                properties: HashMap::from([
+                    ("codemodName".to_string(), manifest.name.clone()),
+                    ("version".to_string(), manifest.version.clone()),
+                    ("cliVersion".to_string(), cli_version.to_string()),
+                ]),
+            },
+            None,
+        );
+    }
 
     println!("✅ Package published successfully!");
     println!("📦 {}", format_package_name(&response.package));
diff --git a/crates/cli/src/commands/run.rs b/crates/cli/src/commands/run.rs
@@ -104,18 +104,23 @@ pub async fn handler(engine: &Engine, args: &Command, telemetry: &PostHogSender)
         &base64::engine::general_purpose::URL_SAFE_NO_PAD,
         execution_id,
     );
-    let _ = telemetry.send_event(
-        BaseEvent {
-            kind: "codemodExecuted".to_string(),
-            properties: HashMap::from([
-                ("codemodName".to_string(), args.package.clone()),
-                ("executionId".to_string(), execution_id.clone()),
-                ("fileCount".to_string(), stats.files_modified.to_string()),
-                ("cliVersion".to_string(), cli_version.to_string()),
-            ]),
-        },
-        None,
-    );
+
+    if std::env::var("DISABLE_ANALYTICS") == Ok("false".to_string())
+        || std::env::var("DISABLE_ANALYTICS").is_err()
+    {
+        let _ = telemetry.send_event(
+            BaseEvent {
+                kind: "codemodExecuted".to_string(),
+                properties: HashMap::from([
+                    ("codemodName".to_string(), args.package.clone()),
+                    ("executionId".to_string(), execution_id.clone()),
+                    ("fileCount".to_string(), stats.files_modified.to_string()),
+                    ("cliVersion".to_string(), cli_version.to_string()),
+                ]),
+            },
+            None,
+        );
+    }
 
     Ok(())
 }
