Added home assistant client

Author: djthorpe

pkg/homeassistant/client.go

@@ -0,0 +1,39 @@
+/*
+homeassistant implements an API client for Home Assistant API
+https://developers.home-assistant.io/docs/api/rest/
+*/
+package homeassistant
+
+import (
+	// Packages
+	"github.com/mutablelogic/go-client/pkg/client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type Client struct {
+	*client.Client
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// LIFECYCLE
+
+func New(endPoint, apiKey string, opts ...client.ClientOpt) (*Client, error) {
+	// Add a final slash to the endpoint
+	if endPoint[len(endPoint)-1] != '/' {
+		endPoint += "/"
+	}
+
+	// Create client
+	client, err := client.New(append(opts, client.OptEndpoint(endPoint), client.OptReqToken(client.Token{
+		Scheme: client.Bearer,
+		Value:  apiKey,
+	}))...)
+	if err != nil {
+		return nil, err
+	}
+
+	// Return the client
+	return &Client{client}, nil
+}

pkg/homeassistant/client_test.go

@@ -0,0 +1,40 @@
+package homeassistant_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	homeassistant "github.com/mutablelogic/go-client/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_client_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+	t.Log(client)
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// ENVIRONMENT
+
+func GetApiKey(t *testing.T) string {
+	key := os.Getenv("HA_API_KEY")
+	if key == "" {
+		t.Skip("HA_API_KEY not set")
+		t.SkipNow()
+	}
+	return key
+}
+
+func GetEndPoint(t *testing.T) string {
+	key := os.Getenv("HA_API_URL")
+	if key == "" {
+		t.Skip("HA_API_URL not set")
+		t.SkipNow()
+	}
+	return key
+}

pkg/homeassistant/events.go

@@ -0,0 +1,26 @@
+package homeassistant
+
+import "github.com/mutablelogic/go-client/pkg/client"
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type Event struct {
+	Event     string `json:"event"`
+	Listeners uint   `json:"listener_count"`
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// Events returns all the events and number of listeners
+func (c *Client) Events() ([]Event, error) {
+	var response []Event
+	payload := client.NewRequest(client.ContentTypeJson)
+	if err := c.Do(payload, &response, client.OptPath("events")); err != nil {
+		return nil, err
+	}
+
+	// Return success
+	return response, nil
+}

pkg/homeassistant/events_test.go

@@ -0,0 +1,24 @@
+package homeassistant_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	homeassistant "github.com/mutablelogic/go-client/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_events_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+
+	events, err := client.Events()
+	assert.NoError(err)
+	assert.NotNil(events)
+
+	t.Log(events)
+}

pkg/homeassistant/health.go

@@ -0,0 +1,24 @@
+package homeassistant
+
+import "github.com/mutablelogic/go-client/pkg/client"
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// ListModels returns all the models
+func (c *Client) Health() (string, error) {
+	// Response schema
+	type responseHealth struct {
+		Message string `json:"message"`
+	}
+
+	// Return the response
+	var response responseHealth
+	payload := client.NewRequest(client.ContentTypeJson)
+	if err := c.Do(payload, &response); err != nil {
+		return "", err
+	}
+
+	// Return success
+	return response.Message, nil
+}

pkg/homeassistant/health_test.go

@@ -0,0 +1,24 @@
+package homeassistant_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	homeassistant "github.com/mutablelogic/go-client/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_health_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+
+	message, err := client.Health()
+	assert.NoError(err)
+	assert.NotEmpty(message)
+
+	t.Log(message)
+}

pkg/homeassistant/states.go

@@ -0,0 +1,42 @@
+package homeassistant
+
+import (
+	"encoding/json"
+	"time"
+
+	"github.com/mutablelogic/go-client/pkg/client"
+)
+
+///////////////////////////////////////////////////////////////////////////////
+// TYPES
+
+type State struct {
+	Entity      string         `json:"entity_id"`
+	LastChanged time.Time      `json:"last_changed"`
+	State       string         `json:"state"`
+	Attributes  map[string]any `json:"attributes"`
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// API CALLS
+
+// States returns all the entities and their state
+func (c *Client) States() ([]State, error) {
+	// Return the response
+	var response []State
+	payload := client.NewRequest(client.ContentTypeJson)
+	if err := c.Do(payload, &response, client.OptPath("states")); err != nil {
+		return nil, err
+	}
+
+	// Return success
+	return response, nil
+}
+
+///////////////////////////////////////////////////////////////////////////////
+// STRINGIFY
+
+func (s State) String() string {
+	data, _ := json.MarshalIndent(s, "", "  ")
+	return string(data)
+}

pkg/homeassistant/states_test.go

@@ -0,0 +1,24 @@
+package homeassistant_test
+
+import (
+	"os"
+	"testing"
+
+	// Packages
+	opts "github.com/mutablelogic/go-client/pkg/client"
+	homeassistant "github.com/mutablelogic/go-client/pkg/homeassistant"
+	assert "github.com/stretchr/testify/assert"
+)
+
+func Test_states_001(t *testing.T) {
+	assert := assert.New(t)
+	client, err := homeassistant.New(GetEndPoint(t), GetApiKey(t), opts.OptTrace(os.Stderr, true))
+	assert.NoError(err)
+	assert.NotNil(client)
+
+	states, err := client.States()
+	assert.NoError(err)
+	assert.NotNil(states)
+
+	t.Log(states)
+}