Contents

# frozen_string_literal: true

require 'set'

module SimpleAnnotation
  module Annotatable
    def self.included(mod)
      mod.instance_variable_set(:@__method_to_annotations, Hash.new { |h, k| h[k] = Set.new })
      mod.instance_variable_set(:@__standby_annotations, [])
      mod.extend ClassMethods
    end

    module ClassMethods
      def annotates(*method_or_annotations, with: nil)
        if with.nil?
          @__standby_annotations.concat(method_or_annotations)
        else
          method_or_annotations.each do |meth|
            @__method_to_annotations[meth.to_sym] << with
          end
        end
      end

      def annotated?(meth, with: nil)
        return @__method_to_annotations.include?(meth.to_sym) if with.nil?

        @__method_to_annotations[meth.to_sym].include? with
      end

      def annotations(meth)
        @__method_to_annotations[meth.to_sym].to_a
      end

      private

      def method_added(meth)
        super

        @__standby_annotations.each do |annotation|
          annotates meth, with: annotation
        end
        @__standby_annotations.clear
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
simple_annotation-0.1.1	lib/simple_annotation/annotatable.rb
simple_annotation-0.1.0	lib/simple_annotation/annotatable.rb