summaryrefslogtreecommitdiff
path: root/app/models
diff options
context:
space:
mode:
Diffstat (limited to 'app/models')
-rw-r--r--app/models/tree.rb5
1 files changed, 3 insertions, 2 deletions
diff --git a/app/models/tree.rb b/app/models/tree.rb
index 5bb1f3cdaf4..07c9a825e24 100644
--- a/app/models/tree.rb
+++ b/app/models/tree.rb
@@ -1,4 +1,6 @@
class Tree
+ include Gitlab::MarkdownHelper
+
attr_accessor :entries, :readme, :contribution_guide
def initialize(repository, sha, path = '/')
@@ -13,8 +15,7 @@ class Tree
# by markup renderer.
if available_readmes.length > 1
supported_readmes = available_readmes.select do |readme|
- Gitlab::MarkdownHelper.gitlab_markdown?(readme.name) ||
- Gitlab::MarkdownHelper.markup?(readme.name)
+ gitlab_markdown?(readme.name) || markup?(readme.name)
end
# Take the first supported readme, or the first available readme, if we