rbs_inline_data 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b590372f180f2c8115503f7d93a8eee62b4a67098a278b9487a24e3f715710f
-  data.tar.gz: 9b17bc5d5b477670ea88acb82f6123f6e7c361a370a0589190e02015a417f4e3
+  metadata.gz: e626cd6506b96f771d5b5e178354e6fcca38b1ee8f9eb1e8c6eaf9191ddb9074
+  data.tar.gz: 79cd75b8df23c3066617a5b0a6186b9dc8617b444c5c6318816714bcfc285cfe
 SHA512:
-  metadata.gz: 4696211d602da66ccbf6c07a2ce2ad6f66e7e42c1f503c560a5c31b5b64bda3c0d46e357106f73ecc584788e09a15fee1fd5756c7a5bf0edac91d7616104a058
-  data.tar.gz: dff46a22e6c68c0d6624e8bead20ca70779752546859cd2fd2086ea02fdee028d10251095c43c89e42c566db62d42a8e6f5bd443197c02ec52320b7efd00c203
+  metadata.gz: 8a648312df3817a4e65e810f12ba3a63dd63746c00db2f289b037883563c61da31fc9d1b208aaa3d7a4149ed651d07171eb62134775b257c7e15e889e95ed3ff
+  data.tar.gz: 31eaf6f6f76f5995b5e9c3ce2e878a756601dfb8a04b16e6699b45e2465d7d879ca4b8ebd2cebf1cf530809ec5aca0ace9abf40c5a390e0266c20a22b0f68910

data/.rubocop.yml

@@ -1,5 +1,6 @@
 AllCops:
-  TargetRubyVersion: 3.0
+  TargetRubyVersion: 3.3
+  NewCops: enable
 
 Style/StringLiterals:
   EnforcedStyle: double_quotes
@@ -9,3 +10,16 @@ Style/StringLiteralsInInterpolation:
 
 Layout/LeadingCommentSpace:
   Enabled: false
+
+Metrics/ClassLength:
+  Exclude:
+    - 'test/**/*'
+
+Metrics/MethodLength:
+  Exclude:
+    - 'test/**/*'
+  Max: 20
+
+Style/Style/Documentation:
+  Enabled: false
+

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.2.0] - 2024-06-18
+
+- Support Hash type https://github.com/euglena1215/rbs_inline_data/pull/1
+
 ## [0.1.0] - 2024-06-09
 
 - Initial release

data/lib/rbs_inline_data/cli.rb

@@ -6,6 +6,7 @@ require "rbs_inline_data/parser"
 require "rbs_inline_data/writer"
 
 module RbsInlineData
+  # Process executed when running the rbs-inline-data command.
   class CLI
     #:: (Array[String]) -> void
     def run(args)
@@ -18,22 +19,29 @@ module RbsInlineData
         end
       end.parse!(args)
 
-      targets = Pathname.glob(args[0]).flat_map do |path|
-        if path.directory?
-          Pathname.glob(path.join("**/*.rb").to_s)
+      get_targets(args[0]).each do |target|
+        result = Prism.parse_file(target.to_s)
+        definitions = Parser.parse(result)
+        Writer.write(definitions, output_path ? (output_path + target).sub_ext(".rbs") : nil)
+      end
+    end
+
+    private
+
+    #:: (String) -> Array[Pathname]
+    def get_targets(path)
+      targets = Pathname.glob(path).flat_map do |pathname|
+        if pathname.directory?
+          Pathname.glob(pathname.join("**/*.rb").to_s)
         else
-          path
+          pathname
         end
       end
 
       targets.sort!
       targets.uniq!
 
-      targets.each do |target|
-        result = Prism.parse_file(target.to_s)
-        definitions = Parser.parse(result)
-        Writer.write(definitions, output_path ? (output_path + target).sub_ext(".rbs") : nil)
-      end
+      targets
     end
   end
 end

data/lib/rbs_inline_data/parser.rb

@@ -14,21 +14,47 @@ module RbsInlineData
       :field_name, #:: String
       :type #:: String
     )
+    # @rbs skip
+    Comments = Data.define(
+      :comment_lines #:: Hash[Integer, String]
+    )
+    class Comments
+      MARKER = "#::"
+
+      #:: (Array[Prism::Comment]) -> RbsInlineData::Parser::Comments
+      def self.from_prism_comments(comments)
+        # @type var comment_lines: Hash[Integer, String]
+        comment_lines = {}
+        comments.each do |comment|
+          sliced = comment.slice
+          next unless sliced.start_with?(MARKER)
+
+          comment_lines[comment.location.start_line] = sliced.sub(MARKER, "").strip
+        end
+
+        new(comment_lines:)
+      end
+    end
 
     # @rbs @definitions: Array[RbsInlineData::Parser::TypedDefinition]
     # @rbs @surronding_class_or_module: Array[Symbol]
+    # @rbs @comments: RbsInlineData::Parser::Comments
 
-    #:: (Array[RbsInlineData::Parser::TypedDefinition]) -> void
-    def initialize(definitions)
+    # rubocop:disable Lint/MissingSuper
+    #:: (Array[RbsInlineData::Parser::TypedDefinition], RbsInlineData::Parser::Comments) -> void
+    def initialize(definitions, comments)
       @definitions = definitions
+      @comments = comments
       @surronding_class_or_module = []
     end
+    # rubocop:enable Lint/MissingSuper
 
     #:: (Prism::ParseResult) -> Array[RbsInlineData::Parser::TypedDefinition]
     def self.parse(result)
       # @type var definitions: Array[RbsInlineData::Parser::TypedDefinition]
       definitions = []
