Fix two-factor authentication problem, close node-gh#240 node-gh#120

henvic · Jan 5, 2014 · ee8103b · ee8103b
1 parent 479683d
commit ee8103b
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/lib/cmds/user.js b/lib/cmds/user.js
@@ -112,6 +112,33 @@ User.createAuthorization = function(opt_callback) {
             });
 
             base.github.authorization.create(payload, function(err, res) {
+                //verify if a two-factor authentication code was missing
+                if (err && err.message && err.message.indexOf('OTP') > 0) {
+                    inquirer.prompt(
+                        [
+                            {
+                                type: 'input',
+                                message: 'Enter your two-factor code',
+                                name: 'otp'
+                            }
+                        ], function(factor) {
+                            payload.headers['X-GitHub-OTP'] = factor.otp;
+
+                            base.github.authorization.create(payload, function(err, res) {
+                                logger.defaultCallback(err, null, 'Authentication succeed.');
+
+                                if (res.token) {
+                                    base.writeGlobalConfigCredentials(answers.user, res.token);
+
+                                    User.authorize();
+                                }
+
+                                opt_callback && opt_callback(err);
+                            });
+                        });
+                    return;
+                }
+
                 logger.defaultCallback(err, null, 'Authentication succeed.');
 
                 if (res.token) {