Added Kraken Exchange support.

This commit is contained in:
Adrian Gallagher
2015-03-05 19:49:42 +11:00
parent 01fcde5110
commit c6852f3436
4 changed files with 589 additions and 2 deletions

View File

@@ -74,6 +74,15 @@
"Enabled": true,
"Verbose": false
},
{
"Name": "Kraken",
"Pairs": "XBTEUR,XBTUSD,XBTGBP,XBTJPY,LTCEUR,LTCUSD,EURXVN,USDXVN,XBTLTC,XBTNMC,XBTSTR,XBTXDG,XBTXRP,XBTXVN",
"APIKey": "Key",
"APISecret": "Secret",
"BaseCurrencies": "EUR,USD,GBP,JPY",
"Enabled": true,
"Verbose": false
},
{
"Name": "LakeBTC",
"Pairs": "BTCCNY,LTCCNY,LTCBTC",

View File

@@ -206,6 +206,7 @@ func IsValidExchange(Exchange string) (bool) {
bot.exchange.btcmarkets.GetName() == Exchange && bot.exchange.btcmarkets.IsEnabled() ||
bot.exchange.huobi.GetName() == Exchange && bot.exchange.huobi.IsEnabled() ||
bot.exchange.itbit.GetName() == Exchange && bot.exchange.itbit.IsEnabled() ||
bot.exchange.kraken.GetName() == Exchange && bot.exchange.kraken.IsEnabled() ||
bot.exchange.lakebtc.GetName() == Exchange && bot.exchange.lakebtc.IsEnabled() ||
bot.exchange.okcoinChina.GetName() == Exchange && bot.exchange.okcoinChina.IsEnabled() ||
bot.exchange.okcoinIntl.GetName() == Exchange && bot.exchange.okcoinIntl.IsEnabled() {

552
kraken.go Normal file
View File

@@ -0,0 +1,552 @@
package main
import (
"log"
"fmt"
"strconv"
"encoding/base64"
"encoding/json"
"crypto/hmac"
"crypto/sha512"
"crypto/sha256"
"errors"
"time"
"strings"
"net/url"
"net/http"
"io/ioutil"
)
const (
KRAKEN_API_URL = "https://api.kraken.com"
KRAKEN_API_VERSION = "0"
KRAKEN_SERVER_TIME = "Time"
KRAKEN_ASSETS = "ssets"
KRAKEN_ASSET_PAIRS = "AssetPairs"
KRAKEN_TICKER = "Ticker"
KRAKEN_OHLC = "OHLC"
KRAKEN_DEPTH = "Depth"
KRAKEN_TRADES = "Trades"
KRAKEN_SPREAD = "Spread"
KRAKEN_BALANCE = "Balance"
KRAKEN_TRADE_BALANCE = "TradeBalance"
KRAKEN_OPEN_ORDERS = "OpenOrders"
KRAKEN_CLOSED_ORDERS = "ClosedOrders"
KRAKEN_QUERY_ORDERS = "QueryOrders"
KRAKEN_TRADES_HISTORY = "TradesHistory"
KRAKEN_QUERY_TRADES = "QueryTrades"
KRAKEN_OPEN_POSITIONS = "OpenPositions"
KRAKEN_LEDGERS = "Ledgers"
KRAKEN_QUERY_LEDGERS = "QueryLedgers"
KRAKEN_TRADE_VOLUME = "TradeVolume"
KRAKEN_ORDER_CANCEL = "CancelOrder"
KRAKEN_ORDER_PLACE = "AddOrder"
)
type Kraken struct {
Name string
Enabled bool
Verbose bool
ClientKey, APISecret string
FiatFee, CryptoFee float64
}
type KrakenResponse struct {
Error []string `json:error`
Result map[string]interface{} `json:result`
}
func (k *Kraken) SetDefaults() {
k.Name = "Kraken"
k.Enabled = true
k.FiatFee = 0.35
k.CryptoFee = 0.10
k.Verbose = false
}
func (k *Kraken) GetName() (string) {
return k.Name
}
func (k *Kraken) SetEnabled(enabled bool) {
k.Enabled = enabled
}
func (k *Kraken) IsEnabled() (bool) {
return k.Enabled
}
func (k *Kraken) SetAPIKeys(apiKey, apiSecret string) {
k.ClientKey = apiKey
k.APISecret = apiSecret
}
func (k *Kraken) GetFee(cryptoTrade bool) (float64) {
if cryptoTrade {
return k.CryptoFee
} else {
return k.FiatFee
}
}
func (k *Kraken) GetServerTime() {
result, err := k.SendKrakenRequest(KRAKEN_SERVER_TIME)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetAssets() {
result, err := k.SendKrakenRequest(KRAKEN_ASSETS)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetAssetPairs() {
result, err := k.SendKrakenRequest(KRAKEN_ASSET_PAIRS)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetTicker(symbol string) interface{} {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendKrakenRequest(KRAKEN_TICKER + "?" + values.Encode())
if err != nil {
log.Println(err)
return ""
}
if strings.Contains(symbol, "LTC") {
return result["XLTCZUSD"]
} else if strings.Contains(symbol, "XBT") {
return result["XXBTZUSD"]
}
return nil
}
func (k *Kraken) GetOHLC(symbol string) {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendKrakenRequest(KRAKEN_OHLC + "?" + values.Encode())
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetDepth(symbol string) {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendKrakenRequest(KRAKEN_DEPTH + "?" + values.Encode())
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetTrades(symbol string) {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendKrakenRequest(KRAKEN_TRADES + "?" + values.Encode())
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetSpread(symbol string) {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendKrakenRequest(KRAKEN_SPREAD + "?" + values.Encode())
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) SendKrakenRequest(method string) (map[string]interface{}, error) {
path := fmt.Sprintf("%s/%s/public/%s", KRAKEN_API_URL, KRAKEN_API_VERSION, method)
resp := KrakenResponse{}
err := SendHTTPRequest(path, true, &resp)
log.Printf("Sending GET request to %s\n", path)
if err != nil {
return nil, err
}
if len(resp.Error) != 0 {
return nil, errors.New(fmt.Sprintf("Kraken error: %s", resp.Error))
}
return resp.Result, nil
}
func (k *Kraken) GetBalance() {
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_BALANCE, url.Values{})
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetTradeBalance(symbol, asset string) {
values := url.Values{}
if len(symbol) > 0 {
values.Set("aclass", symbol)
}
if len(asset) > 0 {
values.Set("asset", asset)
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_TRADE_BALANCE, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetOpenOrders(showTrades bool, userref int64) {
values := url.Values{}
if showTrades {
values.Set("trades", "true")
}
if userref != 0 {
values.Set("userref", strconv.FormatInt(userref, 10))
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_OPEN_ORDERS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetClosedOrders(showTrades bool, userref, start, end, offset int64, closetime string) {
values := url.Values{}
if showTrades {
values.Set("trades", "true")
}
if userref != 0 {
values.Set("userref", strconv.FormatInt(userref, 10))
}
if start != 0 {
values.Set("start", strconv.FormatInt(start, 10))
}
if end != 0 {
values.Set("end", strconv.FormatInt(end, 10))
}
if offset != 0 {
values.Set("ofs", strconv.FormatInt(offset, 10))
}
if len(closetime) > 0 {
values.Set("closetime", closetime)
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_CLOSED_ORDERS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) QueryOrdersInfo(showTrades bool, userref, txid int64) {
values := url.Values{}
if showTrades {
values.Set("trades", "true")
}
if userref != 0 {
values.Set("userref", strconv.FormatInt(userref, 10))
}
if txid != 0 {
values.Set("txid", strconv.FormatInt(userref, 10))
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_QUERY_ORDERS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetTradesHistory(tradeType string, showRelatedTrades bool, start, end, offset int64) {
values := url.Values{}
if len(tradeType) > 0 {
values.Set("aclass", tradeType)
}
if showRelatedTrades {
values.Set("trades", "true")
}
if start != 0 {
values.Set("start", strconv.FormatInt(start, 10))
}
if end != 0 {
values.Set("end", strconv.FormatInt(end, 10))
}
if offset != 0 {
values.Set("offset", strconv.FormatInt(offset, 10))
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_TRADES_HISTORY, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) QueryTrades(txid int64, showRelatedTrades bool) {
values := url.Values{}
values.Set("txid", strconv.FormatInt(txid, 10))
if showRelatedTrades {
values.Set("trades", "true")
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_QUERY_TRADES, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) OpenPositions(txid int64, showPL bool) {
values := url.Values{}
values.Set("txid", strconv.FormatInt(txid, 10))
if showPL {
values.Set("docalcs", "true")
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_OPEN_POSITIONS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetLedgers(symbol, asset, ledgerType string, start, end, offset int64) {
values := url.Values{}
if len(symbol) > 0 {
values.Set("aclass", symbol)
}
if len(asset) > 0 {
values.Set("asset", asset)
}
if len(ledgerType) > 0 {
values.Set("type", ledgerType)
}
if start != 0 {
values.Set("start", strconv.FormatInt(start, 10))
}
if end != 0 {
values.Set("end", strconv.FormatInt(end, 10))
}
if offset != 0 {
values.Set("offset", strconv.FormatInt(offset, 10))
}
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_LEDGERS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) QueryLedgers(id string) {
values := url.Values{}
values.Set("id", id)
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_QUERY_LEDGERS, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) GetTradeVolume(symbol string) {
values := url.Values{}
values.Set("pair", symbol)
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_TRADE_VOLUME, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) AddOrder(symbol, side, orderType string, price, price2, volume, leverage, position float64) {
values := url.Values{}
values.Set("pairs", symbol)
values.Set("type", side)
values.Set("ordertype", orderType)
values.Set("price", strconv.FormatFloat(price, 'f', 2, 64))
values.Set("price2", strconv.FormatFloat(price, 'f', 2, 64))
values.Set("volume", strconv.FormatFloat(volume, 'f', 2, 64))
values.Set("leverage", strconv.FormatFloat(leverage, 'f', 2, 64))
values.Set("position", strconv.FormatFloat(position, 'f', 2, 64))
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_ORDER_PLACE, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) CancelOrder(orderID int64) {
values := url.Values{}
values.Set("txid", strconv.FormatInt(orderID, 10))
result, err := k.SendAuthenticatedHTTPRequest(KRAKEN_ORDER_CANCEL, values)
if err != nil {
log.Println(err)
return
}
log.Println(result)
}
func (k *Kraken) SendAuthenticatedHTTPRequest(method string, values url.Values) (interface{}, error) {
path := fmt.Sprintf("/%s/private/%s", KRAKEN_API_VERSION, method)
values.Set("nonce", strconv.FormatInt(time.Now().UnixNano(), 10))
secret, err := base64.StdEncoding.DecodeString(k.APISecret)
if err != nil {
return nil, err
}
sha := sha256.New()
sha.Write([]byte(values.Get("nonce") + values.Encode()))
shasum := sha.Sum(nil)
hmac := hmac.New(sha512.New, []byte(secret))
hmac.Write(append([]byte(path), shasum...))
signature := base64.StdEncoding.EncodeToString(hmac.Sum(nil))
if k.Verbose {
log.Printf("Sending POST request to %s, path: %s.", KRAKEN_API_URL, path)
}
req, err := http.NewRequest("POST", KRAKEN_API_URL + path, strings.NewReader(values.Encode()))
req.Header.Set("API-Key", k.ClientKey)
req.Header.Set("API-Sign", signature)
client := &http.Client{}
resp, err := client.Do(req)
if err != nil {
return nil, errors.New("SendAuthenticatedHTTPRequest: Unable to send request")
}
contents, _ := ioutil.ReadAll(resp.Body)
if k.Verbose {
log.Printf("Recieved raw: \n%s\n", string(contents))
}
kresp := KrakenResponse{}
err = json.Unmarshal(contents, &kresp)
if err != nil {
return nil, errors.New("Unable to JSON response.")
}
if len(kresp.Error) != 0 {
return nil, errors.New(fmt.Sprintf("Kraken error: %s", kresp.Error))
}
resp.Body.Close()
return kresp.Result, nil
}

29
main.go
View File

@@ -19,6 +19,7 @@ type Exchange struct {
itbit ItBit
lakebtc LakeBTC
huobi HUOBI
kraken Kraken
}
type Bot struct {
@@ -73,6 +74,7 @@ func main() {
log.Printf("Available Exchanges: %d. Enabled Exchanges: %d.\n", len(bot.config.Exchanges), enabledExchanges)
log.Println("Bot Exchange support:")
bot.exchange.kraken.SetDefaults()
bot.exchange.btcchina.SetDefaults()
bot.exchange.bitstamp.SetDefaults()
bot.exchange.bitfinex.SetDefaults()
@@ -210,6 +212,21 @@ func main() {
log.Printf("%s Verbose output disabled.\n", exch.Name)
}
}
} else if bot.exchange.kraken.GetName() == exch.Name {
if !exch.Enabled {
bot.exchange.kraken.SetEnabled(false)
log.Printf("%s disabled.\n", exch.Name)
} else {
log.Printf("%s enabled.\n", exch.Name)
bot.exchange.kraken.SetAPIKeys(exch.APIKey, exch.APISecret)
if exch.Verbose {
bot.exchange.kraken.Verbose = true
log.Printf("%s Verbose output enabled.\n", exch.Name)
} else {
log.Printf("%s Verbose output disabled.\n", exch.Name)
}
}
} else if bot.exchange.lakebtc.GetName() == exch.Name {
if !exch.Enabled {
bot.exchange.lakebtc.SetEnabled(false)
@@ -251,15 +268,23 @@ func main() {
//temp until proper asynchronous method of getting pricing/order books is coded
for {
//spot
if bot.exchange.kraken.IsEnabled() {
go func() {
KrakenBTC := bot.exchange.kraken.GetTicker("XBTUSD")
log.Printf("Kraken BTC: %v\n", KrakenBTC)
}()
go func() {
KrakenLTC := bot.exchange.kraken.GetTicker("LTCUSD")
log.Printf("Kraken LTC: %v\n", KrakenLTC)
}()
}
if bot.exchange.lakebtc.IsEnabled() {
go func() {
LakeBTCTickerResponse := bot.exchange.lakebtc.GetTicker()
log.Printf("LakeBTC USD: Last %f (%f) High %f (%f) Low %f (%f)\n", LakeBTCTickerResponse.USD.Last, LakeBTCTickerResponse.CNY.Last, LakeBTCTickerResponse.USD.High, LakeBTCTickerResponse.CNY.High, LakeBTCTickerResponse.USD.Low, LakeBTCTickerResponse.CNY.Low)
}()
}
if bot.exchange.btcchina.IsEnabled() {
go func() {
BTCChinaBTC := bot.exchange.btcchina.GetTicker("btccny")