Initial AS213905 SDK release
This commit is contained in:
AS213905 Engineering
@@ -0,0 +1,258 @@
|
||||
package as213905
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const defaultBaseURL = "https://as213905.com"
|
||||
|
||||
type Options struct {
|
||||
BaseURL string
|
||||
HTTPClient *http.Client
|
||||
UserAgent string
|
||||
}
|
||||
|
||||
type Client struct {
|
||||
apiKey string
|
||||
base *url.URL
|
||||
http *http.Client
|
||||
userAgent string
|
||||
}
|
||||
|
||||
type APIError struct {
|
||||
Status int
|
||||
Code string `json:"error"`
|
||||
Message string `json:"message"`
|
||||
RequiredScope string `json:"required_scope"`
|
||||
RetryAfter int `json:"retry_after"`
|
||||
}
|
||||
|
||||
func (e *APIError) Error() string {
|
||||
if e.Message != "" {
|
||||
return fmt.Sprintf("as213905: %s (%d): %s", e.Code, e.Status, e.Message)
|
||||
}
|
||||
return fmt.Sprintf("as213905: %s (%d)", e.Code, e.Status)
|
||||
}
|
||||
|
||||
func New(apiKey string) *Client {
|
||||
c, err := NewWithOptions(apiKey, Options{})
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
func NewWithOptions(apiKey string, opts Options) (*Client, error) {
|
||||
if strings.TrimSpace(apiKey) == "" {
|
||||
return nil, errors.New("as213905: api key is required")
|
||||
}
|
||||
base := opts.BaseURL
|
||||
if base == "" {
|
||||
base = defaultBaseURL
|
||||
}
|
||||
u, err := url.Parse(strings.TrimRight(base, "/"))
|
||||
if err != nil || u.Scheme == "" || u.Host == "" {
|
||||
return nil, errors.New("as213905: invalid base URL")
|
||||
}
|
||||
hc := opts.HTTPClient
|
||||
if hc == nil {
|
||||
hc = &http.Client{Timeout: 30 * time.Second}
|
||||
}
|
||||
ua := opts.UserAgent
|
||||
if ua == "" {
|
||||
ua = "as213905-go/1.1.0"
|
||||
}
|
||||
return &Client{apiKey: apiKey, base: u, http: hc, userAgent: ua}, nil
|
||||
}
|
||||
|
||||
func (c *Client) do(ctx context.Context, method, path, organizationID string, body, out any) error {
|
||||
var reader io.Reader
|
||||
if body != nil {
|
||||
payload, err := json.Marshal(body)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
reader = bytes.NewReader(payload)
|
||||
}
|
||||
rel, err := url.Parse(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
u := c.base.ResolveReference(rel)
|
||||
req, err := http.NewRequestWithContext(ctx, method, u.String(), reader)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
req.Header.Set("Authorization", "Bearer "+c.apiKey)
|
||||
req.Header.Set("Accept", "application/json")
|
||||
req.Header.Set("User-Agent", c.userAgent)
|
||||
if body != nil {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
if organizationID != "" {
|
||||
req.Header.Set("X-Organization-ID", organizationID)
|
||||
}
|
||||
res, err := c.http.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer res.Body.Close()
|
||||
data, err := io.ReadAll(io.LimitReader(res.Body, 4<<20))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if res.StatusCode < 200 || res.StatusCode >= 300 {
|
||||
e := &APIError{Status: res.StatusCode, Code: "http_error"}
|
||||
_ = json.Unmarshal(data, e)
|
||||
if e.Code == "" {
|
||||
e.Code = "http_error"
|
||||
}
|
||||
if e.RetryAfter == 0 {
|
||||
e.RetryAfter, _ = strconv.Atoi(res.Header.Get("Retry-After"))
|
||||
}
|
||||
return e
|
||||
}
|
||||
if out == nil || len(data) == 0 {
|
||||
return nil
|
||||
}
|
||||
if err := json.Unmarshal(data, out); err != nil {
|
||||
return fmt.Errorf("as213905: decode response: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func withTrafficQuery(path string, q *TrafficQuery) string {
|
||||
if q == nil {
|
||||
return path
|
||||
}
|
||||
v := url.Values{}
|
||||
if q.Days > 0 {
|
||||
v.Set("days", strconv.Itoa(q.Days))
|
||||
}
|
||||
if q.From != "" {
|
||||
v.Set("from", q.From)
|
||||
}
|
||||
if q.To != "" {
|
||||
v.Set("to", q.To)
|
||||
}
|
||||
if len(v) > 0 {
|
||||
return path + "?" + v.Encode()
|
||||
}
|
||||
return path
|
||||
}
|
||||
|
||||
func (c *Client) Organization(ctx context.Context) (Organization, error) {
|
||||
var r struct {
|
||||
Organization Organization `json:"organization"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/organization", "", nil, &r)
|
||||
return r.Organization, err
|
||||
}
|
||||
func (c *Client) Invoices(ctx context.Context) ([]Invoice, error) {
|
||||
var r struct {
|
||||
Invoices []Invoice `json:"invoices"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/invoices", "", nil, &r)
|
||||
return r.Invoices, err
|
||||
}
|
||||
func (c *Client) Services(ctx context.Context) ([]Service, error) {
|
||||
var r struct {
|
||||
Services []Service `json:"services"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/services", "", nil, &r)
|
||||
return r.Services, err
|
||||
}
|
||||
func (c *Client) Service(ctx context.Context, id string) (Service, error) {
|
||||
var r struct {
|
||||
Service Service `json:"service"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/services/"+url.PathEscape(id), "", nil, &r)
|
||||
return r.Service, err
|
||||
}
|
||||
func (c *Client) ServiceMetrics(ctx context.Context, id, target, metricRange string) (MetricSeries, error) {
|
||||
q := url.Values{}
|
||||
if target != "" {
|
||||
q.Set("target", target)
|
||||
}
|
||||
if metricRange != "" {
|
||||
q.Set("range", metricRange)
|
||||
}
|
||||
path := "/api/v1/services/" + url.PathEscape(id) + "/metrics"
|
||||
if len(q) > 0 {
|
||||
path += "?" + q.Encode()
|
||||
}
|
||||
var r MetricSeries
|
||||
err := c.do(ctx, http.MethodGet, path, "", nil, &r)
|
||||
return r, err
|
||||
}
|
||||
func (c *Client) PowerService(ctx context.Context, id, action string) error {
|
||||
return c.do(ctx, http.MethodPost, "/api/v1/services/"+url.PathEscape(id)+"/power", "", map[string]string{"action": action}, nil)
|
||||
}
|
||||
func (c *Client) Traffic(ctx context.Context, q *TrafficQuery) (Traffic, error) {
|
||||
var r Traffic
|
||||
err := c.do(ctx, http.MethodGet, withTrafficQuery("/api/v1/traffic", q), "", nil, &r)
|
||||
return r, err
|
||||
}
|
||||
func (c *Client) Tunnels(ctx context.Context) ([]Tunnel, error) {
|
||||
var r struct {
|
||||
Tunnels []Tunnel `json:"tunnels"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/tunnels", "", nil, &r)
|
||||
return r.Tunnels, err
|
||||
}
|
||||
func (c *Client) CreateTunnel(ctx context.Context, input CreateTunnelRequest) (Tunnel, error) {
|
||||
var r struct {
|
||||
Tunnel Tunnel `json:"tunnel"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodPost, "/api/v1/tunnels", "", input, &r)
|
||||
return r.Tunnel, err
|
||||
}
|
||||
func (c *Client) DeleteTunnel(ctx context.Context, id string) error {
|
||||
return c.do(ctx, http.MethodDelete, "/api/v1/tunnels/"+url.PathEscape(id), "", nil, nil)
|
||||
}
|
||||
func (c *Client) TunnelTraffic(ctx context.Context, id string, q *TrafficQuery) (Traffic, error) {
|
||||
var r Traffic
|
||||
err := c.do(ctx, http.MethodGet, withTrafficQuery("/api/v1/tunnels/"+url.PathEscape(id)+"/traffic", q), "", nil, &r)
|
||||
return r, err
|
||||
}
|
||||
func (c *Client) Geofeeds(ctx context.Context, org string) (GeofeedList, error) {
|
||||
var r GeofeedList
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/geofeeds", org, nil, &r)
|
||||
return r, err
|
||||
}
|
||||
func (c *Client) CreateGeofeed(ctx context.Context, org, name string) (GeofeedList, error) {
|
||||
var r GeofeedList
|
||||
err := c.do(ctx, http.MethodPost, "/api/v1/geofeeds", org, map[string]string{"name": name}, &r)
|
||||
return r, err
|
||||
}
|
||||
func (c *Client) Geofeed(ctx context.Context, org, id string) (Geofeed, error) {
|
||||
var r struct {
|
||||
Geofeed Geofeed `json:"geofeed"`
|
||||
}
|
||||
err := c.do(ctx, http.MethodGet, "/api/v1/geofeeds/"+url.PathEscape(id), org, nil, &r)
|
||||
return r.Geofeed, err
|
||||
}
|
||||
func (c *Client) RenameGeofeed(ctx context.Context, org, id, name string) error {
|
||||
return c.do(ctx, http.MethodPatch, "/api/v1/geofeeds/"+url.PathEscape(id), org, map[string]string{"name": name}, nil)
|
||||
}
|
||||
func (c *Client) DeleteGeofeed(ctx context.Context, org, id string) error {
|
||||
return c.do(ctx, http.MethodDelete, "/api/v1/geofeeds/"+url.PathEscape(id), org, nil, nil)
|
||||
}
|
||||
func (c *Client) ReplaceGeofeedRecords(ctx context.Context, org, id string, records []GeofeedRecord) error {
|
||||
return c.do(ctx, http.MethodPut, "/api/v1/geofeeds/"+url.PathEscape(id)+"/records", org, map[string]any{"records": records}, nil)
|
||||
}
|
||||
func (c *Client) RotateGeofeedURL(ctx context.Context, org, id string) (GeofeedList, error) {
|
||||
var r GeofeedList
|
||||
err := c.do(ctx, http.MethodPost, "/api/v1/geofeeds/"+url.PathEscape(id)+"/rotate", org, nil, &r)
|
||||
return r, err
|
||||
}
|
||||
Reference in New Issue
Block a user