lib/brief/model.rb in brief-1.2.0 vs lib/brief/model.rb in brief-1.3.0

@@ -13,12 +13,12 @@
       include AccessorMethods
       include Persistence
 
       class_attribute :models, :after_initialization_hooks, :defined_actions
 
-      self.models = Array(self.models).to_set
-      self.defined_actions = Array(self.defined_actions).to_set
+      self.models = Array(models).to_set
+      self.defined_actions = Array(defined_actions).to_set
 
       class << self
         include Enumerable
       end
 
@@ -66,20 +66,28 @@
 
     def self.for_type(type_alias)
       table[type_alias]
     end
 
+    def self.lookup_class_from_args(args = [])
+      args = Array(args)
+
+      if model_class = for_type(args.first)
+        model_class
+      end
+    end
+
     def self.finalize
       Virtus.finalize
       classes.each(&:finalize)
     end
 
     def ==(other)
-      self.path == other.path
+      path == other.path
     end
 
-    def extract_content(options={})
+    def extract_content(options = {})
       document.extract_content(options)
     end
 
     module ClassMethods
       def has_actions?
@@ -88,11 +96,11 @@
 
       def finalize
         klass = self
 
         klass.name ||= klass.to_s.split('::').last.humanize
-        klass.type_alias ||= klass.name.parameterize.gsub(/-/,'_')
+        klass.type_alias ||= klass.name.parameterize.gsub(/-/, '_')
 
         klass.attribute_set.map(&:name).each do |attr|
           unless klass.method_defined?("find_by_#{ attr }")
             klass.define_singleton_method("find_by_#{ attr }") do |value|
               where(attr => value).first
@@ -103,43 +111,39 @@
         klass.definition.apply_config
 
         Brief::Repository.define_document_finder_methods
       end
 
-      def where(*args, &block)
+      def where(*args, &_block)
         Brief::DocumentMapper::Query.new(self).send(:where, *args)
       end
 
       def each(*args, &block)
-        Array(self.models).send(:each, *args, &block)
+        Array(models).send(:each, *args, &block)
       end
 
       def after_initialize(&block)
         (self.after_initialization_hooks ||= []).push(block)
       end
 
-      def name=(value)
-        @name = value
-      end
+      attr_writer :name
 
       def name
-        @name || to_s.split('::').last.underscore.gsub('_',' ').titlecase
+        @name || to_s.split('::').last.underscore.gsub('_', ' ').titlecase
       end
 
-      def type_alias=(value)
-        @type_alias = value
-      end
+      attr_writer :type_alias
 
       def type_alias
-        @type_alias || name.parameterize.gsub(/-/,'_')
+        @type_alias || name.parameterize.gsub(/-/, '_')
       end
 
       def definition
         @definition ||= Brief::Model::Definition.new(name, type_alias: type_alias, model_class: self)
       end
 
-      def definition=(value)
+      def definition=(_value)
         @definition
       end
 
       def section_mapping(*args)
         definition.send(:section_mapping, *args)
@@ -147,22 +151,47 @@
 
       def section_mappings(*args)
         definition.send(:section_mappings, *args)
       end
 
+      def generate_template_content_from(object, include_frontmatter = true)
+        @erb ||= ERB.new(template_body)
+        content = @erb.result(binding)
+        frontmatter = object.slice(*attribute_names)
+
+        base = ''
+        base += frontmatter.to_hash.to_yaml + "---\n" if include_frontmatter
+        base += content
+
+        base
+      end
+
+      def attribute_names
+        attribute_set.map(&:name)
+      end
+
+      def template_body(*args)
+        res = definition.send(:template_body, *args)
+        res.to_s.length == 0 ? example_body : res.to_s.strip
+      end
+
+      def example_body(*args)
+        definition.send(:example_body, *args).to_s.strip
+      end
+
       def method_missing(meth, *args, &block)
-        if %w(meta content actions helpers).include?(meth.to_s)
+        if %w(meta content template example actions helpers).include?(meth.to_s)
           definition.send(meth, *args, &block)
           finalize
         elsif meth.to_s.match(/^on_(.*)_change$/)
-          create_change_handler($1, *args, &block)
+          create_change_handler(Regexp.last_match[1], *args, &block)
         else
           super
         end
       end
 
-      def create_change_handler(attribute, *args, &block)
+      def create_change_handler(_attribute, *_args, &block)
         block.call(self)
       end
     end
 
     module Initializers
@@ -176,11 +205,11 @@
         Array(self.class.after_initialization_hooks).each do |hook|
           hook.call(self)
         end
       end
 
-      def set_slug_from(column=:name)
-        self.slug = send(column).to_s.downcase.parameterize if self.slug.to_s.length == 0
+      def set_slug_from(column = :name)
+        self.slug = send(column).to_s.downcase.parameterize if slug.to_s.length == 0
       end
     end
   end
 end