mirror of
https://github.com/yusing/godoxy.git
synced 2025-07-04 06:04:24 +02:00
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func (cfg *AgentConfig) Do(ctx context.Context, method, endpoint string, body io.Reader) (*http.Response, error) {
|
|
req, err := http.NewRequestWithContext(ctx, method, APIBaseURL+endpoint, body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cfg.httpClient.Do(req)
|
|
}
|
|
|
|
func (cfg *AgentConfig) Forward(req *http.Request, endpoint string) ([]byte, int, error) {
|
|
req = req.WithContext(req.Context())
|
|
req.URL.Host = AgentHost
|
|
req.URL.Scheme = "https"
|
|
req.URL.Path = APIEndpointBase + endpoint
|
|
req.RequestURI = ""
|
|
resp, err := cfg.httpClient.Do(req)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
defer resp.Body.Close()
|
|
data, _ := io.ReadAll(resp.Body)
|
|
return data, resp.StatusCode, nil
|
|
}
|
|
|
|
func (cfg *AgentConfig) Fetch(ctx context.Context, endpoint string) ([]byte, int, error) {
|
|
resp, err := cfg.Do(ctx, "GET", endpoint, nil)
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
defer resp.Body.Close()
|
|
data, _ := io.ReadAll(resp.Body)
|
|
return data, resp.StatusCode, nil
|
|
}
|
|
|
|
func (cfg *AgentConfig) Websocket(ctx context.Context, endpoint string) (*websocket.Conn, *http.Response, error) {
|
|
transport := cfg.Transport()
|
|
dialer := websocket.Dialer{
|
|
NetDialContext: transport.DialContext,
|
|
NetDialTLSContext: transport.DialTLSContext,
|
|
}
|
|
return dialer.DialContext(ctx, APIBaseURL+endpoint, http.Header{
|
|
"Host": {AgentHost},
|
|
})
|
|
}
|