-
-
Notifications
You must be signed in to change notification settings - Fork 511
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge branch 'main' into 644-remove-volumes-on-down
- Loading branch information
Showing
52 changed files
with
949 additions
and
309 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
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,107 @@ | ||
| package testcontainers | ||
|
|
||
| import ( | ||
| "context" | ||
| "encoding/base64" | ||
| "encoding/json" | ||
| "os" | ||
|
|
||
| "github.com/cpuguy83/dockercfg" | ||
| "github.com/docker/docker/api/types" | ||
| "github.com/testcontainers/testcontainers-go/internal/testcontainersdocker" | ||
| ) | ||
|
|
||
| // DockerImageAuth returns the auth config for the given Docker image, extracting first its Docker registry. | ||
| // Finally, it will use the credential helpers to extract the information from the docker config file | ||
| // for that registry, if it exists. | ||
| func DockerImageAuth(ctx context.Context, image string) (string, types.AuthConfig, error) { | ||
| defaultRegistry := defaultRegistry(ctx) | ||
| registry := testcontainersdocker.ExtractRegistry(image, defaultRegistry) | ||
|
|
||
| cfgs, err := getDockerAuthConfigs() | ||
| if err != nil { | ||
| return registry, types.AuthConfig{}, err | ||
| } | ||
|
|
||
| if cfg, ok := cfgs[registry]; ok { | ||
| return registry, cfg, nil | ||
| } | ||
|
|
||
| return registry, types.AuthConfig{}, dockercfg.ErrCredentialsNotFound | ||
| } | ||
|
|
||
| // defaultRegistry returns the default registry to use when pulling images | ||
| // It will use the docker daemon to get the default registry, returning "https://index.docker.io/v1/" if | ||
| // it fails to get the information from the daemon | ||
| func defaultRegistry(ctx context.Context) string { | ||
| p, err := NewDockerProvider() | ||
| if err != nil { | ||
| return testcontainersdocker.IndexDockerIO | ||
| } | ||
|
|
||
| info, err := p.client.Info(ctx) | ||
| if err != nil { | ||
| return testcontainersdocker.IndexDockerIO | ||
| } | ||
|
|
||
| return info.IndexServerAddress | ||
| } | ||
|
|
||
| // getDockerAuthConfigs returns a map with the auth configs from the docker config file | ||
| // using the registry as the key | ||
| func getDockerAuthConfigs() (map[string]types.AuthConfig, error) { | ||
| cfg, err := getDockerConfig() | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| cfgs := map[string]types.AuthConfig{} | ||
| for k, v := range cfg.AuthConfigs { | ||
| ac := types.AuthConfig{ | ||
| Auth: v.Auth, | ||
| Email: v.Email, | ||
| IdentityToken: v.IdentityToken, | ||
| Password: v.Password, | ||
| RegistryToken: v.RegistryToken, | ||
| ServerAddress: v.ServerAddress, | ||
| Username: v.Username, | ||
| } | ||
|
|
||
| if v.Username == "" && v.Password == "" { | ||
| u, p, _ := dockercfg.GetRegistryCredentials(k) | ||
| ac.Username = u | ||
| ac.Password = p | ||
| } | ||
|
|
||
| if v.Auth == "" { | ||
| ac.Auth = base64.StdEncoding.EncodeToString([]byte(ac.Username + ":" + ac.Password)) | ||
| } | ||
|
|
||
| cfgs[k] = ac | ||
| } | ||
|
|
||
| return cfgs, nil | ||
| } | ||
|
|
||
| // getDockerConfig returns the docker config file. It will internally check, in this particular order: | ||
| // 1. the DOCKER_AUTH_CONFIG environment variable, unmarshalling it into a dockercfg.Config | ||
| // 2. the DOCKER_CONFIG environment variable, as the path to the config file | ||
| // 3. else it will load the default config file, which is ~/.docker/config.json | ||
| func getDockerConfig() (dockercfg.Config, error) { | ||
| dockerAuthConfig := os.Getenv("DOCKER_AUTH_CONFIG") | ||
| if dockerAuthConfig != "" { | ||
| cfg := dockercfg.Config{} | ||
| err := json.Unmarshal([]byte(dockerAuthConfig), &cfg) | ||
| if err == nil { | ||
| return cfg, nil | ||
| } | ||
|
|
||
| } | ||
|
|
||
| cfg, err := dockercfg.LoadDefaultConfig() | ||
| if err != nil { | ||
| return cfg, err | ||
| } | ||
|
|
||
| return cfg, nil | ||
| } |
Oops, something went wrong.