mirror of
https://github.com/Gouryella/drip.git
synced 2026-02-23 21:00:44 +00:00
aead68bb62
enhancements - Add adaptive HTTP response handling with automatic streaming for large responses (>1MB) - Implement zero-copy streaming using buffer pools for better performance - Add compression module for reduced bandwidth usage - Add GitHub Container Registry workflow for automated Docker builds - Add production-optimized Dockerfile and docker-compose configuration - Simplify background mode with -d flag and improved daemon management - Update documentation with new command syntax and deployment guides - Clean up unused code and improve error handling - Fix lipgloss style usage (remove unnecessary .Copy() calls)
72 lines
2.5 KiB
Go
72 lines
2.5 KiB
Go
package protocol
|
|
|
|
// MessageType defines the type of tunnel message
|
|
type MessageType string
|
|
|
|
const (
|
|
// TypeRegister is sent when a client connects and gets a subdomain assigned
|
|
TypeRegister MessageType = "register"
|
|
// TypeRequest is sent from server to client when an HTTP request arrives
|
|
TypeRequest MessageType = "request"
|
|
// TypeResponse is sent from client to server with the HTTP response
|
|
TypeResponse MessageType = "response"
|
|
// TypeHeartbeat is sent periodically to keep the connection alive
|
|
TypeHeartbeat MessageType = "heartbeat"
|
|
// TypeError is sent when an error occurs
|
|
TypeError MessageType = "error"
|
|
)
|
|
|
|
// Message represents a tunnel protocol message
|
|
type Message struct {
|
|
Type MessageType `json:"type"`
|
|
ID string `json:"id,omitempty"`
|
|
Subdomain string `json:"subdomain,omitempty"`
|
|
Data map[string]interface{} `json:"data,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
// HTTPRequest represents an HTTP request to be forwarded
|
|
type HTTPRequest struct {
|
|
Method string `json:"method"`
|
|
URL string `json:"url"`
|
|
Headers map[string][]string `json:"headers"`
|
|
Body []byte `json:"body,omitempty"`
|
|
}
|
|
|
|
// HTTPRequestHead represents HTTP request headers for streaming (no body)
|
|
type HTTPRequestHead struct {
|
|
Method string `json:"method"`
|
|
URL string `json:"url"`
|
|
Headers map[string][]string `json:"headers"`
|
|
ContentLength int64 `json:"content_length"` // -1 for unknown/chunked
|
|
}
|
|
|
|
// HTTPResponse represents an HTTP response from the local service
|
|
type HTTPResponse struct {
|
|
StatusCode int `json:"status_code"`
|
|
Status string `json:"status"`
|
|
Headers map[string][]string `json:"headers"`
|
|
Body []byte `json:"body,omitempty"`
|
|
}
|
|
|
|
// HTTPResponseHead represents HTTP response headers for streaming (no body)
|
|
type HTTPResponseHead struct {
|
|
StatusCode int `json:"status_code"`
|
|
Status string `json:"status"`
|
|
Headers map[string][]string `json:"headers"`
|
|
ContentLength int64 `json:"content_length"` // -1 for unknown/chunked
|
|
}
|
|
|
|
// RegisterData contains information sent when a tunnel is registered
|
|
type RegisterData struct {
|
|
Subdomain string `json:"subdomain"`
|
|
URL string `json:"url"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
// ErrorData contains error information
|
|
type ErrorData struct {
|
|
Code string `json:"code"`
|
|
Message string `json:"message"`
|
|
}
|