Render wiki entries using a collection of WikiPage and WikiDirectory objects

app/controllers/projects/wikis_controller.rb

@@ -8,7 +8,7 @@ class Projects::WikisController < Projects::ApplicationController
 
   def pages
     @wiki_pages = Kaminari.paginate_array(@project_wiki.pages).page(params[:page])
-    @wiki_directories = WikiPage.group_by_directory(@wiki_pages)
+    @wiki_entries = WikiPage.group_by_directory(@wiki_pages)
   end
 
   def show
@@ -117,7 +117,7 @@ class Projects::WikisController < Projects::ApplicationController
     # Call #wiki to make sure the Wiki Repo is initialized
     @project_wiki.wiki
 
-    @sidebar_wiki_directories = WikiPage.group_by_directory(@project_wiki.pages.first(15))
+    @sidebar_wiki_entries = WikiPage.group_by_directory(@project_wiki.pages.first(15))
   rescue ProjectWiki::CouldNotCreateWikiError
     flash[:notice] = "Could not create Wiki Repository at this time. Please try again later."
     redirect_to project_path(@project)

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

app/views/projects/wikis/_sidebar_wiki_pages.html.haml

-- 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

app/views/projects/wikis/_wiki_directory.html.haml

+%li
+  = wiki_directory.slug
+  %ul
+    = render wiki_directory.pages, context: context

app/views/projects/wikis/_wiki_pages.html.haml → app/views/projects/wikis/_wiki_page.html.haml

-- 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})

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'