joshuar-go-hass-agent/internal/hass/registration.go

// Copyright 2024 Joshua Rich <joshua.rich@gmail.com>.
// SPDX-License-Identifier: MIT

package hass

import (
	"context"
	"errors"
	"fmt"

	"github.com/joshuar/go-hass-agent/internal/device"
	"github.com/joshuar/go-hass-agent/internal/hass/api"
	"github.com/joshuar/go-hass-agent/internal/preferences"
)

const (
	RegistrationPath = "/api/mobile_app/registrations"
)

var ErrInternalValidationFailed = errors.New("internal validation error")

type registrationRequest struct {
	*device.Device
	Token string `json:"-"`
}

func (r *registrationRequest) Auth() string {
	return r.Token
}

func (r *registrationRequest) RequestBody() any {
	return r.Device
}

// revive:disable:unused-receiver
func (r *registrationRequest) Retry() bool {
	return true
}

func newRegistrationRequest(thisDevice *device.Device, token string) *registrationRequest {
	return &registrationRequest{
		Device: thisDevice,
		Token:  token,
	}
}

func RegisterDevice(ctx context.Context, registration *preferences.Registration) (*preferences.Hass, error) {
	// Validate provided registration details.
	if err := registration.Validate(); err != nil {
		return nil, fmt.Errorf("could not register device: %w", err)
	}

	registrationURL := registration.Server + RegistrationPath
	thisDevice := device.NewDevice()

	// Set device details in preferences.
	if err := preferences.SetDevicePreferences(&preferences.Device{
		Name: thisDevice.Name,
		ID:   thisDevice.ID,
	}); err != nil {
		return nil, fmt.Errorf("could not register device: %w", err)
	}

	// Register the device against the registration endpoint.
	registrationStatus, err := api.Send[preferences.Hass](ctx, registrationURL, newRegistrationRequest(thisDevice, registration.Token))
	if err != nil {
		return nil, fmt.Errorf("could not register device: %w", err)
	}

	return &registrationStatus, nil
}