diff --git a/index.js b/index.js index 341884e..9d5be9a 100644 --- a/index.js +++ b/index.js @@ -3,9 +3,11 @@ var Benchmark = require('./lib/benchmark.js'), DefaultReporter = require('./lib/defaultreporter.js'), fs = require('fs'), + path = require('path'), program = require('commander'), logger = require('./lib/logger'); + program .version('0.0.3') .usage('[options] ') @@ -48,10 +50,17 @@ if (!program.generator) { program.generator = __dirname + '/lib/generator.js'; } -if (program.generator.indexOf('/') !== 0) { - program.generator = process.cwd() + '/' + program.generator; +if(process.platform !== 'win32'){ + if (program.generator.indexOf('/') !== 0) { + program.generator = process.cwd() + '/' + program.generator; + } +} else { + if(!/^[\w]+\:/.test(program.generator)){ + program.generator = process.cwd() + path.sep + program.generator; + } } + if (!program.message) { program.message = 0; }