fix: fix blank background (#181)

IceWhaleTech · Dec 17, 2024 · c6c357e · c6c357e
1 parent bcbc756
commit c6c357e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/internal/background.go b/internal/background.go
@@ -17,8 +17,10 @@ func BackgroundPath(version codegen.Version) string {
 
 func DownloadReleaseBackground(url string, version string) {
 	// to check if the file exist
-	if _, err := os.Stat(BackgroundPath(version)); err == nil {
-		return
+	if fileInfo, err := os.Stat(BackgroundPath(version)); err == nil {
+		if fileInfo.Size() > 0 {
+			return
+		}
 	}
 
 	// if the background url is nil, return
@@ -35,6 +37,7 @@ func DownloadReleaseBackground(url string, version string) {
 	}
 	err := getClient.Get()
 	if err != nil {
+		os.Remove(BackgroundPath(version))
 		logger.Error("error when trying to download background", zap.Error(err))
 	} else {
 		logger.Info("download background success", zap.String("url", url), zap.String("version", version))