proxy

type APIInterceptor func(APITransaction)

type APITransaction struct {
	Request  RequestData
	Response ResponseData
	// ResponseTruncated is true when the captured response body was cut at
	// the proxy's capture limit; the full response was still forwarded.
	ResponseTruncated bool `json:"response_truncated,omitempty"`
}

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

type ProxyServer struct {

	// MaxBodyBytes bounds how much of a response body is captured. The full
	// response is still forwarded to the client. <= 0 means unlimited.
	MaxBodyBytes int64
	// WSInterceptor receives captured WebSocket sessions when connections
	// close. Nil means WebSocket sessions are not stored.
	WSInterceptor WebSocketInterceptor
	// MaxWSMessageBytes bounds how much of a WebSocket message payload is
	// captured per message. The full payload is still forwarded. <= 0 means
	// unlimited.
	MaxWSMessageBytes int64
	// MaxWSMessages bounds how many WebSocket messages are captured per
	// connection; forwarding continues unchanged after the limit. <= 0 means
	// unlimited.
	MaxWSMessages int
	// contains filtered or unexported fields
}

type RequestData struct {
	Method      string
	Path        string
	QueryParams url.Values
	Headers     http.Header
	Body        []byte
	Timestamp   time.Time
}

type ResponseData struct {
	StatusCode int
	Headers    http.Header
	Body       []byte
	Timestamp  time.Time
}

type Storage interface {
	Store(transaction APITransaction) error
	StoreWebSocketSession(session WebSocketSession) error
	GetAll() ([]APITransaction, error)
	GetAllWebSocketSessions() ([]WebSocketSession, error)
	Clear() error
}

type WSFrameHeader struct {
	Fin        bool
	Opcode     byte
	Masked     bool
	MaskKey    [4]byte
	PayloadLen uint64
	Header     []byte
}

type WebSocketInterceptor func(WebSocketSession)

type WebSocketMessage struct {
	Direction string    `json:"direction"`
	Opcode    int       `json:"opcode"`
	Payload   []byte    `json:"payload,omitempty"`
	Truncated bool      `json:"truncated,omitempty"`
	Timestamp time.Time `json:"timestamp,omitempty"`
}

type WebSocketSession struct {
	HandshakeRequest RequestData        `json:"handshake_request"`
	ResponseStatus   int                `json:"response_status"`
	ResponseHeaders  http.Header        `json:"response_headers,omitempty"`
	Subprotocol      string             `json:"subprotocol,omitempty"`
	Messages         []WebSocketMessage `json:"messages,omitempty"`
}