lib/trace_location/generator/csv.rb in trace_location-0.9.3 vs lib/trace_location/generator/csv.rb in trace_location-0.9.3.1

@@ -7,10 +7,11 @@
     class Csv < Base # :nodoc:
       ATTRIBUTES = %w[id event path lineno caller_path caller_lineno owner_with_name hierarchy].freeze
 
       def initialize(events, return_value, options)
         super
+        @current_dir = ::TraceLocation.config.current_dir
         @dest_dir = options.fetch(:dest_dir) { ::TraceLocation.config.dest_dir }
         @file_path = File.join(@dest_dir, "trace_location-#{Time.now.strftime('%Y%m%d%H%m%s')}.csv")
       end
 
       def generate
@@ -19,21 +20,30 @@
         $stdout.puts "Created at #{file_path}"
       end
 
       private
 
-      attr_reader :events, :return_value, :dest_dir, :file_path
+      attr_reader :events, :return_value, :current_dir, :dest_dir, :file_path
 
       def setup_dir
         FileUtils.mkdir_p(dest_dir)
       end
 
       def create_file
         CSV.open(file_path, 'wb+') do |csv|
           csv << ATTRIBUTES
 
           events.each do |event|
-            csv << ATTRIBUTES.map { |attr| event.public_send(attr) }
+            csv << [
+              event.id,
+              event.event,
+              event.path.to_s.gsub(%r{#{current_dir}/}, ''),
+              event.lineno,
+              event.caller_path.to_s.gsub(%r{#{current_dir}/}, ''),
+              event.caller_lineno,
+              event.owner_with_name,
+              event.hierarchy
+            ]
           end
         end
       end
     end
   end