-      instance = new(definitions)
+      comments = Comments.from_prism_comments(result.comments)
+      instance = new(definitions, comments)
       result.value.accept(instance)
       definitions
     end
@@ -78,30 +104,27 @@ module RbsInlineData
 
     #:: (Prism::ConstantWriteNode) -> RbsInlineData::Parser::TypedDefinition?
     def extract_definition(node)
-      source = node.slice
-      _, class_name, field_text = source.match(/\A([a-zA-Z0-9]+) ?= ?Data\.define\(([\n\s\w\W]+)\)\z/).to_a
-      return nil if field_text.nil? || class_name.nil?
-
-      class_name = "#{@surronding_class_or_module.join("::")}::#{class_name}"
-
-      fields = field_text.split("\n").map(&:strip).reject(&:empty?).map do |str|
-        case str
-        when /:(\w+),? #:: ([\w\:\[\]]+)/
-          [::Regexp.last_match(1), ::Regexp.last_match(2)]
-        when /:(\w+),?/
-          [::Regexp.last_match(1), "untyped"]
-        end
-      end.compact.map do |field_name, type|
-        TypedField.new(
-          field_name: field_name,
-          type: type
-        )
+      arguments_node = node.value.arguments
+      if arguments_node
+        typed_fields = arguments_node.arguments.map do |sym_node|
+          return nil unless sym_node.is_a?(Prism::SymbolNode)
+
+          TypedField.new(
+            field_name: sym_node.unescaped,
+            type: type_of(sym_node)
+          )
+        end.compact
       end
 
       TypedDefinition.new(
-        class_name: class_name,
-        fields: fields
+        class_name: "#{@surronding_class_or_module.join("::")}::#{node.name}",
+        fields: typed_fields || []
       )
     end
+
+    #:: (Prism::SymbolNode) -> String
+    def type_of(node)
+      @comments.comment_lines[node.location.start_line] || "untyped"
+    end
   end
 end

data/lib/rbs_inline_data/version.rb

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

data/sig/generated/data/lib/rbs_inline_data/parser.rbs

@@ -15,3 +15,11 @@ class RbsInlineData::Parser::TypedField
               | (**untyped) -> ::RbsInlineData::Parser::TypedField
               | ...
 end
+
+class RbsInlineData::Parser::Comments
+  extend Data::_DataClass
+  attr_reader comment_lines: Hash[Integer, String]
+  def self.new: (*untyped) -> ::RbsInlineData::Parser::Comments
+              | (**untyped) -> ::RbsInlineData::Parser::Comments
+              | ...
+end

data/sig/generated/rbs_inline_data/cli.rbs

@@ -1,8 +1,14 @@
 # Generated from lib/rbs_inline_data/cli.rb with RBS::Inline
 
 module RbsInlineData
+  # Process executed when running the rbs-inline-data command.
   class CLI
     # :: (Array[String]) -> void
     def run: (Array[String]) -> void
+
+    private
+
+    # :: (String) -> Array[Pathname]
+    def get_targets: (String) -> Array[Pathname]
   end
 end

data/sig/generated/rbs_inline_data/parser.rbs

@@ -2,12 +2,22 @@
 
 module RbsInlineData
   class Parser < Prism::Visitor
+    class Comments
+      MARKER: ::String
+
+      # :: (Array[Prism::Comment]) -> RbsInlineData::Parser::Comments
+      def self.from_prism_comments: (Array[Prism::Comment]) -> RbsInlineData::Parser::Comments
+    end
+
     @definitions: Array[RbsInlineData::Parser::TypedDefinition]
 
     @surronding_class_or_module: Array[Symbol]
 
-    # :: (Array[RbsInlineData::Parser::TypedDefinition]) -> void
-    def initialize: (Array[RbsInlineData::Parser::TypedDefinition]) -> void
+    @comments: RbsInlineData::Parser::Comments
+
+    # rubocop:disable Lint/MissingSuper
+    # :: (Array[RbsInlineData::Parser::TypedDefinition], RbsInlineData::Parser::Comments) -> void
+    def initialize: (Array[RbsInlineData::Parser::TypedDefinition], RbsInlineData::Parser::Comments) -> void
 
     # :: (Prism::ParseResult) -> Array[RbsInlineData::Parser::TypedDefinition]
     def self.parse: (Prism::ParseResult) -> Array[RbsInlineData::Parser::TypedDefinition]
@@ -31,5 +41,8 @@ module RbsInlineData
 
     # :: (Prism::ConstantWriteNode) -> RbsInlineData::Parser::TypedDefinition?
     def extract_definition: (Prism::ConstantWriteNode) -> RbsInlineData::Parser::TypedDefinition?
+
+    # :: (Prism::SymbolNode) -> String
+    def type_of: (Prism::SymbolNode) -> String
   end
 end

data/sig/patch.rbs

@@ -8,4 +8,13 @@ module Prism
     def constant_path: () -> Prism::ConstantReadNode
                      | ...
   end
+
+  class Comment
+    def slice: () -> String
+  end
+
+  class ConstantWriteNode
+    def value: () -> CallNode
+             | ...
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbs_inline_data
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Teppei Shintani
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-06-11 00:00:00.000000000 Z
+date: 2024-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prism
@@ -73,6 +73,7 @@ metadata:
   homepage_uri: https://github.com/euglena1215/rbs_inline_data
   source_code_uri: https://github.com/euglena1215/rbs_inline_data
   changelog_uri: https://github.com/euglena1215/rbs_inline_data/blob/main/CHANGELOG.md
+  rubygems_mfa_required: 'true'
 post_install_message:
 rdoc_options: []
 require_paths: