Contents

require 'yaml'

# When included in a document, this module will turn it into a static
# model which sources its records from a yaml file.
module Structure
  class Document
    module Static
      class << self
        private

        def included(base)
          base.key(:_id, Integer)
          base.extend(ClassMethods)
        end
      end

      module ClassMethods
        include Enumerable

        # The path for the data file.
        #
        # This file should contain a YAML representation of the records.
        #
        # Overwrite this reader with an opiniated location to dry.
        attr :data_path

        # Returns all records.
        def all
          @records ||= data.map do |record|
            record['_id'] ||= record.delete('id') || increment_id
            new(record)
          end
        end

        # Yields each record to given block.
        #
        # Other enumerators will be made available by the Enumerable
        # module.
        def each(&block)
          all.each { |record| block.call(record) }
        end

        # Finds a record by its ID.
        def find(id)
          detect { |record| record._id == id }
        end

        # Sets the path for the data file.
        def set_data_path(data_path)
          @data_path = data_path
        end

        private

        def data
          YAML.load_file(@data_path)
        end


        def increment_id
          @increment_id = @increment_id.to_i + 1
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
structure-0.16.0	lib/structure/document/static.rb