Use constants in Gitlab::MarkupHelper

This commit is contained in:
Douwe Maan 2017-04-20 00:39:29 +00:00 committed by Robert Speicher
parent ceb1ebc9d9
commit 3509534a03

View file

@ -1,6 +1,11 @@
module Gitlab module Gitlab
module MarkupHelper module MarkupHelper
module_function extend self
MARKDOWN_EXTENSIONS = %w(mdown mkd mkdn md markdown).freeze
ASCIIDOC_EXTENSIONS = %w(adoc ad asciidoc).freeze
OTHER_EXTENSIONS = %w(textile rdoc org creole wiki mediawiki rst).freeze
EXTENSIONS = MARKDOWN_EXTENSIONS + ASCIIDOC_EXTENSIONS + OTHER_EXTENSIONS
# Public: Determines if a given filename is compatible with GitHub::Markup. # Public: Determines if a given filename is compatible with GitHub::Markup.
# #
@ -8,10 +13,7 @@ module Gitlab
# #
# Returns boolean # Returns boolean
def markup?(filename) def markup?(filename)
gitlab_markdown?(filename) || EXTENSIONS.include?(extension(filename))
asciidoc?(filename) ||
filename.downcase.end_with?(*%w(.textile .rdoc .org .creole .wiki
.mediawiki .rst))
end end
# Public: Determines if a given filename is compatible with # Public: Determines if a given filename is compatible with
@ -21,7 +23,7 @@ module Gitlab
# #
# Returns boolean # Returns boolean
def gitlab_markdown?(filename) def gitlab_markdown?(filename)
filename.downcase.end_with?(*%w(.mdown .mkd .mkdn .md .markdown)) MARKDOWN_EXTENSIONS.include?(extension(filename))
end end
# Public: Determines if the given filename has AsciiDoc extension. # Public: Determines if the given filename has AsciiDoc extension.
@ -30,7 +32,7 @@ module Gitlab
# #
# Returns boolean # Returns boolean
def asciidoc?(filename) def asciidoc?(filename)
filename.downcase.end_with?(*%w(.adoc .ad .asciidoc)) ASCIIDOC_EXTENSIONS.include?(extension(filename))
end end
# Public: Determines if the given filename is plain text. # Public: Determines if the given filename is plain text.
@ -39,12 +41,17 @@ module Gitlab
# #
# Returns boolean # Returns boolean
def plain?(filename) def plain?(filename)
filename.downcase.end_with?('.txt') || extension(filename) == 'txt' || filename.casecmp('readme').zero?
filename.casecmp('readme').zero?
end end
def previewable?(filename) def previewable?(filename)
markup?(filename) markup?(filename)
end end
private
def extension(filename)
File.extname(filename).downcase.delete('.')
end
end end
end end