Merge pull request #2 from neo4j-documentation/issue-1-add-slug-attribute

resolves #1 add page-slug attribute on the training navigation items

Author: adam-cowley

lib/generate-site.js

@@ -12,6 +12,7 @@ const produceRedirects = require('@antora/redirect-producer')
 const publishSite = require('@antora/site-publisher')
 const { resolveAsciiDocConfig } = require('@antora/asciidoc-loader')
 const knowledgeBase = require('./knowledge-base')
+const training = require('./training')
 
 async function generateSite (args, env) {
   const playbook = buildPlaybook(args, env)
@@ -24,6 +25,7 @@ async function generateSite (args, env) {
   knowledgeBase.addCategoryPages(pages, contentCatalog)
   knowledgeBase.addTagPages(pages, contentCatalog)
   const navigationCatalog = buildNavigation(contentCatalog, asciidocConfig)
+  training.attachNavigationSlug(contentCatalog, navigationCatalog)
   const composePage = createPageComposer(playbook, contentCatalog, uiCatalog, env)
   pages.forEach((page) => composePage(page, contentCatalog, navigationCatalog))
   const siteFiles = [...mapSite(playbook, pages), ...produceRedirects(playbook, contentCatalog)]

lib/training.js

@@ -0,0 +1,32 @@
+function attachNavigationSlug (contentCatalog, navigationCatalog) {
+  contentCatalog.getComponents()
+    .filter((component) => {
+      return component.latest &&
+        component.latest.asciidoc &&
+        component.latest.asciidoc.attributes &&
+        component.latest.asciidoc.attributes['page-component'] === 'training'
+    }).forEach((component) => {
+    const componentNavigation = navigationCatalog.getNavigation(component.name, component.latest.version)
+    if (componentNavigation && componentNavigation.length > 0) {
+      componentNavigation[0].items.forEach((item) => {
+        if (item.urlType === 'internal') {
+          const pages = contentCatalog.getPages((page) => page.pub && page.pub.url === item.url)
+          if (pages && pages.length > 0) {
+            const page = pages[0]
+            if (page.asciidoc && page.asciidoc.attributes && page.asciidoc.attributes['page-quiz'] === '') {
+              item['asciidoc'] = {
+                attributes: {
+                  'page-slug': page.asciidoc.attributes['page-slug']
+                }
+              }
+            }
+          }
+        }
+      })
+    }
+  })
+}
+
+module.exports = {
+  attachNavigationSlug: attachNavigationSlug
+}