class RuboCop::Cop::Style::ModuleFunction

def autocorrect(node)
  lambda do |corrector|
    if extend_self_node?(node)
      corrector.replace(node.source_range, 'module_function')
    else
      corrector.replace(node.source_range, 'extend self')
    end
  end
end