forked from shipwright-io/cli
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Local source code feature (using bundle images)
Reference shipwright-io/build#717 Add `bundle` package that contains convenience code to push a local source code directory as a bundle image to a container registry. Add new flags for container image and local directory settings.
- Loading branch information
1 parent
b1b7e40
commit 20e1b46
Showing
12 changed files
with
391 additions
and
50 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,228 @@ | ||
| package bundle | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io/ioutil" | ||
| "net/http" | ||
| "strings" | ||
|
|
||
| "k8s.io/cli-runtime/pkg/genericclioptions" | ||
|
|
||
| "github.com/google/go-containerregistry/pkg/authn" | ||
| "github.com/google/go-containerregistry/pkg/name" | ||
| v1 "github.com/google/go-containerregistry/pkg/v1" | ||
| "github.com/google/go-containerregistry/pkg/v1/empty" | ||
| "github.com/google/go-containerregistry/pkg/v1/remote" | ||
| "github.com/schollz/progressbar/v3" | ||
| "github.com/shipwright-io/build/pkg/reconciler/buildrun/resources/sources" | ||
| ) | ||
|
|
||
| // Push bundles the provided local directory into a container image and pushes | ||
| // it to the given registry. | ||
| func Push(ctx context.Context, io *genericclioptions.IOStreams, localDirectory string, targetImage string) (name.Digest, error) { | ||
| tag, err := name.NewTag(targetImage) | ||
| if err != nil { | ||
| return name.Digest{}, err | ||
| } | ||
|
|
||
| auth, err := authn.DefaultKeychain.Resolve(tag.Context()) | ||
| if err != nil { | ||
| return name.Digest{}, err | ||
| } | ||
|
|
||
| updates := make(chan v1.Update, 1) | ||
| done := make(chan struct{}, 1) | ||
| go func() { | ||
| var progress *progressbar.ProgressBar | ||
| for { | ||
| select { | ||
| case <-ctx.Done(): | ||
| return | ||
|
|
||
| case <-done: | ||
| return | ||
|
|
||
| case update, ok := <-updates: | ||
| if !ok { | ||
| return | ||
| } | ||
|
|
||
| if progress == nil { | ||
| progress = progressbar.NewOptions(int(update.Total), | ||
| progressbar.OptionSetWriter(io.ErrOut), | ||
| progressbar.OptionEnableColorCodes(true), | ||
| progressbar.OptionShowBytes(true), | ||
| progressbar.OptionSetWidth(15), | ||
| progressbar.OptionSetPredictTime(false), | ||
| progressbar.OptionSetDescription("Uploading local source..."), | ||
| progressbar.OptionSetTheme(progressbar.Theme{ | ||
| Saucer: "[green]=[reset]", | ||
| SaucerHead: "[green]>[reset]", | ||
| SaucerPadding: " ", | ||
| BarStart: "[", | ||
| BarEnd: "]"}), | ||
| progressbar.OptionClearOnFinish(), | ||
| ) | ||
| defer progress.Close() | ||
| } | ||
|
|
||
| progress.ChangeMax64(update.Total) | ||
| _ = progress.Set64(update.Complete) | ||
| } | ||
| } | ||
| }() | ||
|
|
||
| fmt.Fprintf(io.Out, "Bundling %q as %q ...\n", localDirectory, targetImage) | ||
| digest, err := sources.Bundle( | ||
| tag, | ||
| localDirectory, | ||
| remote.WithContext(ctx), | ||
| remote.WithAuth(auth), | ||
| remote.WithProgress(updates), | ||
| ) | ||
|
|
||
| done <- struct{}{} | ||
| return digest, err | ||
| } | ||
|
|
||
| func Prune(ctx context.Context, io *genericclioptions.IOStreams, image string) error { | ||
| ref, err := name.ParseReference(image) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| auth, err := authn.DefaultKeychain.Resolve(ref.Context()) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| switch ref.Context().RegistryStr() { | ||
| case "index.docker.io": | ||
| list, err := remote.ListWithContext(ctx, ref.Context(), remote.WithAuth(auth)) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| switch len(list) { | ||
| case 0: | ||
| return nil | ||
|
|
||
| case 1: | ||
| authr, err := auth.Authorization() | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| token, err := dockerHubLogin(authr.Username, authr.Password) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| return dockerHubRepoDelete(token, ref) | ||
|
|
||
| default: | ||
| fmt.Fprintf(io.ErrOut, "removing specific tags is currently not supported for images hosted on %s\n", ref.Context().RegistryStr()) | ||
|
|
||
| // In case the input argument included a digest, the reference | ||
| // needs to be updated to exclude the digest for the empty image | ||
| // override to succeed. | ||
| switch ref.(type) { | ||
| case name.Digest: | ||
| tmp := strings.SplitN(image, "@", 2) | ||
| ref, err = name.NewTag(tmp[0]) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| } | ||
|
|
||
| return remote.Write(ref, empty.Image, remote.WithContext(ctx), remote.WithAuth(auth)) | ||
| } | ||
|
|
||
| default: | ||
| return remote.Delete( | ||
| ref, | ||
| remote.WithContext(ctx), | ||
| remote.WithAuth(auth), | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| func dockerHubLogin(username string, password string) (string, error) { | ||
| type LoginData struct { | ||
| Username string `json:"username"` | ||
| Password string `json:"password"` | ||
| } | ||
|
|
||
| loginData, err := json.Marshal(LoginData{Username: username, Password: password}) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| req, err := http.NewRequest("POST", "https://hub.docker.com/v2/users/login/", bytes.NewReader(loginData)) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| req.Header.Set("Content-Type", "application/json") | ||
|
|
||
| resp, err := http.DefaultClient.Do(req) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| defer resp.Body.Close() | ||
|
|
||
| bodyData, err := ioutil.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| switch resp.StatusCode { | ||
| case http.StatusOK: | ||
| type LoginToken struct { | ||
| Token string `json:"token"` | ||
| } | ||
|
|
||
| var loginToken LoginToken | ||
| if err := json.Unmarshal(bodyData, &loginToken); err != nil { | ||
| return "", err | ||
| } | ||
|
|
||
| return loginToken.Token, nil | ||
|
|
||
| default: | ||
| return "", fmt.Errorf(string(bodyData)) | ||
| } | ||
| } | ||
|
|
||
| func dockerHubRepoDelete(token string, ref name.Reference) error { | ||
| req, err := http.NewRequest("DELETE", fmt.Sprintf("https://hub.docker.com/v2/repositories/%s/", ref.Context().RepositoryStr()), nil) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| req.Header.Set("Authorization", "JWT "+token) | ||
|
|
||
| resp, err := http.DefaultClient.Do(req) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| defer resp.Body.Close() | ||
|
|
||
| respData, err := ioutil.ReadAll(resp.Body) | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| switch resp.StatusCode { | ||
| case http.StatusAccepted: | ||
| return nil | ||
|
|
||
| default: | ||
| return fmt.Errorf("failed with HTTP status code %d: %s", resp.StatusCode, string(respData)) | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.