prism_ext 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: d09f437a427a12d11ba67c28be50da7372493ad3a177938815c5b5abf065b5cf
+  data.tar.gz: c2f18817e9862e578b3bbda235a7040d7b8526654888ae8f200a90fac00db0c3
+SHA512:
+  metadata.gz: 6174bbad0ec343872a8f15d0953c09f8ab642476a8ed6c68161969df4aed8c786cebd6559ba6ae2b2e9fbf0bd897c0ee2a89e55e4f7bda6c952e85b5c85c9e8b
+  data.tar.gz: f49dde3337f979b258633a07fa1825d286e029e474d9fd560203d2e74efd77258a76bcfd90f6fdbc1add4588c1f3e1bcd8430b8e7949bbbd23547d752a9da38e

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/CHANGELOG.md

@@ -0,0 +1,5 @@
+# CHANGELOG
+
+## 0.1.0 (2024-02-10)
+
+* Add some helper methods to `Prism::Node`.

data/Gemfile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in prism_ext.gemspec
+gemspec
+
+gem "rake", "~> 13.0"
+
+gem "rspec", "~> 3.0"

data/Gemfile.lock

@@ -0,0 +1,37 @@
+PATH
+  remote: .
+  specs:
+    prism_ext (0.1.0)
+      prism
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    diff-lcs (1.5.0)
+    prism (0.20.0)
+    rake (13.1.0)
+    rspec (3.12.0)
+      rspec-core (~> 3.12.0)
+      rspec-expectations (~> 3.12.0)
+      rspec-mocks (~> 3.12.0)
+    rspec-core (3.12.2)
+      rspec-support (~> 3.12.0)
+    rspec-expectations (3.12.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-mocks (3.12.6)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-support (3.12.1)
+
+PLATFORMS
+  ruby
+  x86_64-darwin-23
+
+DEPENDENCIES
+  prism_ext!
+  rake (~> 13.0)
+  rspec (~> 3.0)
+
+BUNDLED WITH
+   2.5.3

data/README.md

@@ -0,0 +1,37 @@
+# PrismExt
+
+It 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
+```
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add prism_ext
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install prism_ext
+
+## Usage
+
+```ruby
+require 'prism'
+require 'prism_ext'
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
+
+To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/xinminlabs/prism_ext.

data/Rakefile

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task default: :spec

data/lib/prism_ext/version.rb

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

data/lib/prism_ext.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+require_relative "prism_ext/version"
+require 'prism'
+
+module PrismExt
+  class Error < StandardError; end
+  # Your code goes here...
+end
+
+module Prism
+  class Node
+    def source
+      location.instance_variable_get(:@source).source
+    end
+
+    def keys
+      if is_a?(HashNode)
+        elements.map(&:key)
+      else
+        raise MethodNotSupported, "keys is not supported for #{self}"
+      end
+    end
+
+    def values
+      if is_a?(HashNode)
+        elements.map(&:value)
+      else
+        raise MethodNotSupported, "values is not supported for #{self}"
+      end
+    end
+
+    def hash_element(key)
+      if is_a?(HashNode)
+        elements.find { |element_node| element_node.key.to_value == key }
+      else
+        raise MethodNotSupported, "hash_pair is not supported for #{self}"
+      end
+    end
+
+    def hash_value(key)
+      if is_a?(HashNode)
+        elements.find { |element_node| element_node.key.to_value == key }&.value
+      else
+        raise MethodNotSupported, "hash_value is not supported for #{self}"
+      end
+    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 is_a?(HashNode)
+
+      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 }
+      elsif method_name.to_s.end_with?('_value')
+        key = method_name.to_s[0..-7]
+        return elements.find { |element| element.key.to_value.to_s == key }&.value
+      elsif method_name.to_s.end_with?('_source')
+        key = method_name.to_s[0..-8]
+        return elements.find { |element| element.key.to_value.to_s == key }&.value&.to_source || ''
+      end
+
+      super
+    end
+
+    def respond_to_missing?(method_name, *args)
+      return super unless is_a?(HashNode)
+
+      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 }
+      elsif method_name.to_s.end_with?('_value')
+        key = method_name.to_s[0..-7]
+        return !!elements.find { |element| element.key.to_value.to_s == key }
+      elsif method_name.to_s.end_with?('_source')
+        key = method_name.to_s[0..-8]
+        return !!elements.find { |element| element.key.to_value.to_s == key }
+      end
+
+      super
+    end
+  end
+end

data/sig/prism_ext.rbs

@@ -0,0 +1,4 @@
+module PrismExt
+  VERSION: String
+  # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+end

metadata

@@ -0,0 +1,68 @@
+--- !ruby/object:Gem::Specification
+name: prism_ext
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Richard Huang
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-02-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: prism
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: extend ruby prism
+email:
+- flyerhzm@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- CHANGELOG.md
+- Gemfile
+- Gemfile.lock
+- README.md
+- Rakefile
+- lib/prism_ext.rb
+- lib/prism_ext/version.rb
+- sig/prism_ext.rbs
+homepage: https://github.com/xinminlabs/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
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.7.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.3
+signing_key:
+specification_version: 4
+summary: extend ruby prism
+test_files: []