maccro 0.1.0

data/lib/maccro/dsl/value.rb

@@ -0,0 +1,85 @@
+require_relative "node"
+require_relative "literal"
+
+module Maccro
+  module DSL
+    class LocalVariable < Node
+      # LVAR: local variable in method local scope (method or local arguments)
+      # DVAR: dynamically defined local variable (defined&initialized in local scope)
+      SUB_NODETYPE_LIST = [:LVAR, :DVAR].freeze
+
+      def self.match?(node)
+        SUB_NODETYPE_LIST.include?(node.type)
+      end
+    end
+
+    class GlobalVariable < Node
+      def type; :GVAR; end
+      def self.match?(node); node.type == :GVAR; end
+    end
+
+    class InstanceVariable < Node
+      def type; :IVAR; end
+      def self.match?(node); node.type == :IVAR; end
+    end
+
+    class ClassVariable < Node
+      def type; :CVAR; end
+      def self.match?(node); node.type == :CVAR; end
+    end
+    
+    class Variable < Node
+      SUB_TYPES = [LocalVariable, GlobalVariable, InstanceVariable, ClassVariable].freeze
+
+      def type; :MACCRO_VARIABLE; end
+
+      def self.match?(node)
+        SUB_TYPES.any?{|s| s.match?(node) }
+      end
+    end        
+
+    class NthRefVariable < Node
+      # $1, $2 (regexp group capture reference)
+      def type; :NTH_REF; end
+      def self.match?(node); node.type == :NTH_REF; end
+    end
+
+    class BackRefVariable < Node
+      # $&, $`, ...
+      def type; :BACK_REF; end
+      def self.match?(node); node.type == :BACK_REF; end
+    end
+
+    class SpecialVariable < Node
+      SUB_TYPES = [NthRefVariable, BackRefVariable].freeze
+
+      def type; :MACCRO_SPECIAL_VARIABLE; end
+
+      def self.match?(node)
+        SUB_TYPES.any?{|s| s.match?(node) }
+      end
+    end
+
+    class Constant < Node
+      def type; :CONST; end
+      def self.match?(node); node.type == :CONST; end
+    end
+
+    class Self < Node
+      def type; :SELF; end
+      def self.match?(node); node.type == :SELF; end
+    end
+
+    class Value < Node
+      SUB_TYPES = [
+        Variable, SpecialVariable, Constant,
+        Literal, String, RegexpCompiledOnce, DRegexp, DSymbol,
+        Self, NilValue, TrueValue, FalseValue, Lambda, Array, Hash,
+      ].freeze
+
+      def self.match?(node)
+        SUB_TYPES.any?{|s| s.match?(node) }
+      end
+    end
+  end
+end

data/lib/maccro/kernel_ext.rb

@@ -0,0 +1,15 @@
+require_relative "./code_util"
+
+module Kernel
+  def dirty_require(feature)
+    Maccro::CodeUtil.suppress_warning do
+      require feature
+    end
+  end
+
+  def dirty_load(file, priv = false)
+    Maccro::CodeUtil.suppress_warning do
+      load(file, priv)
+    end
+  end
+end

data/lib/maccro/matched.rb

@@ -0,0 +1,34 @@
+require_relative 'code_util'
+
+module Maccro
+  Match = Struct.new(:rule, :placeholders, :range, keyword_init: true)
+  # placeholders: name => code_range (in method)
+
+  class Matched
+    attr_reader :matches
+
+    def initialize(matches)
+      @matches = matches.sort{|a, b| a.range <=> b.range }
+    end
+
+    def self.get_replace_pairs(entire_source, placeholders) # name => code_ranges
+      replace_pairs = {}
+      placeholders.each_pair do |name, code_range|
+        replace_pairs[name] = CodeUtil.code_range_to_code(entire_source, code_range)
+      end
+      replace_pairs # name => snippets
+    end
+
+    def rewrite(source)
+      # TODO: implement @safe_reference
+      source = source.dup
+      # move tail to head, not to break code positions of unprocessed matches
+      @matches.reverse.each do |match|
+        replace_pairs = self.class.get_replace_pairs(source, match.placeholders)
+        range = CodeUtil.code_range_to_range(source, match.range)
+        source[range] = match.rule.after_code(replace_pairs)
+      end
+      source
+    end
+  end
+end

data/lib/maccro/rewrite_the_world.rb

@@ -0,0 +1,4 @@
+require "maccro/builtin"
+
+Maccro::Builtin.register_all
+Maccro.enable

data/lib/maccro/rule.rb

