Refactor Portfolio balance.

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Front.git] / markets / poloniex.go

diff --git a/markets/poloniex.go b/markets/poloniex.go
index 9aaeafbfcb1cedc3f6812bb0cd34d61a97ff160d..58645b2b2a23750c4adae2633e7c7297bfb71d19 100644 (file)
--- a/markets/poloniex.go
+++ b/markets/poloniex.go
@@ -52,6 +52,22 @@ func NewPoloniex() *Poloniex {
        }
 }
 
+func (p *Poloniex) TestCredentials(apiKey, apiSecret string) error {
+       client, _ := poloniex.NewClient(apiKey, apiSecret)
+
+       _, err := client.TradeReturnDepositAdresses()
+
+       if poloniexInvalidCredentialsError(err) {
+               return utils.Error{InvalidCredentials, "invalid poloniex credentials"}
+       }
+
+       if poloniexRestrictedIPError(err) {
+               return utils.Error{IPRestricted, "IP restricted api key"}
+       }
+
+       return nil
+}
+
 func (p *Poloniex) GetBalance(apiKey, apiSecret string) (Summary, error) {
        client, _ := poloniex.NewClient(apiKey, apiSecret)
        var summary Summary