Merge pull request #16899 from Homebrew/tapioca-compiler-for-tty-rbi

Convert the `utils/tty` RBI generator to a Tapioca compiler

Library/Homebrew/dev-cmd/typecheck.rb

@@ -51,7 +51,6 @@ def run
  tapioca_args = args.update_all? ? ["--all"] : []
 
  ohai "Updating homegrown RBI files..."
- safe_system "bundle", "exec", "ruby", "sorbet/custom_generators/tty.rb"
  safe_system "bundle", "exec", "ruby", "sorbet/custom_generators/env_config.rb"
 
  ohai "Updating Tapioca RBI files..."

Library/Homebrew/sorbet/tapioca/compilers/tty.rb

@@ -0,0 +1,30 @@
+# typed: strict
+# frozen_string_literal: true
+
+require_relative "../../../global"
+require "utils/tty"
+
+module Tapioca
+ module Compilers
+ class Tty < Tapioca::Dsl::Compiler
+ # FIXME: Enable cop again when https://github.com/sorbet/sorbet/issues/3532 is fixed.
+ # rubocop:disable Style/MutableConstant
+ ConstantType = type_member { { fixed: Module } }
+ # rubocop:enable Style/MutableConstant
+
+ sig { override.returns(T::Enumerable[Module]) }
+ def self.gather_constants = [::Tty]
+
+ sig { override.void }
+ def decorate
+ root.create_module(T.must(constant.name)) do |mod|
+ dynamic_methods = ::Tty::COLOR_CODES.keys + ::Tty::STYLE_CODES.keys + ::Tty::SPECIAL_CODES.keys
+
+ dynamic_methods.each do |method|
+ mod.create_method(method.to_s, return_type: "String", class_method: true)
+ end
+ end
+ end
+ end
+ end
+end