prism_ext 0.2.2 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c5fbb378a9c09790397fcdda6460841dda3ee4a743f3fae4d19e6309b2239539
-  data.tar.gz: d52081d91e7b7f133f3fd5241e14168765d0fbbe8cfeafb686dc79adbc285a4a
+  metadata.gz: 9dc2a569a5d46086c7979bf6dd2b6af03b766ba78a7d3909873e568bcf22baa9
+  data.tar.gz: 40a740a9bf6b7f3bd0920a5fc1af08cd27ff0ccaf5e88e172ef2757a690d5f19
 SHA512:
-  metadata.gz: 6c9d6e3142e5ffbabe9c6ef8432e1ce58d1ae779ff8af732ed734cbd46e367b4816e7c5f816dc03cbc504e41b3299d9bad977f8a1127728bc7b34086cc4bcb21
-  data.tar.gz: cc2f194f4e933b1110ddd6f6795c02cb5b50515ec106cc61b77d0051afa077f6a49f7794c3d827a9078e2af50892742a701a9910313ade8ae20a0701cd29df7e
+  metadata.gz: 3a8aa09f730f28d6faf50be52288d3c85f16ab3e724befc09221fd5ff89185c6b086c069324643a46b93e9650a503411433e994c336a33885e59df5e54bd9396
+  data.tar.gz: 27920f2028fa75a70c74a2240d9794c9346d960e22af211e83b76c1cb7ee1ee840c2011e6d0852dbb4af620a9015b9c239218fc2a9f611873ca2a41e7398c7e7

data/CHANGELOG.md

@@ -1,5 +1,14 @@
 # CHANGELOG
 
+## 0.3.0 (2024-04-07)
+
+* Abstract `prism_ext/parent_node_ext`
+* Inject hash helper methods only to `HashNode` and `KeywordHashNode`
+
+## 0.2.3 (2024-02-17)
+
+* Remove `Prism::Node#source` hack
+
 ## 0.2.2 (2024-02-11)
 
 * Reuse `respond_to_elements?`

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    prism_ext (0.2.2)
+    prism_ext (0.3.0)
       prism
 
 GEM

data/README.md

@@ -1,15 +1,9 @@
 # PrismExt
 
