initial commit

Signed-off-by: Richard Case <richard@weave.works>

Author: richardcase

.gitignore

@@ -13,3 +13,5 @@
 
 # Dependency directories (remove the comment below to include it)
 # vendor/
+
+.vscode/

cmd/fl/main.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"context"
+	"log"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+
+	"github.com/weaveworks-experiments/fl/internal/cmd"
+)
+
+func main() {
+	ctx := context.Background()
+
+	cobra.OnInitialize(initConfig)
+
+	rootCmd := cmd.NewRootCmd()
+	if err := rootCmd.ExecuteContext(ctx); err != nil {
+		log.Fatal("failed executing root command")
+	}
+}
+
+func initConfig() {
+	viper.SetEnvPrefix("FL")
+	viper.AutomaticEnv()
+}

go.mod

@@ -0,0 +1,39 @@
+module github.com/weaveworks-experiments/fl
+
+go 1.17
+
+require (
+	github.com/moby/moby v20.10.14+incompatible
+	github.com/spf13/cobra v1.4.0
+	github.com/spf13/pflag v1.0.5
+	github.com/spf13/viper v1.10.1
+	github.com/weaveworks/flintlock/api v0.0.0-20220304105853-8fcb8aa2bafb
+	github.com/yitsushi/macpot v1.0.2
+	go.uber.org/zap v1.21.0
+	google.golang.org/grpc v1.44.0
+	gopkg.in/yaml.v2 v2.4.0
+)
+
+require (
+	github.com/fsnotify/fsnotify v1.5.1 // indirect
+	github.com/golang/protobuf v1.5.2 // indirect
+	github.com/grpc-ecosystem/grpc-gateway/v2 v2.6.0 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/inconshreveable/mousetrap v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.6 // indirect
+	github.com/mitchellh/mapstructure v1.4.3 // indirect
+	github.com/pelletier/go-toml v1.9.4 // indirect
+	github.com/spf13/afero v1.8.2 // indirect
+	github.com/spf13/cast v1.4.1 // indirect
+	github.com/spf13/jwalterweatherman v1.1.0 // indirect
+	github.com/subosito/gotenv v1.2.0 // indirect
+	github.com/weaveworks/flintlock/client v0.0.0-20220304105853-8fcb8aa2bafb // indirect
+	go.uber.org/atomic v1.9.0 // indirect
+	go.uber.org/multierr v1.8.0 // indirect
+	golang.org/x/net v0.0.0-20211209124913-491a49abca63 // indirect
+	golang.org/x/sys v0.0.0-20220405210540-1e041c57c461 // indirect
+	golang.org/x/text v0.3.7 // indirect
+	google.golang.org/genproto v0.0.0-20211208223120-3a66f561d7aa // indirect
+	google.golang.org/protobuf v1.28.0 // indirect
+	gopkg.in/ini.v1 v1.66.4 // indirect
+)

go.sum

