mirror of
https://github.com/Gouryella/drip.git
synced 2026-02-23 21:00:44 +00:00
dd54e79ad7
- Optimized terminal output style and configuration management using libraries such as `lipgloss` and `go-json` - Introduced the `ui` package to unify the display logic of colors, tables, and prompts - Updated the README document structure and installation script links to improve readability and internationalization support - Improved the interaction flow and log display effects of the daemon startup and attach commands - Fixed some command parameter parsing issues, improving program robustness and user onboarding experience
50 lines
1.8 KiB
Go
50 lines
1.8 KiB
Go
package protocol
|
|
|
|
import json "github.com/goccy/go-json"
|
|
|
|
// RegisterRequest is sent by client to register a tunnel
|
|
type RegisterRequest struct {
|
|
Token string `json:"token"` // Authentication token
|
|
CustomSubdomain string `json:"custom_subdomain"` // Optional custom subdomain
|
|
TunnelType TunnelType `json:"tunnel_type"` // http, tcp, udp
|
|
LocalPort int `json:"local_port"` // Local port to forward to
|
|
}
|
|
|
|
// RegisterResponse is sent by server after successful registration
|
|
type RegisterResponse struct {
|
|
Subdomain string `json:"subdomain"` // Assigned subdomain
|
|
Port int `json:"port,omitempty"` // Assigned TCP port (for TCP tunnels)
|
|
URL string `json:"url"` // Full tunnel URL
|
|
Message string `json:"message"` // Success message
|
|
}
|
|
|
|
// ErrorMessage represents an error
|
|
type ErrorMessage struct {
|
|
Code string `json:"code"` // Error code
|
|
Message string `json:"message"` // Error message
|
|
}
|
|
|
|
// DataHeader represents metadata for a data frame
|
|
type DataHeader struct {
|
|
StreamID string `json:"stream_id"` // Unique stream identifier
|
|
RequestID string `json:"request_id"` // Request identifier (for HTTP)
|
|
Type string `json:"type"` // "data", "response", "close", "http_request", "http_response"
|
|
IsLast bool `json:"is_last"` // Is this the last frame for this stream
|
|
}
|
|
|
|
// TCPData represents TCP tunnel data
|
|
type TCPData struct {
|
|
StreamID string `json:"stream_id"` // Stream identifier
|
|
Data []byte `json:"data"` // Raw TCP data
|
|
IsClose bool `json:"is_close"` // Close this stream
|
|
}
|
|
|
|
// Marshal helpers
|
|
func MarshalJSON(v interface{}) ([]byte, error) {
|
|
return json.Marshal(v)
|
|
}
|
|
|
|
func UnmarshalJSON(data []byte, v interface{}) error {
|
|
return json.Unmarshal(data, v)
|
|
}
|