ruby_mod_kit 0.0.1

data/lib/ruby_mod_kit/node/def_node.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class DefNode < Node
+      # @rbs @prism_node: Prism::DefNode
+      # @rbs @parent: Node
+
+      attr_reader :prism_node #: Prism::DefNode
+      attr_reader :parent #: Node
+
+      # @rbs prism_node: Prism::DefNode
+      # @rbs parent: Node
+      # @rbs return: void
+      def initialize(prism_node, parent:)
+        @prism_node = prism_node
+        @parent = parent
+        raise RubyModKit::Error unless prism_node.is_a?(Prism::DefNode)
+
+        super()
+      end
+
+      # @rbs return: Symbol
+      def name
+        @prism_node.name
+      end
+
+      # @rbs return: Prism::Location | nil
+      def body_location
+        prism_node.body&.location
+      end
+
+      # @rbs return: Prism::Location | nil
+      def lparen_loc
+        @prism_node.lparen_loc
+      end
+
+      # @rbs return: Prism::Location | nil
+      def rparen_loc
+        @prism_node.rparen_loc
+      end
+
+      # @rbs return: Prism::Location
+      def name_loc
+        @prism_node.name_loc
+      end
+
+      # @rbs return: Prism::Location | nil
+      def end_keyword_loc
+        @prism_node.end_keyword_loc
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node/parameter_node.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class ParameterNode < Node
+      # @rbs!
+      #   type prism_node = Prism::RequiredParameterNode | Prism::OptionalKeywordParameterNode
+      #                   | Prism::OptionalParameterNode | Prism::RequiredKeywordParameterNode
+      #                   | Prism::RestParameterNode
+
+      # @rbs @prism_node: prism_node
+      # @rbs @parent: Node
+
+      attr_reader :prism_node #: prism_node
+      attr_reader :parent #: Node
+
+      # @rbs prism_node: prism_node
+      # @rbs parent: Node
+      # @rbs return: void
+      def initialize(prism_node, parent:)
+        @prism_node = prism_node
+        @parent = parent
+        case prism_node
+        when Prism::RequiredParameterNode, Prism::OptionalKeywordParameterNode,
+             Prism::OptionalParameterNode, Prism::RequiredKeywordParameterNode,
+             Prism::RestParameterNode
+          super()
+        else
+          raise RubyModKit::Error, "unexpected prism node #{prism_node.class}"
+        end
+      end
+
+      # @rbs return: Symbol | nil
+      def name
+        @prism_node.name
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node/program_node.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class ProgramNode < Node
+      # @rbs @prism_node: Prism::ProgramNode
+
+      attr_reader :prism_node #: Prism::ProgramNode
+
+      # @rbs prism_node: Prism::ProgramNode
+      # @rbs return: void
+      def initialize(prism_node)
+        @prism_node = prism_node
+        raise RubyModKit::Error unless prism_node.is_a?(Prism::ProgramNode)
+
+        super()
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node/statements_node.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class StatementsNode < Node
+      # @rbs @prism_node: Prism::StatementsNode
+      # @rbs @parent: Node
+
+      attr_reader :prism_node #: Prism::StatementsNode
+      attr_reader :parent #: Node
+
+      # @rbs prism_node: Prism::StatementsNode
+      # @rbs parent: Node
+      # @rbs return: void
+      def initialize(prism_node, parent:)
+        @prism_node = prism_node
+        @parent = parent
+        raise RubyModKit::Error unless prism_node.is_a?(Prism::StatementsNode)
+
+        super()
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node/symbol_node.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class SymbolNode < Node
+      # @rbs @prism_node: Prism::SymbolNode
+      # @rbs @parent: Node
+
+      attr_reader :prism_node #: Prism::SymbolNode
+      attr_reader :parent #: Node
+
+      # @rbs prism_node: Prism::SymbolNode
+      # @rbs parent: Node
+      # @rbs return: void
+      def initialize(prism_node, parent:)
+        @prism_node = prism_node
+        @parent = parent
+        raise RubyModKit::Error unless prism_node.is_a?(Prism::SymbolNode)
+
+        super()
+      end
+
+      # @rbs return: nil | Symbol
+      def value
+        return @value if defined?(@value)
+
+        @value = @prism_node.value&.to_sym
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node/untyped_node.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  class Node
+    # Transpiler program node
+    class UntypedNode < Node
+      # @rbs @prism_node: Prism::Node
+      # @rbs @parent: Node
+
+      attr_reader :prism_node #: Prism::Node
+      attr_reader :parent #: Node
+
+      # @rbs prism_node: Prism::Node
+      # @rbs parent: Node
+      # @rbs return: void
+      def initialize(prism_node, parent:)
+        @prism_node = prism_node
+        @parent = parent
+        super()
+      end
+    end
+  end
+end

