syntax_tree_ext 0.1.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d51e4d0fd788f5234eded21764f3e8e7a85dd27038c9c40f0a7f211a1a84dd60
-  data.tar.gz: fb064aee9423515356daf3e935bbfcac7320e56883d25fad0cc179380f78bcc1
+  metadata.gz: b6521c0c1d5ead0d51f0dce8810f7dcb8e080f4ebceb14a1a0c82a1ea9de5a44
+  data.tar.gz: 384efd9248ed1002836b9b481d1ca727d058120f6cb082eecd9b4190cac609a5
 SHA512:
-  metadata.gz: d68490aa8fa813bd72e892acf2db74627142c60dd2c6f6d5f163593ba999b026fda774c3fa0c6d37c0eaa293ebe7baeab4f18aeb0126d5fefc50df3d42cf7edb
-  data.tar.gz: 69554ffc9196a4c0c758c91771a60b7e39b0cb65dfb6535e4f918ce81e93c15e7d5764188c40d9e6b51d7591067cb262e8d2b6f0102d6d43a50ffe6f5933dfe4
+  metadata.gz: 9fea364be6aa7bb2303d5932b050efdae247dd8cb8f6e84d1906759eb6aef3dd0acd245e7f16c2e50e897d32a90a9fef6f79d29a03482a73c79bdfc2b1e40adb
+  data.tar.gz: 49e7e190eaa3772b5bfb29e053bb366ebb00581ed28de823a8bb2aa1500c960d455519cb12ac64d13c395f31086e840c18341e348f5c07b896fac8f329bcb550

data/CHANGELOG.md

@@ -1,5 +1,15 @@
 # CHANGELOG
 
+## 0.3.0 (2023-05-15)
+
+* Support `xxx_assoc` to get assoc node of the hash node
+
+## 0.2.0 (2023-05-14)
+
+* Rename `Node#parent` to `Node#parent_node`
+* Add `Node#to_value` and `Node#to_source`
+* Add `Node#method_missing` for hash `xxx_value` and `xxx_pair`
+
 ## 0.1.0 (2023-04-20)
 
 * Add `SyntaxTree::Node#siblings`

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    syntax_tree_ext (0.1.0)
+    syntax_tree_ext (0.3.0)
       syntax_tree
 
 GEM

data/README.md

@@ -1,24 +1,32 @@
 # SyntaxTreeExt
 
-TODO: Delete this and the text below, and describe your gem
+It adds `parent_node` and `source` methods to the `ParserTree::Node`.
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/syntax_tree_ext`. To experiment with that code, run `bin/console` for an interactive prompt.
+It also adds some helpers
 
-## Installation
+```ruby
+# node is a HashLiteral node
+node.foo_assoc # get the assoc node of hash foo key
+node.foo_value # get the value node of the hash foo key
+node.foo_source # get the source of the value node of the hash foo key
+```
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+## Installation
 
 Install the gem and add to the application's Gemfile by executing:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ bundle add syntax_tree_ext
 
 If bundler is not being used to manage dependencies, install the gem by executing:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+    $ gem install syntax_tree_ext
 
 ## Usage
 
-TODO: Write usage instructions here
+```ruby
+require 'syntax_tree'
+require 'syntax_tree_ext'
+```
 
 ## Development
 
@@ -28,4 +36,4 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/syntax_tree_ext.
+Bug reports and pull requests are welcome on GitHub at https://github.com/xinminlabs/syntax_tree_ext.

data/lib/syntax_tree_ext/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SyntaxTreeExt
-  VERSION = "0.1.0"
+  VERSION = "0.3.0"
 end

data/lib/syntax_tree_ext.rb

@@ -15,31 +15,98 @@ module SyntaxTree
 
     def parse
       node = original_parse
-      node.set_parent_and_source(source)
+      node.set_parent_node_and_source(source)
       node
     end
   end
 
   class Node
-    attr_accessor :parent, :source
+    attr_accessor :parent_node, :source
 
-    def set_parent_and_source(source)
+    def set_parent_node_and_source(source)
       self.source = source
       child_nodes.each do |child_node|
         next unless child_node.is_a?(Node)
 
-        child_node.parent = self
-        child_node.set_parent_and_source(source)
+        child_node.parent_node = self
+        child_node.set_parent_node_and_source(source)
       end
     end
 
     def siblings
-      index = parent.child_nodes.index(self)
-      parent.child_nodes[index + 1...]
+      index = parent_node.child_nodes.index(self)
+      parent_node.child_nodes[index + 1...]
     end
 
-    def source
-      @source[location.start_char..location.end_char]
+    def to_value
+      case self
+      when SymbolLiteral
+        value.value.to_sym
+      when StringLiteral
+        parts.map(&:to_value).join
+      when FloatLiteral
+        value.to_f
+      when Int
+        value.to_i
+      when Kw
+        value == 'true'
+      when VarRef
+        value.to_value
+      when Label, TStringContent
+        value
+      else
+        self
+      end
+    end
+
+    def to_source
+      source[location.start_char...location.end_char]
+    end
+
+    # Respond key value and source for hash node
+    def method_missing(method_name, *args, &block)
+      return super unless respond_to_assocs?
+
+      if method_name.to_s.end_with?('_assoc')
+        key = method_name.to_s[0..-7]
+        return assocs.find { |assoc| assoc_key_equal?(assoc, key) }
+      elsif method_name.to_s.end_with?('_value')
+        key = method_name.to_s[0..-7]
+        return assocs.find { |assoc| assoc_key_equal?(assoc, key) }&.value
+      elsif method_name.to_s.end_with?('_source')
+        key = method_name.to_s[0..-8]
+        return assocs.find { |assoc| assoc_key_equal?(assoc, key) }&.value&.to_source || ''
+      end
+
+      super
+    end
+
+    def respond_to_missing?(method_name, *args)
+      return super unless respond_to_assocs?
+
+      if method_name.to_s.end_with?('_assoc')
+        key = method_name[0..-7]
+        return !!assocs.find { |assoc| assoc_key_equal?(assoc, key) }
+      elsif method_name.to_s.end_with?('_value')
+        key = method_name[0..-7]
+        return !!assocs.find { |assoc| assoc_key_equal?(assoc, key) }
+      elsif method_name.to_s.end_with?('_source')
+        key = method_name.to_s[0..-8]
+        return !!assocs.find { |assoc| assoc_key_equal?(assoc, key) }
+      end
+
+      super
+    end
+
+    private
+
+    def respond_to_assocs?
+      is_a?(HashLiteral) || is_a?(BareAssocHash)
+    end
+
+    def assoc_key_equal?(assoc, key)
+      assoc_key = assoc.key.to_value.to_s
+      assoc_key.end_with?(':') ? assoc_key == "#{key}:" : assoc_key == key
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: syntax_tree_ext
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Richard Huang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-20 00:00:00.000000000 Z
+date: 2023-05-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: syntax_tree