@@ -0,0 +1,60 @@
+package microvm
+
+import (
+	"github.com/moby/moby/pkg/namesgenerator"
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+
+	"github.com/weaveworks-experiments/fl/pkg/app"
+	"github.com/weaveworks-experiments/fl/pkg/flags"
+)
+
+const (
+	defaultNamespace   = "default"
+	defaultVCPU        = 2
+	defaultMemoryMb    = 2048
+	defaultKernelImage = "ghcr.io/weaveworks/flintlock-kernel:5.10.77"
+	defaultKernelFile  = "boot/vmlinux"
+	defaultRootImage   = "ghcr.io/weaveworks/capmvm-kubernetes:1.21.8"
+)
+
+func newCreateCommand() *cobra.Command {
+	createInput := &app.CreateInput{}
+
+	cmd := &cobra.Command{
+		Use:   "create",
+		Short: "create a new microvm",
+		PreRun: func(cmd *cobra.Command, args []string) {
+			flags.BindFlags(cmd)
+		},
+		Run: func(c *cobra.Command, _ []string) {
+			a := app.New(zap.S().With("action", "create"))
+			if err := a.Create(c.Context(), createInput); err != nil {
+				zap.S().Errorw("failed creating microvm", "error", err)
+			}
+		},
+	}
+
+	cmd.Flags().StringVar(&createInput.Host, "host", "", "the flintlock host to create the microvm on")
+	cmd.MarkFlagRequired("host")
+
+	defaultName := namesgenerator.GetRandomName(10)
+	cmd.Flags().StringVar(&createInput.Name, "name", defaultName, "the name of the microvm, auto-generated if not supplied")
+	cmd.Flags().StringVar(&createInput.Namespace, "namespace", defaultNamespace, "the namespace for the microvm")
+	cmd.Flags().IntVar(&createInput.VCPU, "vcpu", defaultVCPU, "the number of vcpus")
+	cmd.Flags().IntVar(&createInput.MemoryInMb, "memory", defaultMemoryMb, "the memory in mb")
+	cmd.Flags().StringVar(&createInput.KernelImage, "kernel-image", defaultKernelImage, "the image to use for the kernel")
+	cmd.Flags().BoolVar(&createInput.KernelAddNetConf, "add-netconf", true, "automatically add network configuration to the kernel cmd line")
+	cmd.Flags().StringVar(&createInput.KernelFileName, "kernel-filename", defaultKernelFile, "name of the kernel file in the image")
+	cmd.Flags().StringVar(&createInput.RootImage, "root-image", defaultRootImage, "the image to use for the root volume")
+	cmd.Flags().StringVar(&createInput.InitrdImage, "initrd-image", "", "the image to use for the initial ramdisk")
+	cmd.Flags().StringVar(&createInput.InitrdFilename, "initrd-filename", "", "name of the file in the image to use for the initial ramdisk")
+	cmd.Flags().StringSliceVar(&createInput.NetworkInterfaces, "network-interface", nil, "specify the network interfaces to attach. In the following format: name:type:[macaddress]:[ipaddress]")
+	cmd.Flags().StringSliceVar(&createInput.MetadataFromFile, "metadata-from-file", nil, "specify metadata to be available to your microvm. In the following format key=pathtofile")
+	cmd.Flags().StringVar(&createInput.Hostname, "hostname", "", "the hostname of the the microvm")
+	cmd.Flags().StringVar(&createInput.SSHKeyFile, "ssh-key-file", "", "an ssh key to use")
+	//TODO: additional command line args for kernel
+	//TODO: add additional volumes
+
+	return cmd
+}

internal/cmd/microvm/create.go

@@ -0,0 +1,45 @@
+package microvm
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+
+	"github.com/weaveworks-experiments/fl/pkg/app"
+	"github.com/weaveworks-experiments/fl/pkg/flags"
+)
+
+const (
+	deleteExamples = `
+# Delete a microvm
+fl microvm delete --host host1:9090 01FZZJV1XD2FKH2KY0NDB4MBRQ
+`
+)
+
+func newDeleteCommand() *cobra.Command {
+	deleteInput := &app.DeleteInput{}
+
+	cmd := &cobra.Command{
+		Use:     "delete",
+		Short:   "delete a microvm from a host",
+		Example: deleteExamples,
+		Args:    cobra.ExactValidArgs(1),
+		PreRun: func(cmd *cobra.Command, args []string) {
+			flags.BindFlags(cmd)
+		},
+		Run: func(c *cobra.Command, args []string) {
+			deleteInput.UID = args[0]
+
+			a := app.New(zap.S().With("action", "delete"))
+			err := a.Delete(c.Context(), deleteInput)
+			if err != nil {
+				zap.S().Errorw("failed deleting microvm", "error", err)
+				return
+			}
+		},
+	}
+
+	cmd.Flags().StringVar(&deleteInput.Host, "host", "", "the flintlock host to get the microvms from")
+	cmd.MarkFlagRequired("host")
+
+	return cmd
+}

