module LadyJosephine
  class Whitelist
    include Singleton

    def initialize
      @whitelist_classes = []
    end

    def whitelist(class_to_add)
      if class_to_add.instance_of? String
        @whitelist_classes << class_to_add
      else
        @whitelist_classes << class_to_add.to_s
      end
    end

    def is_whitelisted?(class_to_check)
      if class_to_check.instance_of? String
        @whitelist_classes.include? class_to_check.classify
      else
        @whitelist_classes.include? class_to_check.to_s
      end
    end

  end
end