kairos/kairos-sdk
1
0
Fork 0
mirror of https://github.com/kairos-io/kairos-sdk.git synced 2025-09-17 23:58:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a method to get the image size

Browse Source 
Signed-off-by: Itxaka <itxaka@kairos.io>
This commit is contained in:
Itxaka
2023-08-07 13:13:59 +02:00
parent 396c5553ca
commit d4a31ab318
1 changed files with 40 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
utils/image.go
View File

@@ -42,28 +42,10 @@ var defaultRetryPredicate = func(err error) bool {
// ExtractOCIImage will extract a given targetImage into a given targetDestination
func ExtractOCIImage(targetImage, targetDestination, targetPlatform string) error {
var platform *v1.Platform
var img v1.Image
var err error
if targetPlatform != "" {
platform, err = v1.ParsePlatform(targetPlatform)
if err != nil {
return err
}
} else {
platform, err = v1.ParsePlatform(fmt.Sprintf("%s/%s", runtime.GOOS, runtime.GOARCH))
if err != nil {
return err
}
}
ref, err := name.ParseReference(targetImage)
if err != nil {
return err
}
img, err = getimage(ref, *platform)
img, err = getimage(targetImage, targetPlatform)
if err != nil {
return err
}
@@ -79,9 +61,28 @@ func ExtractOCIImage(targetImage, targetDestination, targetPlatform string) erro
// image returns the proper image to pull with transport and auth
// tries local daemon first and then fallbacks into remote
func getimage(ref name.Reference, platform v1.Platform) (v1.Image, error) {
func getimage(targetImage, targetPlatform string) (v1.Image, error) {
var platform *v1.Platform
var image v1.Image
var err error
if targetPlatform != "" {
platform, err = v1.ParsePlatform(targetPlatform)
if err != nil {
return image, err
}
} else {
platform, err = v1.ParsePlatform(fmt.Sprintf("%s/%s", runtime.GOOS, runtime.GOARCH))
if err != nil {
return image, err
}
}
ref, err := name.ParseReference(targetImage)
if err != nil {
return image, err
}
tr := transport.NewRetry(http.DefaultTransport,
transport.WithRetryBackoff(defaultRetryBackoff),
transport.WithRetryPredicate(defaultRetryPredicate),
@@ -92,10 +93,28 @@ func getimage(ref name.Reference, platform v1.Platform) (v1.Image, error) {
if err != nil {
image, err = remote.Image(ref,
remote.WithTransport(tr),
remote.WithPlatform(platform),
remote.WithPlatform(*platform),
remote.WithAuthFromKeychain(authn.DefaultKeychain),
)
}
return image, err
}
func GetOCIImageSize(targetImage, targetPlatform string) (int64, error) {
var size int64
var img v1.Image
var err error
img, err = getimage(targetImage, targetPlatform)
if err != nil {
return size, err
}
size, err = img.Size()
if err != nil {
return size, err
}
return size, nil
}