Check for wiki.

Marin Jankovski
1 parent 817399e3
Showing 2 changed files with 13 additions and 3 deletions Show diff stats
app/helpers/gitlab_markdown_helper.rb
lib/redcarpet/render/gitlab_html.rb
@@ -59,10 +59,16 @@ module GitlabMarkdownHelper
     end
   end
-  def create_relative_links(text, project_path_with_namespace, ref)
+  def create_relative_links(text, project_path_with_namespace, ref, wiki = false)
     links = text.split("\n").map { |a| a.scan(/\]\(([^(]+)\)/) }.reject{|b| b.empty? }.flatten.reject{|c| c.include?("http" || "www")}
     links.each do |string|
-      text.gsub!(string, "/#{project_path_with_namespace}/blob/#{ref}/#{string}")
+      new_link = [
+        project_path_with_namespace,
+        wiki ? "wiki":"blob",
+        ref,
+        string
+      ].compact.join("/")
+      text.gsub!(string, "/#{new_link}")
     end
     text
   end
@@ -34,10 +34,14 @@ class Redcarpet::Render::GitlabHTML &lt; Redcarpet::Render::HTML
   end
   def preprocess(full_document)
-    h.create_relative_links(full_document, @project.path_with_namespace, @ref)
+    h.create_relative_links(full_document, @project.path_with_namespace, @ref, is_wiki?)
   end
   def postprocess(full_document)
     h.gfm(full_document)
   end
+
+  def is_wiki?
+    @template.instance_variable_get("@wiki")
+  end
 end