mosaic/lib/mosaic.js

"use strict";

var url = require('url')
var cheerio = require('cheerio')
var program = require('commander')
var path = require('path')
var fs = require('fs')
var Q = require('q')
var CookieAccess = require('cookiejar').CookieAccessInfo


var settings = {
  daemon: 'brix.alibaba-inc.com',
  // daemon: '127.0.0.1:7000',
  port: '5000'
}

var mosaic = exports


mosaic.config = function(p, v) {
  if (typeof v !== 'undefined') {
    settings[p] = v
  }
  else {
    return settings[p]
  }
}

mosaic.home = function() {
  var homedir = process.env.HOME;

  if (!homedir) {
    homedir = process.env.HOMEDRIVE + process.env.HOMEPATH;
  }

  var mohome = path.join(homedir, '.mosaic')

  if (fs.existsSync(mohome)) {
    if (fs.statSync(mohome).isFile()) {
      mosaic.fatal('配置目录创建失败，已存在 ~/.mosaic 文件')
    }
  }
  else {
    fs.mkdirSync(mohome)
  }

  return mohome
}

mosaic.remote = function(path) {
  path = path || ''

  return url.format({
    protocol: 'http',
    host: mosaic.config('daemon'),
    pathname: path.charAt(0) === '/' ? path : 'api/' + path
  })
}

mosaic.login = function(res) {

  return Q
    .fcall(function() {
      var $ = cheerio.load(res.text)
      var props = '_csrf_token action return login_maintain event_submit_do_sso_login'.split(' ')
      var data = {}

      props.forEach(function(p) {
        data[p] = $('[name="' + p + '"]').val()
      })
      data.login_maintain = true

      return data
    })
    .then(function(data) {
      var d = Q.defer()

      program.prompt('Username: ', function(name) {
        data['_fm.l._0.a'] = name
        d.resolve(data)
      })

      return d.promise
    })
    .then(function(data) {
      var d = Q.defer()

      program.password('Password: ', function(password) {
        data['_fm.l._0.p'] = password
        d.resolve(data)
      })

      return d.promise
    })
    .then(function(data) {
      return mosaic.doLogin(res.req.url, data)
    })
    .fail(function(err) {
      throw err
    })
}

mosaic.doLogin = function(remote, data) {
  var d = Q.defer()
  var req = mosaic.agent().post(remote)

  data.login_maintain = 'on'
  // superagent defaults Content-Type to application/json
  // don't ask me how I find this out.
  req.type('form').send(data).end(function(err, res) {
    if (err) return d.reject(err)

    var $ = cheerio.load(res.text)
    var errNode = $('.hint .error')

    if (errNode.length > 0) {
      mosaic.error(errNode.text())
      // Login error occured. Let's resolve the login promise again.
      d.resolve(mosaic.login(res))
    }
    else {
      mosaic.saveCookies()
      d.resolve(res)
    }
  })

  return d.promise
}

mosaic.agent = function() {
  var agent = mosaic._agent
  var fpath = path.join(mosaic.home(), 'cookies.txt')

  if (!agent) agent = mosaic._agent = require('superagent').agent()
  if (fs.existsSync(fpath)) agent.jar.setCookies(fs.readFileSync(fpath, 'utf-8'))

  return agent
}

mosaic.saveCookies = function() {
  var url = require('url').parse(mosaic.remote())
  var access = CookieAccess(url.host, url.pathname, 'https:' === url.protocol)

  fs.writeFileSync(
    path.join(mosaic.home(), 'cookies.txt'),
    mosaic.agent().jar.getCookies(access).toValueString(),
    'utf-8'
  )
}

mosaic.attachCookies = function() {}

/*
 * mix utils into module.exports
 */
var utils = require('./utils')

for (var p in utils) {
  if (utils.hasOwnProperty(p)) {
    mosaic[p] = utils[p]
  }
}

mosaic.config('home', mosaic.home())