shale-builder 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b0eb2f219fcd64c0f89f347f70e892e96f7b18490450a42557a39525e14fabe
-  data.tar.gz: c1f7b8a45efd094bc4f43028e56eae5d0da322357bbf0d6c1e882021a8da92e6
+  metadata.gz: bf1a615a9515f8312498f25d2c15163b42c472cde871ddb73fccabe9b0135854
+  data.tar.gz: b014989e7ece28fe8f88a9b49e6e543c0606e382927b8ec2f86a9f9ddad00210
 SHA512:
-  metadata.gz: c02cb9e08061c9d978f2e64be2ede31a3a874cc4b5f9e420f00bfe39771e69d48930fd6c063d7a8d6d066a8a3bfcd3faf7c2e58983d081d1ee852c8a2d59c377
-  data.tar.gz: 69bd7e9e44ae49c0f9bc2a7c19b5b7472d28a56c841b215b8a4be2bbd49dc42f99b3b43bd2d782a87b86a93d1534957af4356184a9e50926ee423b8277ef4bac
+  metadata.gz: 36d8c81ec9eae051dd80ce8cdcd342fa92eac54962834212e22bda6c9dcef9c0068812dad6e50590a41ab3d135fce072e5a9bfd8900c4c97270ce2d268d76dae
+  data.tar.gz: 7ac199ea4dcdaeba2be4205414d7bb9d2612fe3113bb99b39276107815da02d4db0ff8e98b4c1e0886d0dfb8bc90728adb8137703b8d62eb178ff299ff252ad5

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## [0.1.4] - 2024-05-13
+
+- Add a tapioca compiler for shale and shale builder
+
 ## [0.1.3] - 2023-11-23
 
 - Change shale version dependency from `< 1.0` to `< 2.0`

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    shale-builder (0.1.3)
+    shale-builder (0.1.4)
       shale (< 2.0)
 
 GEM

data/lib/shale/builder/version.rb

@@ -3,6 +3,6 @@
 module Shale
   module Builder
     # @return [String]
-    VERSION = '0.1.3'
+    VERSION = '0.1.4'
   end
 end

data/lib/tapioca/dsl/compilers/shale.rb

@@ -0,0 +1,103 @@
+# typed: true
+# frozen_string_literal: true
+
+module Tapioca
+  module Compilers
+    class Shale < Tapioca::Dsl::Compiler
+      ConstantType = type_member { { fixed: T.class_of(::Shale::Mapper) } }
+
+      class << self
+        sig { override.returns(T::Enumerable[Module]) }
+        def gather_constants
+          # Collect all the classes that inherit from Shale::Mapper
+          all_classes.select { |c| c < ::Shale::Mapper }
+        end
+      end
+
+      sig { override.void }
+      def decorate
+        # Create a RBI definition for each class that inherits from Shale::Mapper
+        root.create_path(constant) do |klass|
+          has_shale_builder = includes_shale_builder(constant)
+
+          # For each attribute defined in the class
+          constant.attributes.each_value do |attribute|
+            non_nilable_type, nilable_type = shale_type_to_sorbet_type(attribute)
+            type = nilable_type
+            if attribute.collection?
+              type = "T.nilable(T::Array[#{non_nilable_type}])"
+            end
+
+            if has_shale_builder && attribute.type < ::Shale::Mapper
+              sigs = T.let([], T::Array[RBI::Sig])
+              # simple getter
+              sigs << klass.create_sig(
+                parameters: { block: 'NilClass' },
+                return_type: type,
+              )
+              # getter with block
+              sigs << klass.create_sig(
+                parameters: { block: "T.proc.params(arg0: #{non_nilable_type}).void" },
+                return_type: non_nilable_type
+              )
+              klass.create_method_with_sigs(
+                attribute.name,
+                sigs: sigs,
+                parameters: [RBI::BlockParam.new('block')],
+              )
+            else
+              klass.create_method(attribute.name, return_type: type)
+            end
+
+            # setter
+            klass.create_method(
+              "#{attribute.name}=",
+              parameters: [create_param('value', type: type)],
+              return_type: type,
+            )
+          end
+        end
+
+      end
+
+      private
+
+      sig { params(klass: Class).returns(T.nilable(T::Boolean)) }
+      def includes_shale_builder(klass)
+        return false unless defined?(::Shale::Builder)
+
+        klass < ::Shale::Builder
+      end
+
+      SHALE_TYPES_MAP = T.let(
+        {
+          ::Shale::Type::Value    => Object,
+          ::Shale::Type::String   => String,
+          ::Shale::Type::Float    => Float,
+          ::Shale::Type::Integer  => Integer,
+          ::Shale::Type::Time     => Time,
+          ::Shale::Type::Date     => Date,
+        }.freeze,
+        T::Hash[Class, Class],
+      )
+
+      sig { params(attribute: ::Shale::Attribute).returns([String, String]) }
+      def shale_type_to_sorbet_type(attribute)
+        return_type = SHALE_TYPES_MAP[attribute.type]
+        return complex_shale_type_to_sorbet_type(attribute) unless return_type
+        return [T.must(return_type.name), T.must(return_type.name)] if attribute.collection? || attribute.default.is_a?(return_type)
+
+        [T.must(return_type.name), "T.nilable(#{return_type.name})"]
+      end
+
+      sig { params(attribute: ::Shale::Attribute).returns([String, String]) }
+      def complex_shale_type_to_sorbet_type(attribute)
+        return [T.cast(attribute.type.to_s, String), "T.nilable(#{attribute.type})"] unless attribute.type.respond_to?(:return_type)
+
+        return_type_string = attribute.type.return_type.to_s
+        [return_type_string, return_type_string]
+      end
+
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: shale-builder
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - Mateusz Drewniak
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-11-23 00:00:00.000000000 Z
+date: 2024-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: shale
@@ -42,6 +42,7 @@ files:
 - Rakefile
 - lib/shale/builder.rb
 - lib/shale/builder/version.rb
+- lib/tapioca/dsl/compilers/shale.rb
 - sig/shale/builder.rbs
 homepage: https://github.com/Verseth/ruby-shale-builder
 licenses: