client.coffee

   1 #!/usr/bin/coffee
   2
   3 fs = require 'fs'
   4 async = require 'async'
   5 af = require './api.coffee'
   6
   7 token_file = "#{process.env.HOME}/.af-coffee-token"
   8
   9 timeout = (ms, f) -> setTimeout f, ms
  10
  11 # a session caches the api access token and prompts for the username and
  12 # password if it goes stale. It re-tries API calls that fail due to
  13 # invalid/expired token
  14
  15 class Session
  16         constructor: ->
  17                 @token = 0
  18
  19         api: (call, args..., callback) ->
  20                 async.waterfall [
  21                         (callback) =>
  22                                 switch @token
  23                                         when 0
  24                                                 get_token callback
  25                                         when -1
  26                                                 login callback
  27                                         else
  28                                                 callback null, @token
  29                         (token, callback) =>
  30                                 @token = token
  31                                 af[call] @token, args..., callback
  32                 ], (err, result) =>
  33                         # eg /app/xxx/stats sometimes returns 404 with wrong auth token
  34                         if err?.code is 403 or err?.code is 404
  35                                 @token = -1
  36                                 @api(call, args..., callback)
  37                         else
  38                                 callback err, result
  39
  40 ask = (opts, callback) ->
  41         process.stdout.write opts.prompt
  42         process.stdin.setEncoding 'utf8'
  43         process.stdin.resume()
  44         process.stdin.once 'data', (line) ->
  45                 if opts.silent
  46                         # send ^[[A^[[2K to move the cursor up one line, then clear that line
  47                         process.stdout.write new Buffer [27, 91, 65, 27, 91, 50, 75]
  48                         process.stdout.write opts.prompt + "***\n"
  49                 process.stdin.pause()
  50                 callback null, (line.substr 0, line.length - 1)
  51
  52 get_token = (callback) ->
  53         fs.readFile token_file, 'utf8', (err, token) ->
  54                 if err?
  55                         login callback
  56                 else
  57                         callback null, token
  58
  59 login = (callback) ->
  60         async.waterfall [
  61                 (callback) -> async.series [
  62                         (callback) -> ask prompt: 'username: ', callback
  63                         (callback) -> ask prompt: 'password: ', silent: true, callback
  64                 ], callback
  65                 ([username, password], callback) ->
  66                         af.login username, password, callback
  67                 (token, callback) ->
  68                         # wait for file write so there's no race condition if get_token gets called soon
  69                         fs.writeFile token_file, token, (err) ->
  70                                 if err
  71                                         process.stderr.write "Warning: couldn't cache auth token in #{token_file}: #{err}\n"
  72                                 # don't pass on error, it's ok if we can't cache it
  73                                 callback null, token
  74         ], callback
  75
  76
  77 exports.new_session = ->
  78         return new Session()
  79
  80 usage = ->
  81         process.stderr.write "usage: #{process.argv[0]} #{process.argv[1]} command [args...]\n"
  82         process.stderr.write "valid commands are:\n"
  83         for k, v of af
  84                 process.stderr.write "\t#{k}\n" unless k is 'login'
  85
  86 # parse and act on commandline arguments unless we were require()d as a module
  87 if require.main is module
  88         args = process.argv[2..]
  89         if args.length is 0
  90                 usage()
  91         else if not af[args[0]]
  92                 process.stderr.write "unknown command \"#{args[0]}\"\n"
  93                 usage()
  94         else
  95                 session = new Session()
  96                 session.api args[0], args[1..], (err, result) ->
  97                         if err?
  98                                 process.stderr.write "Error: #{JSON.stringify err}\n"
  99                         if result?
 100                                 if typeof result is 'string'
 101                                         process.stdout.write result
 102                                 else
 103                                         process.stdout.write JSON.stringify result