client.coffee

   1 fs = require 'fs'
   2 async = require 'async'
   3 af = require './api.coffee'
   4
   5 token_file = "#{process.env.HOME}/.af-coffee-token"
   6
   7 timeout = (ms, f) -> setTimeout f, ms
   8
   9 # a session caches the api access token and prompts for the username and
  10 # password if it goes stale. It re-tries API calls that fail due to
  11 # invalid/expired token
  12
  13 class Session
  14         constructor: ->
  15                 @token = 0
  16
  17         api: (call, args..., callback) ->
  18                 async.waterfall [
  19                         (callback) =>
  20                                 switch @token
  21                                         when 0
  22                                                 get_token callback
  23                                         when -1
  24                                                 login callback
  25                                         else
  26                                                 callback null, @token
  27                         (token, callback) =>
  28                                 @token = token
  29                                 af[call] @token, args..., callback
  30                 ], (err, result) =>
  31                         # eg /app/xxx/stats sometimes returns 404 with wrong auth token
  32                         if err?.code is 403 or err?.code is 404
  33                                 @token = -1
  34                                 @api(call, args..., callback)
  35                         else
  36                                 callback err, result
  37
  38 ask = (opts, callback) ->
  39         process.stdout.write opts.prompt
  40         process.stdin.setEncoding 'utf8'
  41         process.stdin.resume()
  42         process.stdin.once 'data', (line) ->
  43                 if opts.silent
  44                         # send ^[[A^[[2K to move the cursor up one line, then clear that line
  45                         process.stdout.write new Buffer [27, 91, 65, 27, 91, 50, 75]
  46                         process.stdout.write opts.prompt + "***\n"
  47                 process.stdin.pause()
  48                 callback null, (line.substr 0, line.length - 1)
  49
  50 get_token = (callback) ->
  51         fs.readFile token_file, 'utf8', (err, token) ->
  52                 if err?
  53                         login callback
  54                 else
  55                         callback null, token
  56
  57 login = (callback) ->
  58         async.waterfall [
  59                 (callback) -> async.series [
  60                         (callback) -> ask prompt: 'username: ', callback
  61                         (callback) -> ask prompt: 'password: ', silent: true, callback
  62                 ], callback
  63                 ([username, password], callback) ->
  64                         af.login username, password, callback
  65                 (token, callback) ->
  66                         # wait for file write so there's no race condition if get_token gets called soon
  67                         fs.writeFile token_file, token, (err) ->
  68                                 if err
  69                                         console.log "Warning: couldn't cache auth token in #{token_file}: ", err
  70                                 # don't pass on error, it's ok if we can't cache it
  71                                 callback null, token
  72         ], callback
  73
  74
  75 exports.new_session = ->
  76         return new Session()