Check if skippedFiles directory exists everywhere, why not

replicatedhq · Jul 24, 2020 · f85a6fd · f85a6fd
1 parent 866834c
commit f85a6fd
Show file tree

Hide file tree

Showing 5 changed files with 29 additions and 6 deletions.
diff --git a/ffi/main.go b/ffi/main.go
@@ -98,7 +98,11 @@ func UpdateCheck(socket, fromArchivePath, namespace string) {
 			filepath.Join(tmpRoot, "upstream"),
 			filepath.Join(tmpRoot, "base"),
 			filepath.Join(tmpRoot, "overlays"),
-			filepath.Join(tmpRoot, "skippedFiles"),
+		}
+
+		skippedFilesPath := filepath.Join(tmpRoot, "skippedFiles")
+		if _, err := os.Stat(skippedFilesPath); err == nil {
+			paths = append(paths, skippedFilesPath)
 		}
 
 		err = os.Remove(fromArchivePath)

diff --git a/ffi/online.go b/ffi/online.go
@@ -161,7 +161,11 @@ func RewriteVersion(socket, fromArchivePath, outputFile, downstreamsStr, k8sName
 			filepath.Join(tmpRoot, "upstream"),
 			filepath.Join(tmpRoot, "base"),
 			filepath.Join(tmpRoot, "overlays"),
-			filepath.Join(tmpRoot, "skippedFiles"),
+		}
+
+		skippedFilesPath := filepath.Join(tmpRoot, "skippedFiles")
+		if _, err := os.Stat(skippedFilesPath); err == nil {
+			paths = append(paths, skippedFilesPath)
 		}
 
 		if err := tarGz.Archive(paths, outputFile); err != nil {

diff --git a/ffi/updatedownload.go b/ffi/updatedownload.go
@@ -119,7 +119,11 @@ func UpdateDownload(socket, fromArchivePath, namespace, registryJson, cursor str
 			filepath.Join(tmpRoot, "upstream"),
 			filepath.Join(tmpRoot, "base"),
 			filepath.Join(tmpRoot, "overlays"),
-			filepath.Join(tmpRoot, "skippedFiles"),
+		}
+
+		skippedFilesPath := filepath.Join(tmpRoot, "skippedFiles")
+		if _, err := os.Stat(skippedFilesPath); err == nil {
+			paths = append(paths, skippedFilesPath)
 		}
 
 		err = os.Remove(fromArchivePath)
@@ -294,7 +298,11 @@ func UpdateDownloadFromAirgap(socket, fromArchivePath, namespace, registryJson,
 			filepath.Join(tmpRoot, "upstream"),
 			filepath.Join(tmpRoot, "base"),
 			filepath.Join(tmpRoot, "overlays"),
-			filepath.Join(tmpRoot, "skippedFiles"),
+		}
+
+		skippedFilesPath := filepath.Join(tmpRoot, "skippedFiles")
+		if _, err := os.Stat(skippedFilesPath); err == nil {
+			paths = append(paths, skippedFilesPath)
 		}
 
 		err = os.Remove(fromArchivePath)

diff --git a/integration/replicated/generate.go b/integration/replicated/generate.go
@@ -144,7 +144,10 @@ func generateExpectedFilesystem(namespace, rawArchivePath string) ([]byte, error
 		path.Join(tmpRootDir, "upstream"),
 		path.Join(tmpRootDir, "base"),
 		path.Join(tmpRootDir, "overlays"),
-		path.Join(tmpRootDir, "skippedFiles"),
+	}
+	skippedFilesPath := path.Join(tmpRootDir, "skippedFiles")
+	if _, err := os.Stat(skippedFilesPath); err == nil {
+		paths = append(paths, skippedFilesPath)
 	}
 	if err := tarGz.Archive(paths, archiveFile); err != nil {
 		return nil, errors.Wrap(err, "failed to create archive")

diff --git a/pkg/pull/archive.go b/pkg/pull/archive.go
@@ -32,7 +32,11 @@ func writeArchiveAsConfigMap(pullOptions PullOptions, u *upstreamtypes.Upstream,
 		path.Join(pullOptions.RootDir, u.Name, "upstream"),
 		path.Join(pullOptions.RootDir, u.Name, "base"),
 		path.Join(pullOptions.RootDir, u.Name, "overlays"),
-		path.Join(pullOptions.RootDir, u.Name, "skippedFiles"),
+	}
+
+	skippedFilesPath := path.Join(pullOptions.RootDir, "skippedFiles")
+	if _, err := os.Stat(skippedFilesPath); err == nil {
+		paths = append(paths, skippedFilesPath)
 	}
 
 	tempDir, err := ioutil.TempDir("", "kots")