Initial release

Signed-off-by: Marcus Noble <github@marcusnoble.co.uk>

pkg/tailscale/client.go

@@ -0,0 +1,64 @@
+package tailscale
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	ts "github.com/tailscale/tailscale-client-go/tailscale"
+)
+
+type Client struct {
+	tsClient *ts.Client
+	tailnet  string
+	ctx      context.Context
+}
+
+func New() (*Client, error) {
+	apiKey := os.Getenv("TAILSCALE_API_KEY")
+	tailnet := os.Getenv("TAILSCALE_TAILNET")
+
+	if apiKey == "" {
+		return nil, fmt.Errorf("TAILSCALE_API_KEY must be set")
+	}
+	if tailnet == "" {
+		return nil, fmt.Errorf("TAILSCALE_TAILNET must be set")
+	}
+
+	client, err := ts.NewClient(apiKey, tailnet)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Client{
+		tsClient: client,
+		tailnet:  tailnet,
+		ctx:      context.Background(),
+	}, nil
+}
+
+func (c *Client) GetTailnet() string {
+	return c.tailnet
+}
+
+func (c *Client) GetKeys() ([]ts.Key, error) {
+	allKeys := []ts.Key{}
+
+	keys, err := c.tsClient.Keys(c.ctx)
+	if err != nil {
+		return nil, err
+	} else {
+		for _, k := range keys {
+			key, err := c.tsClient.GetKey(c.ctx, k.ID)
+			if err != nil {
+				return nil, err
+			}
+			allKeys = append(allKeys, key)
+		}
+	}
+	return allKeys, nil
+}
+
+func (c *Client) GetDevices() ([]ts.Device, error) {
+	return c.tsClient.Devices(c.ctx)
+}