Contents

# frozen_string_literal: true

require 'delegate'

require 'media_types/version'
require 'media_types/hash'
require 'media_types/object'
require 'media_types/scheme'
require 'media_types/dsl'
require 'media_types/errors'

require 'media_types/views'

module MediaTypes
  def self.set_organisation(mod, organisation)
    @organisation_prefixes ||= {}
    @organisation_prefixes[mod.name] = organisation
  end

  def self.expect_string_keys(mod)
    set_key_expectation(mod, false)
  end

  def self.expect_symbol_keys(mod)
    set_key_expectation(mod, true)
  end

  # Keep track of modules setting their key expectations
  def self.set_key_expectation(mod, expect_symbol_keys)
    @key_expectations ||= {}
    @key_expectations_used ||= {}

    raise KeyExpectationSetError.new(mod: mod) unless @key_expectations[mod.name].nil?
    raise KeyExpectationUsedError.new(mod: mod) if @key_expectations_used[mod.name]

    @key_expectations[mod.name] = expect_symbol_keys
  end

  SYMBOL_KEYS_DEFAULT = true

  def self.get_key_expectation(mod)
    @key_expectations ||= {}
    @key_expectations_used ||= {}

    expect_symbol = find_key_expectation(mod)

    expect_symbol.nil? ? SYMBOL_KEYS_DEFAULT : expect_symbol
  end

  def self.find_key_expectation(mod)
    modules = mod.name.split('::')
    expect_symbol = nil

    while modules.any? && expect_symbol.nil?
      current_module = modules.join('::')
      expect_symbol = @key_expectations[current_module]
      @key_expectations_used[current_module] = true
      modules.pop
    end

    expect_symbol
  end

  def self.get_organisation(mod)
    name = mod.name
    prefixes = @organisation_prefixes.keys.select { |p| name.start_with? p }
    return nil unless prefixes.any?

    best = prefixes.max_by { |p| p.length }

    @organisation_prefixes[best]
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
media_types-2.3.2	lib/media_types.rb
media_types-2.2.0	lib/media_types.rb
media_types-2.1.1	lib/media_types.rb
media_types-2.1.0	lib/media_types.rb