summaryrefslogtreecommitdiffstats
path: root/lib/generate/site/search_indexer.js
blob: b239b8b6f8f6d99ff32ead0c1181bf1ceb643e3c (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
65
66
67
68
69
70
71
var Q = require("q");
var _ = require("lodash");

var lunr = require('lunr');
var marked = require('marked');
var textRenderer = require('marked-text-renderer');


function Indexer() {
    if(!(this instanceof Indexer)) {
        return new Indexer();
    }

    _.bindAll(this);

    // Setup lunr index
    this.idx = lunr(function () {
        this.ref('url');

        this.field('title', { boost: 10 });
        this.field('body');
    });

    this.renderer = textRenderer();
}

Indexer.prototype.text = function(nodes) {
    // Copy section
    var section = _.toArray(nodes);

    // marked's Render expects this, we don't use it yet
    section.links = {};

    var options = _.extend({}, marked.defaults, {
        renderer: this.renderer
    });

    return marked.parser(section, options);
};

Indexer.prototype.addSection = function(path, section) {
    var url = [path, section.id].join('#');

    var title = this.text(
        _.filter(section, {'type': 'heading'})
    );

    var body = this.text(
        _.omit(section, {'type': 'heading'})
    );

    // Add to lunr index
    this.idx.add({
        url: url,
        title: title,
        body: body,
    });
};

Indexer.prototype.add = function(lexedPage, url) {
    var sections = lexedPage;

    _.map(sections, _.partial(this.addSection, url));
};

Indexer.prototype.dump = function() {
    return JSON.stringify(this.idx);
};

// Exports
module.exports = Indexer;