overloader 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 3ae0b6a1e018a6dc49298c7333333f5b780395c08d6050696d466e67071bdaf5
+  data.tar.gz: e796b18cc593dc6ac014185e45a86a8b3f17f7de9764ee5da465ced12e445c10
+SHA512:
+  metadata.gz: d7287ffa69f2c273324acd514fa011f5a4834a16ee970310e452adec5a206325b43a8c3a213332b7046826ff075791e34e2e8ab7391c47c12452842834d4222d
+  data.tar.gz: 0c11df4506fdcf596a5be17f5a067759f513c6ed32bb19b711b564c5a0a66fc0f4091ba85a866e2a6e590777fca6f9897eb68aabb4030d3ed4736f1c9c00b3c2

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/Gemfile.lock

data/Gemfile

@@ -0,0 +1,4 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in overloader.gemspec
+gemspec

data/README.md

@@ -0,0 +1,49 @@
+# Overloader
+
+Overload for Ruby
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'overloader'
+```
+
+And then execute:
+
+    $ bundle install
+
+Or install it yourself as:
+
+    $ gem install overloader
+
+## Usage
+
+```ruby
+require 'overloader'
+
+class A
+  extend Overloader
+  overload do
+    def foo() "no args" end
+    def foo(x) "one arg" end
+    def foo(x, y) "two args" end
+  end
+end
+
+a = A.new
+p a.foo # => "no args"
+p a.foo(1) # => "one args"
+p a.foo(1, 2) # => "two args"
+```
+
+## Development
+
+After checking out the repo, run `bin/setup` to install dependencies. 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 tags, 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/pocke/overloader.

data/Rakefile

@@ -0,0 +1,9 @@
+require "bundler/gem_tasks"
+require 'rake/testtask'
+task :default => :test
+
+Rake::TestTask.new do |test|
+  test.libs << 'test'
+  test.test_files = Dir['test/**/*_test.rb']
+  test.verbose = true
+end

data/bin/console

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+require "bundler/setup"
+require "overloader"
+
+# You can add fixtures and/or initialization code here to make experimenting
+# with your gem easier. You can also use a different console, if you like.
+
+# (If you use this, don't forget to add pry to your Gemfile!)
+# require "pry"
+# Pry.start
+
+require "irb"
+IRB.start(__FILE__)

data/bin/setup

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+set -euo pipefail
+IFS=$'\n\t'
+set -vx
+
+bundle install
+
+# Do any other automated setup that you need to do here

data/lib/overloader/ast_ext.rb

@@ -0,0 +1,62 @@
+module Overloader
+  module AstExt
+    refine RubyVM::AbstractSyntaxTree::Node do
+      def traverse(&block)
+        block.call(self)
+        children.each do |child|
+          child.traverse(&block) if child.is_a?(RubyVM::AbstractSyntaxTree::Node)
+        end
+      end
+
+      def find_nodes(*types)
+        nodes = []
+        traverse do |node|
+          nodes << node if types.include?(node.type)
+        end
+        nodes
+      end
+
+      def to_source(path)
+        file_content(path)[first_index(path)..last_index(path)]
+      end
+
+      # method node ext
+
+      def method_body
+        children[1].children[2]
+      end
+
+      def method_args
+        children[1].children[1]
+      end
+
+      def method_name
+        children[0]
+      end
+
+      private def first_index(path)
+        return first_column if first_lineno == 1
+
+        lines = file_content(path).split("\n")
+        lines[0..(first_lineno - 2)].sum(&:size) +
+          first_lineno - 1 + # For \n
+          first_column
+      end
+
+      private def last_index(path)
+        last_column = self.last_column - 1
+        return last_column if last_lineno == 1
+
+        lines = file_content(path).split("\n")
+        lines[0..(last_lineno - 2)].sum(&:size) +
+          last_lineno - 1 + # For \n
+          last_column
+      end
+
+      private def file_content(path)
+        @file_content ||= {}
+        @file_content[path] ||= File.binread(path)
+      end
+    end
+  end
+end

data/lib/overloader/core.rb

@@ -0,0 +1,58 @@
+module Overloader
+  class Core
+    using AstExt
+
+    def self.define_overload(klass, proc)
+      self.new(klass, proc).define_overload
+    end
+
+    def initialize(klass, proc)
+      @klass = klass
+      @proc = proc
+    end
+
+    def define_overload
+      ast = RubyVM::AbstractSyntaxTree.of(@proc)
+      methods = {}
+      ast.find_nodes(:DEFN).each.with_index do |def_node, index|
+        args = def_node.method_args
+        body = def_node.method_body
+        name = def_node.method_name
+        args_source = args.to_source(absolute_path)
+        args_source = "" if args_source == "(" # RubyVM::AST's bug?
+
+        @klass.class_eval <<~RUBY
+          def __#{name}_#{index}_checker_inner(#{args_source}) end
+          def __#{name}_#{index}_checker(*args)
+            __#{name}_#{index}_checker_inner(*args)
+            true
+          rescue ArgumentError
+            false
+          end
+        RUBY
+
+        @klass.class_eval <<~RUBY, absolute_path, def_node.first_lineno
+          def __#{name}_#{index}(#{args_source})
+          #{body.to_source(absolute_path)}
+          end
+        RUBY
+        (methods[name] ||= []) << index
+      end
+
+      methods.each do |name, indexes|
+        @klass.class_eval <<~RUBY
+          def #{name}(*args, &block)
+            #{indexes.map do |index|
+              "return __#{name}_#{index}(*args, &block) if __#{name}_#{index}_checker(*args)"
+            end.join("\n")}
+            raise ArgumentError
+          end
+        RUBY
+      end
+    end
+
+    private def absolute_path
+      @proc.source_location[0]
+    end
+  end
+end

data/lib/overloader/version.rb

@@ -0,0 +1,3 @@
+module Overloader
+  VERSION = "0.1.0"
+end

data/lib/overloader.rb

@@ -0,0 +1,9 @@
+require "overloader/version"
+require 'overloader/ast_ext'
+require 'overloader/core'
+
+module Overloader
+  def overload(&block)
+    Core.define_overload(self, block)
+  end
+end

data/overloader.gemspec

@@ -0,0 +1,33 @@
+lib = File.expand_path("lib", __dir__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "overloader/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "overloader"
+  spec.version       = Overloader::VERSION
+  spec.authors       = ["Masataka Pocke Kuwabara"]
+  spec.email         = ["kuwabara@pocke.me"]
+
+  spec.summary       = %q{}
+  spec.description   = %q{}
+  spec.homepage      = "https://github.com/pocke/overloader"
+
+  # spec.metadata["allowed_push_host"] = "TODO: Set to 'http://mygemserver.com'"
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  spec.metadata["source_code_uri"] = "https://github.com/pocke/overloader"
+  # spec.metadata["changelog_uri"] = "TODO: Put your gem's CHANGELOG.md URL here."
+
+  # 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(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 2.1.a"
+  spec.add_development_dependency "rake", "~> 12.0"
+  spec.add_development_dependency "minitest", "> 5"
+end

metadata

@@ -0,0 +1,97 @@
+--- !ruby/object:Gem::Specification
+name: overloader
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Masataka Pocke Kuwabara
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2019-04-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.1.a
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.1.a
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '12.0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">"
+      - !ruby/object:Gem::Version
+        version: '5'
+description: ''
+email:
+- kuwabara@pocke.me
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/overloader.rb
+- lib/overloader/ast_ext.rb
+- lib/overloader/core.rb
+- lib/overloader/version.rb
+- overloader.gemspec
+homepage: https://github.com/pocke/overloader
+licenses: []
+metadata:
+  homepage_uri: https://github.com/pocke/overloader
+  source_code_uri: https://github.com/pocke/overloader
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: ''
+test_files: []