class RuboCop::Cop::Migrations::IconButtonComponent

def autocorrect(node)
  return if hash_with_box_value?(node.arguments.first)
  lambda do |corrector|
    corrector.replace(icon_button(node), "Primer::Beta::IconButton")
  end
end

def on_send(node)
  return unless icon_button(node)
  add_offense(node, message: INVALID_MESSAGE)
end