带参数或stdin的node.js脚本

时间:2013-01-05 21:43:52

标签: node.js asynchronous

程序必须获得2个变量。例如,它必须是登录名和密码。 脚本可以从命令行接收它(登录或使用密码登录)。 如果没有一个命令行参数未填充,则用户必须使用stdin输入它。 如果用户只输入了一个agrument - 我需要问用户第二个。 如果用户输入了两个参数 - 无所事事(执行一些函数)。

问题是:如何在node.js(async)上正确构建程序。

关于某些事件或什么?

我写了这段代码,我怎么能改进这个?

if(process.getuid() != 0) {
    console.log('Error!');
    console.log('Required root privileges');
    process.exit(1);
}

var getUser = function(user, callback) {
    if( user == undefined ) {
        ask('Enter user', /.+/, function(user) {
            callback(user, process.argv[3]);
        });
    } else {
        callback(user, process.argv[3]);
    }
}

var getPassword = function(user, password) {
    if( password == undefined ) {
        ask('Enter password', /.+/, function(password) {
            addUser(user, password);
        });
    } else {
        addUser(user, password);
    }
}

function addUser(user, password) {
    console.log('Adding user:');
    console.log(user);
    console.log(password);

    process.exit(0);
}

var user = getUser(process.argv[2], getPassword);

function ask(question, format, callback) {
    var stdin = process.stdin, stdout = process.stdout;

    stdin.resume();
    stdout.write(question + ": ");

    stdin.once('data', function(data) {
        data = data.toString().trim();

        if (format.test(data)) {
            callback(data);
        } else {
            stdout.write("It should match: "+ format +"\n");
            ask(question, format, callback);
        }
    });
}

1 个答案:

答案 0 :(得分:0)

看看这个:

How do I get started with Node.js

有些链接会有你的答案,或者会让你接受它。