func pullImage(client client.APIClient, service *Service, image string) error { distributionRef, err := reference.ParseNamed(image) if err != nil { return err } switch distributionRef.(type) { case reference.Canonical: case reference.NamedTagged: default: distributionRef, err = reference.WithTag(distributionRef, "latest") if err != nil { return err } } repoInfo, err := registry.ParseRepositoryInfo(distributionRef) if err != nil { return err } authConfig := types.AuthConfig{} if service.context.ConfigFile != nil && repoInfo != nil && repoInfo.Index != nil { authConfig = registry.ResolveAuthConfig(service.context.ConfigFile.AuthConfigs, repoInfo.Index) } encodedAuth, err := encodeAuthToBase64(authConfig) if err != nil { return err } options := types.ImagePullOptions{ RegistryAuth: encodedAuth, } responseBody, err := client.ImagePull(context.Background(), distributionRef.String(), options) if err != nil { logrus.Errorf("Failed to pull image %s: %v", image, err) return err } defer responseBody.Close() var writeBuff io.Writer = os.Stdout outFd, isTerminalOut := term.GetFdInfo(os.Stdout) err = jsonmessage.DisplayJSONMessagesStream(responseBody, writeBuff, outFd, isTerminalOut, nil) if err != nil { if jerr, ok := err.(*jsonmessage.JSONError); ok { // If no error code is set, default to 1 if jerr.Code == 0 { jerr.Code = 1 } fmt.Fprintf(os.Stderr, "%s", writeBuff) return fmt.Errorf("Status: %s, Code: %d", jerr.Message, jerr.Code) } } return err }
func pullImage(ctx context.Context, client client.APIClient, service *Service, image string) error { fmt.Fprintf(os.Stderr, "Pulling %s (%s)...\n", service.name, image) distributionRef, err := reference.ParseNamed(image) if err != nil { return err } repoInfo, err := registry.ParseRepositoryInfo(distributionRef) if err != nil { return err } authConfig := service.context.AuthLookup.Lookup(repoInfo) encodedAuth, err := encodeAuthToBase64(authConfig) if err != nil { return err } options := types.ImagePullOptions{ ImageID: distributionRef.String(), Tag: "latest", RegistryAuth: encodedAuth, } if named, ok := distributionRef.(reference.Named); ok { options.ImageID = named.FullName() } if tagged, ok := distributionRef.(reference.NamedTagged); ok { options.Tag = tagged.Tag() } timeoutsRemaining := 3 for i := 0; i < 100; i++ { responseBody, err := client.ImagePull(ctx, options, nil) if err != nil { logrus.Errorf("Failed to pull image %s: %v", image, err) return err } var writeBuff io.Writer = os.Stderr outFd, isTerminalOut := term.GetFdInfo(os.Stderr) err = jsonmessage.DisplayJSONMessagesStream(responseBody, writeBuff, outFd, isTerminalOut, nil) responseBody.Close() if err == nil { return nil } else if strings.Contains(err.Error(), "timed out") { timeoutsRemaining -= 1 if timeoutsRemaining == 0 { return err } continue } else if strings.Contains(err.Error(), "connection") || strings.Contains(err.Error(), "unreachable") { time.Sleep(300 * time.Millisecond) continue } else { if jerr, ok := err.(*jsonmessage.JSONError); ok { // If no error code is set, default to 1 if jerr.Code == 0 { jerr.Code = 1 } fmt.Fprintf(os.Stderr, "%s", writeBuff) return fmt.Errorf("Status: %s, Code: %d", jerr.Message, jerr.Code) } } } return err }