@@ -0,0 +1,106 @@
+require_relative 'dsl'
+require_relative 'matched'
+require_relative 'code_util'
+
+module Maccro
+  class Rule
+    attr_reader :name, :before, :after, :matcher, :under, :safe_reference
+
+    def initialize(name, before, after, under: nil, safe_reference: false)
+      @name = name
+      @before = before
+      @after = after
+      @under = under
+      @safe_reference = safe_reference
+
+      # TODO: check all placeholder in @after exist in @before
+      # (placeholders in @before are not required to exist in @after, because it may be removed)
+      # TODO: check $TARGET exists in under just once
+
+      # TODO: implement a matcher in @before matches in multi times
+
+      @matcher = DSL.matcher(before)
+      @pruner = under && DSL.matcher(under) || nil
+    end
+
+    def match(ast)
+      matches = []
+
+      if @pruner
+        dig_prune(matches, ast)
+      else
+        dig_match(matches, ast)
+      end
+
+      return nil if matches.empty?
+
+      Matched.new(matches)
+    end
+
+    def dig_prune(matches, ast)
+      if @pruner.match?(ast)
+        placeholders = {}
+        @pruner.capture(ast, placeholders)
+        dig_match(matches, placeholders[:__target__])
+      elsif ast.respond_to?(:children)
+        ast.children.each do |c|
+          dig_prune(matches, c)
+        end
+      elsif ast.respond_to?(:each)
+        ast.each do |i|
+          dig_prune(matches, i)
+        end
+      end
+    end
+
+    def dig_match(matches, ast)
+      if @matcher.match?(ast)
+        placeholders = {}
+        @matcher.capture(ast, placeholders)
+        matches << Match.new(rule: self, placeholders: placeholders, range: ast.to_code_range)
+      elsif ast.respond_to?(:children)
+        ast.children.each do |c|
+          dig_match(matches, c)
+        end
+      elsif ast.respond_to?(:each)
+        ast.each do |i|
+          dig_match(matches, i)
+        end
+      end
+    end
+
+    def self.find_placeholder_code_ranges(ast, name)
+      return [] unless ast.is_a? RubyVM::AbstractSyntaxTree::Node
+
+      if ast.type == :VCALL && ast.children.first.to_s == name
+        return [CodeRange.from_node(ast)]
+      end
+
+      ranges = []
+      ast.children.each do |c|
+        rs = find_placeholder_code_ranges(c, name)
+        ranges.concat(rs) unless rs.empty?
+      end
+      ranges.sort
+    end
+
+    def after_code(replace_pairs) # name => snippet
+      code = @after.dup
+      ast = CodeUtil.parse_to_ast(@after)
+      code_range_to_name = {}
+      replace_pairs.each_key do |name|
+        self.class.find_placeholder_code_ranges(ast, name).each do |r|
+          code_range_to_name[r] = name
+        end
+      end
+      # reverse is not to break code position for unprocessed code ranges
+      code_range_to_name.keys.sort.reverse.each do |code_range|
+        name = code_range_to_name[code_range]
+        snippet = replace_pairs[name]
+        range = CodeUtil.code_range_to_range(@after.dup, code_range)
+        code[range] = snippet
+      end
+      code
+    end
+  end
+end

data/lib/maccro/version.rb

@@ -0,0 +1,3 @@
+module Maccro
+  VERSION = "0.1.0"
+end

data/maccro.gemspec

@@ -0,0 +1,27 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "maccro/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "maccro"
+  spec.version       = Maccro::VERSION
+  spec.authors       = ["TAGOMORI Satoshi"]
+  spec.email         = ["tagomoris@gmail.com"]
+
+  spec.summary       = %q{Pure black magic scroll}
+  spec.description   = %q{Macro processor for Ruby 2.6 or later}
+  spec.homepage      = "https://github.com/tagomoris/maccro"
+  spec.license       = "MIT"
+
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "test-unit"
+end

metadata

@@ -0,0 +1,108 @@
+--- !ruby/object:Gem::Specification
+name: maccro
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- TAGOMORI Satoshi
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-04-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Macro processor for Ruby 2.6 or later
+email:
+- tagomoris@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/maccro.rb
+- lib/maccro/builtin.rb
+- lib/maccro/code_range.rb
+- lib/maccro/code_util.rb
+- lib/maccro/dsl.rb
+- lib/maccro/dsl/assign.rb
+- lib/maccro/dsl/expression.rb
+- lib/maccro/dsl/literal.rb
+- lib/maccro/dsl/node.rb
+- lib/maccro/dsl/value.rb
+- lib/maccro/kernel_ext.rb
+- lib/maccro/matched.rb
+- lib/maccro/rewrite_the_world.rb
+- lib/maccro/rule.rb
+- lib/maccro/version.rb
+- maccro.gemspec
+homepage: https://github.com/tagomoris/maccro
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: Pure black magic scroll
+test_files: []