internal/cmd/microvm/delete.go

@@ -0,0 +1,52 @@
+package microvm
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+
+	"github.com/weaveworks-experiments/fl/pkg/app"
+	"github.com/weaveworks-experiments/fl/pkg/flags"
+)
+
+const (
+	examples = `
+# Get all microvms from a host
+fl microvm get --host host1:9090
+
+# Get a microvm with a specific id
+fl microvm get --host host1:9090 01FZZJV1XD2FKH2KY0NDB4MBRQ
+`
+)
+
+func newGetCommand() *cobra.Command {
+	getInput := &app.GetInput{}
+
+	cmd := &cobra.Command{
+		Use:     "get",
+		Short:   "get details of a microvm(s) from a host",
+		Example: examples,
+		Args:    cobra.MaximumNArgs(1),
+		PreRun: func(cmd *cobra.Command, args []string) {
+			flags.BindFlags(cmd)
+		},
+		Run: func(c *cobra.Command, args []string) {
+			if len(args) > 0 {
+				getInput.UID = args[0]
+			}
+
+			a := app.New(zap.S().With("action", "get"))
+			err := a.Get(c.Context(), getInput)
+			if err != nil {
+				zap.S().Errorw("failed getting microvm(s)", "error", err)
+				return
+			}
+		},
+	}
+
+	cmd.Flags().StringVar(&getInput.Host, "host", "", "the flintlock host to get the microvms from")
+	cmd.Flags().StringVar(&getInput.Namespace, "namespace", defaultNamespace, "the namespace to get the microvms from")
+
+	cmd.MarkFlagRequired("host")
+
+	return cmd
+}

internal/cmd/microvm/get.go

@@ -0,0 +1,31 @@
+package microvm
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+func NewCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "microvm",
+		Short: "perform microvm operations",
+		Run: func(c *cobra.Command, _ []string) {
+			if err := c.Help(); err != nil {
+				zap.S().Debugw("ingoring cobra error",
+					"error",
+					err.Error())
+			}
+		},
+	}
+
+	createCmd := newCreateCommand()
+	cmd.AddCommand(createCmd)
+
+	getCmd := newGetCommand()
+	cmd.AddCommand(getCmd)
+
+	deleteCmd := newDeleteCommand()
+	cmd.AddCommand(deleteCmd)
+
+	return cmd
+}

internal/cmd/microvm/microvm.go

@@ -0,0 +1,46 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+
+	"github.com/weaveworks-experiments/fl/internal/cmd/microvm"
+	"github.com/weaveworks-experiments/fl/pkg/logging"
+)
+
+const (
+	logLevelFlag = "log-level"
+)
+
+func NewRootCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "fl",
+		Short: "The experimental cli for flintlock",
+		PersistentPreRunE: func(cmd *cobra.Command, _ []string) error {
+			logLevelFlag, _ := cmd.Flags().GetString(logLevelFlag)
+			err := logging.Configure(logLevelFlag)
+			if err != nil {
+				return err
+			}
+
+			return nil
+		},
+		Run: func(c *cobra.Command, _ []string) {
+			if err := c.Help(); err != nil {
+				zap.S().Debugw("ingoring cobra error",
+					"error",
+					err.Error())
+			}
+		},
+	}
+
+	cmd.PersistentFlags().String(logLevelFlag, "debug", "set the level of the logger")
+
+	versionCmd := newVersionCommand()
+	cmd.AddCommand(versionCmd)
+
+	microvmCmd := microvm.NewCommand()
+	cmd.AddCommand(microvmCmd)
+
+	return cmd
+}

internal/cmd/root.go

@@ -0,0 +1,18 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+)
+
+func newVersionCommand() *cobra.Command {
+	return &cobra.Command{
+		Use:   "version",
+		Short: "display version information",
+		Run: func(c *cobra.Command, _ []string) {
+			fmt.Println("to do, add version information")
+		},
+	}
+
+}