module ZTK::DSL::Core::Actions

  # @author Zachary Patten <zachary AT jovelabs DOT com>
  # @api private
  module Find

    def self.included(base)
      base.class_eval do
        base.send(:extend, ZTK::DSL::Core::Actions::Find::ClassMethods)
      end
    end

    # @author Zachary Patten <zachary AT jovelabs DOT com>
    module ClassMethods

      def all
        dataset
      end

      def find(*args)
        ids = [args].flatten
        all.select{ |data| ids.include?(data.id) }
      end

      def first(*args)
        find(*args).first
      end

      def count
        all.count
      end

    end

  end
end