data/lib/ruby_mod_kit/node.rb

@@ -0,0 +1,164 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  # The class of transpile node.
+  class Node
+    # @rbs @location: Prism::Location
+    # @rbs @children: Array[Node]
+    # @rbs @ancestors: Array[Node]
+
+    attr_reader :location #: Prism::Location
+
+    # @rbs return: void
+    def initialize
+      @location = prism_node.location
+    end
+
+    # @rbs return: Array[Node]
+    def children
+      return @children if @children
+
+      @children = prism_node.child_nodes.compact.map do |prism_child_node|
+        wrap(prism_child_node)
+      end
+    end
+
+    # @rbs prism_node: Prism::Node
+    # @rbs return: Node
+    def wrap(prism_node)
+      case prism_node
+      when Prism::SymbolNode
+        Node::SymbolNode.new(prism_node, parent: self)
+      when Prism::ClassNode
+        Node::ClassNode.new(prism_node, parent: self)
+      when Prism::DefNode
+        Node::DefNode.new(prism_node, parent: self)
+      when Prism::RequiredParameterNode, Prism::OptionalKeywordParameterNode,
+           Prism::OptionalParameterNode, Prism::RequiredKeywordParameterNode,
+           Prism::RestParameterNode
+        Node::ParameterNode.new(prism_node, parent: self)
+      when Prism::StatementsNode
+        Node::StatementsNode.new(prism_node, parent: self)
+      when Prism::CallNode
+        Node::CallNode.new(prism_node, parent: self)
+      else
+        Node::UntypedNode.new(prism_node, parent: self)
+      end
+    end
+
+    # @rbs return: Array[Node]
+    def ancestors
+      return @ancestors if @ancestors
+
+      parent = @parent
+      @ancestors = if parent
+        [parent] + parent.ancestors
+      else
+        []
+      end
+    end
+
+    # @rbs return: Symbol
+    def name
+      raise(RubyModKit::Error, "Expected ParameterNode but #{self.class}:#{prism_node.inspect}")
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: Node | nil
+    def node_at(offset)
+      return nil unless include?(offset)
+
+      child = children.find { _1.include?(offset) }
+      child&.node_at(offset) || self
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: Node::StatementsNode | nil
+    def statements_node_at(offset)
+      node = node_at(offset) || return
+      [node, *node.ancestors].each { return _1 if _1.is_a?(Node::StatementsNode) }
+      nil
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: Node::DefNode | nil
+    def def_node_at(offset)
+      node = node_at(offset) || return
+      [node, *node.ancestors].each { return _1 if _1.is_a?(Node::DefNode) }
+      nil
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: Node::ParameterNode | nil
+    def parameter_node_at(offset)
+      node = node_at(offset) || return
+      [node, *node.ancestors].each { return _1 if _1.is_a?(Node::ParameterNode) }
+      nil
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: Node::ClassNode | nil
+    def class_node_at(offset)
+      node = node_at(offset) || return
+      [node, *node.ancestors].each { return _1 if _1.is_a?(Node::ClassNode) }
+      nil
+    end
+
+    # @rbs offset: Integer
+    # @rbs return: bool
+    def include?(offset)
+      self.offset <= offset && offset <= prism_node.location.end_offset
+    end
+
+    # @rbs return: Prism::Node & Prism::_Node
+    def prism_node
+      raise RubyModKit::Error
+    end
+
+    # @rbs return: Integer
+    def offset
+      location.start_offset
+    end
+
+    # @rbs return: String
+    def inspect
+      str = +"#<#{self.class} "
+      first = true
+      instance_variables.each do |ivar_name|
+        case ivar_name
+        when :@children, :@ancestors, :@parent
+          next
+        end
+
+        if first
+          first = false
+        else
+          str << ", "
+        end
+        str << "#{ivar_name}="
+        value = instance_variable_get(ivar_name)
+        str << (
+          case value
+          when Prism::Node
+            "#<#{value.class} location=#{value.location.inspect}>"
+          else
+            value.inspect
+          end
+        )
+      end
+      str << ">"
+      str
+    end
+  end
+end
+
+require_relative "node/symbol_node"
+require_relative "node/class_node"
+require_relative "node/call_node"
+require_relative "node/def_node"
+require_relative "node/parameter_node"
+require_relative "node/program_node"
+require_relative "node/statements_node"
+require_relative "node/untyped_node"

