From 5fab99f5b76588c95395e56c0a081fdfcfd85e96 Mon Sep 17 00:00:00 2001
From: Mauro Morales <mauro.morales@spectrocloud.com>
Date: Wed, 27 Sep 2023 16:25:34 +0200
Subject: [PATCH] Take symlinks into account for partition size

Signed-off-by: Mauro Morales <mauro.morales@spectrocloud.com>
---
 pkg/config/spec.go | 38 ++++++++++++++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 4 deletions(-)

diff --git a/pkg/config/spec.go b/pkg/config/spec.go
index 84096ec..cfce3e2 100644
--- a/pkg/config/spec.go
+++ b/pkg/config/spec.go
@@ -19,6 +19,7 @@ package config
 import (
 	"fmt"
 	"io/fs"
+	"os"
 	"path/filepath"
 	"reflect"
 	"strings"
@@ -480,11 +481,40 @@ func GetSourceSize(config *Config, source *v1.ImageSource) (int64, error) {
 			if err != nil {
 				return err
 			}
-			info, err := d.Info()
-			if err != nil {
-				return err
+			if d.Type()&fs.ModeSymlink != 0 {
+				// If it's a symlink, get its target and calculate its size.
+				linkTarget, err := os.Readlink(path)
+				if err != nil {
+					return err
+				}
+
+				if !filepath.IsAbs(linkTarget) {
+					// If it's a relative path, join it with the base directory path.
+					linkTarget = filepath.Join(filepath.Dir(path), linkTarget)
+				}
+
+				_, err = os.Stat(linkTarget)
+				if os.IsNotExist(err) {
+					size += 0
+				} else if err != nil {
+					return err
+				} else {
+					linkInfo, err := os.Stat(linkTarget)
+					if err != nil {
+						return err
+					}
+					size += linkInfo.Size()
+				}
+			} else if !d.IsDir() {
+				// If it's a regular file, add its size to the total.
+				fileInfo, err := d.Info()
+				if err != nil {
+					return err
+				}
+
+				size += fileInfo.Size()
 			}
-			size += info.Size()
+
 			return nil
 		})