PKGBUILD/cmd/sensor/sensor.go

package sensor

import (
	"fmt"
	"os"

	"github.com/go-flucky/flucky/pkg/cli"
	"github.com/go-flucky/flucky/pkg/config"
	"github.com/go-flucky/flucky/pkg/types"
	"github.com/spf13/cobra"
)

var (
	enabled      bool
	gpioIn       string
	i2cAddress   uint8
	i2cBus       int
	location     string
	tickDuration string
	wireID       string
)

// InitCmd initialize all sensor subcommands
func InitCmd(cmd *cobra.Command) error {
	sensorCmd := &cobra.Command{
		Use:   "sensor",
		Short: "Manage Sensors",
	}

	addSensorCmd := &cobra.Command{
		Use:     "add",
		Short:   "Add Sensor",
		Aliases: []string{"append"},
		Args:    cobra.ExactArgs(2),
		Example: `flucky sensor add --gpio GPIO14 indoor DHT11
flucky sensor add --wire-id 28-011432f0bb3d outdoor DS18B20
flucky sensor add --i2c-bus 1 --i2c-address 0x76 wetter-station BME280`,
		RunE: addSensor,
	}
	addSensorCmd.Flags().BoolVar(&enabled, "enabled", true, "Enable new sensor")
	addSensorCmd.Flags().StringVar(&gpioIn, "gpio", "", "Defines the GPIO port")
	addSensorCmd.Flags().Uint8Var(&i2cAddress, "i2c-address", 0, "Defines the I2C address on the I2C bus")
	addSensorCmd.Flags().IntVar(&i2cBus, "i2c-bus", 0, "Defines the I2C bus")
	addSensorCmd.Flags().StringVar(&location, "location", "", "Location of the sensor")
	addSensorCmd.Flags().StringVar(&tickDuration, "tick-duration", "1m", "Controls how often values should be read from the sensor when running flucky in daemon mode")
	addSensorCmd.Flags().StringVar(&wireID, "wire-id", "", "Defines the Wire-ID")

	disableSensorCmd := &cobra.Command{
		Use:     "disable",
		Short:   "Disable Sensor",
		Args:    cobra.ExactArgs(1),
		Example: "flucky sensor disable outdoor",
		RunE:    deleteSensor,
	}

	enableSensorCmd := &cobra.Command{
		Use:     "enable",
		Short:   "Enable Sensor",
		Example: "flucky sensor enable outdoor",
		Args:    cobra.ExactArgs(1),
		RunE:    enableSensor,
	}

	listSensorCmd := &cobra.Command{
		Use:     "list",
		Short:   "List Sensors",
		Aliases: []string{"ls"},
		RunE:    listSensors,
	}

	removeSensorCmd := &cobra.Command{
		Use:     "remove",
		Short:   "Remove Sensor",
		Aliases: []string{"rm"},
		Example: "flucky sensor remove outdoor",
		Args:    cobra.ExactArgs(1),
		RunE:    enableSensor,
	}

	renameSensorCmd := &cobra.Command{
		Use:   "rename",
		Short: "Rename Sensor",
		Args:  cobra.ExactArgs(2),
		Example: `flucky sensor rename indoor outdoor
flucky sensor rename f98b00ea-a9b2-4e00-924f-113859d0af2d outdoor`,
		RunE: renameSensor,
	}

	sensorCmd.AddCommand(addSensorCmd)
	sensorCmd.AddCommand(disableSensorCmd)
	sensorCmd.AddCommand(enableSensorCmd)
	sensorCmd.AddCommand(listSensorCmd)
	sensorCmd.AddCommand(removeSensorCmd)
	sensorCmd.AddCommand(renameSensorCmd)
	cmd.AddCommand(sensorCmd)

	return nil
}

func addSensor(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	sensorModel, err := types.SelectSensorModel(args[1])
	if err != nil {
		return err
	}

	sensor := &types.Sensor{
		Name:         args[0],
		Model:        sensorModel,
		Location:     location,
		Enabled:      enabled,
		TickDuration: tickDuration,
	}

	// determine gpio port if set
	if gpioIn != "" &&
		i2cAddress == 0 &&
		i2cBus == 0 &&
		wireID == "" {
		gpio, err := types.StringToGPIO(gpioIn)
		if err != nil {
			return err
		}
		sensor.GPIONumber = &gpio
	}

	// set i2c connection settings
	if gpioIn == "" &&
		i2cAddress != 0 &&
		i2cBus != 0 &&
		wireID == "" {
		sensor.I2CAddress = &i2cAddress
		sensor.I2CBus = &i2cBus
	}

	// set wire connection settings
	if gpioIn == "" &&
		i2cAddress == 0 &&
		i2cBus == 0 &&
		wireID != "" {
		sensor.WireID = &wireID
	}

	// add sensor entry to list
	err = cnf.AddSensor(sensor)
	if err != nil {
		return err
	}

	// save new configuration
	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}

	return nil
}

func deleteSensor(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	err = cnf.DisableSensor(args[0])
	if err != nil {
		return err
	}

	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}

	return nil
}

func enableSensor(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	err = cnf.EnableSensor(args[0])
	if err != nil {
		return err
	}

	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}
	return nil
}

func listSensors(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	err = cli.PrintSensors(cnf, os.Stdout)
	if err != nil {
		return err
	}

	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}

	return nil
}

func removeSensor(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	err = cnf.RemoveSensor(args[0])
	if err != nil {
		return err
	}

	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}

	return nil
}

func renameSensor(cmd *cobra.Command, args []string) error {

	configFile, err := cmd.Flags().GetString("config")
	if err != nil {
		return fmt.Errorf("No config file defined")
	}

	cnf, err := config.Read(configFile)
	if err != nil {
		return err
	}

	err = cnf.RenameSensor(args[0], args[1])
	if err != nil {
		return err
	}

	err = config.Write(cnf, configFile)
	if err != nil {
		return err
	}

	return nil
}