lib/librarian/helpers/debug.rb in librarian-0.0.9 vs lib/librarian/helpers/debug.rb in librarian-0.0.10

@@ -6,27 +6,27 @@
 
       include Support::AbstractMethod
 
       LIBRARIAN_PATH = Pathname.new('../../../../').expand_path(__FILE__)
 
-      abstract_method :root_module
+      abstract_method :environment
 
     private
 
       def relative_path_to(path)
-        root_module.project_relative_path_to(path)
+        environment.project_relative_path_to(path)
       end
 
       def debug
-        if root_module.ui
-          if root_module.ui.respond_to? :debug_line_numbers and root_module.ui.debug_line_numbers
+        if ui = environment.ui
+          if ui.respond_to? :debug_line_numbers and ui.debug_line_numbers
             loc = caller.find{|l| !(l =~ /in `debug'$/)}
             if loc =~ /^(.+):(\d+):in `(.+)'$/
               loc = "#{Pathname.new($1).relative_path_from(LIBRARIAN_PATH)}:#{$2}:in `#{$3}'"
             end
-            root_module.ui.debug { "[Librarian] #{yield} [#{loc}]" }
+            ui.debug { "[Librarian] #{yield} [#{loc}]" }
           else
-            root_module.ui.debug { "[Librarian] #{yield}" }
+            ui.debug { "[Librarian] #{yield}" }
           end
         end
       end
 
     end