From 9d9f7d71de6dc5ea3c3732b8fbdd2a627560ed09 Mon Sep 17 00:00:00 2001
From: Ildar Kayumov <kaayumov@gmail.com>
Date: Sat, 10 Aug 2024 18:25:32 +0200
Subject: [PATCH] feat: allow to use configurable logger

---
 lib/trailblazer/developer.rb     | 10 ++++++++++
 lib/trailblazer/developer/wtf.rb |  2 +-
 test/docs/developer_test.rb      |  6 +++---
 test/trace/wtf_test.rb           | 10 +++++-----
 4 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/lib/trailblazer/developer.rb b/lib/trailblazer/developer.rb
index 61935af..d827e40 100644
--- a/lib/trailblazer/developer.rb
+++ b/lib/trailblazer/developer.rb
@@ -1,8 +1,18 @@
 require "trailblazer/developer/version"
+require "logger"
 
 module Trailblazer
   module Developer
     # Your code goes here...
+    class << self
+      attr_writer :logger
+
+      def logger
+        @logger ||= Logger.new($stdout, formatter: proc { |*, msg| "#{msg}\n" }).tap do |log|
+          log.progname = self.name
+        end
+      end
+    end
   end
 end
 
diff --git a/lib/trailblazer/developer/wtf.rb b/lib/trailblazer/developer/wtf.rb
index 6aaac98..f46a6c4 100644
--- a/lib/trailblazer/developer/wtf.rb
+++ b/lib/trailblazer/developer/wtf.rb
@@ -60,7 +60,7 @@ def invoke(activity, (ctx, flow_options), present_options: {}, **circuit_options
         &local_present_options_block
       )
 
-      puts output
+      Trailblazer::Developer.logger.debug(output)
 
       raise raise_exception if raise_exception
       return signal, [ctx, flow_options], circuit_options, output, returned_args
diff --git a/test/docs/developer_test.rb b/test/docs/developer_test.rb
index 2646a13..e363b7b 100644
--- a/test/docs/developer_test.rb
+++ b/test/docs/developer_test.rb
@@ -71,7 +71,7 @@ class Create < Trailblazer::Operation
 
 
   it "#wtf?" do
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       result = Memo::Operation::Create.wtf?(seq: [], params: {})
     end
 
@@ -95,7 +95,7 @@ class Create < Trailblazer::Operation
 =end
 
   #@ exception
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       assert_raises ArgumentError do
         result = Memo::Operation::Create.wtf?(seq: [], raise_exception: true)
       end
@@ -138,7 +138,7 @@ class Create < Trailblazer::Activity::Railway # Note that this is not an {Operat
   end
 
   it "{#wtf?} with {Activity}" do
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       result = Trailblazer::Developer.wtf?(B::Memo::Operation::Create, [{seq: [], params: {}}, {}])
     end
 
diff --git a/test/trace/wtf_test.rb b/test/trace/wtf_test.rb
index f46ce5b..ef3eea9 100644
--- a/test/trace/wtf_test.rb
+++ b/test/trace/wtf_test.rb
@@ -42,7 +42,7 @@ def <<(value)
     exception = nil
     returned_args = nil
 
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       exception = assert_raises RuntimeError do
         returned_args = Trailblazer::Developer.wtf?(alpha, [{ seq: Raiser.new(raise_in: :c) }])
       end
@@ -66,7 +66,7 @@ def <<(value)
   it "traces until charlie, 3-level and step takes left track" do
     returned_args = nil
 
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       returned_args = Trailblazer::Developer.wtf?(alpha, [{ seq: [], c: false }])
     end
 
@@ -96,7 +96,7 @@ def <<(value)
   end
 
   it "traces alpha and it's subprocesses, for successful execution" do
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       Trailblazer::Developer.wtf?(alpha, [{ seq: [] }])
     end
 
@@ -125,7 +125,7 @@ def <<(value)
 
     signal, ctx, flow_options, circuit_options, output, returned_present_args = nil
 
-    captured_output, _ = capture_io do
+    captured_output, _ = capture_subprocess_io do
       signal, (ctx, flow_options), circuit_options, output, returned_present_args = Trailblazer::Developer.wtf?(alpha, [{seq: []}],
         present_options: {render_method: my_renderer})
     end
@@ -153,7 +153,7 @@ class PresentCreate < Trailblazer::Activity::Railway
   end
 
   it "overrides default color map of entities" do
-    output, _ = capture_io do
+    output, _ = capture_subprocess_io do
       Trailblazer::Developer.wtf?(
         alpha,
         [