-
Notifications
You must be signed in to change notification settings - Fork 7
/
publish.js
99 lines (82 loc) · 2.09 KB
/
publish.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/**
* jsdoc2diagram
*/
'use strict';
var fs = require('fs');
var path = require('path');
var template = require('lodash.template');
/**
* Create the tree structure need for the d3 tree diagram
* - original code from http://bl.ocks.org/
*
* @param {Array.<Object>} data
* @return {Array}
*/
var createTreeStructure = function(data) {
var dataMap = data.reduce(function(map, node) {
map[node.name] = node;
return map;
}, {});
// create the tree array
var treeData = [];
data.forEach(function(node) {
// add to parent
var parent = dataMap[node.parent];
if (parent) {
// create child array if it doesn't exist
(parent.children || (parent.children = []))
// add node to child array
.push(node);
} else {
// parent is null or missing
treeData.push(node);
}
});
return treeData;
};
/**
@param {TAFFY} data
@param {object} opts
*/
exports.publish = function(data, opt) {
data({undocumented: true}).remove();
data({kind: 'package'}).remove();
//an array of Doclet objects
var docs = data().get();
/**
*
* @param {string} text
* @return {string}
*/
var cleanName = function(text) {
return (text || '').trim().replace(/(\#|\/|\~|\-|\:)/g, '.');
};
var dataSet = [];
docs.forEach(function(doc) {
var path = cleanName(doc.longname);
var memberOf = cleanName(doc.memberof);
var name = path.split('.').pop();
dataSet.push({
name: path,
shortName: name,
parent: memberOf,
type: doc.kind || ''
});
});
var mainObj = {
shortName: 'root',
type: 'root',
children: createTreeStructure(dataSet)
};
var indexContent = fs.readFileSync(__dirname + '/tpl/index.tpl', 'utf-8');
var mainJsContent = fs.readFileSync(__dirname + '/tpl/index.js', 'utf-8');
var dataHtmlFile = path.join(opt.destination, '/diagram.html');
var html = template(indexContent)({
mainJS: mainJsContent,
treeData: JSON.stringify(mainObj)
});
if (!fs.existsSync(opt.destination)) {
fs.mkdirSync(opt.destination);
}
fs.writeFileSync(dataHtmlFile, html, 'utf-8');
};