package trader import ( "context" "encoding/json" "fmt" "io" "math" "nofx/logger" "net/http" "strconv" "strings" "sync" "time" bybit "github.com/bybit-exchange/bybit.go.api" ) // BybitTrader Bybit USDT Perpetual Futures Trader type BybitTrader struct { client *bybit.Client // Balance cache cachedBalance map[string]interface{} balanceCacheTime time.Time balanceCacheMutex sync.RWMutex // Position cache cachedPositions []map[string]interface{} positionsCacheTime time.Time positionsCacheMutex sync.RWMutex // Trading pair precision cache (symbol -> qtyStep) qtyStepCache map[string]float64 qtyStepCacheMutex sync.RWMutex // Cache duration (15 seconds) cacheDuration time.Duration } // NewBybitTrader creates a Bybit trader func NewBybitTrader(apiKey, secretKey string) *BybitTrader { const src = "Up000938" client := bybit.NewBybitHttpClient(apiKey, secretKey, bybit.WithBaseURL(bybit.MAINNET)) // Set HTTP transport if client != nil && client.HTTPClient != nil { defaultTransport := client.HTTPClient.Transport if defaultTransport == nil { defaultTransport = http.DefaultTransport } client.HTTPClient.Transport = &headerRoundTripper{ base: defaultTransport, refererID: src, } } trader := &BybitTrader{ client: client, cacheDuration: 15 * time.Second, qtyStepCache: make(map[string]float64), } logger.Infof("🔵 [Bybit] Trader initialized") return trader } // headerRoundTripper HTTP RoundTripper for adding custom headers type headerRoundTripper struct { base http.RoundTripper refererID string } func (h *headerRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) { req.Header.Set("Referer", h.refererID) return h.base.RoundTrip(req) } // GetBalance retrieves account balance func (t *BybitTrader) GetBalance() (map[string]interface{}, error) { // Check cache t.balanceCacheMutex.RLock() if t.cachedBalance != nil && time.Since(t.balanceCacheTime) < t.cacheDuration { balance := t.cachedBalance t.balanceCacheMutex.RUnlock() return balance, nil } t.balanceCacheMutex.RUnlock() // Call API params := map[string]interface{}{ "accountType": "UNIFIED", } result, err := t.client.NewUtaBybitServiceWithParams(params).GetAccountWallet(context.Background()) if err != nil { return nil, fmt.Errorf("failed to get Bybit balance: %w", err) } if result.RetCode != 0 { return nil, fmt.Errorf("Bybit API error: %s", result.RetMsg) } // Extract balance information resultData, ok := result.Result.(map[string]interface{}) if !ok { return nil, fmt.Errorf("Bybit balance return format error") } list, _ := resultData["list"].([]interface{}) var totalEquity, availableBalance, totalWalletBalance, totalPerpUPL float64 = 0, 0, 0, 0 if len(list) > 0 { account, _ := list[0].(map[string]interface{}) if equityStr, ok := account["totalEquity"].(string); ok { totalEquity, _ = strconv.ParseFloat(equityStr, 64) } if availStr, ok := account["totalAvailableBalance"].(string); ok { availableBalance, _ = strconv.ParseFloat(availStr, 64) } // Bybit UNIFIED account wallet balance field if walletStr, ok := account["totalWalletBalance"].(string); ok { totalWalletBalance, _ = strconv.ParseFloat(walletStr, 64) } // Bybit perpetual contract unrealized PnL if uplStr, ok := account["totalPerpUPL"].(string); ok { totalPerpUPL, _ = strconv.ParseFloat(uplStr, 64) } } // If no totalWalletBalance, use totalEquity if totalWalletBalance == 0 { totalWalletBalance = totalEquity } balance := map[string]interface{}{ "totalEquity": totalEquity, "totalWalletBalance": totalWalletBalance, "availableBalance": availableBalance, "totalUnrealizedProfit": totalPerpUPL, "balance": totalEquity, // Compatible with other exchange formats } // Update cache t.balanceCacheMutex.Lock() t.cachedBalance = balance t.balanceCacheTime = time.Now() t.balanceCacheMutex.Unlock() return balance, nil } // GetPositions retrieves all positions func (t *BybitTrader) GetPositions() ([]map[string]interface{}, error) { // Check cache t.positionsCacheMutex.RLock() if t.cachedPositions != nil && time.Since(t.positionsCacheTime) < t.cacheDuration { positions := t.cachedPositions t.positionsCacheMutex.RUnlock() return positions, nil } t.positionsCacheMutex.RUnlock() // Call API params := map[string]interface{}{ "category": "linear", "settleCoin": "USDT", } result, err := t.client.NewUtaBybitServiceWithParams(params).GetPositionList(context.Background()) if err != nil { return nil, fmt.Errorf("failed to get Bybit positions: %w", err) } if result.RetCode != 0 { return nil, fmt.Errorf("Bybit API error: %s", result.RetMsg) } resultData, ok := result.Result.(map[string]interface{}) if !ok { return nil, fmt.Errorf("Bybit positions return format error") } list, _ := resultData["list"].([]interface{}) var positions []map[string]interface{} for _, item := range list { pos, ok := item.(map[string]interface{}) if !ok { continue } sizeStr, _ := pos["size"].(string) size, _ := strconv.ParseFloat(sizeStr, 64) // Skip empty positions if size == 0 { continue } entryPriceStr, _ := pos["avgPrice"].(string) entryPrice, _ := strconv.ParseFloat(entryPriceStr, 64) unrealisedPnlStr, _ := pos["unrealisedPnl"].(string) unrealisedPnl, _ := strconv.ParseFloat(unrealisedPnlStr, 64) leverageStr, _ := pos["leverage"].(string) leverage, _ := strconv.ParseFloat(leverageStr, 64) // Mark price markPriceStr, _ := pos["markPrice"].(string) markPrice, _ := strconv.ParseFloat(markPriceStr, 64) // Liquidation price liqPriceStr, _ := pos["liqPrice"].(string) liqPrice, _ := strconv.ParseFloat(liqPriceStr, 64) // Position created/updated time (milliseconds timestamp) createdTimeStr, _ := pos["createdTime"].(string) createdTime, _ := strconv.ParseInt(createdTimeStr, 10, 64) updatedTimeStr, _ := pos["updatedTime"].(string) updatedTime, _ := strconv.ParseInt(updatedTimeStr, 10, 64) positionSide, _ := pos["side"].(string) // Buy = LONG, Sell = SHORT // Convert to unified format side := "LONG" positionAmt := size if positionSide == "Sell" { side = "SHORT" positionAmt = -size } position := map[string]interface{}{ "symbol": pos["symbol"], "side": side, "positionAmt": positionAmt, "entryPrice": entryPrice, "markPrice": markPrice, "unRealizedProfit": unrealisedPnl, "unrealizedPnL": unrealisedPnl, "liquidationPrice": liqPrice, "leverage": leverage, "createdTime": createdTime, // Position open time (ms) "updatedTime": updatedTime, // Position last update time (ms) } positions = append(positions, position) } // Update cache t.positionsCacheMutex.Lock() t.cachedPositions = positions t.positionsCacheTime = time.Now() t.positionsCacheMutex.Unlock() return positions, nil } // OpenLong opens a long position func (t *BybitTrader) OpenLong(symbol string, quantity float64, leverage int) (map[string]interface{}, error) { // Set leverage first if err := t.SetLeverage(symbol, leverage); err != nil { logger.Infof("⚠️ [Bybit] Failed to set leverage: %v", err) } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": "Buy", "orderType": "Market", "qty": qtyStr, "positionIdx": 0, // One-way position mode } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return nil, fmt.Errorf("Bybit open long failed: %w", err) } // Clear cache t.clearCache() return t.parseOrderResult(result) } // OpenShort opens a short position func (t *BybitTrader) OpenShort(symbol string, quantity float64, leverage int) (map[string]interface{}, error) { // Set leverage first if err := t.SetLeverage(symbol, leverage); err != nil { logger.Infof("⚠️ [Bybit] Failed to set leverage: %v", err) } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": "Sell", "orderType": "Market", "qty": qtyStr, "positionIdx": 0, // One-way position mode } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return nil, fmt.Errorf("Bybit open short failed: %w", err) } // Clear cache t.clearCache() return t.parseOrderResult(result) } // CloseLong closes a long position func (t *BybitTrader) CloseLong(symbol string, quantity float64) (map[string]interface{}, error) { // If quantity = 0, get current position quantity if quantity == 0 { positions, err := t.GetPositions() if err != nil { return nil, err } for _, pos := range positions { if pos["symbol"] == symbol && pos["side"] == "LONG" { quantity = pos["positionAmt"].(float64) break } } } if quantity <= 0 { return nil, fmt.Errorf("no long position to close") } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": "Sell", // Close long with Sell "orderType": "Market", "qty": qtyStr, "positionIdx": 0, "reduceOnly": true, } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return nil, fmt.Errorf("Bybit close long failed: %w", err) } // Clear cache t.clearCache() return t.parseOrderResult(result) } // CloseShort closes a short position func (t *BybitTrader) CloseShort(symbol string, quantity float64) (map[string]interface{}, error) { // If quantity = 0, get current position quantity if quantity == 0 { positions, err := t.GetPositions() if err != nil { return nil, err } for _, pos := range positions { if pos["symbol"] == symbol && pos["side"] == "SHORT" { quantity = -pos["positionAmt"].(float64) // Short position is negative break } } } if quantity <= 0 { return nil, fmt.Errorf("no short position to close") } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": "Buy", // Close short with Buy "orderType": "Market", "qty": qtyStr, "positionIdx": 0, "reduceOnly": true, } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return nil, fmt.Errorf("Bybit close short failed: %w", err) } // Clear cache t.clearCache() return t.parseOrderResult(result) } // SetLeverage sets leverage func (t *BybitTrader) SetLeverage(symbol string, leverage int) error { params := map[string]interface{}{ "category": "linear", "symbol": symbol, "buyLeverage": fmt.Sprintf("%d", leverage), "sellLeverage": fmt.Sprintf("%d", leverage), } result, err := t.client.NewUtaBybitServiceWithParams(params).SetPositionLeverage(context.Background()) if err != nil { // If leverage is already at target value, Bybit will return an error, ignore this case if strings.Contains(err.Error(), "leverage not modified") { return nil } return fmt.Errorf("failed to set leverage: %w", err) } if result.RetCode != 0 && result.RetCode != 110043 { // 110043 = leverage not modified return fmt.Errorf("failed to set leverage: %s", result.RetMsg) } return nil } // SetMarginMode sets position margin mode func (t *BybitTrader) SetMarginMode(symbol string, isCrossMargin bool) error { tradeMode := 1 // Isolated margin if isCrossMargin { tradeMode = 0 // Cross margin } params := map[string]interface{}{ "category": "linear", "symbol": symbol, "tradeMode": tradeMode, } result, err := t.client.NewUtaBybitServiceWithParams(params).SwitchPositionMargin(context.Background()) if err != nil { if strings.Contains(err.Error(), "Cross/isolated margin mode is not modified") { return nil } return fmt.Errorf("failed to set margin mode: %w", err) } if result.RetCode != 0 && result.RetCode != 110026 { // already in target mode return fmt.Errorf("failed to set margin mode: %s", result.RetMsg) } return nil } // GetMarketPrice retrieves market price func (t *BybitTrader) GetMarketPrice(symbol string) (float64, error) { params := map[string]interface{}{ "category": "linear", "symbol": symbol, } result, err := t.client.NewUtaBybitServiceWithParams(params).GetMarketTickers(context.Background()) if err != nil { return 0, fmt.Errorf("failed to get market price: %w", err) } if result.RetCode != 0 { return 0, fmt.Errorf("API error: %s", result.RetMsg) } resultData, ok := result.Result.(map[string]interface{}) if !ok { return 0, fmt.Errorf("return format error") } list, _ := resultData["list"].([]interface{}) if len(list) == 0 { return 0, fmt.Errorf("price data not found for %s", symbol) } ticker, _ := list[0].(map[string]interface{}) lastPriceStr, _ := ticker["lastPrice"].(string) lastPrice, err := strconv.ParseFloat(lastPriceStr, 64) if err != nil { return 0, fmt.Errorf("failed to parse price: %w", err) } return lastPrice, nil } // SetStopLoss sets stop loss order func (t *BybitTrader) SetStopLoss(symbol string, positionSide string, quantity, stopPrice float64) error { side := "Sell" // LONG stop loss uses Sell if positionSide == "SHORT" { side = "Buy" // SHORT stop loss uses Buy } // Get current price to determine triggerDirection currentPrice, err := t.GetMarketPrice(symbol) if err != nil { return err } triggerDirection := 2 // Price fall trigger (default long stop loss) if stopPrice > currentPrice { triggerDirection = 1 // Price rise trigger (short stop loss) } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": side, "orderType": "Market", "qty": qtyStr, "triggerPrice": fmt.Sprintf("%v", stopPrice), "triggerDirection": triggerDirection, "triggerBy": "LastPrice", "reduceOnly": true, } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return fmt.Errorf("failed to set stop loss: %w", err) } if result.RetCode != 0 { return fmt.Errorf("failed to set stop loss: %s", result.RetMsg) } logger.Infof(" ✓ [Bybit] Stop loss order set: %s @ %.2f", symbol, stopPrice) return nil } // SetTakeProfit sets take profit order func (t *BybitTrader) SetTakeProfit(symbol string, positionSide string, quantity, takeProfitPrice float64) error { side := "Sell" // LONG take profit uses Sell if positionSide == "SHORT" { side = "Buy" // SHORT take profit uses Buy } // Get current price to determine triggerDirection currentPrice, err := t.GetMarketPrice(symbol) if err != nil { return err } triggerDirection := 1 // Price rise trigger (default long take profit) if takeProfitPrice < currentPrice { triggerDirection = 2 // Price fall trigger (short take profit) } // Use FormatQuantity to format quantity qtyStr, _ := t.FormatQuantity(symbol, quantity) params := map[string]interface{}{ "category": "linear", "symbol": symbol, "side": side, "orderType": "Market", "qty": qtyStr, "triggerPrice": fmt.Sprintf("%v", takeProfitPrice), "triggerDirection": triggerDirection, "triggerBy": "LastPrice", "reduceOnly": true, } result, err := t.client.NewUtaBybitServiceWithParams(params).PlaceOrder(context.Background()) if err != nil { return fmt.Errorf("failed to set take profit: %w", err) } if result.RetCode != 0 { return fmt.Errorf("failed to set take profit: %s", result.RetMsg) } logger.Infof(" ✓ [Bybit] Take profit order set: %s @ %.2f", symbol, takeProfitPrice) return nil } // CancelStopLossOrders cancels stop loss orders func (t *BybitTrader) CancelStopLossOrders(symbol string) error { return t.cancelConditionalOrders(symbol, "StopLoss") } // CancelTakeProfitOrders cancels take profit orders func (t *BybitTrader) CancelTakeProfitOrders(symbol string) error { return t.cancelConditionalOrders(symbol, "TakeProfit") } // CancelAllOrders cancels all pending orders func (t *BybitTrader) CancelAllOrders(symbol string) error { params := map[string]interface{}{ "category": "linear", "symbol": symbol, } _, err := t.client.NewUtaBybitServiceWithParams(params).CancelAllOrders(context.Background()) if err != nil { return fmt.Errorf("failed to cancel all orders: %w", err) } return nil } // CancelStopOrders cancels all stop loss and take profit orders func (t *BybitTrader) CancelStopOrders(symbol string) error { if err := t.CancelStopLossOrders(symbol); err != nil { logger.Infof("⚠️ [Bybit] Failed to cancel stop loss orders: %v", err) } if err := t.CancelTakeProfitOrders(symbol); err != nil { logger.Infof("⚠️ [Bybit] Failed to cancel take profit orders: %v", err) } return nil } // getQtyStep retrieves the quantity step for a trading pair func (t *BybitTrader) getQtyStep(symbol string) float64 { // Check cache first t.qtyStepCacheMutex.RLock() if step, ok := t.qtyStepCache[symbol]; ok { t.qtyStepCacheMutex.RUnlock() return step } t.qtyStepCacheMutex.RUnlock() // Call public API directly to get contract information url := fmt.Sprintf("https://api.bybit.com/v5/market/instruments-info?category=linear&symbol=%s", symbol) resp, err := http.Get(url) if err != nil { logger.Infof("⚠️ [Bybit] Failed to get precision info for %s: %v", symbol, err) return 1 // Default to integer } defer resp.Body.Close() body, err := io.ReadAll(resp.Body) if err != nil { return 1 } var result struct { RetCode int `json:"retCode"` Result struct { List []struct { LotSizeFilter struct { QtyStep string `json:"qtyStep"` } `json:"lotSizeFilter"` } `json:"list"` } `json:"result"` } if err := json.Unmarshal(body, &result); err != nil { return 1 } if result.RetCode != 0 || len(result.Result.List) == 0 { return 1 } qtyStep, _ := strconv.ParseFloat(result.Result.List[0].LotSizeFilter.QtyStep, 64) if qtyStep <= 0 { qtyStep = 1 } // Cache result t.qtyStepCacheMutex.Lock() t.qtyStepCache[symbol] = qtyStep t.qtyStepCacheMutex.Unlock() logger.Infof("🔵 [Bybit] %s qtyStep: %v", symbol, qtyStep) return qtyStep } // FormatQuantity formats quantity func (t *BybitTrader) FormatQuantity(symbol string, quantity float64) (string, error) { // Get qtyStep for this symbol qtyStep := t.getQtyStep(symbol) // Align quantity according to qtyStep (round down to nearest step) alignedQty := math.Floor(quantity/qtyStep) * qtyStep // Calculate required decimal places decimals := 0 if qtyStep < 1 { stepStr := strconv.FormatFloat(qtyStep, 'f', -1, 64) if idx := strings.Index(stepStr, "."); idx >= 0 { decimals = len(stepStr) - idx - 1 } } // Format format := fmt.Sprintf("%%.%df", decimals) formatted := fmt.Sprintf(format, alignedQty) return formatted, nil } // Helper methods func (t *BybitTrader) clearCache() { t.balanceCacheMutex.Lock() t.cachedBalance = nil t.balanceCacheMutex.Unlock() t.positionsCacheMutex.Lock() t.cachedPositions = nil t.positionsCacheMutex.Unlock() } func (t *BybitTrader) parseOrderResult(result *bybit.ServerResponse) (map[string]interface{}, error) { if result.RetCode != 0 { return nil, fmt.Errorf("order placement failed: %s", result.RetMsg) } resultData, ok := result.Result.(map[string]interface{}) if !ok { return nil, fmt.Errorf("return format error") } orderId, _ := resultData["orderId"].(string) return map[string]interface{}{ "orderId": orderId, "status": "NEW", }, nil } // GetOrderStatus retrieves order status func (t *BybitTrader) GetOrderStatus(symbol string, orderID string) (map[string]interface{}, error) { params := map[string]interface{}{ "category": "linear", "symbol": symbol, "orderId": orderID, } result, err := t.client.NewUtaBybitServiceWithParams(params).GetOrderHistory(context.Background()) if err != nil { return nil, fmt.Errorf("failed to get order status: %w", err) } if result.RetCode != 0 { return nil, fmt.Errorf("API error: %s", result.RetMsg) } resultData, ok := result.Result.(map[string]interface{}) if !ok { return nil, fmt.Errorf("return format error") } list, _ := resultData["list"].([]interface{}) if len(list) == 0 { return nil, fmt.Errorf("order %s not found", orderID) } order, _ := list[0].(map[string]interface{}) // Parse order data status, _ := order["orderStatus"].(string) avgPriceStr, _ := order["avgPrice"].(string) cumExecQtyStr, _ := order["cumExecQty"].(string) cumExecFeeStr, _ := order["cumExecFee"].(string) avgPrice, _ := strconv.ParseFloat(avgPriceStr, 64) executedQty, _ := strconv.ParseFloat(cumExecQtyStr, 64) commission, _ := strconv.ParseFloat(cumExecFeeStr, 64) // Convert status to unified format unifiedStatus := status switch status { case "Filled": unifiedStatus = "FILLED" case "New", "Created": unifiedStatus = "NEW" case "Cancelled", "Rejected": unifiedStatus = "CANCELED" case "PartiallyFilled": unifiedStatus = "PARTIALLY_FILLED" } return map[string]interface{}{ "orderId": orderID, "status": unifiedStatus, "avgPrice": avgPrice, "executedQty": executedQty, "commission": commission, }, nil } func (t *BybitTrader) cancelConditionalOrders(symbol string, orderType string) error { // First get all conditional orders params := map[string]interface{}{ "category": "linear", "symbol": symbol, "orderFilter": "StopOrder", // Conditional orders } result, err := t.client.NewUtaBybitServiceWithParams(params).GetOpenOrders(context.Background()) if err != nil { return fmt.Errorf("failed to get conditional orders: %w", err) } if result.RetCode != 0 { return nil // No orders } resultData, ok := result.Result.(map[string]interface{}) if !ok { return nil } list, _ := resultData["list"].([]interface{}) // Cancel matching orders for _, item := range list { order, ok := item.(map[string]interface{}) if !ok { continue } orderId, _ := order["orderId"].(string) stopOrderType, _ := order["stopOrderType"].(string) // Filter by type shouldCancel := false if orderType == "StopLoss" && (stopOrderType == "StopLoss" || stopOrderType == "Stop") { shouldCancel = true } if orderType == "TakeProfit" && (stopOrderType == "TakeProfit" || stopOrderType == "PartialTakeProfit") { shouldCancel = true } if shouldCancel && orderId != "" { cancelParams := map[string]interface{}{ "category": "linear", "symbol": symbol, "orderId": orderId, } t.client.NewUtaBybitServiceWithParams(cancelParams).CancelOrder(context.Background()) } } return nil }