node_mutation 1.16.0 → 1.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 14e76d23c3a0be62554de359db928e7f0e70fe75b6c13ecb0655add5af735333
-  data.tar.gz: 1777be0d1e8c4044fa175dcc8f9f9e3431283fdf039392cba9f8fe4fd656c676
+  metadata.gz: 7df63f45429db131b4023d42d1fcab4fdac401358376f0d2ed16722627484bf0
+  data.tar.gz: a9fdadb2c0dd0d82cd709040b2c154b4e2bd75b95ca71082e00e5768e15f476a
 SHA512:
-  metadata.gz: ef64c7156966a2d80c4bf823e0c3330d8934cf52306a9f179bac7105b52c82e78bb9b0d958510165a541fd1fe550b3eec7c7e78eaf3aa7f3207a5b0df1f9e4e3
-  data.tar.gz: 969399df03dc8e3f5c15e55f37a3664bf0945a41e5ff244542319b8a2c20d6afff54dc3aee72b6b4e6c90c6ce84428b6008d14d3dc81e343c68296e78b387218
+  metadata.gz: 7e6194b0d8d80e3b8e553e06c287663a4595bd259f91dbdc9c469c83f4fbabe0781b1ef959fb57341e4d5499993444f21c7fd09283b22f67908495c770446bb2
+  data.tar.gz: 5b99e2f025960eb624f9b82a40a37a1befc63e4787a8cc3a3cd2067e378e9d4b9dfe5f71f7cee2ec922a7171658704a29e0ff799a16389c62b5f27307e818327

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # NodeMutation
 
+## 1.17.0 (2023-05-15)
+
+* Add `SyntaxTreeAdapter`
+
 ## 1.16.0 (2023-05-10)
 
 * Support {key}_value for a hash node

data/Gemfile

@@ -11,5 +11,6 @@ gem "rspec", "~> 3.0"
 
 gem "parser"
 gem "parser_node_ext"
+gem "syntax_tree_ext"
 gem "guard"
 gem "guard-rspec"

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    node_mutation (1.16.0)
+    node_mutation (1.17.0)
 
 GEM
   remote: https://rubygems.org/
@@ -34,10 +34,11 @@ GEM
     notiffany (0.1.3)
       nenv (~> 0.1)
       shellany (~> 0.0)
-    parser (3.2.1.0)
+    parser (3.2.2.1)
       ast (~> 2.4.1)
-    parser_node_ext (1.0.0)
+    parser_node_ext (1.1.0)
       parser
+    prettier_print (1.2.1)
     pry (0.14.1)
       coderay (~> 1.1)
       method_source (~> 1.0)
@@ -59,6 +60,10 @@ GEM
       rspec-support (~> 3.11.0)
     rspec-support (3.11.0)
     shellany (0.0.1)
+    syntax_tree (6.1.1)
+      prettier_print (>= 1.2.0)
+    syntax_tree_ext (0.3.0)
+      syntax_tree
     thor (1.2.1)
 
 PLATFORMS
@@ -74,6 +79,7 @@ DEPENDENCIES
   parser_node_ext
   rake (~> 13.0)
   rspec (~> 3.0)
+  syntax_tree_ext
 
 BUNDLED WITH
    2.3.7

data/README.md

