Adding input and output file parameters to minio-encode.

cmd/minio-encode/main.go

@@ -14,17 +14,58 @@ func main() {
 	app := cli.NewApp()
 	app.Name = "minio-encode"
 	app.Usage = "erasure encode a byte stream"
-	app.Action = func(c *cli.Context) {
+	app.Flags = []cli.Flag{
-		erasureParameters, _ := erasure.ValidateParams(10, 5, 8, erasure.CAUCHY)
+		cli.StringFlag{
-
+			Name:  "input,i",
-		encoder := erasure.NewEncoder(erasureParameters)
+			Value: "",
-		input, err := ioutil.ReadAll(os.Stdin)
+			Usage: "Input file",
-		if err != nil {
+		},
-			log.Fatal("Error reading stdin")
+		cli.StringFlag{
+			Name:  "output,o",
+			Value: "",
+			Usage: "Output file",
+		},
 	}
+	app.Action = func(c *cli.Context) {
+		// check if minio-encode called without parameters
+		if len(c.Args()) == 1 {
+			cli.ShowAppHelp(c)
+		}
+
+		// get input path
+		if c.String("input") == "" {
+			log.Fatal("No input specified")
+		}
+		inputFilePath := c.String("input")
+
+		// get output path
+		outputFilePath := inputFilePath
+		if c.String("output") != "" {
+			outputFilePath = c.String("output")
+		}
+
+		// get file
+		inputFile, err := os.Open(inputFilePath)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		// read file
+		input, err := ioutil.ReadAll(inputFile)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		// set up encoder
+		erasureParameters, _ := erasure.ValidateParams(10, 5, 8, erasure.CAUCHY)
+		encoder := erasure.NewEncoder(erasureParameters)
+
+		// encode data
 		encodedData, _ := encoder.Encode(input)
+
+		// write encoded data out
 		for key, data := range encodedData {
-			ioutil.WriteFile("output."+strconv.Itoa(key), data, 0600)
+			ioutil.WriteFile(outputFilePath+"."+strconv.Itoa(key), data, 0600)
 		}
 	}
 	app.Run(os.Args)