class ActiveFlagPlugin

typed: strict

Public Instance Methods

active_flag_keys(model_class) click to toggle source
# File lib/sorbet-rails/gem_plugins/active_flag_plugin.rb, line 43
def active_flag_keys(model_class)
  T.unsafe(model_class).active_flags.keys
end
generate(root) click to toggle source
# File lib/sorbet-rails/gem_plugins/active_flag_plugin.rb, line 4
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)

  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