Sha256: d67eec1dad3b3aca526740c80d4a87d071987cdc9176387829e171769218033f
Contents?: true
Size: 957 Bytes
Versions: 28
Compression:
Stored size: 957 Bytes
Contents
# frozen_string_literal: true
require_relative 'node'
require_relative 'sexp_extensions'
module Reek
module AST
# Maps AST node types to sublasses of ASTNode extended with the relevant
# utility modules.
#
class ASTNodeClassMap
def initialize
@klass_map = {}
end
def klass_for(type)
klass_map[type] ||= Class.new(Node).tap do |klass|
extension = extension_map[type]
# TODO: map node type to constant directly.
klass.send :include, extension if extension
end
end
def extension_map
@extension_map ||=
begin
assoc = SexpExtensions.constants.map do |const|
[
const.to_s.sub(/Node$/, '').downcase.to_sym,
SexpExtensions.const_get(const)
]
end
Hash[assoc]
end
end
private
attr_reader :klass_map
end
end
end
Version data entries
28 entries across 26 versions & 2 rubygems