Contents

module Brief
  class Document::ContentExtractor
    def initialize(model_type, document)
      @model_type = model_type
      @document = document
    end

    attr_reader :document

    def model_class
      document.model_class
    end

    def content_schema_attributes
      model_class.definition.content_schema.attributes
    end

    def extracted_content_data
      me = self
      content_schema_attributes.keys.reduce({}.to_mash) do |memo, attr|
        val = me.send(attr) rescue nil
        memo[attr] = val if val
        memo
      end
    end

    def respond_to?(meth)
      content_schema_attributes.key?(meth) || super
    end

    def method_missing(meth, *_args, &_block)
      if settings = content_schema_attributes.fetch(meth, nil)
        if settings.args.length == 1 && settings.args.first.is_a?(String)
          selector = settings.args.first
          matches = document.css(selector)

          if matches.length > 1
            selector.match(/first-of-type/) ? matches.first.text : matches.map(&:text)
          else
            matches.first.try(:text)
          end
        elsif settings.args.first.to_s.match(/code/i) && (settings.args.last.serialize rescue nil)
          selector = settings.args.first
          opts = settings.args.last

          matches = document.css(selector)

          val = if matches.length > 1
            selector.match(/first-of-type/) ? matches.first.text : matches.map(&:text)
          else
            matches.first.try(:text)
          end

          if val && opts.serialize == :yaml
            return (YAML.load(val) rescue {}).to_mash
          end

          if val && opts.serialize == :json
            return (JSON.parse(val) rescue {}).to_mash
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
brief-1.6.1	lib/brief/document/content_extractor.rb
brief-1.6.0	lib/brief/document/content_extractor.rb