mirror of
https://github.com/linuxkit/linuxkit.git
synced 2025-10-24 11:26:51 +00:00
ff941778e4
- the `public` option was not previously implemented - add `replace` only for GCP images which will error otherwise. Only recommended for use in development, in production use the `--name` option to provide a different name eaxch time. Note only applies to GCP images, will document these options properly soon. - add a `family` option; this allows you to upload many images and the user can select the latest using the `family` option instead of a specific image. Signed-off-by: Justin Cormack <justin.cormack@docker.com>
101 lines
2.0 KiB
Go
101 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
|
|
"cloud.google.com/go/storage"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func uploadGS(filename, project, bucket string, public bool) error {
|
|
if project != "" {
|
|
err := os.Setenv("GOOGLE_CLOUD_PROJECT", project)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if os.Getenv("GOOGLE_CLOUD_PROJECT") == "" {
|
|
return errors.New("GOOGLE_CLOUD_PROJECT environment variable must be set or project specified in config")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
client, err := storage.NewClient(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
f, err := os.Open(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
obj := client.Bucket(bucket).Object(filename)
|
|
wc := obj.NewWriter(ctx)
|
|
_, err = io.Copy(wc, f)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = wc.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if public {
|
|
err = obj.ACL().Set(ctx, storage.AllUsers, storage.RoleReader)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
fmt.Println("gs://" + bucket + "/" + filename)
|
|
|
|
return nil
|
|
}
|
|
|
|
func imageGS(filename, project, storage, family string, replace bool) error {
|
|
if project != "" {
|
|
err := os.Setenv("GOOGLE_CLOUD_PROJECT", project)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if os.Getenv("GOOGLE_CLOUD_PROJECT") == "" {
|
|
return errors.New("GOOGLE_CLOUD_PROJECT environment variable must be set or project specified in config")
|
|
}
|
|
|
|
// TODO do not shell out to gcloud tool, use the API
|
|
|
|
gcloud, err := exec.LookPath("gcloud")
|
|
if err != nil {
|
|
return errors.New("Please install the gcloud binary")
|
|
}
|
|
|
|
if replace {
|
|
args := []string{"compute", "images", "delete", filename}
|
|
cmd := exec.Command(gcloud, args...)
|
|
// ignore failures; it may not exist
|
|
_ = cmd.Run()
|
|
}
|
|
|
|
args := []string{"compute", "images", "create", "--source-uri", storage}
|
|
if family != "" {
|
|
args = append(args, "--family", family)
|
|
}
|
|
args = append(args, filename)
|
|
cmd := exec.Command(gcloud, args...)
|
|
|
|
out, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
return fmt.Errorf("Image creation failed: %v - %s", err, string(out))
|
|
}
|
|
|
|
fmt.Println(filename)
|
|
|
|
return nil
|
|
}
|