package main

import (
	"github.com/nlf/dlite/utils"
)

type RebuildCommand struct {
	Disk   int    `short:"d" long:"disk" description:"size of disk in GiB to create" default:"30"`
	SSHKey string `short:"s" long:"ssh-key" description:"path to public ssh key" default:"$HOME/.ssh/id_rsa.pub"`
}

func (c *RebuildCommand) Execute(args []string) error {
	fmap := utils.FunctionMap{}
	fmap["Rebuilding disk image"] = func() error {
		return utils.CreateDisk(c.SSHKey, c.Disk)
	}

	return utils.Spin(fmap)
}

func init() {
	var rebuildCommand RebuildCommand
	cmd.AddCommand("rebuild", "rebuild your vm", "rebuild the disk for your vm to reset any modifications. this will DESTROY ALL DATA inside your vm.", &rebuildCommand)
}

package main

import (
	"github.com/nlf/dlite/utils"
)

type RebuildCommand struct {
	Disk   int    `short:"d" long:"disk" description:"size of disk to create" default:"30"`
	SSHKey string `short:"s" long:"ssh-key" description:"path to public ssh key" default:"$HOME/.ssh/id_rsa.pub"`
}

func (c *RebuildCommand) Execute(args []string) error {
	fmap := utils.FunctionMap{}
	fmap["Rebuilding disk image"] = func() error {
		return utils.CreateDisk(c.SSHKey, c.Disk)
	}

	return utils.Spin(fmap)
}

func init() {
	var rebuildCommand RebuildCommand
	cmd.AddCommand("rebuild", "rebuild your vm", "rebuild the disk for your vm to reset any modifications. this will DESTROY ALL DATA inside your vm.", &rebuildCommand)
}

--- 
+++ 
@@ -5,7 +5,7 @@
 )
 
 type RebuildCommand struct {
-	Disk   int    `short:"d" long:"disk" description:"size of disk to create" default:"30"`
+	Disk   int    `short:"d" long:"disk" description:"size of disk in GiB to create" default:"30"`
 	SSHKey string `short:"s" long:"ssh-key" description:"path to public ssh key" default:"$HOME/.ssh/id_rsa.pub"`
 }
 
Use GiB as default disk size unit

This is already done in `install.go` but not here.
