45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/ESilva15/BeamNGMockOg/mockserver"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func recordAction(cmd *cobra.Command, args []string) {
|
|
outputFile, _ := cmd.Flags().GetString("output")
|
|
address, _ := cmd.Flags().GetString("address")
|
|
port, _ := cmd.Flags().GetInt("port")
|
|
|
|
recorder, err := mockserver.NewRecorder(outputFile, address, port)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to create new BeamNG recorder: %+v", err))
|
|
}
|
|
defer recorder.Close()
|
|
|
|
// NOTE: is this doing anything at all??
|
|
ctx := context.Background()
|
|
if err := recorder.Record(ctx); err != nil {
|
|
fmt.Printf("Something went wrong while recording the file: %v", err)
|
|
}
|
|
}
|
|
|
|
// shelf
|
|
var recordCmd = &cobra.Command{
|
|
Use: "record",
|
|
Short: "record -o <path-to-bin-file>",
|
|
Long: `record will store the data from the given UDP server to the
|
|
filepath given by -i`,
|
|
Args: nil,
|
|
Run: recordAction,
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(recordCmd)
|
|
|
|
recordCmd.PersistentFlags().StringP("output", "o", "output.bin", "output file for recording")
|
|
}
|