class RuboCop::Cop::Migrations::TruncateComponent

def autocorrect(node)
  return if hash_with_inline_value?(node.arguments.first)
  lambda do |corrector|
    if node.arguments.first.nil? == false
      corrector.replace(node.children.first, "Primer::Beta::Truncate")
      corrector.insert_after(node.arguments.first, ", tag: :div") unless truncate_with_tag?(node.arguments.first)
    else
      corrector.replace(node.loc.expression, "Primer::Beta::Truncate.new(tag: :div)")
    end
  end
end

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