-
Notifications
You must be signed in to change notification settings - Fork 0
/
glob-html.js
104 lines (86 loc) · 2.77 KB
/
glob-html.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
100
101
102
103
104
var cheerio = require('cheerio'),
path = require('path'),
_ = require('lodash'),
glob = require('glob');
//var isDir = function (path) {
// return grunt.util._.endsWith(path, '/');
//};
/**
* Expand element with glob to matched values.
* @param {object} $element Element with glob.
* @param {string} attr Resource attr. (src, href).
* @param {string} base Base directory to search in.
* @param {object} opts Options
* @return {Array<object>} matched values.
* */
var expandElement = function ($element, attr, base, opts) {
var path = $element.attr(attr),
expandedPaths = glob.sync(path, { cwd : base });
//grunt.log.writeln('Expand ' + path.cyan + ' to ' + expandedPaths.length.toString().cyan + ' files');
return expandedPaths.map(function (path) {
if (typeof opts.basePath === 'string' && path.indexOf(opts.basePath) === 0) {
path = path.substring(opts.basePath.length);
}
return $element.clone().attr(attr, path).toString();
});
};
/**
* Checks whether element should be handled and return appropriate attr in accordance with element type:
* link - href, script - src.
* @param {object} $element element to handle.
* @return {string} attr to handle. ('src', 'href', undefined)
* */
var getAttrToHandle = function ($element) {
var attr;
if ($element.is('script')) {
attr = 'src';
} else if ($element.is('link')) {
attr = 'href';
}
if (($element.attr(attr) || '').indexOf('*') !== -1) {
return attr;
}
};
/**
* Fetches element's indent.
* @param {object} $element element to handle.
* @return {string} indent;
* */
var fetchIndent = function ($element) {
var indent = '';
// Check whether prev element is whitespace.
if (/^\s+$/.test($element[0].prev.data)) {
indent = $element[0].prev.data;
// Cut line break.
indent = _.last(indent.split(/\n|\r/));
}
return '\n' + indent;
};
var processFile = function (content, base, opts) {
var $ = cheerio.load(content),
$elements = $('script,link'),
noElementsFound = true;
$elements.each(function (i, element) {
var $element = $(element), indent,
attr = getAttrToHandle($element);
if (!attr) {
return;
}
noElementsFound = false;
indent = fetchIndent($element);
// Expand initial script to matched ones.
expandElement($element, attr, base, opts).forEach(function (element, i) {
if (i === 0) {
$element.before(element);
} else {
$element.before(indent + element);
}
});
$element.remove();
});
if (noElementsFound) {
return content;
}
return $.html();
};
module.exports = processFile;