130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
// Copyright (c) 2024 Joshua Rich <joshua.rich@gmail.com>
|
|
//
|
|
// This software is released under the MIT License.
|
|
// https://opensource.org/licenses/MIT
|
|
|
|
package preferences
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"github.com/gofrs/uuid/v5"
|
|
|
|
"github.com/joshuar/go-hass-agent/internal/device"
|
|
)
|
|
|
|
const (
|
|
devicePrefPrefix = "device"
|
|
prefDeviceID = devicePrefPrefix + ".id"
|
|
prefDeviceName = devicePrefPrefix + ".name"
|
|
)
|
|
|
|
// Device contains the device-specific preferences.
|
|
type Device struct {
|
|
ID string `toml:"id" json:"device_id" validate:"required,ascii"`
|
|
AppID string `toml:"-" json:"app_id"`
|
|
AppName string `toml:"-" json:"app_name"`
|
|
AppVersion string `toml:"-" json:"app_version"`
|
|
Name string `toml:"name" json:"device_name" validate:"required,ascii"`
|
|
Manufacturer string `toml:"-" json:"manufacturer"`
|
|
Model string `toml:"-" json:"model"`
|
|
OsName string `toml:"-" json:"os_name"`
|
|
OsVersion string `toml:"-" json:"os_version"`
|
|
AppData AppData `toml:"-" json:"app_data,omitempty"`
|
|
SupportsEncryption bool `toml:"-" json:"supports_encryption"`
|
|
}
|
|
|
|
type AppData struct {
|
|
PushWebsocket bool `toml:"-" json:"push_websocket_channel"`
|
|
}
|
|
|
|
var ErrSetDevicePreference = errors.New("could not set device preference")
|
|
|
|
// SetDevicePreferences sets the device preferences.
|
|
func SetDevicePreferences(dev *Device) error {
|
|
if err := prefsSrc.Set(prefDeviceID, dev.ID); err != nil {
|
|
return fmt.Errorf("%w: %w", ErrSetDevicePreference, err)
|
|
}
|
|
|
|
if err := prefsSrc.Set(prefDeviceName, dev.Name); err != nil {
|
|
return fmt.Errorf("%w: %w", ErrSetDevicePreference, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// DeviceID retrieves the device ID from the preferences.
|
|
func DeviceID() string {
|
|
return prefsSrc.String(prefDeviceID)
|
|
}
|
|
|
|
// DeviceName retrieves the device name from the preferences.
|
|
func DeviceName() string {
|
|
return prefsSrc.String(prefDeviceName)
|
|
}
|
|
|
|
func NewDevice() *Device {
|
|
dev := &Device{
|
|
AppName: AppName,
|
|
AppVersion: AppVersion(),
|
|
AppID: AppID(),
|
|
}
|
|
|
|
// Retrieve the name as the device name.
|
|
name, err := device.GetHostname(true)
|
|
if err != nil {
|
|
slog.Warn("Unable to determine device hostname.",
|
|
slog.Any("error", err))
|
|
}
|
|
|
|
dev.Name = name
|
|
|
|
// Generate a new unique Device ID
|
|
id, err := newDeviceID()
|
|
if err != nil {
|
|
slog.Warn("Unable to generate a device ID.",
|
|
slog.Any("error", err))
|
|
}
|
|
|
|
dev.ID = id
|
|
|
|
// Retrieve the OS name and version.
|
|
osName, osVersion, err := device.GetOSID()
|
|
if err != nil {
|
|
slog.Warn("Unable to determine OS details.",
|
|
slog.Any("error", err))
|
|
}
|
|
|
|
dev.OsName = osName
|
|
dev.OsVersion = osVersion
|
|
|
|
// Retrieve the hardware model and manufacturer.
|
|
model, manufacturer, err := device.GetHWProductInfo()
|
|
if err != nil {
|
|
slog.Warn("Unable to determine device hardware details.",
|
|
slog.Any("error", err))
|
|
}
|
|
|
|
dev.Model = model
|
|
dev.Manufacturer = manufacturer
|
|
|
|
if err := SetDevicePreferences(dev); err != nil {
|
|
slog.Warn("Unable to set device ID in preferences.",
|
|
slog.Any("error", err))
|
|
}
|
|
|
|
return dev
|
|
}
|
|
|
|
// newDeviceID create a new device ID. It will be a randomly generated UUIDv4.
|
|
func newDeviceID() (string, error) {
|
|
deviceID, err := uuid.NewV4()
|
|
if err != nil {
|
|
return "", fmt.Errorf("could not retrieve a machine ID: %w", err)
|
|
}
|
|
|
|
return deviceID.String(), nil
|
|
}
|