data/lib/ruby_mod_kit/offset_diff.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+require "sorted_set"
+
+module RubyModKit
+  # The class of offset differences.
+  class OffsetDiff
+    # @rbs @diffs: SortedSet[[Integer, Integer, Integer]]
+
+    # @rbs return: void
+    def initialize
+      @diffs = SortedSet.new
+    end
+
+    # @rbs src_offset: Integer
+    # @rbs return: Integer
+    def [](src_offset)
+      dst_offset = src_offset
+      @diffs.each do |(offset, _, diff)|
+        break if offset > src_offset
+        break if offset == src_offset && diff < 0
+
+        dst_offset += diff
+      end
+      dst_offset
+    end
+
+    # @rbs src_offset: Integer
+    # @rbs new_diff: Integer
+    # @rbs return: void
+    def insert(src_offset, new_diff)
+      @diffs << [src_offset, @diffs.size, new_diff]
+    end
+  end
+end

data/lib/ruby_mod_kit/transpiler.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  # The class of transpiler.
+  class Transpiler
+    # @rbs src: String
+    # @rbs filename: String | nil
+    # @rbs return: String
+    def transpile(src, filename: nil)
+      generation = Generation.new(src.dup, filename: filename)
+      until generation.completed?
+        generation.resolve
+        generation = generation.succ
+      end
+      generation.script
+    end
+  end
+end

data/lib/ruby_mod_kit/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+module RubyModKit
+  VERSION = "0.0.1" #: String
+end

data/lib/ruby_mod_kit.rb

@@ -0,0 +1,59 @@
+# frozen_string_literal: true
+
+# rbs_inline: enabled
+
+require "rbconfig"
+require "prism"
+
+# The root namespace for ruby_mod_kit.
+module RubyModKit
+  class Error < StandardError; end
+  class SyntaxError < ::SyntaxError; end
+
+  class << self
+    # @rbs file: String
+    # @rbs *args: String
+    # @rbs return: void
+    def execute_file(file, *args)
+      rb_file = transpile_file(file)
+      execute_rb_file(rb_file, *args)
+    end
+
+    # @rbs file: String
+    # @rbs return: String
+    def transpile_file(file)
+      rb_src = transpile(File.read(file), filename: file)
+      rb_path = rb_path(file)
+      File.write(rb_path, rb_src)
+      rb_path
+    end
+
+    # @rbs src: String
+    # @rbs filename: String | nil
+    # @rbs return: String
+    def transpile(src, filename: nil)
+      Transpiler.new.transpile(src, filename: filename)
+    end
+
+    # @rbs file: String
+    # @rbs *args: String
+    # @rbs return: void
+    def execute_rb_file(file, *args)
+      system(RbConfig.ruby, file, *args)
+    end
+
+    # @rbs path: String
+    # @rbs return: String
+    def rb_path(path)
+      path.sub(/(?:\.rbm)?$/, ".rb")
+    end
+  end
+end
+
+require_relative "ruby_mod_kit/version"
+require_relative "ruby_mod_kit/transpiler"
+require_relative "ruby_mod_kit/generation"
+require_relative "ruby_mod_kit/memo"
+require_relative "ruby_mod_kit/node"
+require_relative "ruby_mod_kit/mission"
+require_relative "ruby_mod_kit/offset_diff"

