prism_ext 0.1.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d09f437a427a12d11ba67c28be50da7372493ad3a177938815c5b5abf065b5cf
-  data.tar.gz: c2f18817e9862e578b3bbda235a7040d7b8526654888ae8f200a90fac00db0c3
+  metadata.gz: 5579e7f3f46ecbe569deead151875c5a0b4c55ca6b1694c281c206b63c02a806
+  data.tar.gz: c8ea4497448ce416a62261bd0fa217a8080fbdfe07ca4cfd6cf6524d288e1789
 SHA512:
-  metadata.gz: 6174bbad0ec343872a8f15d0953c09f8ab642476a8ed6c68161969df4aed8c786cebd6559ba6ae2b2e9fbf0bd897c0ee2a89e55e4f7bda6c952e85b5c85c9e8b
-  data.tar.gz: f49dde3337f979b258633a07fa1825d286e029e474d9fd560203d2e74efd77258a76bcfd90f6fdbc1add4588c1f3e1bcd8430b8e7949bbbd23547d752a9da38e
+  metadata.gz: c6ec0f9615e24af0e70423988e26b2c5dd92028e4b6efeb7199ca4566c5ec6109c4fccaf7e23c2c0b4e329305dc7ad8a8be3a97be2d2e3a456a9036dc9a03e51
+  data.tar.gz: 9365bc64d9f4dcb77e09e6a6cab9199b332268c3357117fad1305e69521f5468745aecde543e8d73453f721b32c9539614cbbf8755c6cc0be392ff88c0c6c051

data/CHANGELOG.md

@@ -1,5 +1,13 @@
 # CHANGELOG
 
+## 0.2.1 (2024-02-11)
+
+* Check both `HashNode` and `KeywordHashNode`.
+
+## 0.2.0 (2024-02-10)
+
+* Set `parent_node` to `Prism::Node`.
+
 ## 0.1.0 (2024-02-10)
 
 * Add some helper methods to `Prism::Node`.

data/Gemfile.lock

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

data/README.md

@@ -1,6 +1,8 @@
 # PrismExt
 
-It adds some helpers to prism node.
+It adds `parent_node` to the `Prism::Node`.
+
+It also adds some helpers to prism node.
 
 ```ruby
 # node is a HashNode

data/lib/prism_ext/version.rb

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

data/lib/prism_ext.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require_relative "prism_ext/version"
+
 require 'prism'
 
 module PrismExt
@@ -9,7 +10,37 @@ 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
@@ -73,7 +104,7 @@ module Prism
 
     # Respond key value and source for hash node
     def method_missing(method_name, *args, &block)
-      return super unless is_a?(HashNode)
+      return super unless respond_to_elements?
 
       if method_name.to_s.end_with?('_element')
         key = method_name.to_s[0..-9]
@@ -90,7 +121,7 @@ module Prism
     end
 
     def respond_to_missing?(method_name, *args)
-      return super unless is_a?(HashNode)
+      return super unless respond_to_elements?
 
       if method_name.to_s.end_with?('_element')
         key = method_name.to_s[0..-9]
@@ -105,5 +136,11 @@ module Prism
 
       super
     end
+
+    private
+
+    def respond_to_elements?
+      is_a?(HashNode) || is_a?(KeywordHashNode)
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: prism_ext
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Richard Huang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-02-10 00:00:00.000000000 Z
+date: 2024-02-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prism