Fix clie requirements

Author: WoodProgrammer

go.mod

@@ -4,12 +4,17 @@ replace github.com/WoodProgrammer/firecracker-vmbuilder/src => ./src
 
 go 1.24.1
 
-require github.com/WoodProgrammer/firecracker-vmbuilder/src v0.0.0-00010101000000-000000000000
+require (
+	github.com/WoodProgrammer/firecracker-vmbuilder/src v0.0.0-00010101000000-000000000000
+	github.com/rs/zerolog v1.33.0
+	github.com/spf13/cobra v1.9.1
+)
 
 require (
+	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.19 // indirect
-	github.com/rs/zerolog v1.33.0 // indirect
+	github.com/spf13/pflag v1.0.6 // indirect
 	golang.org/x/sys v0.12.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

go.sum

@@ -1,5 +1,8 @@
 github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
+github.com/cpuguy83/go-md2man/v2 v2.0.6/go.mod h1:oOW0eioCTA6cOiMLiUPZOpcVxMig6NIQQ7OS05n1F4g=
 github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
+github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
+github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
 github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
 github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
 github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
@@ -9,6 +12,11 @@ github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINE
 github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
 github.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8=
 github.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
+github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
+github.com/spf13/cobra v1.9.1 h1:CXSaggrXdbHK9CF+8ywj8Amf7PBRmPCOJugH954Nnlo=
+github.com/spf13/cobra v1.9.1/go.mod h1:nDyEzZ8ogv936Cinf6g1RU9MRY64Ir93oCnqb9wxYW0=
+github.com/spf13/pflag v1.0.6 h1:jFzHGLGAlb3ruxLB8MhbI6A8+AQX/2eW4qeyNZXNp2o=
+github.com/spf13/pflag v1.0.6/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
 golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.12.0 h1:CM0HF96J0hcLAwsHPJZjfdNzs0gftsLfgKt57wWHJ0o=
@@ -17,3 +25,4 @@ gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
 gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

main.go

@@ -1,8 +1,15 @@
 package main
 
 import (
+	"os"
+
 	src "github.com/WoodProgrammer/firecracker-vmbuilder/src"
 	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+var (
+	configFile string
 )
 
 func newBuildClient() src.Builder {
@@ -16,12 +23,13 @@ func newParserClient() src.Parser {
 func newRootFSClient() src.RootFS {
 	return &src.RootFSHandler{}
 }
-func main() {
+
+func HandleRootFS() {
 	parserClient := newParserClient()
 	buildCLient := newBuildClient()
 	rootFsClient := newRootFSClient()
 
-	result, err := parserClient.ParseYamlFile("config.yaml")
+	result, err := parserClient.ParseYamlFile(configFile)
 	if err != nil {
 		log.Err(err).Msg("Error while running parserCli.ParseYamlFile()")
 	}
@@ -42,3 +50,21 @@ func main() {
 	}
 
 }
+
+func main() {
+	var rootCmd = &cobra.Command{
+		Use:   "rootfsCreator",
+		Short: "CLI tool to manage RootFS for firecracker micro VMs",
+		Run: func(cmd *cobra.Command, args []string) {
+			HandleRootFS()
+		},
+	}
+	rootCmd.Flags().StringVarP(&configFile, "config", "C", "config.yaml", "Config file of RootFS creation")
+
+	rootCmd.MarkFlagRequired("config")
+
+	if err := rootCmd.Execute(); err != nil {
+		log.Err(err).Msg("rootfsCreator execution failed")
+		os.Exit(1)
+	}
+}

src/rootfs.go

@@ -1,7 +1,6 @@
 package src
 
 import (
-	"fmt"
 	"os"
 	"os/exec"
 
@@ -26,7 +25,7 @@ func (rootfs *RootFSHandler) FormatandMountFileSystem(path, targetDirectory stri
 
 	log.Info().Msgf("Output: %s", string(output))
 
-	cmd = exec.Command(fmt.Sprintf("mount", path, targetDirectory))
+	cmd = exec.Command("mount", path, targetDirectory)
 	output, err = cmd.CombinedOutput()
 	if err != nil {
 		log.Err(err).Msg("Error running mount")