lib/brief/model.rb in brief-1.17.9 vs lib/brief/model.rb in brief-1.17.10

@@ -9,15 +9,16 @@
       end
 
       include Virtus.model(finalize: false)
       include Initializers
       include AccessorMethods
+      include LoggerMethods
       include Persistence
       include Serializers
       include Reports
 
-      class_attribute :models, :after_initialization_hooks
+      class_attribute :models, :after_initialization_hooks, :logger
 
       self.models = Array(models).to_set
 
       class << self
         include Enumerable
@@ -25,9 +26,23 @@
 
       attribute :path, Pathname
       attribute :document, Brief::Document
 
       Brief::Model.classes << self
+    end
+
+    module LoggerMethods
+      def log message
+        (!briefcase.nil? && briefcase.log(message))
+      end
+
+      def debug message
+        (!briefcase.nil? && briefcase.debug(message))
+      end
+
+      def puts message
+        (!briefcase.nil? && briefcase.log(message)) || super
+      end
     end
 
     module AccessorMethods
       def title
         document_title