6 files changed, 166 insertions, 3 deletions

diff --git a/api/const.go b/api/const.go
index 2edd6f4..1b22355 100644
--- a/api/const.go
+++ b/api/const.go
@@ -6,15 +6,19 @@ import "net/http"
 type Status uint32
 type ErrorCode uint32
 
+const EXTERNAL_SERVICE_TIMEOUT_SECONDS = 10
+
 const (
         OK Status = iota
         NOK
 
         BadRequest ErrorCode = iota + 1
         EmailExists
+        ExternalServiceTimeout
         InternalError
         InvalidCredentials
         InvalidEmail
+        InvalidMarketCredentials
         InvalidOtp
         InvalidPassword
         NeedOtpValidation
@@ -31,7 +35,7 @@ func StatusToHttpCode(status Status, code ErrorCode) int {
         }
 
         switch code {
-        case BadRequest, InvalidPassword, InvalidEmail:
+        case BadRequest, InvalidPassword, InvalidEmail, InvalidMarketCredentials:
                 return http.StatusBadRequest
 
         case InvalidCredentials, InvalidOtp:
@@ -45,6 +49,9 @@ func StatusToHttpCode(status Status, code ErrorCode) int {
 
         case NotFound:
                 return http.StatusNotFound
+
+        case ExternalServiceTimeout:
+                return http.StatusGatewayTimeout
         }
 
         return http.StatusInternalServerError
diff --git a/api/const_string.go b/api/const_string.go
index 611db40..e4b9e50 100644
--- a/api/const_string.go
+++ b/api/const_string.go
@@ -15,9 +15,9 @@ func (i Status) String() string {
         return _Status_name[_Status_index[i]:_Status_index[i+1]]
 }
 
-const _ErrorCode_name = "BadRequestEmailExistsInternalErrorInvalidCredentialsInvalidEmailInvalidOtpInvalidPasswordNeedOtpValidationNotAuthorizedNotFoundOtpAlreadySetupOtpNotSetupUserNotConfirmed"
+const _ErrorCode_name = "BadRequestEmailExistsExternalServiceTimeoutInternalErrorInvalidCredentialsInvalidEmailInvalidMarketCredentialsInvalidOtpInvalidPasswordNeedOtpValidationNotAuthorizedNotFoundOtpAlreadySetupOtpNotSetupUserNotConfirmed"
 
-var _ErrorCode_index = [...]uint8{0, 10, 21, 34, 52, 64, 74, 89, 106, 119, 127, 142, 153, 169}
+var _ErrorCode_index = [...]uint8{0, 10, 21, 43, 56, 74, 86, 110, 120, 135, 152, 165, 173, 188, 199, 215}
 
 func (i ErrorCode) String() string {
         i -= 3
diff --git a/api/external_services.go b/api/external_services.go
new file mode 100644
index 0000000..c467171
--- /dev/null
+++ b/api/external_services.go
@@ -0,0 +1,40 @@
+package api
+
+import (
+        "context"
+        "fmt"
+        "time"
+)
+
+// Use this to call external services. It will handle timeout and request cancellation gracefully.
+func CallExternalService(tag string, timeout time.Duration, routine func() *Error) *Error {
+        routineDone := make(chan *Error)
+
+        go func() {
+                routineDone <- routine()
+        }()
+
+        ctx, cancel := context.WithTimeout(context.Background(), timeout)
+        defer cancel()
+
+        select {
+        case err := <-routineDone:
+                return err
+        case <-ctx.Done():
+                return &Error{ExternalServiceTimeout, "external service timeout", fmt.Errorf("'%v' routine timeouted", tag)}
+        }
+}
+
+var ErrorChan chan error
+
+func ErrorMonitoring() {
+        for {
+                err := <-ErrorChan
+                log.Errorf("error: %v", err)
+        }
+}
+
+func init() {
+        ErrorChan = make(chan error)
+        go ErrorMonitoring()
+}
diff --git a/api/market_config.go b/api/market_config.go
index 3fd10ae..d85af4d 100644
--- a/api/market_config.go
+++ b/api/market_config.go
@@ -2,7 +2,13 @@ package api
 
 import (
         "fmt"
+        "strings"
+        "time"
 
+        "immae.eu/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Front/markets"
+
+        "github.com/jloup/utils"
+        "github.com/shopspring/decimal"
         "immae.eu/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Front/db"
 )
 
@@ -42,6 +48,73 @@ func (q MarketConfigQuery) Run() (interface{}, *Error) {
         return config.Config, nil
 }
 
+type MarketBalanceQuery struct {
+        In struct {
+                User     db.User
+                Market   string
+                Currency string
+        }
+}
+
+func (q MarketBalanceQuery) ValidateParams() *Error {
+        if q.In.Market != "poloniex" {
+                return &Error{BadRequest, "invalid market name", fmt.Errorf("'%v' is not a valid market name", q.In.Market)}
+        }
+
+        // TODO: we should request market for available currencies.
+        if q.In.Currency != "BTC" && q.In.Currency != "USDT" && q.In.Currency != "ETH" {
+                return &Error{BadRequest, "invalid currency, accept [BTC, USDT, ETH]", fmt.Errorf("'%v' is not a valid currency", q.In.Currency)}
+        }
+
+        return nil
+}
+
+func (q MarketBalanceQuery) Run() (interface{}, *Error) {
+        config, err := db.GetUserMarketConfig(q.In.User.Id, q.In.Market)
+        if err != nil {
+                return nil, NewInternalError(err)
+        }
+
+        if config.Config["key"] == "" || config.Config["secret"] == "" {
+                return nil, &Error{BadRequest, "your credentials for this market are not setup", fmt.Errorf("'%v' credentials are not setup", q.In.Market)}
+        }
+
+        result := struct {
+                Value         decimal.Decimal            `json:"value"`
+                ValueCurrency string                     `json:"valueCurrency"`
+                Balance       map[string]decimal.Decimal `json:"balance"`
+        }{}
+
+        resultErr := CallExternalService(fmt.Sprintf("'%s' GetBalanceValue", q.In.Market), EXTERNAL_SERVICE_TIMEOUT_SECONDS*time.Second, func() *Error {
+                balance, err := Poloniex.GetBalance(config.Config["key"], config.Config["secret"])
+
+                if utils.ErrIs(err, markets.InvalidCredentials) {
+                        return &Error{InvalidMarketCredentials, "wrong market credentials", fmt.Errorf("wrong '%v' market credentials", q.In.Market)}
+                }
+
+                if err != nil {
+                        return NewInternalError(err)
+                }
+
+                value, err := Poloniex.ComputeAccountBalanceValue(balance, q.In.Currency)
+                if err != nil {
+                        return NewInternalError(err)
+                }
+
+                result.Balance = balance
+                result.ValueCurrency = q.In.Currency
+                result.Value = value.Round(8)
+
+                return nil
+        })
+
+        if resultErr != nil {
+                return nil, resultErr
+        }
+
+        return &result, nil
+}
+
 type UpdateMarketConfigQuery struct {
         In struct {
                 User   db.User
@@ -56,6 +129,9 @@ func (q UpdateMarketConfigQuery) ValidateParams() *Error {
                 return &Error{BadRequest, "invalid market name", fmt.Errorf("'%v' is not a valid market name", q.In.Market)}
         }
 
+        q.In.Secret = strings.TrimSpace(q.In.Secret)
+        q.In.Key = strings.TrimSpace(q.In.Key)
+
         return nil
 }
 
diff --git a/api/markets.go b/api/markets.go
new file mode 100644
index 0000000..60fb912
--- /dev/null
+++ b/api/markets.go
@@ -0,0 +1,29 @@
+package api
+
+import (
+        "immae.eu/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Front/markets"
+)
+
+var Poloniex *markets.Poloniex
+
+func OpenMarketsConnection() error {
+        for {
+                err := Poloniex.StartTicker()
+                if err != nil {
+                        return err
+                }
+                log.Warn("connection to poloniex stream ended, restarting it...")
+        }
+}
+
+func init() {
+        Poloniex = markets.NewPoloniex()
+
+        // We open markets connections in the background as it can take time.
+        go func() {
+                err := OpenMarketsConnection()
+                if err != nil {
+                        ErrorChan <- err
+                }
+        }()
+}
diff --git a/api/routes.go b/api/routes.go
index d7e712c..cdf3dd9 100644
--- a/api/routes.go
+++ b/api/routes.go
@@ -41,6 +41,7 @@ var Groups = []Group{
                 []Route{
                         {"GET", []gin.HandlerFunc{GetMarketConfig}, "/:name"},
                         {"POST", []gin.HandlerFunc{UpdateMarketConfig}, "/:name/update"},
+                        {"GET", []gin.HandlerFunc{GetMarketBalance}, "/:name/balance"},
                 },
         },
 }
@@ -111,6 +112,16 @@ func GetMarketConfig(c *gin.Context) {
         RunQuery(query, c)
 }
 
+func GetMarketBalance(c *gin.Context) {
+        query := &MarketBalanceQuery{}
+
+        query.In.User = GetUser(c)
+        query.In.Market = c.Param("name")
+        query.In.Currency = c.Query("currency")
+
+        RunQuery(query, c)
+}
+
 func UpdateMarketConfig(c *gin.Context) {
         query := &UpdateMarketConfigQuery{}