Merge pull request #1 from WoodProgrammer/oci-export

oci-export: Implementation for OCI image export

Author: WoodProgrammer

go.mod

@@ -1,3 +1,20 @@
 module firecracker-vmbuilder
 
+replace github.com/WoodProgrammer/firecracker-vmbuilder/src => ./src
+
 go 1.24.1
+
+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/spf13/pflag v1.0.6 // indirect
+	golang.org/x/sys v0.12.0 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // indirect
+)

go.sum

@@ -0,0 +1,28 @@
+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=
+github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
+github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+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=
+golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
+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,7 +1,70 @@
 package main
 
-import "fmt"
+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 {
+	return &src.BuildHandler{}
+}
+
+func newParserClient() src.Parser {
+	return &src.ParseHandler{}
+}
+
+func newRootFSClient() src.RootFS {
+	return &src.RootFSHandler{}
+}
+
+func HandleRootFS() {
+	parserClient := newParserClient()
+	buildCLient := newBuildClient()
+	rootFsClient := newRootFSClient()
+
+	result, err := parserClient.ParseYamlFile(configFile)
+	if err != nil {
+		log.Err(err).Msg("Error while running parserCli.ParseYamlFile()")
+	}
+
+	err = rootFsClient.CreateFileDD(10, "ops")
+	if err != nil {
+		log.Err(err).Msg("Error while running rootFsClient.CreateFileDD()")
+	}
+
+	fsErr := rootFsClient.FormatandMountFileSystem("ops", result.TargetDirectory)
+	if fsErr != nil {
+		log.Err(fsErr).Msg("Error while running rootFsClient.FormatFileSystem()")
+	}
+
+	err = buildCLient.BuildExportDockerImage(result.Context, result.DockerfilePath, result.TargetDirectory)
+	if err != nil {
+		log.Err(err).Msg("Error while running buildCLient.BuildExportDockerImage()")
+	}
+
+}
 
 func main() {
-	fmt.Println("This is firecracker vmbuilder 0.0.1")
+	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/container.go

@@ -0,0 +1,25 @@
+package src
+
+import (
+	"os/exec"
+
+	"github.com/rs/zerolog/log"
+)
+
+type Builder interface {
+	BuildExportDockerImage(context, dockerFile, targetDirectory string) error
+}
+
+type BuildHandler struct{}
+
+func (buildClient *BuildHandler) BuildExportDockerImage(context, dockerFile, targetDirectory string) error {
+	cmd := exec.Command("docker", "buildx", "build", "-f", dockerFile, "--output", targetDirectory, context)
+
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		log.Err(err).Msgf("docker buildx error : %s", output)
+		return err
+	}
+	log.Info().Msgf("Build Result %s", string(output))
+	return nil
+}

src/go.mod

@@ -0,0 +1,3 @@
+module firecracker-vmbuilder/src
+
+go 1.24.1

src/parser.go

@@ -0,0 +1,40 @@
+package src
+
+import (
+	"os"
+
+	"github.com/rs/zerolog/log"
+	yaml "gopkg.in/yaml.v2"
+)
+
+type RootFSManifest struct {
+	Image           string `yaml:"image"`
+	TargetDirectory string `yaml:"target_directory"`
+	RootFsRegistry  string `yaml:"rootfs_registry"`
+	Context         string `yaml:"context"`
+	DockerfilePath  string `yaml:"docker_file"`
+}
+
+type Parser interface {
+	ParseYamlFile(configFile string) (RootFSManifest, error)
+}
+
+type ParseHandler struct{}
+
+func (parser *ParseHandler) ParseYamlFile(configFile string) (RootFSManifest, error) {
+	var config RootFSManifest
+	data, err := os.ReadFile("config.yaml")
+
+	if err != nil {
+		log.Err(err).Msg("Error reading YAML file:")
+		return config, err
+	}
+
+	err = yaml.Unmarshal(data, &config)
+	if err != nil {
+		log.Err(err).Msg("Error parsing YAML:")
+		return config, err
+	}
+
+	return config, nil
+}

src/rootfs.go

@@ -0,0 +1,62 @@
+package src
+
+import (
+	"os"
+	"os/exec"
+
+	"github.com/rs/zerolog/log"
+)
+
+type RootFS interface {
+	CreateFileDD(size int64, fileName string) error
+	FormatandMountFileSystem(path, targetDirectory string) error
+}
+
+type RootFSHandler struct{}
+
+func (rootfs *RootFSHandler) FormatandMountFileSystem(path, targetDirectory string) error {
+	cmd := exec.Command("mkfs.ext4", path)
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		log.Err(err).Msg("Error running mkfs.ext4:")
+		log.Info().Msgf("Output: %s", string(output))
+		return err
+	}
+
+	log.Info().Msgf("Output: %s", string(output))
+
+	cmd = exec.Command("mount", path, targetDirectory)
+	output, err = cmd.CombinedOutput()
+	if err != nil {
+		log.Err(err).Msg("Error running mount")
+		log.Info().Msgf("Mount exec output: %s", string(output))
+		return err
+	}
+
+	log.Info().Msgf("Mount exec output: %s", string(output))
+	return nil
+}
+
+func (rootfs *RootFSHandler) CreateFileDD(size int64, fileName string) error {
+
+	size = int64(size * 1024 * 1024) // in MiB files
+
+	file, err := os.Create(fileName)
+	if err != nil {
+		log.Err(err).Msg("Error creating file:")
+		return err
+	}
+	defer file.Close()
+
+	zeroBlock := make([]byte, 1024*1024) // 1 MiB bufferred files in there
+	for i := 0; i < 50; i++ {
+		_, err := file.Write(zeroBlock)
+		if err != nil {
+			log.Err(err).Msg("Error writing to file:")
+			return err
+		}
+	}
+
+	log.Info().Msg("rootfs.ext4 created successfully")
+	return nil
+}