feat: bt improve

pkg/util/path.go

@@ -0,0 +1,81 @@
+package util
+
+import (
+	"errors"
+	"io"
+	"os"
+	"path/filepath"
+)
+
+func Dir(path string) string {
+	dir := filepath.Dir(path)
+	if dir == "." {
+		return ""
+	}
+	return dir
+}
+
+func Filepath(path string, originName string, customName string) string {
+	if customName == "" {
+		customName = originName
+	}
+	return filepath.Join(path, customName)
+}
+
+// SafeRemove remove file safely, ignoring errors if the path does not exist.
+func SafeRemove(name string) error {
+	if err := os.Remove(name); err != nil && !errors.Is(err, os.ErrNotExist) {
+		return err
+	}
+	return nil
+}
+
+// SafeRemoveAll remove file and parent directories safely
+func SafeRemoveAll(path string, names []string) error {
+	for _, name := range names {
+		err := SafeRemove(filepath.Join(path, name))
+		if err != nil {
+			return err
+		}
+		if err := safeRemoveParent(path, name); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func safeRemoveParent(path string, subPath string) error {
+	currPath := filepath.Dir(subPath)
+	if currPath == "." {
+		return nil
+	}
+	// if directory is empty, remove it
+	dir := filepath.Join(path, filepath.Dir(subPath))
+	empty, err := isEmpty(dir)
+	if err != nil {
+		return err
+	}
+	if empty {
+		err = SafeRemove(dir)
+		if err != nil {
+			return err
+		}
+		return safeRemoveParent(path, currPath)
+	}
+	return nil
+}
+
+// check directory is empty
+func isEmpty(name string) (bool, error) {
+	f, err := os.Open(name)
+	if err != nil {
+		return false, err
+	}
+	defer f.Close()
+
+	_, err = f.Readdirnames(1) // Or f.Readdir(1)
+	if err == io.EOF {
+		return true, nil
+	}
+	return false, err // Either not empty or error, suits both cases
+}

pkg/util/path_test.go

@@ -0,0 +1,198 @@
+package util
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestDir(t *testing.T) {
+	type args struct {
+		path string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "empty path",
+			args: args{
+				path: ".",
+			},
+			want: "",
+		},
+		{
+			name: "normal path case 1",
+			args: args{
+				path: "./a/b/c/1.txt",
+			},
+			want: "a/b/c",
+		},
+		{
+			name: "normal path case 2",
+			args: args{
+				path: "a/b/c/1.txt",
+			},
+			want: "a/b/c",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := Dir(tt.args.path); got != tt.want {
+				t.Errorf("Dir() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestFilepath(t *testing.T) {
+	type args struct {
+		path       string
+		originName string
+		customName string
+	}
+	tests := []struct {
+		name string
+		args args
+		want string
+	}{
+		{
+			name: "origin name",
+			args: args{
+				path:       "/Downloads",
+				originName: "1.txt",
+				customName: "",
+			},
+			want: "/Downloads/1.txt",
+		},
+		{
+			name: "origin name",
+			args: args{
+				path:       "/Downloads",
+				originName: "1.txt",
+				customName: "2.txt",
+			},
+			want: "/Downloads/2.txt",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := Filepath(tt.args.path, tt.args.originName, tt.args.customName); got != tt.want {
+				t.Errorf("Filepath() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestSafeRemove(t *testing.T) {
+	name := "test_safe_remove.data"
+	_, err := os.Create(name)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := SafeRemove(name); err != nil {
+		t.Fatal(err)
+	}
+	if _, err := os.Stat(name); err == nil {
+		t.Fatal(err)
+	}
+
+	if err := SafeRemove("test_safe_remove_not_exist.data"); err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestSafeRemoveAll(t *testing.T) {
+	testDir := "test_dir"
+	err := os.Mkdir(testDir, 0755)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.RemoveAll(testDir)
+
+	doSafeRemoveAll(t, testDir, []string{
+		"1.txt",
+	}, []string{})
+
+	doSafeRemoveAll(t, testDir, []string{
+		"1.txt",
+	}, []string{
+		"2.txt",
+	}, "2.txt")
+
+	doSafeRemoveAll(t, testDir, []string{
+		"1.txt",
+		"a/b/c/1.txt",
+		"a/b/1.txt",
+		"a/1.txt",
+	}, []string{})
+
+	doSafeRemoveAll(t, testDir, []string{
+		"1.txt",
+		"a/b/c/1.txt",
+		"a/b/1.txt",
+		"a/1.txt",
+	}, []string{
+		"a/b/2.txt",
+	}, "a/b/2.txt")
+
+	doSafeRemoveAll(t, testDir, []string{
+		"1.txt",
+		"a/b/c/1.txt",
+		"a/b/1.txt",
+		"a/1.txt",
+	}, []string{
+		"a/2.txt",
+	}, "a/2.txt")
+}
+
+func doSafeRemoveAll(t *testing.T, path string, downloadNames []string, otherNames []string, exist ...string) {
+	preCreate(t, path, downloadNames)
+	preCreate(t, path, otherNames)
+
+	if err := SafeRemoveAll(path, downloadNames); err != nil {
+		t.Fatal(err)
+	}
+
+	if len(exist) == 0 {
+		for _, name := range downloadNames {
+			filePath := filepath.Join(path, name)
+			if isExist(filePath) {
+				t.Fatalf("file %s should not exist", filePath)
+			}
+			subDirPath := filepath.Dir(name)
+			if subDirPath != "." {
+				dirPath := filepath.Join(path, subDirPath)
+				if isExist(dirPath) {
+					t.Fatalf("dir %s should not exist", dirPath)
+				}
+			}
+		}
+		return
+	}
+
+	for _, name := range exist {
+		if !isExist(filepath.Join(path, name)) {
+			t.Fatalf("file %s should exist", name)
+		}
+	}
+}
+
+func preCreate(t *testing.T, path string, names []string) {
+	for _, name := range names {
+		fullPath := filepath.Join(path, name)
+		if err := os.MkdirAll(filepath.Dir(fullPath), 0755); err != nil {
+			t.Fatal(err)
+		}
+		_, err := os.Create(fullPath)
+		if err != nil {
+			t.Fatal(err)
+		}
+	}
+}
+
+func isExist(path string) bool {
+	_, err := os.Stat(path)
+	return err == nil || os.IsExist(err)
+}

pkg/util/timer.go

@@ -16,10 +16,6 @@ func (t *Timer) Pause() {
 	t.used += time.Now().UnixNano() - t.t
 }
 
-func (t *Timer) Continue() {
-	t.t = time.Now().UnixNano()
-}
-
 func (t *Timer) Used() int64 {
 	return (time.Now().UnixNano() - t.t) + t.used
 }