summaryrefslogtreecommitdiffstats
path: root/lib/parse/navigation.js
blob: ae4eb9d2874d0e295fce14afc76dd6afa546fc25 (plain)
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
var _ = require('lodash');

// Cleans up an article/chapter object
// remove 'articles' attributes
function clean(obj) {
    return obj && _.omit(obj, ['articles']);
}

function flattenChapters(chapters) {
    return _.reduce(chapters, function(accu, chapter) {
        return accu.concat([clean(chapter)].concat(flattenChapters(chapter.articles)));
    }, []);
}

// Returns from a summary a map of
/*
    {
        "file/path.md": {
            prev: ...,
            next: ...,
        },
        ...
    }
*/
function navigation(summary, files) {
    // Support single files as well as list
    files = _.isArray(files) ? files : (_.isString(files) ? [files] : null);

    // List of all navNodes
    // Flatten chapters, then add in default README node if ndeeded etc ...
    var navNodes = flattenChapters(summary.chapters);
    var prevNodes = [null].concat(navNodes.slice(0, -1));
    var nextNodes = navNodes.slice(1).concat([null]);

    // Mapping of prev/next for a give path
    var mapping = _.chain(_.zip(navNodes, prevNodes, nextNodes))
    .map(function(nodes) {
        var current = nodes[0], prev = nodes[1], next = nodes[2];

        // Skip if no path
        if(!current.path) return null;

        return [current.path, {
            title: current.title,
            prev: prev,
            next: next,
            level: current.level,
        }];
    })
    .filter()
    .object()
    .value();

    // Filter for only files we want
    if(files) {
        return _.pick(mapping, files);
    }

    return mapping;
}


// Exports
module.exports = navigation;