12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- var path = require('path')
- var fs = require('fs')
- var redeyed = require('..')
- var vm = require('vm')
- var samplePath = path.join(__dirname, 'sources', 'log.js')
- var origCode = fs.readFileSync(samplePath, 'utf-8')
- var kinds = [ 'silly', 'info', 'warn', 'error' ]
- function replaceConsole(s, info) {
- var code = info.code
- var idx = info.tokenIndex
- var tokens = info.tokens
- var next = tokens[idx + 1].value
- var kind = tokens[idx + 2].value
- var openParen = tokens[idx + 3].value
- var firstArgTkn = tokens[idx + 4]
- var argIdx = idx + 3
- var open
- var tkn
- if (kind === 'log') kind = 'silly'
- // not a console.xxx(...) statement? -> just return original
- if (next !== '.' || !~kinds.indexOf(kind) || openParen !== '(') return s
- // skip past arguments to console.xxx all args from ( to )
- open = 1
- while (open) {
- tkn = tokens[++argIdx]
- // count open parens vs. closed ones to handle things like console.log(new Error('..'));
- if (tkn.value === '(') open++
- if (tkn.value === ')') open--
- }
- // tkn now is the last closing paren
- var argsIncludingClosingParen = code.slice(firstArgTkn.range[0], tkn.range[1])
- var result = 'log.' + kind + '("main-logger", ' + argsIncludingClosingParen
- // tell redeyed to skip the entire console.xxx(..) statement since we are replacing it all
- return { replacement: result, skipPastToken: tkn }
- }
- function transformAndRun() {
- var config = {
- Identifier: { console: replaceConsole }
- }
- var code = redeyed(origCode, config).code
- var context = vm.createContext({ require: require })
- console.log('Original code:\n', origCode)
- console.log('\nlog calls replaced:\n', code)
- console.log('\nLets run it:')
- vm.runInContext(code, context, 'transformed-log.vm')
- }
- transformAndRun()
|