feat(settings): show categorized storage usage on demand

internal/api/storage_test.go

@@ -0,0 +1,81 @@
+package api
+
+import (
+	"context"
+	"errors"
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestStorageCountsOwnedRootsOnce(t *testing.T) {
+	for _, layout := range []string{"nested-vault", "external-vault", "nested-data", "same-root"} {
+		t.Run(layout, func(t *testing.T) {
+			root := t.TempDir()
+			data, vault := filepath.Join(root, "data"), filepath.Join(root, "data", "vault")
+			switch layout {
+			case "external-vault":
+				vault = filepath.Join(root, "external")
+			case "nested-data":
+				vault, data = root, filepath.Join(root, "app")
+			case "same-root":
+				vault = data
+			}
+			write := func(base, rel string, size int) {
+				t.Helper()
+				path := filepath.Join(base, rel)
+				if err := os.MkdirAll(filepath.Dir(path), 0755); err != nil {
+					t.Fatal(err)
+				}
+				if err := os.WriteFile(path, make([]byte, size), 0600); err != nil {
+					t.Fatal(err)
+				}
+			}
+			write(vault, "note.md", 5)
+			write(vault, "dir/assets/a.png", 7)
+			write(vault, "old.assets/b.png", 11)
+			write(vault, ".rawmark/trash/id/meta.json", 13)
+			write(vault, ".rawmark/trash/id/assets/a.png", 17)
+			write(data, "rawmark.db", 19)
+			write(data, "index.db-wal", 23)
+			write(data, "fonts/a/file.woff2", 29)
+			write(data, "themes/a/theme.css", 31)
+			write(data, "plugins/a/main.js", 37)
+			write(data, "misc.bin", 41)
+			outside := t.TempDir()
+			write(outside, "outside.md", 1000)
+			if err := os.Symlink(outside, filepath.Join(vault, "link")); err != nil {
+				t.Fatal(err)
+			}
+			usage, err := scanStorage(context.Background(), data, vault)
+			if err != nil {
+				t.Fatal(err)
+			}
+			want := map[string]int64{"vault": 53, "databases": 42, "fonts": 29, "themes": 31, "plugins": 37, "other": 41}
+			if layout == "same-root" {
+				want["vault"] += 41
+				want["other"] = 0
+			}
+			for _, category := range usage.Categories {
+				if category.Size != want[category.ID] {
+					t.Errorf("%s = %d, want %d", category.ID, category.Size, want[category.ID])
+				}
+			}
+			if usage.TotalSize != 233 || usage.AssetsSize != 18 || usage.TrashSize != 30 || usage.CalculatedAt <= 0 {
+				t.Fatalf("总量或子项不符: %+v", usage)
+			}
+		})
+	}
+}
+
+func TestStorageDoesNotReturnPartialSuccess(t *testing.T) {
+	root := t.TempDir()
+	if _, err := scanStorage(context.Background(), root, filepath.Join(root, "missing")); !errors.Is(err, os.ErrNotExist) {
+		t.Fatalf("仓库缺失应报错: %v", err)
+	}
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+	if _, err := scanStorage(ctx, root, root); !errors.Is(err, context.Canceled) {
+		t.Fatalf("请求取消应停止扫描: %v", err)
+	}
+}