package db import ( "context" "fmt" "net/url" "github.com/Masterminds/semver" "github.com/volker-raschek/flucky/cmd/internal" "github.com/volker-raschek/flucky/pkg/config" "github.com/volker-raschek/flucky/pkg/storage/db" "github.com/volker-raschek/flucky/pkg/types" "github.com/spf13/cobra" ) var ( semanticVersion *semver.Version ) // InitCmd initialize all convert subcommands func InitCmd(cmd *cobra.Command, semver *semver.Version) error { semanticVersion = semver databaseCmd := &cobra.Command{ Use: "db", Short: "Interagte with the database configured as storage endpoint", } updateDatabaseCmd := &cobra.Command{ Use: "init", Short: "Initialize the database scheme for the version of the flucky binary", Args: cobra.NoArgs, Example: "flucky db update", RunE: initializeDatabase, } databaseCmd.AddCommand(updateDatabaseCmd) cmd.AddCommand(databaseCmd) return nil } func initializeDatabase(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 } storageEndpointURL, err := url.Parse(cnf.StorageEndpoint) if err != nil { return fmt.Errorf("Failed to parse storage endpoint: %v", err) } logLevel, err := cmd.Flags().GetString("loglevel") if err != nil { return fmt.Errorf("No loglevel defined: %v", err) } flogger := internal.InitializeLogger(logLevel) database, err := db.New(storageEndpointURL) if err != nil { return err } defer database.Close() ctx := context.Background() flogger.Debug("Initialize database scheme") err = database.Schema(ctx, semanticVersion) if err != nil { return err } _, err = database.SelectDeviceByID(ctx, cnf.Device.ID) if err != nil { flogger.Debug("Add device into database") devices := []*types.Device{cnf.Device} if err := database.InsertDevices(ctx, devices); err != nil { return err } } else { flogger.Debug("Device already availabe in the database") } for _, sensor := range cnf.Sensors { _, err = database.SelectSensorByID(ctx, sensor.ID) if err != nil { flogger.Debug("Add sensor %v into the database", sensor.FullName()) sensors := []*types.Sensor{sensor} if err := database.InsertSensors(ctx, sensors); err != nil { return err } } else { flogger.Debug("Sensor %v already available in the database", sensor.FullName()) } } return nil }