func (command *RestoreCommand) Run(args []string) int {
flagSet := flag.NewFlagSet("restore", flag.ExitOnError)
flagSet.Usage = func() { fmt.Println(command.Help()) }
sudo := flagSet.Bool("sudo", false, "Use sudo while invoking tar")
if err := flagSet.Parse(args); err != nil {
log.Errorln(err)
return -1
}
args = flagSet.Args()
if len(args) != 2 {
log.Errorln(errors.New("Insufficient argument. Please pass container name and image file name"))
return -1
}
if err := specification.DecompressImage(args[0], args[1], *sudo); err != nil {
log.Errorf("Failed to restore container. Error: %s\n", err)
return -1
}
if err := specification.UpdateUTS(args[0]); err != nil {
log.Errorf("Failed to restore container. Error: %s\n", err)
return -1
}
return 0
}
func (command *FetchCommand) Run(args []string) int {
flagSet := flag.NewFlagSet("fetch", flag.ExitOnError)
flagSet.Usage = func() { fmt.Println(command.Help()) }
name := flagSet.String("name", "", "Name of the container (Default: random UUID)")
bucket := flagSet.String("bucket", "", "S3 bucket")
key := flagSet.String("key", "", "S3 key")
region := flagSet.String("region", "us-west-1", "S3 region")
sudo := flagSet.Bool("sudo", false, "Use sudo during decompression of image")
flagSet.Parse(args)
if *bucket == "" {
log.Errorf("Must provide the s3 bucket name")
return -1
}
if *key == "" {
log.Errorf("Must provide the s3 key name")
return -1
}
svc := s3.New(session.New(), aws.NewConfig().WithRegion(*region))
fo, err := ioutil.TempFile(os.TempDir(), "nut")
if err != nil {
log.Error(err)
return -1
}
params := &s3.GetObjectInput{
Bucket: aws.String(*bucket),
Key: aws.String(*key),
}
resp, downloadErr := svc.GetObject(params)
if downloadErr != nil {
log.Error(downloadErr)
return -1
}
defer resp.Body.Close()
if _, copyError := io.Copy(fo, resp.Body); copyError != nil {
log.Errorln(copyError)
return -1
}
log.Infof("Image written to: %s\n", fo.Name())
fo.Close()
if *name == "" {
uuid, err := specification.UUID()
if err != nil {
log.Errorln(err)
return -1
}
name = &uuid
}
if err := specification.DecompressImage(*name, fo.Name(), *sudo); err != nil {
log.Errorln(err)
return -1
}
if err := specification.UpdateUTS(*name); err != nil {
log.Errorln(err)
return -1
}
return 0
}
