bridge

type BridgeConnectionInfo struct {
	ExpiresAt    *time.Time `json:"expires_at"`
	WebsocketURL string     `json:"websocket_url"`
	StreamURL    string     `json:"stream_url"`
	ClientURL    string     `json:"client_url"`
	RepliesURL   string     `json:"replies_url"`
	RefreshURL   string     `json:"refresh_url"`
}

type Client struct {
	// contains filtered or unexported fields
}

type Handler interface {
	// OnConnect is called when the bridge client is connected to the bridge server
	OnConnect(client *Client)

	// OnDisconnect is called when the bridge client is disconnected from the bridge server
	OnDisconnect(client *Client)

	// OnHeader is called when a header is received from the bridge. this will only be called once before any data is sent.
	OnHeader(client *Client, id string, headers map[string]string)

	// OnData is called when a data is received from the bridge. this will be called multiple times if the data is large.
	OnData(client *Client, id string, data []byte)

	// OnClose is called when the bridge request is completed and no more data will be sent
	OnClose(client *Client, id string)

	// OnError is called when an error occurs at any point in the bridge client
	OnError(client *Client, err error)
}

type HandlerCallback struct {
	OnConnectFunc    func(client *Client)
	OnDisconnectFunc func(client *Client)
	OnHeaderFunc     func(client *Client, id string, headers map[string]string)
	OnDataFunc       func(client *Client, id string, data []byte)
	OnCloseFunc      func(client *Client, id string)
	OnErrorFunc      func(client *Client, err error)
}

type Options struct {
	// Context is the context for the bridge client (optional)
	Context context.Context
	// Logger is the logger for the bridge client (optional)
	Logger logger.Logger
	// URL is the URL of the bridge (optional)
	URL string
	// APIKey is the API key for the bridge (optional, will use AGENTUITY_API_KEY environment variable if not set)
	APIKey string
	// Handler is the handler for the bridge client (required)
	Handler Handler
	// ConnectionInfo is the connection info for the bridge client (optional) which can be used to pre-populate the connection info
	ConnectionInfo *BridgeConnectionInfo
}