mirror of
https://github.com/discountry/ritmex-bot.git
synced 2026-09-09 08:18:07 +00:00
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/elliottech/lighter-go/signer"
|
|
"github.com/elliottech/lighter-go/types"
|
|
)
|
|
|
|
var (
|
|
// DefaultExpireTime is a public var, so it can be changed directly in the SDK if required.
|
|
// The encouraged behaviour is the manually specify the TX deadline in types.TransactOpts.ExpiredAt
|
|
DefaultExpireTime = time.Minute*10 - time.Second // we need to give a second margin, to eliminate millisecond differences
|
|
)
|
|
|
|
type TxClient struct {
|
|
apiClient MinimalHTTPClient
|
|
chainId uint32
|
|
keyManager signer.KeyManager
|
|
accountIndex int64
|
|
apiKeyIndex uint8
|
|
}
|
|
|
|
// NewTxClient is linked to a specific (account, apiKey) pair
|
|
// apiKeyPrivateKey should be hex-encoded bytes generated using `hexutil.Encode(TxClient.GetKeyManager().PrvKeyBytes())`
|
|
func NewTxClient(apiClient MinimalHTTPClient, apiKeyPrivateKey string, accountIndex int64, apiKeyIndex uint8, chainId uint32) (*TxClient, error) {
|
|
// remove 0x from private key, if any, and parse to bytes
|
|
if len(apiKeyPrivateKey) < 2 {
|
|
return nil, fmt.Errorf("empty private key")
|
|
}
|
|
if apiKeyPrivateKey[:2] == "0x" {
|
|
apiKeyPrivateKey = apiKeyPrivateKey[2:]
|
|
}
|
|
|
|
b, err := hex.DecodeString(apiKeyPrivateKey)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
keyManager, err := signer.NewKeyManager(b)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &TxClient{
|
|
apiClient: apiClient,
|
|
apiKeyIndex: apiKeyIndex,
|
|
accountIndex: accountIndex,
|
|
chainId: chainId,
|
|
keyManager: keyManager,
|
|
}, nil
|
|
}
|
|
|
|
// FullFillDefaultOps returns a usable TransactOpts object if none was provided.
|
|
// This should not the be case for sharedlib, except for the nonce, which is optional.
|
|
// Still, the behaviour is implemented, so it can be extended easily by extending the code GO SDK.
|
|
func (c *TxClient) FullFillDefaultOps(ops *types.TransactOpts) (*types.TransactOpts, error) {
|
|
if ops == nil {
|
|
ops = new(types.TransactOpts)
|
|
}
|
|
if ops.ExpiredAt == 0 {
|
|
ops.ExpiredAt = time.Now().Add(DefaultExpireTime).UnixMilli()
|
|
}
|
|
if ops.FromAccountIndex == nil {
|
|
ops.FromAccountIndex = &c.accountIndex
|
|
}
|
|
if ops.ApiKeyIndex == nil {
|
|
ops.ApiKeyIndex = &c.apiKeyIndex
|
|
}
|
|
if ops.Nonce == nil || *ops.Nonce == -1 {
|
|
if c.apiClient == nil {
|
|
return nil, fmt.Errorf("nonce was not provided & HTTPClient is nil. Either provide the nonce or enable HTTPClient to get the nonce from Lighter")
|
|
}
|
|
nonce, err := c.apiClient.GetNextNonce(*ops.FromAccountIndex, *ops.ApiKeyIndex)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ops.Nonce = &nonce
|
|
}
|
|
|
|
return ops, nil
|
|
}
|
|
|
|
func (c *TxClient) GetChainId() uint32 {
|
|
return c.chainId
|
|
}
|
|
|
|
func (c *TxClient) GetKeyManager() signer.KeyManager {
|
|
return c.keyManager
|
|
}
|
|
|
|
func (c *TxClient) GetAccountIndex() int64 {
|
|
return c.accountIndex
|
|
}
|
|
|
|
func (c *TxClient) GetApiKeyIndex() uint8 {
|
|
return c.apiKeyIndex
|
|
}
|
|
|
|
func (c *TxClient) HTTP() MinimalHTTPClient {
|
|
return c.apiClient
|
|
}
|