mirror of
https://github.com/c9s/bbgo.git
synced 2024-09-20 08:11:08 +00:00
kucoin: implement api client
This commit is contained in:
c9s
parent
0c7bbba675
commit
cd69994647
70
examples/kucoin-subaccount/main.go
Normal file
70
examples/kucoin-subaccount/main.go
Normal file
|
|
@ -0,0 +1,70 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/exchange/kucoin/kucoinapi"
|
||||
"github.com/joho/godotenv"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/spf13/viper"
|
||||
)
|
||||
|
||||
func init() {
|
||||
rootCmd.PersistentFlags().String("kucoin-api-key", "", "okex api key")
|
||||
rootCmd.PersistentFlags().String("kucoin-api-secret", "", "okex api secret")
|
||||
rootCmd.PersistentFlags().String("kucoin-api-passphrase", "", "okex api secret")
|
||||
}
|
||||
|
||||
var rootCmd = &cobra.Command{
|
||||
Use: "kucoin-subaccount",
|
||||
Short: "kucoin subaccount",
|
||||
|
||||
// SilenceUsage is an option to silence usage when an error occurs.
|
||||
SilenceUsage: true,
|
||||
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
subAccounts, err := client.AccountService.QuerySubAccounts()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
log.Infof("subAccounts: %+v", subAccounts)
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
var client *kucoinapi.RestClient = nil
|
||||
|
||||
func main() {
|
||||
if _, err := os.Stat(".env.local"); err == nil {
|
||||
if err := godotenv.Load(".env.local"); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
viper.AutomaticEnv()
|
||||
viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
|
||||
|
||||
if err := viper.BindPFlags(rootCmd.PersistentFlags()); err != nil {
|
||||
log.WithError(err).Error("bind pflags error")
|
||||
}
|
||||
|
||||
client = kucoinapi.NewClient()
|
||||
|
||||
key, secret, passphrase := viper.GetString("kucoin-api-key"),
|
||||
viper.GetString("kucoin-api-secret"),
|
||||
viper.GetString("kucoin-api-passphrase")
|
||||
|
||||
if len(key) == 0 || len(secret) == 0 || len(passphrase) == 0 {
|
||||
log.Fatal("empty key, secret or passphrase")
|
||||
}
|
||||
|
||||
client.Auth(key, secret, passphrase)
|
||||
|
||||
if err := rootCmd.ExecuteContext(context.Background()); err != nil {
|
||||
log.WithError(err).Error("cmd error")
|
||||
}
|
||||
}
|
||||
47
pkg/exchange/kucoin/exchange.go
Normal file
47
pkg/exchange/kucoin/exchange.go
Normal file
|
|
@ -0,0 +1,47 @@
|
|||
package kucoin
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/types"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
// OKB is the platform currency of OKEx, pre-allocate static string here
|
||||
const KCS = "KCS"
|
||||
|
||||
var log = logrus.WithFields(logrus.Fields{
|
||||
"exchange": "kucoin",
|
||||
})
|
||||
|
||||
type Exchange struct {
|
||||
key, secret, passphrase string
|
||||
}
|
||||
|
||||
func (e *Exchange) NewStream() types.Stream {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (e *Exchange) QueryMarkets(ctx context.Context) (types.MarketMap, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (e *Exchange) QueryTicker(ctx context.Context, symbol string) (*types.Ticker, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (e *Exchange) QueryTickers(ctx context.Context, symbol ...string) (map[string]types.Ticker, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func (e *Exchange) QueryKLines(ctx context.Context, symbol string, interval types.Interval, options types.KLineQueryOptions) ([]types.KLine, error) {
|
||||
panic("implement me")
|
||||
}
|
||||
|
||||
func New(key, secret, passphrase string) *Exchange {
|
||||
return &Exchange{
|
||||
key: key,
|
||||
secret: secret,
|
||||
passphrase: passphrase,
|
||||
}
|
||||
}
|
||||
37
pkg/exchange/kucoin/kucoinapi/account.go
Normal file
37
pkg/exchange/kucoin/kucoinapi/account.go
Normal file
|
|
@ -0,0 +1,37 @@
|
|||
package kucoinapi
|
||||
|
||||
|
||||
type AccountService struct {
|
||||
client *RestClient
|
||||
}
|
||||
|
||||
type SubAccount struct {
|
||||
UserID string `json:"userId"`
|
||||
Name string `json:"subName"`
|
||||
Type string `json:"type"`
|
||||
Remark string `json:"remarks"`
|
||||
}
|
||||
|
||||
func (s *AccountService) QuerySubAccounts() ([]SubAccount, error) {
|
||||
req, err := s.client.newAuthenticatedRequest("GET", "/api/v1/sub/user", nil, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := s.client.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var apiResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data []SubAccount `json:"data"`
|
||||
}
|
||||
|
||||
if err := response.DecodeJSON(&apiResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return apiResponse.Data, nil
|
||||
}
|
||||
413
pkg/exchange/kucoin/kucoinapi/client.go
Normal file
413
pkg/exchange/kucoin/kucoinapi/client.go
Normal file
|
|
@ -0,0 +1,413 @@
|
|||
package kucoinapi
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/hmac"
|
||||
"crypto/sha256"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/c9s/bbgo/pkg/fixedpoint"
|
||||
"github.com/c9s/bbgo/pkg/types"
|
||||
"github.com/c9s/bbgo/pkg/util"
|
||||
"github.com/pkg/errors"
|
||||
log "github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
const defaultHTTPTimeout = time.Second * 15
|
||||
const RestBaseURL = "https://api.kucoin.com/api"
|
||||
const SandboxRestBaseURL = "https://openapi-sandbox.kucoin.com/api"
|
||||
|
||||
type SideType string
|
||||
|
||||
const (
|
||||
SideTypeBuy SideType = "buy"
|
||||
SideTypeSell SideType = "sell"
|
||||
)
|
||||
|
||||
type OrderType string
|
||||
|
||||
const (
|
||||
OrderTypeMarket OrderType = "market"
|
||||
OrderTypeLimit OrderType = "limit"
|
||||
OrderTypePostOnly OrderType = "post_only"
|
||||
OrderTypeFOK OrderType = "fok"
|
||||
OrderTypeIOC OrderType = "ioc"
|
||||
)
|
||||
|
||||
type InstrumentType string
|
||||
|
||||
const (
|
||||
InstrumentTypeSpot InstrumentType = "SPOT"
|
||||
InstrumentTypeSwap InstrumentType = "SWAP"
|
||||
InstrumentTypeFutures InstrumentType = "FUTURES"
|
||||
InstrumentTypeOption InstrumentType = "OPTION"
|
||||
)
|
||||
|
||||
type OrderState string
|
||||
|
||||
const (
|
||||
OrderStateCanceled OrderState = "canceled"
|
||||
OrderStateLive OrderState = "live"
|
||||
OrderStatePartiallyFilled OrderState = "partially_filled"
|
||||
OrderStateFilled OrderState = "filled"
|
||||
)
|
||||
|
||||
type RestClient struct {
|
||||
BaseURL *url.URL
|
||||
|
||||
client *http.Client
|
||||
|
||||
Key, Secret, Passphrase string
|
||||
KeyVersion string
|
||||
|
||||
AccountService *AccountService
|
||||
// TradeService *TradeService
|
||||
// PublicDataService *PublicDataService
|
||||
// MarketDataService *MarketDataService
|
||||
}
|
||||
|
||||
func NewClient() *RestClient {
|
||||
u, err := url.Parse(RestBaseURL)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
client := &RestClient{
|
||||
BaseURL: u,
|
||||
KeyVersion: "2",
|
||||
client: &http.Client{
|
||||
Timeout: defaultHTTPTimeout,
|
||||
},
|
||||
}
|
||||
|
||||
client.AccountService = &AccountService{ client: client }
|
||||
// client.TradeService = &TradeService{client: client}
|
||||
// client.PublicDataService = &PublicDataService{client: client}
|
||||
// client.MarketDataService = &MarketDataService{client: client}
|
||||
return client
|
||||
}
|
||||
|
||||
func (c *RestClient) Auth(key, secret, passphrase string) {
|
||||
c.Key = key
|
||||
c.Secret = secret
|
||||
c.Passphrase = passphrase
|
||||
}
|
||||
|
||||
// NewRequest create new API request. Relative url can be provided in refURL.
|
||||
func (c *RestClient) newRequest(method, refURL string, params url.Values, body []byte) (*http.Request, error) {
|
||||
rel, err := url.Parse(refURL)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if params != nil {
|
||||
rel.RawQuery = params.Encode()
|
||||
}
|
||||
|
||||
pathURL := c.BaseURL.ResolveReference(rel)
|
||||
return http.NewRequest(method, pathURL.String(), bytes.NewReader(body))
|
||||
}
|
||||
|
||||
// sendRequest sends the request to the API server and handle the response
|
||||
func (c *RestClient) sendRequest(req *http.Request) (*util.Response, error) {
|
||||
resp, err := c.client.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// newResponse reads the response body and return a new Response object
|
||||
response, err := util.NewResponse(resp)
|
||||
if err != nil {
|
||||
return response, err
|
||||
}
|
||||
|
||||
// Check error, if there is an error, return the ErrorResponse struct type
|
||||
if response.IsError() {
|
||||
return response, errors.New(string(response.Body))
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
|
||||
// newAuthenticatedRequest creates new http request for authenticated routes.
|
||||
func (c *RestClient) newAuthenticatedRequest(method, refURL string, params url.Values, payload interface{}) (*http.Request, error) {
|
||||
if len(c.Key) == 0 {
|
||||
return nil, errors.New("empty api key")
|
||||
}
|
||||
|
||||
if len(c.Secret) == 0 {
|
||||
return nil, errors.New("empty api secret")
|
||||
}
|
||||
|
||||
rel, err := url.Parse(refURL)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if params != nil {
|
||||
rel.RawQuery = params.Encode()
|
||||
}
|
||||
|
||||
pathURL := c.BaseURL.ResolveReference(rel)
|
||||
path := pathURL.Path
|
||||
if rel.RawQuery != "" {
|
||||
path += "?" + rel.RawQuery
|
||||
}
|
||||
|
||||
// set location to UTC so that it outputs "2020-12-08T09:08:57.715Z"
|
||||
t := time.Now().In(time.UTC)
|
||||
// timestamp := t.Format("2006-01-02T15:04:05.999Z07:00")
|
||||
timestamp := strconv.FormatInt(t.UnixMilli(), 10)
|
||||
|
||||
var body []byte
|
||||
|
||||
if payload != nil {
|
||||
switch v := payload.(type) {
|
||||
case string:
|
||||
body = []byte(v)
|
||||
|
||||
case []byte:
|
||||
body = v
|
||||
|
||||
default:
|
||||
body, err = json.Marshal(v)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
signKey := timestamp + strings.ToUpper(method) + path + string(body)
|
||||
signature := sign(c.Secret, signKey)
|
||||
|
||||
req, err := http.NewRequest(method, pathURL.String(), bytes.NewReader(body))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req.Header.Add("Content-Type", "application/json")
|
||||
req.Header.Add("Accept", "application/json")
|
||||
req.Header.Add("KC-API-KEY", c.Key)
|
||||
req.Header.Add("KC-API-SIGN", signature)
|
||||
req.Header.Add("KC-API-TIMESTAMP", timestamp)
|
||||
req.Header.Add("KC-API-PASSPHRASE", sign(c.Secret, c.Passphrase))
|
||||
req.Header.Add("KC-API-KEY-VERSION", c.KeyVersion)
|
||||
|
||||
log.Infof("%+v", req.Header)
|
||||
return req, nil
|
||||
}
|
||||
|
||||
type BalanceDetail struct {
|
||||
Currency string `json:"ccy"`
|
||||
Available fixedpoint.Value `json:"availEq"`
|
||||
CashBalance fixedpoint.Value `json:"cashBal"`
|
||||
OrderFrozen fixedpoint.Value `json:"ordFrozen"`
|
||||
Frozen fixedpoint.Value `json:"frozenBal"`
|
||||
Equity fixedpoint.Value `json:"eq"`
|
||||
EquityInUSD fixedpoint.Value `json:"eqUsd"`
|
||||
UpdateTime types.MillisecondTimestamp `json:"uTime"`
|
||||
UnrealizedProfitAndLoss fixedpoint.Value `json:"upl"`
|
||||
}
|
||||
|
||||
type Account struct {
|
||||
TotalEquityInUSD fixedpoint.Value `json:"totalEq"`
|
||||
UpdateTime string `json:"uTime"`
|
||||
Details []BalanceDetail `json:"details"`
|
||||
}
|
||||
|
||||
func (c *RestClient) AccountBalances() (*Account, error) {
|
||||
req, err := c.newAuthenticatedRequest("GET", "/api/v5/account/balance", nil, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := c.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var balanceResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data []Account `json:"data"`
|
||||
}
|
||||
|
||||
if err := response.DecodeJSON(&balanceResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(balanceResponse.Data) == 0 {
|
||||
return nil, errors.New("empty account data")
|
||||
}
|
||||
|
||||
return &balanceResponse.Data[0], nil
|
||||
}
|
||||
|
||||
type AssetBalance struct {
|
||||
Currency string `json:"ccy"`
|
||||
Balance fixedpoint.Value `json:"bal"`
|
||||
Frozen fixedpoint.Value `json:"frozenBal,omitempty"`
|
||||
Available fixedpoint.Value `json:"availBal,omitempty"`
|
||||
}
|
||||
|
||||
type AssetBalanceList []AssetBalance
|
||||
|
||||
func (c *RestClient) AssetBalances() (AssetBalanceList, error) {
|
||||
req, err := c.newAuthenticatedRequest("GET", "/api/v5/asset/balances", nil, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := c.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var balanceResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data AssetBalanceList `json:"data"`
|
||||
}
|
||||
if err := response.DecodeJSON(&balanceResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return balanceResponse.Data, nil
|
||||
}
|
||||
|
||||
type AssetCurrency struct {
|
||||
Currency string `json:"ccy"`
|
||||
Name string `json:"name"`
|
||||
Chain string `json:"chain"`
|
||||
CanDeposit bool `json:"canDep"`
|
||||
CanWithdraw bool `json:"canWd"`
|
||||
CanInternal bool `json:"canInternal"`
|
||||
MinWithdrawalFee fixedpoint.Value `json:"minFee"`
|
||||
MaxWithdrawalFee fixedpoint.Value `json:"maxFee"`
|
||||
MinWithdrawalThreshold fixedpoint.Value `json:"minWd"`
|
||||
}
|
||||
|
||||
func (c *RestClient) AssetCurrencies() ([]AssetCurrency, error) {
|
||||
req, err := c.newAuthenticatedRequest("GET", "/api/v5/asset/currencies", nil, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := c.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var currencyResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data []AssetCurrency `json:"data"`
|
||||
}
|
||||
|
||||
if err := response.DecodeJSON(¤cyResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return currencyResponse.Data, nil
|
||||
}
|
||||
|
||||
type MarketTicker struct {
|
||||
InstrumentType string `json:"instType"`
|
||||
InstrumentID string `json:"instId"`
|
||||
|
||||
// last traded price
|
||||
Last fixedpoint.Value `json:"last"`
|
||||
|
||||
// last traded size
|
||||
LastSize fixedpoint.Value `json:"lastSz"`
|
||||
|
||||
AskPrice fixedpoint.Value `json:"askPx"`
|
||||
AskSize fixedpoint.Value `json:"askSz"`
|
||||
|
||||
BidPrice fixedpoint.Value `json:"bidPx"`
|
||||
BidSize fixedpoint.Value `json:"bidSz"`
|
||||
|
||||
Open24H fixedpoint.Value `json:"open24h"`
|
||||
High24H fixedpoint.Value `json:"high24H"`
|
||||
Low24H fixedpoint.Value `json:"low24H"`
|
||||
Volume24H fixedpoint.Value `json:"vol24h"`
|
||||
VolumeCurrency24H fixedpoint.Value `json:"volCcy24h"`
|
||||
|
||||
// Millisecond timestamp
|
||||
Timestamp types.MillisecondTimestamp `json:"ts"`
|
||||
}
|
||||
|
||||
func (c *RestClient) MarketTicker(instId string) (*MarketTicker, error) {
|
||||
// SPOT, SWAP, FUTURES, OPTION
|
||||
var params = url.Values{}
|
||||
params.Add("instId", instId)
|
||||
|
||||
req, err := c.newRequest("GET", "/api/v5/market/ticker", params, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := c.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tickerResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data []MarketTicker `json:"data"`
|
||||
}
|
||||
if err := response.DecodeJSON(&tickerResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(tickerResponse.Data) == 0 {
|
||||
return nil, fmt.Errorf("ticker of %s not found", instId)
|
||||
}
|
||||
|
||||
return &tickerResponse.Data[0], nil
|
||||
}
|
||||
|
||||
func (c *RestClient) MarketTickers(instType InstrumentType) ([]MarketTicker, error) {
|
||||
// SPOT, SWAP, FUTURES, OPTION
|
||||
var params = url.Values{}
|
||||
params.Add("instType", string(instType))
|
||||
|
||||
req, err := c.newRequest("GET", "/api/v5/market/tickers", params, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
response, err := c.sendRequest(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var tickerResponse struct {
|
||||
Code string `json:"code"`
|
||||
Message string `json:"msg"`
|
||||
Data []MarketTicker `json:"data"`
|
||||
}
|
||||
if err := response.DecodeJSON(&tickerResponse); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return tickerResponse.Data, nil
|
||||
}
|
||||
|
||||
func sign(secret, payload string) string {
|
||||
var sig = hmac.New(sha256.New, []byte(secret))
|
||||
_, err := sig.Write([]byte(payload))
|
||||
if err != nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
return base64.StdEncoding.EncodeToString(sig.Sum(nil))
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user