bugfix: Clean all temporary files before today

version-fox · Jan 23, 2024 · c738772 · c738772
1 parent 5763ac1
commit c738772
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/internal/config/config.go b/internal/config/config.go
@@ -38,7 +38,7 @@ var (
 func NewConfigWithPath(p string) (*Config, error) {
 	if !util.FileExists(p) {
 		content, err := yaml.Marshal(defaultConfig)
-		if err != nil {
+		if err == nil {
 			_ = os.WriteFile(p, content, 0644)
 			return defaultConfig, nil
 		}

diff --git a/internal/tmp.go b/internal/tmp.go
@@ -20,6 +20,8 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strconv"
+	"strings"
 
 	"github.com/version-fox/vfox/internal/util"
 )
@@ -30,7 +32,27 @@ type Temp struct {
 }
 
 func (t *Temp) Remove() {
-	_ = os.RemoveAll(t.CurProcessPath)
+	dir, err := os.ReadDir(t.dirPath)
+	if err == nil {
+		_ = os.RemoveAll(t.CurProcessPath)
+		for _, file := range dir {
+			if !file.IsDir() {
+				continue
+			}
+			names := strings.SplitN(file.Name(), "-", 2)
+			if len(names) != 2 {
+				continue
+			}
+			timestamp := names[0]
+			i, err := strconv.ParseInt(timestamp, 10, 64)
+			if err != nil {
+				continue
+			}
+			if util.IsBeforeToday(i) {
+				_ = os.Remove(filepath.Join(t.dirPath, file.Name()))
+			}
+		}
+	}
 }
 
 func NewTemp(dirPath string, pid int) (*Temp, error) {