forked from artsy/benv
-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
120 lines (106 loc) · 3.71 KB
/
index.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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
var jsdom = require('jsdom');
var rewire = require('rewire');
var path = require('path');
var fs = require('fs');
var domGlobals = [
'navigator',
'document',
'location',
'getComputedStyle',
'btoa'
];
var globals = {};
// Exposes a stubbed browser API and benv.globals into the node.js global namespace
// so the current process can act like a browser environment.
//
// @param {Function} callback
module.exports.setup = function(callback) {
if (typeof window != 'undefined') return callback && callback();
jsdom.env({
html: "<html><body></body></html>",
done: function(errs, w) {
global.window = w;
domGlobals.forEach(function(varName) {
global[varName] = w[varName] || function(){};
});
if (callback) callback();
}
})
}
// Pass in common client-side dependencies and expose them globally.
//
// @param {Object} globals
module.exports.expose = function(_globals) {
for(var key in _globals) {
global[key] = globals[key] = _globals[key];
}
}
// Deletes the stubbed browser API, benv.globals, and cleans things up so other
// tests can run without being harmed.
module.exports.teardown = function(clearDOM) {
if (clearDOM !== false) {
delete global.window;
domGlobals.forEach(function(varName) {
delete global[varName];
});
}
for(var key in globals) {
delete global[key];
}
}
// Require non-commonjs modules by specifying their global variable.
//
// @param {String} filename Path to non-commonjs file
// @param {String} globalVarName Exposed global like Zepto or GMaps
module.exports.require = function(filename, globalVarName) {
var fullPath = path.resolve(path.dirname(module.parent.filename), filename);
if (!fs.existsSync(fullPath)) fullPath = require.resolve(filename);
var mod = rewire(fullPath);
var w = mod.__get__('window');
return globalVarName ? (w[globalVarName] || mod.__get__(globalVarName)) : mod;
}
// Renders a server-side template into a fake browser's body.
// Will strip out all script tag first to avoid jsdom trying to run scripts.
//
// @param {String} filename
// @param {Object} data data passed into template like jade locals
module.exports.render = function(filename, data, callback) {
if (!window) throw Error('You must run benv.setup first.');
if (!filename.match('.jade')) throw Error('Could not identify template type');
var fullPath = path.resolve(path.dirname(module.parent.filename), filename);
var html = require('jade').compile(
fs.readFileSync(fullPath),
{ filename: fullPath }
)(data);
jsdom.env(html, function(err, w) {
var scriptEls = w.document.getElementsByTagName('script');
Array.prototype.forEach.call(scriptEls, function(el) {
el.parentNode.removeChild(el);
});
var bodyHtml = w.document.getElementsByTagName('body')[0].innerHTML;
document.getElementsByTagName('body')[0].innerHTML = bodyHtml;
if (callback) callback();
});
}
// Rewires jadeify templates to work in node again.
//
// @param {String} filename
// @param {Array} varNames Strings of template variable names
module.exports.requireWithJadeify = function(filename, varNames) {
var fullPath = path.resolve(path.dirname(module.parent.filename), filename);
var mod = rewire(fullPath);
var dir = path.dirname(mod.__get__('module').filename);
varNames.forEach(function(varName) {
var section = mod.__get__(varName).toString()
.match(/require\('(.*).jade'\)/);
if(!section) section = mod.__get__(varName).toString()
.match(/require\("(.*).jade"\)/);
var tmplFilename = section[1] + '.jade'
tmplFilename = path.resolve(dir, tmplFilename);
mod.__set__(varName, require('jade').compile(
fs.readFileSync(tmplFilename),
{ filename: tmplFilename }
));
});
return mod;
}