PKGBUILD/cmd/temperature/read.go

package temperature

import (
	"fmt"
	"log"
	"os"

	"github.com/go-flucky/flucky/pkg/rgbled"

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

var logs bool
var temperatureUnit string

var readTemperatureCmd = &cobra.Command{
	Use:     "read",
	Short:   "read temperature from sensor",
	Example: fmt.Sprintf("flucky temperature read\nflucky temperature read outdoor"),
	Run: func(cmd *cobra.Command, args []string) {

		// read configuration
		cnf, err := config.Read(configFile)
		if err != nil {
			log.Fatalln(err)
		}

		rgbLEDs := cnf.GetRGBLEDs(config.ENABLED)
		rgbled.Run(rgbLEDs)

		// fetch all temperature sensors or sensors by args
		temperatureSensors := make([]sensor.TemperatureSensor, 0)
		if len(args) == 0 {
			temperatureSensors = cnf.GetTemperatureSensors(config.ENABLED)
		} else {
			temperatureSensors = cnf.GetTemperatureSensorsByName(args)
		}

		measurementUnit, err := sensor.SelectTemperatureMeasurementUnit(temperatureUnit)
		if err != nil {
			rgbled.Error(rgbLEDs)
			log.Fatalf("Can not parse temperature unit: %v", temperatureUnit)
		}

		temperatures, err := sensor.ReadTemperatures(temperatureSensors, measurementUnit, round)
		if err != nil {
			rgbled.Error(rgbLEDs)
			log.Fatalln(err)
		}

		// print temperatures on stdout
		cli.PrintTemperatures(temperatures, cnf, os.Stdout)

		if logs {
			temperatureLogfile := logfile.New(cnf.Device.TemperatureLogfile)
			err := logfile.AppendTemperatures(temperatureLogfile, compression, temperatures)
			if err != nil {
				rgbled.Error(rgbLEDs)
				log.Fatalln(err)
			}
		}

		rgbled.Off(rgbLEDs)
	},
}

func init() {
	temperatureCmd.AddCommand(readTemperatureCmd)
	readTemperatureCmd.Flags().BoolVar(&logs, "logs", true, "Log temperature")
	readTemperatureCmd.Flags().BoolVar(&compression, "compression", true, "Compress measured with logged temperatures")
	readTemperatureCmd.Flags().Float64VarP(&round, "round", "r", 0.25, "Round values. The value 0 deactivates the function")
	readTemperatureCmd.Flags().StringVar(&temperatureUnit, "temperature-unit", "celsius", "Temperature unit")
}