data/ruby_mod_kit.gemspec

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require_relative "lib/ruby_mod_kit/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "ruby_mod_kit"
+  spec.authors = ["wanabe"]
+  spec.email = ["s.wanabe@gmail.com"]
+
+  version = RubyModKit::VERSION
+  tag_version = `git tag --points-at HEAD`.chomp.sub("v", "")
+  version += "-#{`git rev-parse --short HEAD`}" if version != tag_version
+  spec.version = version
+
+  spec.summary = "ruby_mod_kit"
+  spec.description = "ruby_mod_kit"
+  spec.homepage = "https://github.com/wanabe/ruby_mod_kit"
+  spec.license = "MIT"
+  spec.required_ruby_version = ">= 2.7.0"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = spec.homepage
+  spec.metadata["changelog_uri"] = "#{spec.homepage}/blob/main/CHANGELOG.md"
+
+  spec.metadata["rubygems_mfa_required"] = "true"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files = Dir.chdir(__dir__) do
+    `git ls-files -z`.split("\x0").reject do |f|
+      next true if File.expand_path(f) == __FILE__
+      next false if f =~ %r{\Aexamples/.*rbm?\z}
+      next true if f.start_with?(*%w[bin/ test/ spec/ features/ coverage/ bin/ .git])
+      next true if %w[Gemfile Gemfile.lock .rspec .rubocop.yml Rakefile Steepfile].include?(f)
+      next true if f.end_with?(".rbm")
+
+      false
+    end
+  end
+  spec.bindir = "exe"
+  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  # Uncomment to register a new dependency of your gem
+  spec.add_dependency "prism", "~> 1.0"
+  spec.add_dependency "sorted_set", "~> 1.0"
+  spec.add_dependency "thor", "~> 1.3"
+
+  # For more information and examples about making a new gem, check out our
+  # guide at: https://bundler.io/guides/creating_gem.html
+end

data/sig/generated/examples/user.rbs

@@ -0,0 +1,60 @@
+# Generated from examples/user.rb with RBS::Inline
+
+# Example class
+class User
+  @email: String
+
+  @tel: nil | String
+
+  @nick: nil | String
+
+  attr_reader email: String
+
+  attr_reader name: nil | String
+
+  attr_reader nick: nil | String
+
+  # @rbs email: String
+  # @rbs tel: nil | String
+  # @rbs name: nil | String
+  # @rbs nick: nil | String
+  # @rbs return: void
+  def initialize: (String email, ?nil | String tel, ?name: nil | String, ?nick: nil | String) -> void
+end
+
+# Example class
+class Pos
+  @x: Integer
+
+  @y: Integer
+
+  @z: Integer
+
+  attr_reader x: Integer
+
+  attr_reader y: Integer
+
+  attr_reader z: Integer
+
+  # @rbs x: Integer
+  # @rbs y: Integer
+  # @rbs z: Integer
+  # @rbs return: void
+  def initialize: (Integer x, Integer y, ?Integer z) -> void
+
+  # @rbs (Pos) -> Integer
+  #    | (Integer) -> Pos
+  def *: (Pos) -> Integer
+       | (Integer) -> Pos
+
+  # @rbs other: Pos
+  # @rbs return: Integer
+  def _mul__overload0: (Pos other) -> Integer
+
+  # @rbs n: Integer
+  # @rbs return: Pos
+  def _mul__overload1: (Integer n) -> Pos
+
+  # @rbs return: String
+  def to_s: () -> String
+end

data/sig/generated/ruby_mod_kit/cli.rbs

@@ -0,0 +1,14 @@
+# Generated from lib/ruby_mod_kit/cli.rb with RBS::Inline
+
+module RubyModKit
+  # This class provides CLI commands of ruby_mod_kit.
+  class CLI < Thor
+    # @rbs *args: String
+    # @rbs return: void
+    def exec: (*String args) -> void
+
+    # @rbs *args: String
+    # @rbs return: void
+    def transpile: (*String args) -> void
+  end
+end