Adds input-path flag to load data locally (#182)

Flag allows data to be read from file instead
of collected from clusters using data gatherers

Signed-off-by: Jamie Leppard <[email protected]>

Co-authored-by: Jamie Leppard <[email protected]>

Author: JammyL

cmd/agent.go

@@ -69,7 +69,14 @@ func init() {
 		"output-path",
 		"",
 		"",
-		"Output file path, if used will write data to a local file instead of uploading to the preflight server",
+		"Output file path, if used, it will write data to a local file instead of uploading to the preflight server",
+	)
+	agentCmd.PersistentFlags().StringVarP(
+		&agent.InputPath,
+		"input-path",
+		"",
+		"",
+		"Input file path, if used, it will read data from a local file instead of gathering data from clusters",
 	)
 
 }

pkg/agent/run.go

@@ -37,6 +37,9 @@ var CredentialsPath string
 // OutputPath is where the agent will write data to locally if specified
 var OutputPath string
 
+// InputPath is where the agent will read data from instead of gathering from clusters if specified
+var InputPath string
+
 // Run starts the agent process
 func Run(cmd *cobra.Command, args []string) {
 	ctx := context.Background()
@@ -134,7 +137,21 @@ func getConfiguration(ctx context.Context) (Config, *client.PreflightClient) {
 }
 
 func gatherAndOutputData(ctx context.Context, config Config, preflightClient *client.PreflightClient) {
-	readings := gatherData(ctx, config)
+	var readings []*api.DataReading
+	if InputPath != "" {
+		log.Println("Reading data from", InputPath)
+		data, err := ioutil.ReadFile(InputPath)
+		if err != nil {
+			log.Fatal(err)
+		}
+		err = json.Unmarshal(data, &readings)
+		if err != nil {
+			log.Fatal(err)
+		}
+		log.Println("Data read successfully.")
+	} else {
+		readings = gatherData(ctx, config)
+	}
 
 	if OutputPath != "" {
 		data, err := json.MarshalIndent(readings, " ", " ")