Poloniex connection.

6 files changed, 191 insertions, 51 deletions

diff --git a/cmd/web/js/api.js b/cmd/web/js/api.js
index e2acd1d..5c19fdf 100644
--- a/cmd/web/js/api.js
+++ b/cmd/web/js/api.js
@@ -53,6 +53,17 @@ var ApiEndpoints = {
       return '/market/' + params.name;
     }
   },
+  'MARKET_BALANCE': {
+    'type': 'GET',
+    'auth': true,
+    'parameters': [
+      {'name': 'name',     'mandatory': true, 'inquery': false},
+      {'name': 'currency', 'mandatory': true, 'inquery': true},
+    ],
+    'buildUrl': function(params) {
+      return '/market/' + params.name + '/balance';
+    }
+  },
   'UPDATE_MARKET': {
     'type': 'POST',
     'auth': true,
diff --git a/cmd/web/js/main.jsx b/cmd/web/js/main.jsx
index eb53057..e5e505d 100644
--- a/cmd/web/js/main.jsx
+++ b/cmd/web/js/main.jsx
@@ -1,15 +1,17 @@
-var SignupForm    = require('./signup.js').SignupForm;
-var SigninForm    = require('./signin.js').SigninForm;
-var OtpEnrollForm = require('./otp.js').OtpEnrollForm;
-var PoloniexForm  = require('./poloniex.js').PoloniexForm;
-var App           = require('./app.js');
-var Api           = require('./api.js').Api;
-var cookies       = require('./cookies.js');
+var SignupForm         = require('./signup.js').SignupForm;
+var SigninForm         = require('./signin.js').SigninForm;
+var OtpEnrollForm      = require('./otp.js').OtpEnrollForm;
+var PoloniexController = require('./poloniex.js').PoloniexController;
+var App                = require('./app.js');
+var Api                = require('./api.js').Api;
+var cookies            = require('./cookies.js');
 
 var Logo = React.createClass({
   render: function() {
-    return (<div id='logo'>
-              <a href='/'>Cryptoportfolio</a>
+    return (<div className='row'>
+              <div id='logo' className='offset-4 col-4'>
+                <a href='/'>Cryptoportfolio</a>
+              </div>
             </div>);
   }
 });
@@ -49,21 +51,12 @@ App.page('/signout', true, function(context) {
 });
 
 App.page('/me', true, function(context) {
-  Api.Call('MARKET', {'name': 'poloniex'}, function(err, status, data) {
-    if (err) {
-      console.error(err, data);
-      return;
-    }
-
-    App.mount(
-      <div>
-        <Logo />
-        <p>Poloniex</p>
-        <PoloniexForm apiKey={data.key} apiSecret={data.secret}/>
-      </div>
-    );
-
-  }.bind(this));
+  App.mount(
+    <div>
+      <Logo />
+      <PoloniexController/>
+    </div>
+  );
 });
 
 App.page('/otp/setup', true, function(context) {
diff --git a/cmd/web/js/otp.jsx b/cmd/web/js/otp.jsx
index 2717d9f..a0ee5cc 100644
--- a/cmd/web/js/otp.jsx
+++ b/cmd/web/js/otp.jsx
@@ -53,12 +53,12 @@ module.exports.OtpEnrollForm = React.createClass({
       );
     }
     return (
-        <div className='row otp-enroll justify-content-center'>
-          <div className='col-lg-offset-4 col-lg-4 col-md-offset-4 col-md-4 col-sm-offset-4 col-sm-4 col-xs-offset-1 col-xs-10'>
+        <div className='row otp-enroll'>
+          <div className='offset-4 col-4 col-xs-offset-1 col-xs-10 text-center'>
             {qrCode}
             <div className='row justify-content-center'>
               <form role='form' onSubmit={this.handleSubmit}>
-                <input className='form-control' type='pass' placeholder='pass' onChange={this.handlePassChange} />
+                <input className='form-control' type='pass' placeholder='code' onChange={this.handlePassChange} />
                 <input className='form-control submit' type='submit' value='Validate' />
                 <div className={cName} ref='message'>{this.state.msg}</div>
               </form>
diff --git a/cmd/web/js/poloniex.jsx b/cmd/web/js/poloniex.jsx
index 877198d..8b577b4 100644
--- a/cmd/web/js/poloniex.jsx
+++ b/cmd/web/js/poloniex.jsx
@@ -1,49 +1,185 @@
 var Api        = require('./api.js').Api;
-var App        = require('./app.js');
 var classNames = require('classnames');
 
-module.exports.PoloniexForm = React.createClass({
+module.exports.PoloniexController = React.createClass({
   getInitialState: function() {
-    return {'hideMsg': true, 'msg': '', 'msgOk': false, 'apiSecret': this.props.apiSecret, 'apiKey': this.props.apiKey};
+    return {'apiKey': '', 'apiSecret': '', 'flag': 'loading', 'valueCurrency': null, 'balanceValue': null, 'balance': null};
   },
-  handleSubmit: function(e) {
+  handleCredentialsChange: function(key, secret) {
+    this.setState({'apiKey': key, 'apiSecret': secret});
+  },
+  handleCredentialsSubmit: function() {
+    if (!this.state.apiKey || !this.state.apiSecret) {
+      return;
+    }
     Api.Call('UPDATE_MARKET', {'key': this.state.apiKey, 'secret': this.state.apiSecret, 'name': 'poloniex'}, function(err, status, data) {
       if (err) {
         console.error(err, data);
-        this.displayMessage(App.errorCodeToMessage(err.code), false);
         return;
       }
 
-      this.displayMessage('OK', true);
+      this.setState({'flag': 'loading', 'valueCurrency': null, 'balanceValue': null, 'balance': null});
+      this.loadBalance();
+    }.bind(this));
+  },
+  loadBalance: function() {
+    Api.Call('MARKET_BALANCE', {'name': 'poloniex', 'currency': 'BTC'}, function(err, status, data) {
+      if (err) {
+        console.error(err, data);
+        if (err.code === 'invalid_market_credentials') {
+          this.setState({'flag': 'invalidCredentials', 'valueCurrency': null, 'balanceValue': null, 'balance': null});
+        }
+        return;
+      }
 
+      this.setState({'flag': 'ok', 'valueCurrency': data.valueCurrency, 'balanceValue': data.value, 'balance': data.balance});
     }.bind(this));
+  },
+  componentDidMount: function() {
+    Api.Call('MARKET', {'name': 'poloniex'}, function(err, status, data) {
+      if (err) {
+        console.error(err, data);
+        return;
+      }
+
+      var flag = this.state.flag;
+      if (!data.key || !data.secret) {
+        flag = 'emptyCredentials';
+      } else {
+        this.loadBalance();
+      }
+
+      this.setState({'apiKey': data.key, 'apiSecret': data.secret, 'flag': flag});
+    }.bind(this));
+  },
+  render: function() {
+    var displayText = null;
+    switch (this.state.flag) {
+      case 'loading':
+        displayText = 'Loading data from poloniex...';
+        break;
+      case 'invalidCredentials':
+        displayText = 'Invalid poloniex credentials';
+        break;
+      case 'emptyCredentials':
+        displayText = 'Please provide poloniex credentials';
+        break;
+      default:
+        displayText = null;
+    }
+    return (
+      <div>
+        <PoloniexBalance  balanceCurrency={this.state.valueCurrency}
+                          balanceValue={this.state.balanceValue}
+                          balance={this.state.balance}
+                          displayText={displayText}/>
+        <PoloniexCredentialsForm onLoadCredentials={this.onLoadCredentials}
+                                 onCredentialsSubmit={this.handleCredentialsSubmit}
+                                 onCredentialsChange={this.handleCredentialsChange}
+                                 apiSecret={this.state.apiSecret}
+                                 apiKey={this.state.apiKey}/>
+      </div>
+    );
+  }
+});
+
+var PoloniexBalance = React.createClass({
+  getInitialState: function() {
+    return {'hideMsg': true, 'msg': '', 'msgOk': false};
+  },
+  render: function() {
+    var dashboard = null;
+
+    if (this.props.balanceValue !== null) {
+
+      var balance = Object.keys(this.props.balance).map(function(currency) {
+        return <div key={currency}><i className={classNames('cc', currency)}></i> {this.props.balance[currency]}</div>;
+      }.bind(this));
+
+      dashboard = (
+        <div className='row'>
+          <div className='col-4 align-self-center h-100'>
+            <div>
+              {balance}
+            </div>
+          </div>
+          <div className='offset-1 col-7 h-100 align-self-center'>
+            <div className='text-center'>
+              Balance ({this.props.balanceCurrency}): <span>{this.props.balanceValue} <i className={classNames('cc', this.props.balanceCurrency)}></i></span>
+            </div>
+          </div>
+        </div>
+      );
+    } else {
+      dashboard = (
+        <div className='row'>
+          <div className='col-12 text-center'>
+           <span>{this.props.displayText}</span>
+          </div>
+        </div>
+
+      );
+    }
+
+    return (
+      <div className='row'>
+        <div className='box offset-2 col-8'>
+          <div className='row'>
+            <div className='col-4'>Portfolio</div>
+          </div>
+          <hr/>
+          {dashboard}
+        </div>
+      </div>
+    );
+  }
+});
+
+module.exports.PoloniexBalance = PoloniexBalance;
+
+var PoloniexCredentialsForm = React.createClass({
+  getInitialState: function() {
+    return {'hideMsg': true, 'msg': '', 'editMode': false, 'msgOk': false};
+  },
+  handleSubmit: function(e) {
+    this.props.onCredentialsSubmit();
+    this.setState({'editMode': false});
     e.preventDefault();
   },
   handleApiKeyChange: function(event) {
-    this.setState({'apiKey': event.target.value});
+    this.props.onCredentialsChange(event.target.value, this.props.apiSecret);
   },
   handleApiSecretChange: function(event) {
-    this.setState({'apiSecret': event.target.value});
+    this.props.onCredentialsChange(this.props.apiKey, event.target.value);
   },
-  hideMessage: function() {
-    this.setState({'hideMsg': true});
-  },
-  displayMessage: function(msg, ok) {
-    this.setState({'msg': msg, 'msgOk': ok, 'hideMsg': false});
+  onEditClick: function() {
+    this.setState({'editMode': true});
   },
   render: function() {
-    var cName = classNames('form-message', {'hidden': this.state.hideMsg, 'message-ok': this.state.msgOk});
+    var submitType      = (this.state.editMode === true) ? 'submit' : 'hidden';
+    var buttonDisplay   = (this.state.editMode === true) ? 'none' : 'inline';
+    var secretDisplayed = (this.state.editMode === true) ? this.props.apiSecret : 'XXXXXXX';
+    var keyDisplayed    = (this.state.editMode === true) ? this.props.apiKey : 'XXXXXXX';
+
     return (
-        <div className='row justify-content-center api-credentials-form'>
-          <div className='col-lg-offset-4 col-lg-4 col-md-offset-4 col-md-4 col-sm-offset-4 col-sm-4 col-xs-offset-1 col-xs-10'>
+        <div className='row api-credentials-form'>
+          <div className='offset-3 col-6 box'>
+            <span className='text-center'>Poloniex credentials</span>
+            <hr/>
             <form role='form' onSubmit={this.handleSubmit}>
-              <input className='form-control' type='text' placeholder='apiKey' value={this.state.apiKey} onChange={this.handleApiKeyChange} />
-              <input className='form-control' type='text' placeholder='apiSecret' value={this.state.apiSecret} onChange={this.handleApiSecretChange} />
-              <input className='form-control submit' type='submit' value='Save' />
-              <div className={cName} ref='message'>{this.state.msg}</div>
+              <label className='w-100'>Key:
+                <input className='form-control' type='text' placeholder='key' value={keyDisplayed} onChange={this.handleApiKeyChange} disabled={!this.state.editMode}/>
+              </label>
+              <label className='w-100'>Secret:
+                <input className='form-control' type='text' placeholder='secret' value={secretDisplayed} onChange={this.handleApiSecretChange} disabled={!this.state.editMode}/>
+              </label>
+              <input className='form-control submit' type={submitType} value='Save' />
+              <button className='form-control submit' style={{display: buttonDisplay}} onSubmit={null} onClick={this.onEditClick} type='button'>Show/Edit</button>
             </form>
           </div>
         </div>
        );
   }
 });
+
+module.exports.PoloniexCredentialsForm = PoloniexCredentialsForm;
diff --git a/cmd/web/js/signin.jsx b/cmd/web/js/signin.jsx
index 443a461..a2cfd1b 100644
--- a/cmd/web/js/signin.jsx
+++ b/cmd/web/js/signin.jsx
@@ -35,8 +35,8 @@ module.exports.SigninForm = React.createClass({
   render: function() {
     var cName = classNames('form-message', {'hidden': this.state.hideMsg, 'message-ok': this.state.msgOk});
     return (
-        <div className='row justify-content-center sign-in'>
-          <div className='col-lg-offset-4 col-lg-4 col-md-offset-4 col-md-4 col-sm-offset-4 col-sm-4 col-xs-offset-1 col-xs-10'>
+        <div className='row sign-in'>
+          <div className='offset-4 col-4 col-xs-offset-1 col-xs-10 text-center'>
             <form role='form' onSubmit={this.handleSubmit}>
               <input className='form-control' type='email' placeholder='email' onChange={this.handleEmailChange} />
               <input className='form-control' type='password' placeholder='password' onChange={this.handlePasswordChange} />
diff --git a/cmd/web/js/signup.jsx b/cmd/web/js/signup.jsx
index 149125a..404a828 100644
--- a/cmd/web/js/signup.jsx
+++ b/cmd/web/js/signup.jsx
@@ -40,8 +40,8 @@ module.exports.SignupForm = React.createClass({
   render: function() {
     var cName = classNames('form-message', {'hidden': this.state.hideMsg, 'message-ok': this.state.msgOk});
     return (
-        <div className='row justify-content-center sign-in'>
-          <div className='col-lg-offset-4 col-lg-4 col-md-offset-4 col-md-4 col-sm-offset-4 col-sm-4 col-xs-offset-1 col-xs-10'>
+        <div className='row sign-in'>
+          <div className='offset-4 col-4 col-xs-offset-1 col-xs-10 text-center'>
             <form role='form' onSubmit={this.handleSubmit}>
               <input className='form-control' type='email' placeholder='email' onChange={this.handleEmailChange} />
               <input className='form-control' type='password' placeholder='password' onChange={this.handlePasswordChange} />