Contents

# frozen_string_literal: true

#
# Copyright (c) 2019-present, Blue Marble Payroll, LLC
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
#

module Hashematics
  # A Category is an index of objects.  It holds two things:
  # 1. list of top-level objects
  # 2. list of top-level objects cross-referenced by a parent.
  class Category
    attr_reader :id_key, :include_blank, :parent_key

    def initialize(id_key:, include_blank: false, parent_key: nil)
      raise ArgumentError, 'id_key is required' unless id_key

      @default_parent_id  = Id.default
      @id_key             = Key.get(id_key)
      @include_blank      = include_blank || false
      @lookup             = {}
      @parent_key         = Key.get(parent_key)

      freeze
    end

    def records(parent_record = nil)
      parent_id = parent_record&.id(parent_key) || default_parent_id

      get(parent_id).values
    end

    def add(record)
      return self if skip_record?(record)

      set(
        record.id(parent_key),
        record.id(id_key),
        record
      )
    end

    private

    attr_reader :default_parent_id, :lookup

    def skip_record?(record)
      !include_record?(record)
    end

    def include_record?(record)
      include_blank || record.id?(id_key)
    end

    def get(parent_id)
      lookup[parent_id] ||= {}
    end

    def set(parent_id, id, record)
      get(parent_id)[id] = record

      self
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
hashematics-1.1.0	lib/hashematics/category.rb
hashematics-1.0.0	lib/hashematics/category.rb