HyperKit instance plugin (#489)

Signed-off-by: David Chung <[email protected]>

Author: David Chung

Makefile

@@ -132,6 +132,7 @@ endif
 	$(call build_binary,infrakit-instance-vagrant,github.com/docker/infrakit/examples/instance/vagrant)
 	$(call build_binary,infrakit-instance-maas,github.com/docker/infrakit/examples/instance/maas)
 	$(call build_binary,infrakit-instance-docker,github.com/docker/infrakit/examples/instance/docker)
+	$(call build_binary,infrakit-instance-hyperkit,github.com/docker/infrakit/cmd/instance/hyperkit)
 	$(call build_binary,infrakit-event-time,github.com/docker/infrakit/examples/event/time)
 
 cli: build-cli

cmd/cli/instance/instance.go

@@ -126,22 +126,26 @@ func Command(plugins func() discovery.Plugins) *cobra.Command {
 	///////////////////////////////////////////////////////////////////////////////////
 	// destroy
 	destroy := &cobra.Command{
-		Use:   "destroy <instance ID>",
+		Use:   "destroy <instance ID>...",
 		Short: "Destroy the instance",
 		RunE: func(cmd *cobra.Command, args []string) error {
 
-			if len(args) != 1 {
+			if len(args) < 1 {
 				cmd.Usage()
 				os.Exit(1)
 			}
 
-			instanceID := instance.ID(args[0])
-			err := instancePlugin.Destroy(instanceID)
+			for _, a := range args {
 
-			if err == nil {
+				instanceID := instance.ID(a)
+				err := instancePlugin.Destroy(instanceID)
+
+				if err != nil {
+					return err
+				}
 				fmt.Println("destroyed", instanceID)
 			}
-			return err
+			return nil
 		},
 	}
 

cmd/instance/hyperkit/main.go

@@ -0,0 +1,72 @@
+package main
+
+import (
+	"os"
+	"os/user"
+	"path/filepath"
+
+	"github.com/docker/infrakit/pkg/cli"
+	logutil "github.com/docker/infrakit/pkg/log"
+	"github.com/docker/infrakit/pkg/plugin/instance/hyperkit"
+	"github.com/docker/infrakit/pkg/plugin/metadata"
+	instance_plugin "github.com/docker/infrakit/pkg/rpc/instance"
+	metadata_plugin "github.com/docker/infrakit/pkg/rpc/metadata"
+	instance_spi "github.com/docker/infrakit/pkg/spi/instance"
+	"github.com/spf13/cobra"
+)
+
+var log = logutil.New("module", "instance/hyperkit")
+
+func init() {
+	logutil.Configure(&logutil.ProdDefaults)
+}
+
+func main() {
+
+	cmd := &cobra.Command{
+		Use:   os.Args[0],
+		Short: "HyperKit instance plugin",
+	}
+
+	defaultVMDir := filepath.Join(getHome(), ".infrakit/hyperkit-vms")
+
+	name := cmd.Flags().String("name", "instance-hyperkit", "Plugin name to advertise for discovery")
+	logLevel := cmd.Flags().Int("log", cli.DefaultLogLevel, "Logging level. 0 is least verbose. Max is 5")
+
+	vmDir := cmd.Flags().String("vm-dir", defaultVMDir, "Directory where to store VM state")
+	hyperkitCmd := cmd.Flags().String("hyperkit-cmd", "hyperkit", "Path to HyperKit executable")
+	vpnkitSock := cmd.Flags().String("vpnkit-sock", "auto", "Path to VPNKit UNIX domain socket")
+	listen := cmd.Flags().String("listen", "localhost:24865", "Listens on port")
+
+	cmd.RunE = func(c *cobra.Command, args []string) error {
+
+		os.MkdirAll(*vmDir, os.ModePerm)
+
+		cli.SetLogLevel(*logLevel)
+		cli.RunListener(*listen, *name,
+			instance_plugin.PluginServer(hyperkit.NewPlugin(*vmDir, *hyperkitCmd, *vpnkitSock)),
+			metadata_plugin.PluginServer(metadata.NewPluginFromData(
+				map[string]interface{}{
+					"version":    cli.Version,
+					"revision":   cli.Revision,
+					"implements": instance_spi.InterfaceSpec,
+				},
+			)),
+		)
+		return nil
+	}
+
+	cmd.AddCommand(cli.VersionCommand())
+
+	if err := cmd.Execute(); err != nil {
+		log.Crit("Error", "err", err)
+		os.Exit(1)
+	}
+}
+
+func getHome() string {
+	if usr, err := user.Current(); err == nil {
+		return usr.HomeDir
+	}
+	return os.Getenv("HOME")
+}

docs/playbooks/intro/darwin/index.ikb

@@ -0,0 +1,6 @@
+
+# Start the plugin as local daemon
+start-daemon : start-plugin.ikt
+
+# Create a single instance
+provision-instance : provision-instance.ikt

docs/playbooks/intro/darwin/provision-instance.ikt

@@ -0,0 +1,23 @@
+{{/* Input to create instance using the HyperKit instance plugin */}}
+{{/* =% instanceProvision "instance-hyperkit" true  %= */}}
+
+{{ $user := flag "user" "string" "username" | prompt "Please enter your user name:" "string" (env "USER")}}
+{{ $linuxkitPath := flag "linuxkit-dir" "string" "Linuxkit directory" | prompt "Linuxkit directory?" "string" (cat (env "HOME") "/projects/src/github.com/linuxkit/linuxkit" | nospace) }}
+{{ $bundle := flag "bundle" "string" "Bundle prefix" | prompt "Bundle prefix?" "string"}}
+{{ $privateIP := flag "private-ip" "string" "Private IP" | prompt "Private IP address (IPv4)?" "string" "192.168.65.101" nil }}
+LogicalID: {{ $privateIP }}
+Tags:
+  infrakit.created: {{ now | htmlDate }}
+  infrakit.user: {{ $user }}
+
+Init: |
+  #!/bin/bash
+  echo Hello world!
+
+Properties:
+  kernel: {{ cat $linuxkitPath "/" $bundle "-bzImage" | nospace }}
+  initrd: {{ cat $linuxkitPath "/" $bundle "-initrd.img" | nospace }}
+  cpus: 1
+  memory: 1024
+  disk_size: 100
+  cmdline: {{ include (cat "file://" $linuxkitPath "/" $bundle "-cmdline" | nospace) }}

docs/playbooks/intro/darwin/start-plugin.ikt

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+{{/* =% sh %= */}}
+
+{{ $hyperkit := flag "start-hyperkit" "bool" "Start HYPERKIT plugin" | prompt "Start HYPERKIT plugin?" "bool" "no" }}
+
+{{ if $hyperkit }}
+
+echo "Starting HYPERKIT plugin.  This must be running on the Mac as a daemon and not as a container"
+
+infrakit-instance-hyperkit --log 5
+
+{{ end }}

docs/playbooks/intro/index.ikb

@@ -12,3 +12,4 @@ events         : events/index.ikb
 
 aws : aws/index.ikb
 gcp : gcp/index.ikb
+darwin : darwin/index.ikb

docs/playbooks/intro/start-infrakit.ikt

@@ -43,3 +43,5 @@ docker run -d --volumes-from infrakit --name manager \
 {{ source "aws/start-plugin.ikt" }}
 
 {{ source "gcp/start-plugin.ikt" }}
+
+{{ source "darwin/start-plugin.ikt" }}

pkg/cli/context.go

@@ -28,13 +28,14 @@ const (
 
 // Context is the context for the running module
 type Context struct {
-	cmd     *cobra.Command
-	src     string
-	input   io.Reader
-	exec    bool
-	run     func(string) error
-	script  string
-	plugins func() discovery.Plugins
+	cmd      *cobra.Command
+	src      string
+	input    io.Reader
+	exec     bool
+	template *template.Template
+	run      func(string) error
+	script   string
+	plugins  func() discovery.Plugins
 }
 
 // NewContext creates a context
@@ -250,6 +251,24 @@ func (c *Context) Funcs() []template.Function {
 				return c.defineFlag(n, ftype, desc, defaultValue)
 			},
 		},
+		{
+			Name: "include",
+			Func: func(p string, opt ...interface{}) (string, error) {
+				if c.exec {
+					return c.template.Include(p, opt...)
+				}
+				return "", nil
+			},
+		},
+		{
+			Name: "source",
+			Func: func(p string, opt ...interface{}) (string, error) {
+				if c.exec {
+					return c.template.Source(p, opt...)
+				}
+				return "", nil
+			},
+		},
 		{
 			Name: "cond",
 			Func: func(b interface{}, optional ...interface{}) func() (bool, interface{}) {
@@ -378,7 +397,7 @@ func (c *Context) loadBackends() error {
 				if err != nil {
 					return err
 				}
-				fmt.Println(id)
+				fmt.Println(*id)
 				return nil
 			}
 			return ""
@@ -497,6 +516,7 @@ func (c *Context) Execute() error {
 	}
 
 	c.exec = true
+	c.template = t
 	script, err := ConfigureTemplate(t, c.plugins).Render(c)
 	if err != nil {
 		return err