Merge pull request #20 from esantelises/add-submodules-parent

add parent to submodules

Author: toddjordan

index.js

@@ -11,6 +11,7 @@ let fetch = require('./lib/fetch')
 let readDocs = require('./lib/read-docs')
 let addSinceTags = require('./lib/add-since-tags')
 let addInheritedItems = require('./lib/add-inherited-items')
+let addSubModulesParent = require('./lib/add-sub-modules-parent')
 let putClassesInCouch = require('./lib/classes-in-couch')
 let createVersionIndex = require('./lib/create-version-index')
 let normalizeEmberDependencies = require('./lib/normalize-ember-dependencies')
@@ -35,6 +36,9 @@ function transformProjectFiles (projectName) {
   console.log('reading docs for ' + projectName)
   let promise = RSVP.resolve(readDocs(projectName))
     .then((stuff) => {
+      console.log('adding sub modules parent for ' + projectName)
+      return addSubModulesParent(stuff)
+    }).then((stuff) => {
       console.log('adding since tags for ' + projectName)
       return addSinceTags(stuff)
     }).then((stuff) => {

lib/add-sub-modules-parent.js

@@ -0,0 +1,18 @@
+'use strict'
+
+let _ = require('lodash')
+let RSVP = require('rsvp')
+
+module.exports = function addSubModulesParent (docSets) {
+  docSets.forEach(function (versionData) {
+    let modules = _.values(versionData.data.modules)
+
+    modules.forEach(function (mod) {
+      if (mod.is_submodule) {
+        mod.parent = mod.module
+      }
+    })
+  })
+
+  return RSVP.resolve(docSets)
+}

test/add-sub-modules-parent-test.js

@@ -0,0 +1,41 @@
+'use strict'
+
+let addSubModulesParent = require('../lib/add-sub-modules-parent')
+let assert = require('chai').assert
+let _ = require('lodash')
+
+describe('addSubModulesParent', function () {
+  beforeEach(function () {
+    this.yuiDocSets = _.range(3).map(i => {
+      return {
+        version: 'v1.0.' + i,
+        data: {
+          modules: [
+            {
+              is_submodule: 1,
+              module: 'foo'
+            },
+            {
+              is_submodule: 1,
+              module: 'foo'
+            },
+            {
+              is_submodule: 0,
+              module: 'bar'
+            }
+          ]
+        }
+      }
+    })
+    addSubModulesParent(this.yuiDocSets)
+  })
+
+  it('adds a parent attribute to sub modules', function () {
+    this.yuiDocSets.forEach(docSet => {
+      let subModules = _.filter(docSet.data.modules, (mod) => (mod.is_submodule))
+      subModules.forEach((moduleItem) => {
+        assert.equal(moduleItem.parent, 'foo')
+      })
+    })
+  })
+})