handle image id parsing from docker load output (#6)

Co-authored-by: Saurabh <saurabh.kumar@deepfence.io>
deepfence · Dec 15, 2021 · 120f432 · 120f432
1 parent d59ca74
commit 120f432
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/docker/docker.go b/docker/docker.go
@@ -73,9 +73,12 @@ func (d Docker) ExtractFileSystem(imageTarPath string, outputTarPath string, ima
 	scanner := bufio.NewScanner(imageMsg)
 	for scanner.Scan() {
 		line := scanner.Text()
-		if strings.Contains(line, "Loaded image: ") {
-			imageId = strings.TrimSpace(strings.Replace(line,"Loaded image: ", "", 1))
-			break
+		if strings.Contains(line, "Loaded image") {
+			splits := strings.SplitAfterN(line,":", 2)
+			if len(splits) > 1 {
+				imageId = strings.TrimSpace(splits[1])
+				break
+			}
 		}
 	}
 	if imageId == "" {