forked from artsy/benv
/
index.js
129 lines (115 loc) · 4.02 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
121
122
123
124
125
126
127
128
129
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
// @param {Object} options Further key/value pairs to be passed to the
// jsdom environment. This parameter is optional.
module.exports.setup = function(callback, options) {
if (typeof window != 'undefined') return callback && callback();
var 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();
}
};
if (options !== undefined) {
for(var key in options) {
env[key] = options[key];
}
}
jsdom.env(env);
}
// 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 === true) {
delete global.window;
domGlobals.forEach(function(varName) {
delete global[varName];
});
}
for(var key in globals) {
delete global[key];
}
if (typeof document != 'undefined') document.body.innerHTML = '';
}
// 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;
}