pyroscope 0.3.1-arm64-darwin → 0.3.2-arm64-darwin

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b7eb0f69ff31b3fc5a6ac437cb4c4190da57cdfea4e81e87956259fcc2f19de
-  data.tar.gz: 9eab2b1c0dd80f29ccf88d9a6e52c37093b0effd70d54eb6fb718546f375453e
+  metadata.gz: 3b416a6d4822ae46d7b09e82268653941e78c327d7f2d8bd6acf10d950cc5539
+  data.tar.gz: 81c775c9dc50ba60d0b05a36e3f2ba2f58e4cb911056da3839c56376e2ddd217
 SHA512:
-  metadata.gz: 95f3c03936e81ffb9b1ee5fe3632c01e822548305d691db28ba721411e4ccba3964197aeee71b7d17ed6e771991441da03f50015108786ad0e05b5dd6d961905
-  data.tar.gz: 0d411ee574f66eb488bc05f905ec52142061374e2bcdde83d028cb120b4f58167444b88dfa24b8fb5f343f1263ad12771b794f0da73f79a636d8f97ed3df4b3d
+  metadata.gz: 0a2e0b59a43d73e4840744ca285138364fe241e70ae8c2289ed2f629bfb89e15641e5e714d1101ef7d7eac5c0295c3d65ff8f8b598d99fe55a79eeea6eca9364
+  data.tar.gz: ad2f447c129dbe4cab53b586a27313fdbb33e28a285686913bae4bdd72296b78aa2c6d8977c48b7068b3fb241088564ee3343306cc65794644cfab545b91b9fd

data/ext/rbspy/src/lib.rs

@@ -1,11 +1,80 @@
 use ffikit::Signal;
-use pyroscope::backend::Tag;
+use pyroscope::backend::{Report, StackFrame, Tag};
 use pyroscope::PyroscopeAgent;
 use pyroscope_rbspy::{rbspy_backend, RbspyConfig};
 use std::collections::hash_map::DefaultHasher;
 use std::ffi::CStr;
 use std::hash::Hasher;
 use std::os::raw::c_char;
+use std::env;
+
+pub fn transform_report(report: Report) -> Report {
+    let cwd = env::current_dir().unwrap();
+    let cwd = cwd.to_str().unwrap_or("");
+
+    let data = report
+        .data
+        .iter()
+        .map(|(stacktrace, count)| {
+            let new_frames = stacktrace
+                .frames
+                .iter()
+                .map(|frame| {
+                    let frame = frame.to_owned();
+                    let mut s = frame.filename.unwrap();
+                    match s.find(cwd) {
+                        Some(i) => {
+                            s = s[(i+cwd.len()+1)..].to_string();
+                        }
+                        None => {
+                            match s.find("/gems/") {
+                                Some(i) => {
+                                    s = s[(i+1)..].to_string();
+                                }
+                                None => {
+                                    match s.find("/ruby/") {
+                                        Some(i) => {
+                                            s = s[(i+6)..].to_string();
+                                            match s.find("/") {
+                                                Some(i) => {
+                                                    s = s[(i+1)..].to_string();
+                                                }
+                                                None => {
+                                                }
+                                            }
+                                        }
+                                        None => {
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+
+                    // something
+                    StackFrame::new(
+                        frame.module,
+                        frame.name,
+                        Some(s.to_string()),
+                        frame.relative_path,
+                        frame.absolute_path,
+                        frame.line,
+                    )
+                })
+                .collect();
+
+            let mut mystack = stacktrace.to_owned();
+
+            mystack.frames = new_frames;
+
+            (mystack, count.to_owned())
+        })
+        .collect();
+
+    let new_report = Report::new(data).metadata(report.metadata.clone());
+
+    new_report
+}
 
 #[no_mangle]
 pub extern "C" fn initialize_agent(
@@ -52,6 +121,7 @@ pub extern "C" fn initialize_agent(
 
     let mut agent_builder = PyroscopeAgent::builder(server_address, application_name)
         .backend(rbspy)
+        .func(transform_report)
         .tags(tags);
 
     if auth_token != "" {

data/lib/pyroscope/version.rb

@@ -1,3 +1,3 @@
 module Pyroscope
-  VERSION = '0.3.1'.freeze
+  VERSION = '0.3.2'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pyroscope
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: arm64-darwin
 authors:
 - Pyroscope Team
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-07-19 00:00:00.000000000 Z
+date: 2022-07-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi