From c9d4a89f7c0ad2b3956efac4b4b03614c2ba743f Mon Sep 17 00:00:00 2001
From: Ufuk Kayserilioglu <ufuk@paralaus.com>
Date: Tue, 8 Mar 2022 00:33:39 +0200
Subject: [PATCH 1/2] Add test for loading custom compilers

---
 spec/tapioca/cli/dsl_spec.rb | 59 ++++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/spec/tapioca/cli/dsl_spec.rb b/spec/tapioca/cli/dsl_spec.rb
index 116621472..cbaf592d9 100644
--- a/spec/tapioca/cli/dsl_spec.rb
+++ b/spec/tapioca/cli/dsl_spec.rb
@@ -1279,6 +1279,65 @@ class Post
           @project.remove("sorbet/rbi/dsl")
         end
       end
+
+      describe "custom compilers" do
+        it "must be able to load custom compilers without a full require" do
+          @project.bundle_install
+
+          @project.write("lib/post.rb", <<~RB)
+            class Post
+            end
+          RB
+
+          @project.write("lib/compilers/post_compiler.rb", <<~RB)
+            require "post"
+            require "tapioca/dsl"
+
+            class PostCompiler < Tapioca::Dsl::Compiler
+              extend T::Sig
+
+              ConstantType = type_member(fixed: T.class_of(::Post))
+
+              sig { override.void }
+              def decorate
+                root.create_path(constant) do |klass|
+                  klass.create_module("GeneratedBar")
+                  klass.create_include("GeneratedBar")
+                end
+              end
+
+              sig { override.returns(T::Enumerable[Module]) }
+              def self.gather_constants
+                [::Post]
+              end
+            end
+          RB
+
+          @project.write("bin/generate", <<~RB)
+            require_relative "../config/environment"
+
+            file = RBI::File.new(strictness: "strong")
+            pipeline = Tapioca::Dsl::Pipeline.new(requested_constants: [])
+            PostCompiler.new(pipeline, file.root, Post).decorate
+            puts Tapioca::DEFAULT_RBI_FORMATTER.print_file(file)
+          RB
+
+          result = @project.bundle_exec("ruby bin/generate")
+
+          assert_equal(<<~OUT, result.out)
+            # typed: strong
+
+            class Post
+              include GeneratedBar
+
+              module GeneratedBar; end
+            end
+          OUT
+
+          assert_empty_stderr(result)
+          assert_success_status(result)
+        end
+      end
     end
   end
 end

From a4a6cde19201378cb099b028012941c158456c71 Mon Sep 17 00:00:00 2001
From: Ufuk Kayserilioglu <ufuk@paralaus.com>
Date: Tue, 8 Mar 2022 00:34:06 +0200
Subject: [PATCH 2/2] Add `tapioca/dsl` entrypoint for custom DSL compilers

---
 lib/tapioca/dsl.rb | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 lib/tapioca/dsl.rb

diff --git a/lib/tapioca/dsl.rb b/lib/tapioca/dsl.rb
new file mode 100644
index 000000000..0124bd350
--- /dev/null
+++ b/lib/tapioca/dsl.rb
@@ -0,0 +1,6 @@
+# typed: true
+# frozen_string_literal: true
+
+require "tapioca"
+require "tapioca/runtime/reflection"
+require "tapioca/dsl/compiler"