WIP

bin/ssu

@@ -4,19 +4,47 @@
 
 var wreck = require('wreck')
 var argv = process.argv.slice(2)
+var Promise = require('es6-promise')
+var pgp = require('pg-promise')({
+  promiseLib: Promise
+})
+
+var fs = require('fs')
 
 var cmd = argv[0]
-var fingerprint = argv[1]
 
-if (!cmd || !fingerprint) {
+if (!cmd) bail()
+
+function bail () {
   console.log('Command line utility for lamassu-server')
-  console.log('\nUsage: ssu reboot <machine-fingerprint>')
+  console.log('\nssu reboot <machine-fingerprint>')
   console.log('This will remotely reboot your lamassu-machine.')
+  console.log('\nssu crypto <code> <ticker-plugin> <wallet-plugin>')
+  console.log('This will configure a new cryptocurrency.')
   process.exit(1)
 }
 
-var opts = {json: true}
-wreck.get('http://localhost:7070/pid?fingerprint=' + fingerprint, opts, function (err, res, payload) {
+switch (cmd) {
+  case 'reboot':
+    reboot()
+    break
+  case 'crypto':
+    crypto()
+    break
+  default:
+
+}
+
+function reboot () {
+  var fingerprint = argv[1]
+
+  if (!fingerprint) {
+    console.log('Fingerprint required')
+    process.exit(1)
+  }
+
+  var opts = {json: true}
+  wreck.get('http://localhost:7070/pid?fingerprint=' + fingerprint, opts, function (err, res, payload) {
     if (err) {
       console.log('Please make sure that lamassu-server is running on this box.')
       process.exit(2)
@@ -80,4 +108,45 @@ wreck.get('http://localhost:7070/pid?fingerprint=' + fingerprint, opts, function
         })
       }, 5000)
     })
-})
+  })
+}
+
+function crypto () {
+  var code = argv[1]
+  var tickerPlugin = argv[2]
+  var walletPlugin = argv[3]
+
+  if (!code || !tickerPlugin || !walletPlugin) {
+    console.log('\nssu crypto <code> <ticker-plugin> <wallet-plugin>')
+    console.log('This will configure a new cryptocurrency.')
+    process.exit(1)
+  }
+
+  code = code.toUpperCase()
+
+  var psqlUrl
+  try {
+    psqlUrl = process.env.DATABASE_URL || JSON.parse(fs.readFileSync('/etc/lamassu.json')).postgresql
+  } catch (ex) {
+    psqlUrl = 'psql://lamassu:lamassu@localhost/lamassu'
+  }
+  var db = pgp(psqlUrl)
+
+  return db.one('select data from user_config where type=$1', 'exchanges')
+  .then(function (data) {
+    var config = data.data
+    config.exchanges.plugins.current[code] = {
+      ticker: tickerPlugin,
+      transfer: walletPlugin
+    }
+    return db.none('update user_config set data=$1 where type=$2', [config, 'exchanges'])
+  })
+  .then(function () {
+    console.log('success')
+    pgp.end()
+  })
+  .catch(function (err) {
+    console.log(err.stack)
+    pgp.end()
+  })
+}

lib/plugins.js

@@ -2,7 +2,10 @@
 
 var _ = require('lodash')
 var async = require('async')
+
 var BigNumber = require('bignumber.js')
+BigNumber.config({DECIMAL_PLACES: 40})
+
 var logger = require('./logger')
 var argv = require('minimist')(process.argv.slice(2))
 

lib/plugins/kraken.js

@@ -1,8 +1,9 @@
 require('es6-promise').polyfill()
 var axios = require('axios')
 var _ = require('lodash')
+
 var BigNumber = require('bignumber.js')
-BigNumber.config({DECIMAL_PLACES: 30})
+BigNumber.config({DECIMAL_PLACES: 40})
 
 exports.NAME = 'Kraken'
 exports.SUPPORTED_MODULES = ['ticker']

package.json

@@ -35,7 +35,9 @@
     "lodash": "^2.4.1",
     "minimist": "0.0.8",
     "node-uuid": "^1.4.2",
-    "pg": "~2.11.1",
+    "pg": "^4.5.1",
+    "pg-promise": "^3.4.3",
+    "web3": "^0.15.3",
     "wreck": "5.1.0"
   },
   "repository": {