bundles.go

package jitorpc

import (
	"encoding/json"
	"fmt"
	"math/rand"
)

type TipAccount struct {
	Address string `json:"address"`
}

type BundleStatusResponse struct {
	Context struct {
		Slot int64 `json:"slot"`
	} `json:"context"`
	Value []struct {
		BundleID           string   `json:"bundle_id"`
		Transactions       []string `json:"transactions"`
		Slot               int64    `json:"slot"`
		ConfirmationStatus string   `json:"confirmation_status"`
		Err                struct {
			Ok interface{} `json:"Ok"`
		} `json:"err"`
	} `json:"value"`
}

func (c *JitoJsonRpcClient) GetTipAccounts() (json.RawMessage, error) {
	endpoint := "/bundles"
	if c.UUID != "" {
		endpoint = fmt.Sprintf("%s?uuid=%s", endpoint, c.UUID)
	}
	return c.sendRequest(endpoint, "getTipAccounts", nil)
}

func (c *JitoJsonRpcClient) GetRandomTipAccount() (*TipAccount, error) {
	rawResponse, err := c.GetTipAccounts()
	if err != nil {
		return nil, err
	}

	var tipAddresses []string
	err = json.Unmarshal(rawResponse, &tipAddresses)
	if err != nil {
		return nil, fmt.Errorf("failed to unmarshal tip accounts: %w", err)
	}

	if len(tipAddresses) == 0 {
		return nil, fmt.Errorf("no tip accounts available")
	}

	randomIndex := rand.Intn(len(tipAddresses))
	return &TipAccount{Address: tipAddresses[randomIndex]}, nil
}

func (c *JitoJsonRpcClient) GetBundleStatuses(bundleIds []string) (*BundleStatusResponse, error) {
	endpoint := "/bundles"
	if c.UUID != "" {
		endpoint = fmt.Sprintf("%s?uuid=%s", endpoint, c.UUID)
	}
	params := [][]string{bundleIds}
	responseBody, err := c.sendRequest(endpoint, "getBundleStatuses", params)
	if err != nil {
		return nil, err
	}

	var response BundleStatusResponse
	if err := json.Unmarshal(responseBody, &response); err != nil {
		return nil, fmt.Errorf("failed to unmarshal bundle statuses: %w", err)
	}

	return &response, nil
}

func (c *JitoJsonRpcClient) SendBundle(params interface{}) (json.RawMessage, error) {
	endpoint := "/bundles"
	if c.UUID != "" {
		endpoint = fmt.Sprintf("%s?uuid=%s", endpoint, c.UUID)
	}
	return c.sendRequest(endpoint, "sendBundle", params)
}

func (c *JitoJsonRpcClient) GetInflightBundleStatuses(params interface{}) (json.RawMessage, error) {
	endpoint := "/bundles"
	if c.UUID != "" {
		endpoint = fmt.Sprintf("%s?uuid=%s", endpoint, c.UUID)
	}
	return c.sendRequest(endpoint, "getInflightBundleStatuses", params)
}