PKGBUILD/cli/sensor/sensor.go

369 lines
7.7 KiB
Go

package sensor
import (
"fmt"
"net/url"
"os"
"git.cryptic.systems/volker.raschek/flucky/pkg/cli"
"git.cryptic.systems/volker.raschek/flucky/pkg/config"
"git.cryptic.systems/volker.raschek/flucky/pkg/repository"
"git.cryptic.systems/volker.raschek/flucky/pkg/types"
"git.cryptic.systems/volker.raschek/go-logger"
uuid "github.com/satori/go.uuid"
"github.com/spf13/cobra"
)
// 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().Bool("enabled", true, "Enable new sensor")
addSensorCmd.Flags().String("gpio", "", "Defines the GPIO port")
addSensorCmd.Flags().Uint8("i2c-address", 0, "Defines the I2C address on the I2C bus")
addSensorCmd.Flags().Int("i2c-bus", 0, "Defines the I2C bus")
addSensorCmd.Flags().String("location", "", "Location of the sensor")
addSensorCmd.Flags().String("tick-duration", "1m", "Specifies how often measured values should be read from the sensor")
addSensorCmd.Flags().String("wire-id", "", "Defines the Wire-ID")
disableSensorCmd := &cobra.Command{
Use: "disable",
Short: "Disable Sensor",
Args: cobra.MinimumNArgs(1),
Example: "flucky sensor disable outdoor",
RunE: disableSensor,
}
enableSensorCmd := &cobra.Command{
Use: "enable",
Short: "Enable Sensor",
Example: "flucky sensor enable outdoor",
Args: cobra.MinimumNArgs(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.MinimumNArgs(1),
RunE: removeSensor,
}
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,
}
for _, subCommand := range []*cobra.Command{
addSensorCmd,
disableSensorCmd,
enableSensorCmd,
listSensorCmd,
removeSensorCmd,
renameSensorCmd,
} {
sensorCmd.AddCommand(subCommand)
}
cmd.AddCommand(sensorCmd)
return nil
}
func addSensor(cmd *cobra.Command, args []string) error {
sensor := &types.Sensor{
ID: uuid.NewV4().String(),
Name: args[0],
Model: args[1],
}
sensorLocation, err := cmd.Flags().GetString("location")
if err != nil {
return err
}
sensor.Location = sensorLocation
sensorEnabled, err := cmd.Flags().GetBool("enabled")
if err != nil {
return err
}
sensor.Enabled = sensorEnabled
sensorGPIO, err := cmd.Flags().GetString("gpio")
if err != nil {
return err
}
if len(sensorGPIO) > 0 {
sensor.GPIONumber = sensorGPIO
}
sensorI2CAddress, err := cmd.Flags().GetUint8("i2c-address")
if err != nil {
return err
}
if sensorI2CAddress > 0 {
sensor.I2CAddress = &sensorI2CAddress
}
sensorI2CBus, err := cmd.Flags().GetInt("i2c-bus")
if err != nil {
return err
}
if sensorI2CBus > 0 {
sensor.I2CBus = &sensorI2CBus
}
sensorTickDuration, err := cmd.Flags().GetString("tick-duration")
if err != nil {
return err
}
sensor.TickDuration = sensorTickDuration
sensorWireID, err := cmd.Flags().GetString("wire-id")
if err != nil {
return err
}
if len(sensorWireID) > 0 {
sensor.WireID = &sensorWireID
}
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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
// add sensor entry to list
err = repo.AddSensors(sensor)
if err != nil {
return err
}
// save new configuration
// err = config.Write(cnf, configFile)
// if err != nil {
// return err
// }
return nil
}
func disableSensor(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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
return repo.DisableSensorsByNames(args...)
}
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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
return repo.EnableSensorsByNames(args...)
}
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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
// add sensor entry to list
sensors, err := repo.GetSensors()
if err != nil {
return err
}
err = cli.PrintSensors(sensors, os.Stdout)
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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
return repo.RemoveSensorsByNames(args...)
}
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
}
dsnURL, err := url.Parse(cnf.DSN)
if err != nil {
return err
}
// loglevel, err := cmd.Flags().GetString("loglevel")
// if err != nil {
// return fmt.Errorf("No loglevel defined")
// }
flogger := logger.NewLogger(logger.LogLevelDebug)
repo, err := repository.New(dsnURL, flogger)
if err != nil {
return err
}
return repo.RenameSensors(args[0], args[1])
}