Add method to retrieve posture attributes

Signed-off-by: hosom <[email protected]>

Author: hosom

v2/devices.go

@@ -67,6 +67,11 @@ type Device struct {
 	UpdateAvailable           bool     `json:"updateAvailable"`
 }
 
+type DevicePostureAttributes struct {
+	Attributes map[string]any  `json:"attributes"`
+	Expiries   map[string]Time `json:"expiries"`
+}
+
 // Get gets the [Device] identified by deviceID.
 func (dr *DevicesResource) Get(ctx context.Context, deviceID string) (*Device, error) {
 	req, err := dr.buildRequest(ctx, http.MethodGet, dr.buildURL("device", deviceID))
@@ -77,6 +82,15 @@ func (dr *DevicesResource) Get(ctx context.Context, deviceID string) (*Device, e
 	return body[Device](dr, req)
 }
 
+func (dr *DevicesResource) GetPostureAttributes(ctx context.Context, deviceID string) (*DevicePostureAttributes, error) {
+	req, err := dr.buildRequest(ctx, http.MethodGet, dr.buildURL("device", deviceID, "attributes"))
+	if err != nil {
+		return nil, err
+	}
+
+	return body[DevicePostureAttributes](dr, req)
+}
+
 // List lists every [Device] in the tailnet.
 func (dr *DevicesResource) List(ctx context.Context) ([]Device, error) {
 	req, err := dr.buildRequest(ctx, http.MethodGet, dr.buildTailnetURL("devices"))

v2/devices_test.go

@@ -77,6 +77,35 @@ func TestClient_Devices_Get(t *testing.T) {
 	assert.EqualValues(t, expectedDevice, actualDevice)
 }
 
+func TestClient_Devices_GetPostureAttributes(t *testing.T) {
+	t.Parallel()
+
+	expectedAttributes := &tsclient.DevicePostureAttributes{
+		Attributes: map[string]interface{}{
+			"custom:key":          "value",
+			"node:os":             "linux",
+			"node:osVersion":      "5.19.0-42-generic",
+			"node:tsReleaseTrack": "stable",
+			"node:tsVersion":      "1.40.0",
+			"node:tsAutoUpdate":   false,
+		},
+		Expiries: map[string]tsclient.Time{
+			"custom:key": {time.Date(2022, 2, 10, 11, 50, 23, 0, time.UTC)},
+		},
+	}
+
+	client, server := NewTestHarness(t)
+	server.ResponseCode = http.StatusOK
+	server.ResponseBody = expectedAttributes
+
+	actualAttributes, err := client.Devices().GetPostureAttributes(context.Background(), "testid")
+	assert.NoError(t, err)
+	assert.Equal(t, http.MethodGet, server.Method)
+	assert.Equal(t, "/api/v2/device/testid/attributes", server.Path)
+
+	assert.EqualValues(t, expectedAttributes, actualAttributes)
+}
+
 func TestClient_Devices_List(t *testing.T) {
 	t.Parallel()