Render wiki entries using a collection of WikiPage and WikiDirectory objects

5 files changed, 11 insertions, 22 deletions

diff --git a/app/views/projects/wikis/_sidebar.html.haml b/app/views/projects/wikis/_sidebar.html.haml
index b7464180a0c..e3fddfba689 100644
--- a/app/views/projects/wikis/_sidebar.html.haml
+++ b/app/views/projects/wikis/_sidebar.html.haml
@@ -12,15 +12,7 @@
   .blocks-container
     .block.block-first
       %ul.wiki-pages
-        - @sidebar_wiki_directories.each do |wiki_directory, wiki_pages|
-          - if wiki_directory == '/'
-            = render 'sidebar_wiki_pages', wiki_pages: wiki_pages
-          - else
-            %li
-              = wiki_directory
-              %ul
-                = render 'sidebar_wiki_pages', wiki_pages: wiki_pages
-
+        = render @sidebar_wiki_entries, context: 'sidebar'
 
     .block
       = link_to namespace_project_wikis_pages_path(@project.namespace, @project), class: 'btn btn-block' do
diff --git a/app/views/projects/wikis/_sidebar_wiki_pages.html.haml b/app/views/projects/wikis/_sidebar_wiki_pages.html.haml
deleted file mode 100644
index 65453a384d2..00000000000
--- a/app/views/projects/wikis/_sidebar_wiki_pages.html.haml
+++ /dev/null
@@ -1,4 +0,0 @@
-- wiki_pages.each do |wiki_page|
-  %li{ class: params[:id] == wiki_page.slug ? 'active' : '' }
-    = link_to namespace_project_wiki_path(@project.namespace, @project, wiki_page) do
-      = wiki_page.title.capitalize
diff --git a/app/views/projects/wikis/_wiki_directory.html.haml b/app/views/projects/wikis/_wiki_directory.html.haml
new file mode 100644
index 00000000000..0e5f32ed859
--- /dev/null
+++ b/app/views/projects/wikis/_wiki_directory.html.haml
@@ -0,0 +1,4 @@
+%li
+  = wiki_directory.slug
+  %ul
+    = render wiki_directory.pages, context: context
diff --git a/app/views/projects/wikis/_wiki_pages.html.haml b/app/views/projects/wikis/_wiki_page.html.haml
index ac98599d96b..cea27388a0d 100644
--- a/app/views/projects/wikis/_wiki_pages.html.haml
+++ b/app/views/projects/wikis/_wiki_page.html.haml
@@ -1,4 +1,8 @@
-- wiki_pages.each do |wiki_page|
+- if context == 'sidebar'
+  %li{ class: params[:id] == wiki_page.slug ? 'active' : '' }
+    = link_to namespace_project_wiki_path(@project.namespace, @project, wiki_page) do
+      = wiki_page.title.capitalize
+- else
   %li
     = link_to wiki_page.title, namespace_project_wiki_path(@project.namespace, @project, wiki_page)
     %small (#{wiki_page.format})
diff --git a/app/views/projects/wikis/pages.html.haml b/app/views/projects/wikis/pages.html.haml
index 28dd81e5c3f..5fba2b1a5ae 100644
--- a/app/views/projects/wikis/pages.html.haml
+++ b/app/views/projects/wikis/pages.html.haml
@@ -14,13 +14,6 @@
         Clone repository
 
   %ul.wiki-pages-list.content-list
-    - @wiki_directories.each do |wiki_directory, wiki_pages|
-      - if wiki_directory == '/'
-        = render 'wiki_pages', wiki_pages: wiki_pages
-      - else
-        %li
-          = wiki_directory
-          %ul
-            = render 'wiki_pages', wiki_pages: wiki_pages
+    = render @wiki_entries, context: 'pages'
 
   = paginate @wiki_pages, theme: 'gitlab'