cmd

var DecodeCmd = &cobra.Command{
	Use:    "decode",
	Short:  "Decodes a binary Aim stream",
	Hidden: true,
	RunE: func(cmd *cobra.Command, args []string) error {
		f := os.Stdin
		if len(args) > 0 {
			var err error
			f, err = os.Open(args[0])
			if err != nil {
				return err
			}

			defer f.Close()
		}

		decoder := encoding.NewDecoder(f)
		fmt.Println("decoded Aim stream data:")
		for {
			data, err := decoder.Next()
			if len(data) > 0 {
				for key, value := range data {
					fmt.Printf("%s: %#v\n", key, value)
				}
			}
			if err != nil {
				if errors.Is(err, io.EOF) {
					break
				}
				return eris.Wrap(err, "error decoding binary AIM stream")
			}
		}
		return nil
	},
}

var ImportCmd = &cobra.Command{
	Use:   "import",
	Short: "Copies an input database to an output database",
	Long: `The import command will transfer the contents of the input
         database to the output database. Please make sure that the
         FasttrackML server is not currently connected to the input
         database.`,
	RunE: importCmd,
}

var RootCmd = &cobra.Command{
	Use:   filepath.Base(os.Args[0]),
	Short: "Experiment tracking server focused on speed and scalability",
	Long: `FastTrackML is an experiment tracking server focused on speed and scalability.
It aims at being 100% compatible with the MLFlow client library and should be a
drop-in replacement.`,
	Version:           version.Version,
	PersistentPreRunE: initCmd,
	SilenceUsage:      true,
	SilenceErrors:     true,
}

var ServerCmd = &cobra.Command{
	Use:   "server",
	Short: "Starts the tracking server",
	RunE:  serverCmd,
}