imp(ruby): add regex support

Author: omarabid

pyroscope_ffi/ruby/ext/rbspy/Cargo.toml

@@ -14,6 +14,7 @@ pyroscope = { path = "../../../../" }
 pyroscope_rbspy = { path = "../../../../pyroscope_backends/pyroscope_rbspy" }
 ffikit = { path = "../../../ffikit" }
 pretty_env_logger = "0.4.0"
+regex = "1"
 
 [patch.crates-io]
 read-process-memory = {git = "https://github.com/omarabid/read-process-memory.git", branch = "0.1.4-fix"}

pyroscope_ffi/ruby/ext/rbspy/src/lib.rs

@@ -10,8 +10,8 @@ use std::os::raw::c_char;
 #[no_mangle]
 pub extern "C" fn initialize_agent(
     application_name: *const c_char, server_address: *const c_char, auth_token: *const c_char,
-    sample_rate: u32, detect_subprocesses: bool, on_cpu: bool, report_pid: bool,
-    report_thread_id: bool, tags: *const c_char,
+    directory_name: *const c_char, sample_rate: u32, detect_subprocesses: bool, on_cpu: bool,
+    report_pid: bool, report_thread_id: bool, tags: *const c_char,
 ) -> bool {
     // Initialize FFIKit
     let recv = ffikit::initialize_ffi().unwrap();
@@ -31,6 +31,11 @@ pub extern "C" fn initialize_agent(
         .unwrap()
         .to_string();
 
+    let directory_name = unsafe { CStr::from_ptr(directory_name) }
+        .to_str()
+        .unwrap()
+        .to_string();
+
     let tags_string = unsafe { CStr::from_ptr(tags) }
         .to_str()
         .unwrap()
@@ -52,6 +57,7 @@ pub extern "C" fn initialize_agent(
 
     let mut agent_builder = PyroscopeAgent::builder(server_address, application_name)
         .backend(rbspy)
+        .regex(regex::Regex::new(&directory_name).unwrap())
         .tags(tags);
 
     if auth_token != "" {

pyroscope_ffi/ruby/lib/pyroscope.rb

@@ -4,7 +4,7 @@ module Pyroscope
   module Rust
     extend FFI::Library
     ffi_lib File.expand_path(File.dirname(__FILE__)) + "/rbspy/rbspy.#{RbConfig::CONFIG["DLEXT"]}"
-    attach_function :initialize_agent, [:string, :string, :string, :int, :bool, :bool, :bool, :bool, :string], :bool
+    attach_function :initialize_agent, [:string, :string, :string, :string, :int, :bool, :bool, :bool, :bool, :string], :bool
     attach_function :add_thread_tag, [:uint64, :string, :string], :bool
     attach_function :remove_thread_tag, [:uint64, :string, :string], :bool
     attach_function :add_global_tag, [:string, :string], :bool
@@ -18,11 +18,12 @@ module Utils
     attach_function :thread_id, [], :uint64
   end
 
-  Config = Struct.new(:application_name, :app_name, :server_address, :auth_token, :sample_rate, :detect_subprocesses, :on_cpu, :report_pid, :report_thread_id, :log_level, :tags) do
+  Config = Struct.new(:application_name, :app_name, :server_address, :auth_token, :directory_name, :sample_rate, :detect_subprocesses, :on_cpu, :report_pid, :report_thread_id, :log_level, :tags) do
     def initialize(*)
       self.application_name = ''
       self.server_address = 'http://localhost:4040'
       self.auth_token = ''
+      self.directory_name = __dir__
       self.sample_rate = 100
       self.detect_subprocesses = false
       self.on_cpu = true
@@ -45,6 +46,7 @@ def configure
         @config.app_name || @config.application_name || "",
         @config.server_address || "",
         @config.auth_token || "",
+        @config.directory_name || __dir__,
         @config.sample_rate || 100,
         @config.detect_subprocesses || false,
         @config.on_cpu || false,