Contents

# typed: strict
class ActiveFlagPlugin < SorbetRails::ModelPlugins::Base
  sig { params(root: Parlour::RbiGenerator::Namespace).void.override }
  def generate(root)
    # method added here: https://github.com/kenn/active_flag/blob/master/lib/active_flag.rb#L13
    return unless model_class.respond_to?(:active_flags)

    module_name = self.model_module_name("GeneratedActiveFlagMethods")
    module_rbi = root.create_module(module_name)
    module_rbi.create_extend("T::Sig")

    model_class_rbi = root.create_class(self.model_class_name)
    model_class_rbi.create_include(module_name)

    module_rbi.create_method(
      "active_flags",
      return_type: "T::Hash[Symbol, ActiveFlag::Definition]",
      class_method: true
    )

    active_flag_keys(model_class).each do |flag|
      module_rbi.create_method(
        flag.to_s,
        return_type: "::ActiveFlag::Value"
      )

      module_rbi.create_method(
        "#{flag}=",
        parameters: [
          Parameter.new("value", type: "T::Array[Symbol]")
        ],
        return_type: nil
      )

      module_rbi.create_method(
        flag.to_s.pluralize,
        return_type: "::ActiveFlag::Definition",
        class_method: true
      )
    end
  end

  sig { params(model_class: T.class_of(ActiveRecord::Base)).returns(T::Array[Symbol]) }
  def active_flag_keys(model_class)
    T.unsafe(model_class).active_flags.keys
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
sorbet-rails-0.6.3	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb
sorbet-rails-0.6.2	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb
sorbet-rails-0.6.1	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb
sorbet-rails-0.6.0	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb
sorbet-rails-0.5.9.1	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb
sorbet-rails-0.5.9	lib/sorbet-rails/gem_plugins/active_flag_plugin.rb