97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
// Copyright (C) 2019 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
package cli
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto/tls"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/syncthing/syncthing/lib/config"
|
|
)
|
|
|
|
type APIClient struct {
|
|
http.Client
|
|
cfg config.GUIConfiguration
|
|
apikey string
|
|
}
|
|
|
|
func getClient(cfg config.GUIConfiguration) *APIClient {
|
|
httpClient := http.Client{
|
|
Transport: &http.Transport{
|
|
TLSClientConfig: &tls.Config{
|
|
InsecureSkipVerify: true,
|
|
},
|
|
DialContext: func(_ context.Context, _, _ string) (net.Conn, error) {
|
|
return net.Dial(cfg.Network(), cfg.Address())
|
|
},
|
|
},
|
|
}
|
|
return &APIClient{
|
|
Client: httpClient,
|
|
cfg: cfg,
|
|
apikey: cfg.APIKey,
|
|
}
|
|
}
|
|
|
|
func (c *APIClient) Endpoint() string {
|
|
if c.cfg.Network() == "unix" {
|
|
return "http://unix/"
|
|
}
|
|
url := c.cfg.URL()
|
|
if !strings.HasSuffix(url, "/") {
|
|
url += "/"
|
|
}
|
|
return url
|
|
}
|
|
|
|
func (c *APIClient) Do(req *http.Request) (*http.Response, error) {
|
|
req.Header.Set("X-API-Key", c.apikey)
|
|
resp, err := c.Client.Do(req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return resp, checkResponse(resp)
|
|
}
|
|
|
|
func (c *APIClient) Get(url string) (*http.Response, error) {
|
|
request, err := http.NewRequest("GET", c.Endpoint()+"rest/"+url, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.Do(request)
|
|
}
|
|
|
|
func (c *APIClient) Post(url, body string) (*http.Response, error) {
|
|
request, err := http.NewRequest("POST", c.Endpoint()+"rest/"+url, bytes.NewBufferString(body))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return c.Do(request)
|
|
}
|
|
|
|
func checkResponse(response *http.Response) error {
|
|
if response.StatusCode == 404 {
|
|
return errors.New("invalid endpoint or API call")
|
|
} else if response.StatusCode == 403 {
|
|
return errors.New("invalid API key")
|
|
} else if response.StatusCode != 200 {
|
|
data, err := responseToBArray(response)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
body := strings.TrimSpace(string(data))
|
|
return fmt.Errorf("unexpected HTTP status returned: %s\n%s", response.Status, body)
|
|
}
|
|
return nil
|
|
}
|