mirror of https://github.com/fatedier/frp.git
141 lines
3.2 KiB
Go
141 lines
3.2 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/fatedier/frp/client"
|
|
httppkg "github.com/fatedier/frp/pkg/util/http"
|
|
)
|
|
|
|
type Client struct {
|
|
address string
|
|
authUser string
|
|
authPwd string
|
|
}
|
|
|
|
func New(host string, port int) *Client {
|
|
return &Client{
|
|
address: net.JoinHostPort(host, strconv.Itoa(port)),
|
|
}
|
|
}
|
|
|
|
func (c *Client) SetAuth(user, pwd string) {
|
|
c.authUser = user
|
|
c.authPwd = pwd
|
|
}
|
|
|
|
func (c *Client) GetProxyStatus(ctx context.Context, name string) (*client.ProxyStatusResp, error) {
|
|
req, err := http.NewRequestWithContext(ctx, "GET", "http://"+c.address+"/api/status", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
content, err := c.do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
allStatus := make(client.StatusResp)
|
|
if err = json.Unmarshal([]byte(content), &allStatus); err != nil {
|
|
return nil, fmt.Errorf("unmarshal http response error: %s", strings.TrimSpace(content))
|
|
}
|
|
for _, pss := range allStatus {
|
|
for _, ps := range pss {
|
|
if ps.Name == name {
|
|
return &ps, nil
|
|
}
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("no proxy status found")
|
|
}
|
|
|
|
func (c *Client) GetAllProxyStatus(ctx context.Context) (client.StatusResp, error) {
|
|
req, err := http.NewRequestWithContext(ctx, "GET", "http://"+c.address+"/api/status", nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
content, err := c.do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
allStatus := make(client.StatusResp)
|
|
if err = json.Unmarshal([]byte(content), &allStatus); err != nil {
|
|
return nil, fmt.Errorf("unmarshal http response error: %s", strings.TrimSpace(content))
|
|
}
|
|
return allStatus, nil
|
|
}
|
|
|
|
func (c *Client) Reload(ctx context.Context, strictMode bool) error {
|
|
v := url.Values{}
|
|
if strictMode {
|
|
v.Set("strictConfig", "true")
|
|
}
|
|
queryStr := ""
|
|
if len(v) > 0 {
|
|
queryStr = "?" + v.Encode()
|
|
}
|
|
req, err := http.NewRequestWithContext(ctx, "GET", "http://"+c.address+"/api/reload"+queryStr, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = c.do(req)
|
|
return err
|
|
}
|
|
|
|
func (c *Client) Stop(ctx context.Context) error {
|
|
req, err := http.NewRequestWithContext(ctx, "POST", "http://"+c.address+"/api/stop", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = c.do(req)
|
|
return err
|
|
}
|
|
|
|
func (c *Client) GetConfig(ctx context.Context) (string, error) {
|
|
req, err := http.NewRequestWithContext(ctx, "GET", "http://"+c.address+"/api/config", nil)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return c.do(req)
|
|
}
|
|
|
|
func (c *Client) UpdateConfig(ctx context.Context, content string) error {
|
|
req, err := http.NewRequestWithContext(ctx, "PUT", "http://"+c.address+"/api/config", strings.NewReader(content))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = c.do(req)
|
|
return err
|
|
}
|
|
|
|
func (c *Client) setAuthHeader(req *http.Request) {
|
|
if c.authUser != "" || c.authPwd != "" {
|
|
req.Header.Set("Authorization", httppkg.BasicAuth(c.authUser, c.authPwd))
|
|
}
|
|
}
|
|
|
|
func (c *Client) do(req *http.Request) (string, error) {
|
|
c.setAuthHeader(req)
|
|
|
|
resp, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != 200 {
|
|
return "", fmt.Errorf("api status code [%d]", resp.StatusCode)
|
|
}
|
|
buf, err := io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(buf), nil
|
|
}
|