@@ -74,7 +74,7 @@ we define an [Adapter](https://github.com/xinminlabs/node-mutation-ruby/blob/mai
 if you implement the Adapter interface, you can set it as NodeMutation's adapter.
 
 ```ruby
-NodeMutation.configure(adapter: ParserAdapter.new) // default is ParserAdapter
+NodeMutation.configure(adapter: SyntaxTreeAdapter.new) // default is ParserAdapter
 ```
 
 ### strategy

data/lib/node_mutation/{parser_adapter.rb → adapter/parser.rb}

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-INDEX_REGEXP = /\A-?\d+\z/
-
 class NodeMutation::ParserAdapter < NodeMutation::Adapter
   def get_source(node)
     if node.is_a?(Array)
@@ -111,24 +109,6 @@ class NodeMutation::ParserAdapter < NodeMutation::Adapter
         NodeMutation::Struct::Range.new(node.loc.begin.begin_pos, node.loc.end.end_pos)
       end
     else
-      if node.type == :hash && child_name.to_s.end_with?('_pair')
-        pair_node = node.pairs.find { |pair| pair.key.to_value.to_s == child_name.to_s[0..-6] }
-        raise NodeMutation::MethodNotSupported,
-              "#{direct_child_name} is not supported for #{get_source(node)}" unless pair_node
-        return child_node_range(pair, nested_child_name) if nested_child_name
-
-        return NodeMutation::Struct::Range.new(pair_node.loc.expression.begin_pos, pair_node.loc.expression.end_pos)
-      end
-
-      if node.type == :hash && child_name.to_s.end_with?('_value')
-        pair_node = node.pairs.find { |pair| pair.key.to_value.to_s == child_name.to_s[0..-7] }
-        raise NodeMutation::MethodNotSupported,
-              "#{direct_child_name} is not supported for #{get_source(node)}" unless pair_node
-        return child_node_range(pair.value, nested_child_name) if nested_child_name
-
-        return NodeMutation::Struct::Range.new(pair_node.value.loc.expression.begin_pos, pair_node.value.loc.expression.end_pos)
-      end
-
       raise NodeMutation::MethodNotSupported,
             "#{direct_child_name} is not supported for #{get_source(node)}" unless node.respond_to?(direct_child_name)
 
@@ -206,24 +186,6 @@ class NodeMutation::ParserAdapter < NodeMutation::Adapter
       return child_node
     end
 
-    if node.is_a?(Parser::AST::Node) && node.type == :hash && direct_child_name.end_with?('_pair')
-      pair_node = node.pairs.find { |pair| pair.key.to_value.to_s == direct_child_name[0..-6] }
-      raise NodeMutation::MethodNotSupported,
-            "#{direct_child_name} is not supported for #{get_source(node)}" unless pair_node
-      return child_node_by_name(pair_node, nested_child_name) if nested_child_name
-
-      return pair_node
-    end
-
-    if node.is_a?(Parser::AST::Node) && node.type == :hash && direct_child_name.end_with?('_value')
-      pair_node = node.pairs.find { |pair| pair.key.to_value.to_s == direct_child_name[0..-7] }
-      raise NodeMutation::MethodNotSupported,
-            "#{direct_child_name} is not supported for #{get_source(node)}" unless pair_node
-      return child_node_by_name(pair_node.value, nested_child_name) if nested_child_name
-
-      return pair_node.value
-    end
-
     if node.respond_to?(direct_child_name)
       child_node = node.send(direct_child_name)
     elsif direct_child_name.include?('(') && direct_child_name.include?(')')

data/lib/node_mutation/adapter/syntax_tree.rb

@@ -0,0 +1,115 @@
+# frozen_string_literal: true
+
+class NodeMutation::SyntaxTreeAdapter < NodeMutation::Adapter
+  def get_source(node)
+    if node.is_a?(Array)
+      return node.first.source[node.first.location.start_char...node.last.location.end_char]
+    end
+
+    node.source[node.location.start_char...node.location.end_char]
+  end
+
+  def rewritten_source(node, code)
+    code.gsub(/{{(.+?)}}/m) do
+      old_code = Regexp.last_match(1)
+      evaluated = child_node_by_name(node, old_code)
+      case evaluated
+      when SyntaxTree::Node
+        get_source(evaluated)
+      when Array
+        if evaluated.size > 0
+          source = get_source(evaluated)
+          lines = source.split "\n"
+          lines_count = lines.length
+          if lines_count > 1 && lines_count == evaluated.size
+            new_code = []
+            lines.each_with_index { |line, index|
+              new_code << (index == 0 ? line : line[evaluated.first.indent - 2..-1])
+            }
+            new_code.join("\n")
+          else
+            source
+          end
+        end
+      when String, Symbol, Integer, Float
+        evaluated
+      when NilClass
+        ''
+      else
+        raise "can not parse \"#{code}\""
+      end
+    end
+  end
+
+  def child_node_range(node, child_name)
+    child_node = child_node_by_name(node, child_name)
+    return nil if child_node.nil?
+
+    if child_node.is_a?(Array)
+      return NodeMutation::Struct::Range.new(child_node.first.location.start_char, child_node.last.location.end_char)
+    end
+
+    return NodeMutation::Struct::Range.new(child_node.location.start_char, child_node.location.end_char)
+  end
+
+  def get_start(node, child_name = nil)
+    node = child_node_by_name(node, child_name) if child_name
+    node.location.start_char
+  end
+
+  def get_end(node, child_name = nil)
+    node = child_node_by_name(node, child_name) if child_name
+    node.location.end_char
+  end
+
+  def get_start_loc(node, child_name = nil)
+    node = child_node_by_name(node, child_name) if child_name
+    NodeMutation::Struct::Location.new(node.location.start_line, node.location.start_column)
+  end
+
+  def get_end_loc(node, child_name = nil)
+    node = child_node_by_name(node, child_name) if child_name
+    NodeMutation::Struct::Location.new(node.location.end_line, node.location.end_column)
+  end
+
+  def get_indent(node)
+    node.location.start_column
+  end
+
+  private
+
+  def child_node_by_name(node, child_name)
+    direct_child_name, nested_child_name = child_name.to_s.split('.', 2)
+
+    if node.is_a?(Array)
+      if direct_child_name =~ INDEX_REGEXP
+        child_node = node[direct_child_name.to_i]
+        raise NodeMutation::MethodNotSupported,
+              "#{direct_child_name} is not supported for #{get_source(node)}" unless child_node
+        return child_node_by_name(child_node, nested_child_name) if nested_child_name
+
+        return child_node
+      end
+
+      raise NodeMutation::MethodNotSupported,
+            "#{direct_child_name} is not supported for #{get_source(node)}" unless node.respond_to?(direct_child_name)
+
+      child_node = node.send(direct_child_name)
+      return child_node_by_name(child_node, nested_child_name) if nested_child_name
+
+      return child_node
+    end
+
+    if node.respond_to?(direct_child_name)
+      child_node = node.send(direct_child_name)
+    elsif direct_child_name.include?('(') && direct_child_name.include?(')')
+      child_node = node.instance_eval(direct_child_name)
+    else
+      raise NodeMutation::MethodNotSupported, "#{direct_child_name} is not supported for #{get_source(node)}"
+    end
+
+    return child_node_by_name(child_node, nested_child_name) if nested_child_name
+
+    child_node
+  end
+end

data/lib/node_mutation/adapter.rb

@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 class NodeMutation::Adapter
+  INDEX_REGEXP = /\A-?\d+\z/
+
   # Get source code of the ast node
   # @param node [Node] ast node
   # @return [String] source code

data/lib/node_mutation/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class NodeMutation
-  VERSION = "1.16.0"
+  VERSION = "1.17.0"
 end

data/lib/node_mutation.rb

@@ -7,7 +7,8 @@ class NodeMutation
   class ConflictActionError < StandardError; end
 
   autoload :Adapter, "node_mutation/adapter"
-  autoload :ParserAdapter, "node_mutation/parser_adapter"
+  autoload :ParserAdapter, "node_mutation/adapter/parser"
+  autoload :SyntaxTreeAdapter, "node_mutation/adapter/syntax_tree"
   autoload :Action, 'node_mutation/action'
   autoload :AppendAction, 'node_mutation/action/append_action'
   autoload :DeleteAction, 'node_mutation/action/delete_action'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: node_mutation
 version: !ruby/object:Gem::Version
-  version: 1.16.0
+  version: 1.17.0
 platform: ruby
 authors:
 - Richard Huang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-10 00:00:00.000000000 Z
+date: 2023-05-15 00:00:00.000000000 Z
 dependencies: []
 description: ast node mutation apis
 email:
@@ -37,7 +37,8 @@ files:
 - lib/node_mutation/action/replace_action.rb
 - lib/node_mutation/action/replace_with_action.rb
 - lib/node_mutation/adapter.rb
-- lib/node_mutation/parser_adapter.rb
+- lib/node_mutation/adapter/parser.rb
+- lib/node_mutation/adapter/syntax_tree.rb
 - lib/node_mutation/result.rb
 - lib/node_mutation/strategy.rb
 - lib/node_mutation/struct.rb