protod 0.1.0

data/Gemfile.lock

@@ -0,0 +1,165 @@
+PATH
+  remote: .
+  specs:
+    protod (0.1.0)
+      activemodel (>= 5.2.0)
+      activesupport
+      google-protobuf
+      rbs (>= 3.2.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actionmailer (0.6.1)
+      actionpack (>= 0.9.5)
+    actionpack (1.4.0)
+    activemodel (7.0.7.2)
+      activesupport (= 7.0.7.2)
+    activerecord (7.0.7.2)
+      activemodel (= 7.0.7.2)
+      activesupport (= 7.0.7.2)
+    activesupport (7.0.7.2)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 1.6, < 2)
+      minitest (>= 5.1)
+      tzinfo (~> 2.0)
+    ast (2.4.2)
+    base64 (0.1.1)
+    binding_of_caller (1.0.0)
+      debug_inspector (>= 0.0.1)
+    byebug (11.1.3)
+    coderay (1.1.3)
+    concurrent-ruby (1.2.2)
+    debug_inspector (1.1.0)
+    diff-lcs (1.5.0)
+    factory_bot (6.3.0)
+      activesupport (>= 5.0.0)
+    faker (3.2.1)
+      i18n (>= 1.8.11, < 2)
+    google-protobuf (3.24.3-x86_64-linux)
+    i18n (1.14.1)
+      concurrent-ruby (~> 1.0)
+    json (2.6.3)
+    language_server-protocol (3.17.0.3)
+    method_source (1.0.0)
+    minitest (5.19.0)
+    parallel (1.23.0)
+    parser (3.2.2.3)
+      ast (~> 2.4.1)
+      racc
+    proc_to_ast (0.1.0)
+      coderay
+      parser
+      unparser
+    pry (0.14.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    pry-byebug (3.10.1)
+      byebug (~> 11.0)
+      pry (>= 0.13, < 0.15)
+    pry-doc (1.4.0)
+      pry (~> 0.11)
+      yard (~> 0.9.11)
+    racc (1.7.1)
+    rack (3.0.8)
+    rails (0.9.5)
+      actionmailer (>= 0.6.1)
+      actionpack (>= 1.4.0)
+      activerecord (>= 1.6.0)
+      rake (>= 0.4.15)
+    rainbow (3.1.1)
+    rake (13.0.6)
+    rbs (3.2.1)
+    regexp_parser (2.8.1)
+    rexml (3.2.6)
+    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-parameterized (1.0.0)
+      rspec-parameterized-core (< 2)
+      rspec-parameterized-table_syntax (< 2)
+    rspec-parameterized-core (1.0.0)
+      parser
+      proc_to_ast
+      rspec (>= 2.13, < 4)
+      unparser
+    rspec-parameterized-table_syntax (1.0.0)
+      binding_of_caller
+      rspec-parameterized-core (< 2)
+    rspec-support (3.12.1)
+    rubocop (1.56.2)
+      base64 (~> 0.1.1)
+      json (~> 2.3)
+      language_server-protocol (>= 3.17.0)
+      parallel (~> 1.10)
+      parser (>= 3.2.2.3)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.28.1, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.29.0)
+      parser (>= 3.2.1.0)
+    rubocop-capybara (2.18.0)
+      rubocop (~> 1.41)
+    rubocop-factory_bot (2.23.1)
+      rubocop (~> 1.33)
+    rubocop-performance (1.19.0)
+      rubocop (>= 1.7.0, < 2.0)
+      rubocop-ast (>= 0.4.0)
+    rubocop-rails (2.20.2)
+      activesupport (>= 4.2.0)
+      rack (>= 1.1)
+      rubocop (>= 1.33.0, < 2.0)
+    rubocop-rake (0.6.0)
+      rubocop (~> 1.0)
+    rubocop-rspec (2.23.2)
+      rubocop (~> 1.33)
+      rubocop-capybara (~> 2.17)
+      rubocop-factory_bot (~> 2.22)
+    rubocop-rubycw (0.1.6)
+      rubocop (~> 1.0)
+    ruby-progressbar (1.13.0)
+    shoulda-matchers (5.3.0)
+      activesupport (>= 5.2.0)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
+    unicode-display_width (2.4.2)
+    unparser (0.6.8)
+      diff-lcs (~> 1.3)
+      parser (>= 3.2.0)
+    yard (0.9.34)
+
+PLATFORMS
+  x86_64-linux
+
+DEPENDENCIES
+  factory_bot
+  faker
+  protod!
+  pry-byebug
+  pry-doc
+  rails
+  rake
+  rspec
+  rspec-parameterized
+  rubocop
+  rubocop-performance
+  rubocop-rails
+  rubocop-rake
+  rubocop-rspec
+  rubocop-rubycw
+  shoulda-matchers
+
+BUNDLED WITH
+   2.4.19

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2023 TODO: Write your name
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,37 @@
+# Protod
+
+Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/protod`. To experiment with that code, run `bin/console` for an interactive prompt.
+
+TODO: Delete this and the text above, and describe your gem
+
+## Installation
+
+Install the gem and add to the application's Gemfile by executing:
+
+    $ bundle add protod
+
+If bundler is not being used to manage dependencies, install the gem by executing:
+
+    $ gem install protod
+
+## Usage
+
+TODO: Write usage instructions here
+
+## 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/[USERNAME]/protod. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/protod/blob/master/CODE_OF_CONDUCT.md).
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+
+## Code of Conduct
+
+Everyone interacting in the Protod project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/protod/blob/master/CODE_OF_CONDUCT.md).

data/Rakefile

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+require "rubocop/rake_task"
+
+RuboCop::RakeTask.new
+
+task default: %i[spec rubocop]

data/lib/generators/protod/gruf_generator.rb

@@ -0,0 +1,43 @@
+require 'rails'
+
+class Protod
+  class GrufGenerator < Rails::Generators::Base
+    def create_gruf_controller
+      create_file "app/rpc/protod/handler_controller.rb", <<~RUBY
+        # frozen_string_literal: true
+
+        class Protod
+          class HandlerController < ::Gruf::Controllers::Base
+            include ::Gruf::Loggable
+
+            def self.package
+              @package ||= Protod::Rpc::Handler.find_package
+            end
+
+            bind Protod::Rpc::Handler.find_service_in(package).pb_const
+
+            def handle
+              return enum_for(:handle) unless block_given?
+
+              logger.info("protod/handle start")
+              handler = Protod::Rpc::Handler.new(self.class.package)
+
+              request.messages.each do |m|
+                logger.debug("protod/handle receive : #\{m\}")
+                yield handler.handle(m).tap { logger.debug("protod/handle send : #\{_1\}") }
+              end
+              logger.info("protod/handle finished")
+            rescue Protod::Rpc::Handler::InvalidArgument => e
+              logger.debug("protod/handle failed : #\{e.message\}")
+              fail!(:invalid_argument, :invalid_argument, "ERROR: #\{e.message\}")
+            rescue Exception => e
+              logger.error("protod/handle failed : #\{e.message\}\\n#\{e.backtrace.join("\\n")\}")
+              set_debug_info(e.message, e.backtrace[0..4])
+              fail!(:internal, :internal, "ERROR: #\{e.message\}")
+            end
+          end
+        end
+      RUBY
+    end
+  end
+end

data/lib/generators/protod/task_generator.rb

@@ -0,0 +1,23 @@
+require 'rails'
+
+class Protod
+  class TaskGenerator < Rails::Generators::Base
+    def create_raketask
+      create_file "lib/tasks/protod.rake", <<~RUBY
+        # frozen_string_literal: true
+
+        begin
+          require 'protod/rake_task'
+
+          Protod::RakeTask::Builder.new do |builder|
+            # If you want to change the rake task namespace, comment in it.
+            # default: :protod
+            # builder.name = :pd
+          end.build
+        rescue LoadError => e
+          $stderr.puts "#\{e\}\\nYou might should remove 'lib/tasks/protod.rake'."
+        end
+      RUBY
+    end
+  end
+end

data/lib/protod/configuration.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+class Protod
+  class Configuration
+    attr_accessor :proto_root_dir, :pb_root_dir
+
+    def initialize
+      @proto_root_dir = './proto'
+      @pb_root_dir    = './lib'
+      @builders       = {}
+    end
+
+    def setup_rbs_environment_loader(&body)
+      @setup_rbs_environment_loader = body
+    end
+
+    def define_package(ident, for_ruby: nil, for_java: nil, &body)
+      bkup = @current_root_package
+
+      raise NotImplementedError, "Unsupported nested package not yet!" if bkup
+
+      @current_root_package = Protod.find_or_register_package(ident, for_ruby: for_ruby, for_java: for_java)
+
+      body.call
+    ensure
+      @current_root_package = bkup
+    end
+
+    def derive_from(const_name, abstruct: false, &body)
+      raise NotImplementedError, "You need to call #define_package before" unless @current_root_package
+
+      bkup = @current_const
+      @current_const = bkup ? "#{bkup}::#{const_name.to_s.delete_prefix('::')}" : const_name.to_s.delete_prefix('::')
+
+      Protod::Rpc::Request::Receiver.register_for(@current_const, force: false, ignore: true)
+      Protod::Rpc::Response::Receiver.register_for(@current_const, force: false, ignore: true)
+
+      unless abstruct
+        builder = @builders[@current_root_package.full_ident] ||= Protod::Proto::Builder.new(@current_root_package)
+        builder.push_receiver(@current_const)
+      end
+
+      body.call
+    ensure
+      @current_const = bkup
+    end
+
+    def define_rpc(*names, singleton: false)
+      raise NotImplementedError, "You need to call #define_package before" unless @current_root_package
+      raise NotImplementedError, "You need to call #derive_from before" unless @current_const
+
+      Protod::Rpc::Request.find_by(@current_const).push_procedure(*names, singleton: singleton)
+      Protod::Rpc::Response.find_by(@current_const).push_procedure(*names, singleton: singleton)
+    end
+
+    def register_interpreter_for(*const_names, with: nil, force: true, &body)
+      raise NotImplementedError, "You need to call #define_package before" unless @current_root_package
+
+      Protod::Interpreter.register_for(*const_names, with: with, force: force, &body)
+    end
+
+    def builders
+      @builders.values
+    end
+
+    def rbs_environment
+      @rbs_environment ||= RBS::EnvironmentLoader.new().tap do
+        @setup_rbs_environment_loader&.call(_1)
+      end.then do
+        RBS::Environment.from_loader(_1).resolve_type_names
+      end
+    end
+
+    def rbs_definition_builder
+      @rbs_definition_builder ||= RBS::DefinitionBuilder.new(env: rbs_environment)
+    end
+  end
+end

data/lib/protod/interpreter/active_record.rb

@@ -0,0 +1,161 @@
+# frozen_string_literal: true
+
+class Protod
+  class Interpreter
+    class ActiveRecord
+      def self.setup!
+        setup_active_model! if defined?(::ActiveModel::Attributes)
+        setup_active_record! if defined?(::ActiveRecord::Attributes)
+      end
+
+      private
+
+      def self.setup_active_model!
+        Interpreter.register_for('ActiveModel::Attributes', force: false, ignore: true) do
+          def proto_message
+            i = const.new
+
+            Protod::Proto::Message.new(
+              ident: proto_ident,
+              fields: const.attribute_names.map do |name|
+                Protod::Proto::Field.build_from(
+                  resolve_type_from(const.attribute_types.fetch(name)) || 'RBS::Types::Bases::Any',
+                  ident: name,
+                  required: false,
+                  optional: true,
+                  repeated: i.method(name).call.is_a?(::Array) ? true : false
+                )
+              end
+            )
+          end
+
+          def resolve_type_from(attribute_type)
+            case attribute_type
+            when ::ActiveModel::Type::Boolean
+              'RBS::Types::Bases::Bool'
+            when ::ActiveModel::Type::String, ::ActiveModel::Type::ImmutableString
+              'String'
+            when ::ActiveModel::Type::Decimal
+              'BigDecimal'
+            when ::ActiveModel::Type::Integer, ::ActiveModel::Type::BigInteger
+              'Integer'
+            when ::ActiveModel::Type::Float
+              'Numeric'
+            when ::ActiveModel::Type::Date
+              'Date'
+            when ::ActiveModel::Type::Time, ::ActiveModel::Type::DateTime
+              'Time'
+            when ::ActiveModel::Type::Binary
+              'Protod::Types::Binary'
+            end
+          end
+
+          def to_pb_from(rb)
+            acceptable_names = const.attribute_names.index_with(true)
+
+            attributes = proto_message.fields.filter_map do |f|
+              name = f.ident
+
+              next unless acceptable_names.key?(name)
+
+              [
+                name.to_sym,
+                if f.repeated
+                  rb.attributes.fetch(name).map { f.interpreter.to_pb_from(_1) }
+                else
+                  f.interpreter.to_pb_from(rb.attributes.fetch(name))
+                end
+              ]
+            end.to_h
+
+            pb_const.new(**attributes)
+          end
+
+          def to_rb_from(pb)
+            acceptable_names = const.attribute_names.index_with(true)
+
+            attributes = proto_message.fields.map do |f|
+              name = f.ident
+
+              next unless acceptable_names.key?(name)
+              next if Protod::Proto.omits_field?(pb, name)
+
+              [
+                name.to_sym,
+                if f.repeated
+                  pb.public_send(name)&.map { f.interpreter.to_rb_from(_1) }
+                else
+                  f.interpreter.to_rb_from(pb.public_send(name))
+                end
+              ]
+            end.compact.to_h
+
+            const.new(**attributes)
+          end
+        end
+      end
+
+      def self.setup_active_record!
+        Interpreter.register_for('ActiveRecord::Attributes', with: 'ActiveModel::Attributes', force: false, ignore: true) do
+          def resolve_type_from(attribute_type)
+            case attribute_type
+            when ::ActiveRecord::Type::Boolean
+              'RBS::Types::Bases::Bool'
+            when ::ActiveRecord::Type::String, ::ActiveRecord::Type::ImmutableString, ::ActiveRecord::Type::Text
+              'String'
+            when ::ActiveRecord::Type::Decimal, ::ActiveRecord::Type::DecimalWithoutScale
+              'BigDecimal'
+            when ::ActiveRecord::Type::UnsignedInteger
+              'Protod::Types::UnsignedInteger'
+            when ::ActiveRecord::Type::Integer, ::ActiveRecord::Type::BigInteger
+              'Integer'
+            when ::ActiveRecord::Type::Float
+              'Numeric'
+            when ::ActiveRecord::Type::Date
+              'Date'
+            when ::ActiveRecord::Type::Time, ::ActiveRecord::Type::DateTime
+              'Time'
+            when ::ActiveRecord::Type::Json
+              'Protod::Types::Json'
+            when ::ActiveRecord::Type::Binary
+              'Protod::Types::Binary'
+            else
+              case attribute_type.type
+              when :boolean
+                'RBS::Types::Bases::Bool'
+              when :string
+                'String'
+              when :integer
+                'Integer'
+              when :datetime
+                'Time'
+              else
+                super
+              end
+            end
+          end
+        end
+
+        Interpreter.register_for('ActiveRecord::Relation', force: false, ignore: true) do
+          def proto_ident
+            Protod::Proto::Ident.build_from('ActiveRecord::Relation')
+          end
+
+          def proto_message
+            Protod::Proto::Message.new(
+              ident: proto_ident
+            )
+          end
+
+          def to_pb_from(rb)
+            pb_const.new
+          end
+
+          def to_rb_from(pb)
+            pb
+          end
+        end
+      end
+    end
+  end
+end