token authorization

.gitignore

@@ -1,4 +1,5 @@
 .pitconnect.txt
+.pituser.txt
 
 # Logs
 logs

package.json

@@ -31,6 +31,8 @@
   },
   "homepage": "https://github.com/mozilla/snakepit-client#readme",
   "dependencies": {
-    "commander": "^2.14.1"
+    "commander": "^2.14.1",
+    "password-prompt": "^1.0.3",
+    "readline-sync": "^1.4.9"
   }
 }

src/pit.js

@@ -4,14 +4,16 @@ const os = require('os')
 const path = require('path')
 const program = require('commander')
 const request = require('request')
+const prompt = require('password-prompt')
+const readlineSync = require('readline-sync')
 
 
-function runCommand(verb, resource, content, params) {
+function runCommand(verb, resource, content, callback, params) {
     var connectFile = '.pitconnect.txt'
     if(!fs.existsSync(connectFile)) {
         connectFile = path.join(os.homedir(), connectFile)
         if(!fs.existsSync(connectFile)) {
-            console.error('Unable to find connectivity info about your pit.')
+            console.error('Unable to find connectivity info about your pvar it.')
             console.error('If you know your pit\'s URL, use "pit connect <URL>" to configure the connection.')
             console.error('If your pit admin provided a ".pitconnect.txt" file, place it either in your home directory (as default pit) or the (overruling) project root.')
             process.exit(1)
@@ -24,15 +26,94 @@ function runCommand(verb, resource, content, params) {
     if (connectContent.length > 0) {
         agentOptions = { ca: connectContent.join('\n') }
     }
-    var resourceUrl = pitUrl + '/' + resource
-    request[verb]({
-        url: resourceUrl, agentOptions: agentOptions,
-        json: true
-    }, function(error, response, body) {
-        console.log(response.statusCode)
-        console.log(response.headers['content-type'])
-        console.log(body)
-    })
+
+    var userFile = '.pituser.txt'
+    var username
+    var token = ''
+
+    function sendRequest(verb, resource, content, callback, params) {
+        var resourceUrl = pitUrl + '/' + resource
+        request[verb]({
+            url: resourceUrl,
+            agentOptions: agentOptions,
+            headers: { 'X-Auth-Token': token },
+            json: !!content,
+            body: content
+        }, callback)
+    }
+
+    function authenticate(username, password, callback) {
+        sendRequest('post', 'users/' + username + '/authenticate', { password: password }, function(error, response, body) {
+            if (response.statusCode == 200) {
+                token = body.token
+                fs.writeFile(userFile, username + '\n' + token, function(err) {
+                    if(err) {
+                        console.error('Unable to store user info: ' + err)
+                        process.exit(1)
+                    } else {
+                        callback()
+                    }
+                })
+            } else {
+                console.error('Unable to authenticate.')
+                process.exit(1)
+            }
+        })
+    }
+
+    function loadUser() {
+        var userContent = fs.readFileSync(userFile, 'utf-8').split('\n')
+        username = userContent[0]
+        token = userContent[1]
+    }
+
+    function sendCommand() {
+        sendRequest(verb, resource, content, callback, params)
+    }
+
+    if(!fs.existsSync(userFile)) {
+        userFile = path.join(os.homedir(), userFile)
+        if(!fs.existsSync(userFile)) {
+            userFile = '.pituser.txt'
+            console.log('No user info found. Seems like a new user or first time login from this machine.')
+            username = readlineSync.question('Please enter an existing or new username: ')
+            var userPath = 'users/' + username
+            sendRequest('get', userPath + '/exists', undefined, function(error, response, body) {
+                if (response.statusCode == 200) {
+                    console.log('The user already exists.')
+                    var password = readlineSync.question('Please enter password (or Ctrl-C to abort): ', { hideEchoBack: true })
+                    authenticate(username, password, sendCommand)
+                } else if(response.statusCode == 404) {
+                    console.log('Found no user of that name.')
+                    var register = readlineSync.question('Do you want to register this usename (yes|no)? ', { trueValue: ['yes', 'y'], falseValue: ['no', 'n'] })
+                    if (register) {
+                        var fullname = readlineSync.question('Full name: ')
+                        var email = readlineSync.questionEMail('E-Mail address: ')
+                        var password = readlineSync.questionNewPassword('New password: ')
+                        var user = { fullname: fullname, email: email, password: password }
+                        sendRequest('put', userPath, user, function(error, response, body) {
+                            if (response.statusCode == 200) {
+                                authenticate(username, password, sendCommand)
+                            } else {
+                                console.error('Unable to register user.')
+                                exit(1)
+                            }
+                        })
+                    } else {
+                        exit(0)
+                    }
+                } else {
+                    console.error(response.statusCode)
+                }
+            })
+        } else {
+            loadUser()
+            sendCommand()
+        }
+    } else {
+        loadUser()
+        sendCommand()
+    }
 }
 
 program
@@ -48,4 +129,6 @@ program
 
 program.parse(process.argv)
 
-runCommand('get', 'users')
+runCommand('get', 'users', null, function(error, response, body) {
+    console.log(body)
+})