154 lines
3.2 KiB
Go
154 lines
3.2 KiB
Go
package client
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/sqshq/sampler/metadata"
|
|
"io/ioutil"
|
|
"net/http"
|
|
)
|
|
|
|
const (
|
|
backendUrl = "http://localhost/api/v1"
|
|
installationPath = "/telemetry/installation"
|
|
statisticsPath = "/telemetry/statistics"
|
|
crashPath = "/telemetry/crash"
|
|
registrationPath = "/license/registration"
|
|
verificationPath = "/license/verification"
|
|
)
|
|
|
|
// BackendClient is used to verify license and to send telemetry
|
|
// for analyses (anonymous usage data statistics and crash reports)
|
|
type BackendClient struct {
|
|
client http.Client
|
|
}
|
|
|
|
func NewBackendClient() *BackendClient {
|
|
return &BackendClient{
|
|
client: http.Client{},
|
|
}
|
|
}
|
|
|
|
func (c *BackendClient) ReportInstallation(statistics *metadata.Statistics) {
|
|
|
|
buf := new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(statistics)
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), statistics)
|
|
}
|
|
|
|
_, err = sendRequest(backendUrl+installationPath, buf)
|
|
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), statistics)
|
|
}
|
|
}
|
|
|
|
func (c *BackendClient) ReportStatistics(statistics *metadata.Statistics) {
|
|
|
|
buf := new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(statistics)
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), statistics)
|
|
}
|
|
|
|
_, err = sendRequest(backendUrl+statisticsPath, buf)
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), statistics)
|
|
}
|
|
}
|
|
|
|
func (c *BackendClient) ReportCrash(error string, statistics *metadata.Statistics) {
|
|
|
|
req := struct {
|
|
Error string
|
|
Statistics *metadata.Statistics
|
|
}{
|
|
error,
|
|
statistics,
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(req)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
_, _ = sendRequest(backendUrl+crashPath, buf)
|
|
}
|
|
|
|
func (c *BackendClient) RegisterLicenseKey(licenseKey string, statistics *metadata.Statistics) (*metadata.License, error) {
|
|
|
|
req := struct {
|
|
LicenseKey string
|
|
Statistics *metadata.Statistics
|
|
}{
|
|
licenseKey,
|
|
statistics,
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(req)
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), statistics)
|
|
}
|
|
|
|
response, err := sendRequest(backendUrl+registrationPath, buf)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if response.StatusCode != 200 {
|
|
body, _ := ioutil.ReadAll(response.Body)
|
|
return nil, errors.New(string(body))
|
|
}
|
|
|
|
var license metadata.License
|
|
json.NewDecoder(response.Body).Decode(&license)
|
|
|
|
return &license, nil
|
|
}
|
|
|
|
func (c *BackendClient) VerifyLicenseKey(licenseKey string) (*metadata.License, error) {
|
|
|
|
req := struct {
|
|
LicenseKey string
|
|
}{
|
|
licenseKey,
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
err := json.NewEncoder(buf).Encode(req)
|
|
if err != nil {
|
|
c.ReportCrash(err.Error(), nil)
|
|
}
|
|
|
|
response, err := sendRequest(backendUrl+verificationPath, buf)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if response.StatusCode != 200 {
|
|
body, _ := ioutil.ReadAll(response.Body)
|
|
return nil, errors.New(string(body))
|
|
}
|
|
|
|
var license metadata.License
|
|
json.NewDecoder(response.Body).Decode(&license)
|
|
|
|
return &license, nil
|
|
}
|
|
|
|
func sendRequest(url string, body *bytes.Buffer) (resp *http.Response, err error) {
|
|
c := http.DefaultClient
|
|
req, err := http.NewRequest("POST", url, body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
req.Header.Set("Content-Type", "application/json")
|
|
return c.Do(req)
|
|
}
|