-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
This adds a `docker` registry option, enabling plugins to be written in languages other than Go by providing a way to distribute and run plugins in a cross-platform manner via Docker.
- Loading branch information
1 parent
caf9b52
commit 5bc1f60
Showing
7 changed files
with
478 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
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
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,113 @@ | ||
| package managedplugin | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/json" | ||
| "fmt" | ||
| "io" | ||
|
|
||
| "github.com/docker/docker/api/types" | ||
| "github.com/docker/docker/api/types/filters" | ||
| "github.com/docker/docker/client" | ||
| "github.com/schollz/progressbar/v3" | ||
| ) | ||
|
|
||
| type dockerProgressReader struct { | ||
| decoder *json.Decoder | ||
| bar *progressbar.ProgressBar | ||
| downloadedByID map[string]int64 | ||
| totalBytes int64 | ||
| } | ||
|
|
||
| type dockerProgressInfo struct { | ||
| Status string `json:"status"` | ||
| Progress string `json:"progress"` | ||
| ProgressData struct { | ||
| Current int64 `json:"current"` | ||
| Total int64 `json:"total"` | ||
| } `json:"progressDetail"` | ||
| ID string `json:"id"` | ||
| } | ||
|
|
||
| func (pr *dockerProgressReader) Read(_ []byte) (n int, err error) { | ||
| var progress dockerProgressInfo | ||
| err = pr.decoder.Decode(&progress) | ||
| if err != nil { | ||
| if err == io.EOF { | ||
| return 0, io.EOF | ||
| } | ||
| return 0, fmt.Errorf("failed to decode JSON: %v", err) | ||
| } | ||
| if progress.Status == "Downloading" { | ||
| if pr.bar == nil { | ||
| pr.bar = downloadProgressBar(1, "Downloading") | ||
| _ = pr.bar.RenderBlank() | ||
| } | ||
| if _, seen := pr.downloadedByID[progress.ID]; !seen { | ||
| pr.downloadedByID[progress.ID] = 0 | ||
| pr.totalBytes += progress.ProgressData.Total | ||
| pr.bar.ChangeMax64(pr.totalBytes) | ||
| } | ||
| pr.downloadedByID[progress.ID] = progress.ProgressData.Current | ||
| total := int64(0) | ||
| for _, v := range pr.downloadedByID { | ||
| total += v | ||
| } | ||
| if total < pr.totalBytes { | ||
| // progressbar stops responding if it reaches 100%, so as a workaround we don't update | ||
| // the bar if we're at 100%, because there may be more layers of the image | ||
| // coming that we don't know about. | ||
| _ = pr.bar.Set64(total) | ||
| } | ||
| } | ||
| return 0, nil | ||
| } | ||
|
|
||
| func isDockerImageAvailable(ctx context.Context, imageName string) (bool, error) { | ||
| cli, err := client.NewClientWithOpts(client.FromEnv) | ||
| if err != nil { | ||
| return false, fmt.Errorf("failed to create Docker client: %w", err) | ||
| } | ||
|
|
||
| // List the images matching the given name | ||
| images, err := cli.ImageList(ctx, types.ImageListOptions{ | ||
| Filters: filters.NewArgs(filters.Arg("reference", imageName)), | ||
| }) | ||
| if err != nil { | ||
| return false, fmt.Errorf("failed to list Docker images: %w", err) | ||
| } | ||
|
|
||
| // Check if the image with the specified name was found | ||
| return len(images) > 0, nil | ||
| } | ||
|
|
||
| func pullDockerImage(ctx context.Context, imageName string) error { | ||
| cli, err := client.NewClientWithOpts(client.FromEnv) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create Docker client: %v", err) | ||
| } | ||
|
|
||
| // Pull the image | ||
| out, err := cli.ImagePull(ctx, imageName, types.ImagePullOptions{}) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to pull Docker image: %v", err) | ||
| } | ||
| defer out.Close() | ||
|
|
||
| // Create a progress reader to display the download progress | ||
| pr := &dockerProgressReader{ | ||
| decoder: json.NewDecoder(out), | ||
| downloadedByID: map[string]int64{}, | ||
| bar: nil, | ||
| } | ||
| _, err = io.Copy(io.Discard, pr) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to copy image pull output: %v", err) | ||
| } | ||
| if pr.bar != nil { | ||
| _ = pr.bar.Finish() | ||
| pr.bar.Close() | ||
| } | ||
|
|
||
| return nil | ||
| } |
Oops, something went wrong.