-It adds `parent_node` to the `Prism::Node`.
+[![Build Status](https://github.com/synvert-hq/prism_ext/actions/workflows/main.yml/badge.svg)](https://github.com/synvert-hq/prism_ext/actions/workflows/main.yml)
+[![Gem Version](https://img.shields.io/gem/v/prism_ext.svg)](https://rubygems.org/gems/prism_ext)
 
-It also adds some helpers to prism node.
-
-```ruby
-# node is a HashNode
-node.foo_element # get the element node of hash foo key
-node.foo_value # get the value of of the hash foo key
-node.foo_source # get the source of the value node of the hash foo key
-```
+It adds some helpers to prism node.
 
 ## Installation
 
@@ -26,6 +20,17 @@ If bundler is not being used to manage dependencies, install the gem by executin
 ```ruby
 require 'prism'
 require 'prism_ext'
+
+# node is a HashNode or KeywordHashNode
+node.foo_element # get the element node of hash foo key
+node.foo_value # get the value of of the hash foo key
+node.foo_source # get the source of the value node of the hash foo key
+node.keys # get key nodes of the hash node
+node.values # get value nodes of the hash node
+
+# all nodes
+node.to_value # get the value of the node, like `true`, `false`, `nil`, `1`, `"foo"`
+node.to_source # get the source code of the node
 ```
 
 ## Development
@@ -36,4 +41,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/xinminlabs/prism_ext.
+Bug reports and pull requests are welcome on GitHub at https://github.com/synvert-hq/prism_ext.

data/lib/prism_ext/parent_node_ext.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Prism
+  class << self
+    alias_method :original_parse, :parse
+
+    def parse(source)
+      result = original_parse(source)
+      result.value.set_parent_node
+      result
+    end
+  end
+
+  class Node
+    attr_accessor :parent_node
+
+    def set_parent_node
+      self.compact_child_nodes.each do |child_node|
+        if child_node.is_a?(Array)
+          child_node.each do |child_child_node|
+            child_child_node.parent_node = self
+            child_child_node.set_parent_node
+          end
+        else
+          child_node.parent_node = self
+          child_node.set_parent_node
+        end
+      end
+    end
+  end
+end

data/lib/prism_ext/version.rb

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

data/lib/prism_ext.rb

@@ -10,102 +10,25 @@ module PrismExt
 end
 
 module Prism
-  class << self
-    alias_method :original_parse, :parse
-
-    def parse(source)
-      result = original_parse(source)
-      result.value.set_parent_node
-      result
-    end
-  end
-
-  class Node
-    attr_accessor :parent_node
-
-    def set_parent_node
-      self.child_nodes.each do |child_node|
-        if child_node.is_a?(Array)
-          child_node.each do |child_child_node|
-            next unless child_child_node.is_a?(Node)
-
-            child_child_node.parent_node = self
-            child_child_node.set_parent_node
-          end
-        end
-
-        next unless child_node.is_a?(Node)
-
-        child_node.parent_node = self
-        child_node.set_parent_node
-      end
-    end
-
-    def source
-      location.instance_variable_get(:@source).source
-    end
-
+  module HashNodeExt
     def keys
-      if respond_to_elements?
-        elements.map(&:key)
-      else
-        raise MethodNotSupported, "keys is not supported for #{self}"
-      end
+      elements.map(&:key)
     end
 
     def values
-      if respond_to_elements?
-        elements.map(&:value)
-      else
-        raise MethodNotSupported, "values is not supported for #{self}"
-      end
+      elements.map(&:value)
     end
 
     def hash_element(key)
-      if respond_to_elements?
-        elements.find { |element_node| element_node.key.to_value == key }
-      else
-        raise MethodNotSupported, "hash_pair is not supported for #{self}"
-      end
+      elements.find { |element_node| element_node.key.to_value == key }
     end
 
     def hash_value(key)
-      if respond_to_elements?
-        elements.find { |element_node| element_node.key.to_value == key }&.value
-      else
-        raise MethodNotSupported, "hash_value is not supported for #{self}"
-      end
+      elements.find { |element_node| element_node.key.to_value == key }&.value
     end
 
-    def to_value
-      case self
-      when SymbolNode
-        value.to_sym
-      when StringNode
-        content
-      when FloatNode
-        value
-      when IntegerNode
-        value
-      when TrueNode
-        true
-      when FalseNode
-        false
-      when NilNode
-        nil
-      when ArrayNode
-        elements.map { |element| element.to_value }
-      else
-        self
-      end
-    end
-
-    alias :to_source :slice
-
     # Respond key value and source for hash node
     def method_missing(method_name, *args, &block)
-      return super unless respond_to_elements?
-
       if method_name.to_s.end_with?('_element')
         key = method_name.to_s[0..-9]
         return elements.find { |element| element.key.to_value.to_s == key }
@@ -121,8 +44,6 @@ module Prism
     end
 
     def respond_to_missing?(method_name, *args)
-      return super unless respond_to_elements?
-
       if method_name.to_s.end_with?('_element')
         key = method_name.to_s[0..-9]
         return !!elements.find { |element| element.key.to_value.to_s == key }
@@ -136,11 +57,40 @@ module Prism
 
       super
     end
+  end
 
-    private
+  class HashNode
+    include HashNodeExt
+  end
 
-    def respond_to_elements?
-      is_a?(HashNode) || is_a?(KeywordHashNode)
+  class KeywordHashNode
+    include HashNodeExt
+  end
+
+  class Node
+    def to_value
+      case self
+      when SymbolNode
+        value.to_sym
+      when StringNode
+        content
+      when FloatNode
+        value
+      when IntegerNode
+        value
+      when TrueNode
+        true
+      when FalseNode
+        false
+      when NilNode
+        nil
+      when ArrayNode
+        elements.map { |element| element.to_value }
+      else
+        self
+      end
     end
+
+    alias :to_source :slice
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: prism_ext
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.0
 platform: ruby
 authors:
 - Richard Huang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-11 00:00:00.000000000 Z
+date: 2024-04-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prism
@@ -38,14 +38,15 @@ files:
 - README.md
 - Rakefile
 - lib/prism_ext.rb
+- lib/prism_ext/parent_node_ext.rb
 - lib/prism_ext/version.rb
 - sig/prism_ext.rbs
-homepage: https://github.com/xinminlabs/prism_ext
+homepage: https://github.com/synvert-hq/prism_ext
 licenses: []
 metadata:
-  homepage_uri: https://github.com/xinminlabs/prism_ext
-  source_code_uri: https://github.com/xinminlabs/prism_ext
-  changelog_uri: https://github.com/xinminlabs/prism_ext/CHANGELOG
+  homepage_uri: https://github.com/synvert-hq/prism_ext
+  source_code_uri: https://github.com/synvert-hq/prism_ext
+  changelog_uri: https://github.com/synvert-hq/prism_ext/CHANGELOG
 post_install_message:
 rdoc_options: []
 require_paths: