141 lines
4.3 KiB
Go
141 lines
4.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 upgrade
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/adrg/xdg"
|
|
"github.com/pelletier/go-toml/v2"
|
|
|
|
"github.com/joshuar/go-hass-agent/internal/preferences"
|
|
)
|
|
|
|
const (
|
|
oldAppID = "com.github.joshuar.go-hass-agent"
|
|
registryFile = "sensor.reg"
|
|
)
|
|
|
|
var (
|
|
oldPrefsPath = filepath.Join(xdg.ConfigHome, oldAppID)
|
|
oldRegistryPath = filepath.Join(xdg.ConfigHome, oldAppID, "sensorRegistry")
|
|
newPrefsPath = filepath.Join(xdg.ConfigHome, preferences.AppID)
|
|
newRegistryPath = filepath.Join(xdg.ConfigHome, preferences.AppID, "sensorRegistry")
|
|
|
|
ErrNoPrevConfig = errors.New("no directory from previous version found")
|
|
)
|
|
|
|
type oldPreferences struct {
|
|
WebhookID string `toml:"hass.webhookid"`
|
|
RemoteUIURL string `toml:"hass.remoteuiurl,omitempty"`
|
|
DeviceName string `toml:"device.name"`
|
|
MQTTTopicPrefix string `toml:"mqtt.topic_prefix,omitempty"`
|
|
Host string `toml:"registration.host"`
|
|
RestAPIURL string `toml:"hass.apiurl,omitempty"`
|
|
Token string `toml:"registration.token"`
|
|
CloudhookURL string `toml:"hass.cloudhookurl,omitempty"`
|
|
MQTTPassword string `toml:"mqtt.password,omitempty"`
|
|
Version string `toml:"agent.version"`
|
|
DeviceID string `toml:"device.id"`
|
|
WebsocketURL string `toml:"hass.websocketurl"`
|
|
MQTTServer string `toml:"mqtt.server,omitempty"`
|
|
MQTTUser string `toml:"mqtt.user,omitempty"`
|
|
Secret string `toml:"hass.secret,omitempty"`
|
|
MQTTEnabled bool `toml:"mqtt.enabled"`
|
|
Registered bool `toml:"hass.registered"`
|
|
}
|
|
|
|
//nolint:cyclop
|
|
//revive:disable:function-length
|
|
func Run(ctx context.Context) error {
|
|
// If there is no old preferences directory, exit.
|
|
if _, err := os.Stat(oldPrefsPath); errors.Is(err, fs.ErrNotExist) {
|
|
return errors.Join(ErrNoPrevConfig, err)
|
|
}
|
|
|
|
oldData, err := os.ReadFile(filepath.Join(oldPrefsPath, "preferences.toml"))
|
|
if err != nil {
|
|
return fmt.Errorf("cannot read old preferences: %w", err)
|
|
}
|
|
|
|
var oldPrefs oldPreferences
|
|
|
|
err = toml.Unmarshal(oldData, &oldPrefs)
|
|
if err != nil {
|
|
return fmt.Errorf("cannot read old preferences: %w", err)
|
|
}
|
|
|
|
newPrefs, err := preferences.Load(ctx)
|
|
if err != nil && !errors.Is(err, preferences.ErrNoPreferences) || newPrefs == nil {
|
|
return fmt.Errorf("cannot initialize new preferences: %w", err)
|
|
}
|
|
|
|
// Map old preferences to new preferences.
|
|
newPrefs.Registered = oldPrefs.Registered
|
|
// MQTT preferences.
|
|
if oldPrefs.MQTTEnabled {
|
|
newPrefs.MQTT.MQTTEnabled = true
|
|
newPrefs.MQTT.MQTTServer = oldPrefs.MQTTServer
|
|
newPrefs.MQTT.MQTTUser = oldPrefs.MQTTUser
|
|
newPrefs.MQTT.MQTTPassword = oldPrefs.MQTTPassword
|
|
newPrefs.MQTT.MQTTTopicPrefix = oldPrefs.MQTTTopicPrefix
|
|
}
|
|
// Hass preferences.
|
|
newPrefs.Hass.WebsocketURL = oldPrefs.WebsocketURL
|
|
newPrefs.Hass.WebhookID = oldPrefs.WebhookID
|
|
newPrefs.Hass.CloudhookURL = oldPrefs.CloudhookURL
|
|
newPrefs.Hass.RemoteUIURL = oldPrefs.RemoteUIURL
|
|
newPrefs.Hass.RestAPIURL = oldPrefs.RestAPIURL
|
|
newPrefs.Hass.Secret = oldPrefs.Secret
|
|
// Registration preferences.
|
|
newPrefs.Registration.Token = oldPrefs.Token
|
|
newPrefs.Registration.Server = oldPrefs.Host
|
|
// Device preferences.
|
|
newPrefs.Device.ID = oldPrefs.DeviceID
|
|
newPrefs.Device.Name = oldPrefs.DeviceName
|
|
|
|
err = newPrefs.Save()
|
|
if err != nil {
|
|
return fmt.Errorf("unable to save new preferences: %w", err)
|
|
}
|
|
|
|
// Create a directory for the registry.
|
|
err = os.Mkdir(newRegistryPath, os.ModePerm)
|
|
if err != nil && !errors.Is(err, os.ErrExist) {
|
|
return fmt.Errorf("unable to create new registry directory: %w", err)
|
|
}
|
|
// If the directory already exists, we've been previously upgraded. Exit.
|
|
if errors.Is(err, os.ErrExist) {
|
|
return nil
|
|
}
|
|
|
|
// Attempt to copy over the old registry.
|
|
oldReg, err := os.Open(filepath.Join(oldRegistryPath, registryFile))
|
|
if err != nil {
|
|
return fmt.Errorf("unable to open old registry: %w", err)
|
|
}
|
|
defer oldReg.Close()
|
|
|
|
newReg, err := os.Create(filepath.Join(newRegistryPath, registryFile))
|
|
if err != nil {
|
|
return fmt.Errorf("unable to create new registry file: %w", err)
|
|
}
|
|
defer newReg.Close()
|
|
|
|
_, err = io.Copy(newReg, oldReg)
|
|
if err != nil {
|
|
return fmt.Errorf("could not copy old registry contents: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|