From ee154ce0abe26df2dcad0a9a7c12dae0d1ef14f2 Mon Sep 17 00:00:00 2001
From: Winfried Plappert <18740761+wplapper@users.noreply.github.com>
Date: Sat, 31 Jan 2026 19:58:29 +0000
Subject: [PATCH] restic rewrite --include

added function Count() to the *TreeWriter methods
---
 internal/data/tree.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/internal/data/tree.go b/internal/data/tree.go
index 54f2cb131..1d8ef28af 100644
--- a/internal/data/tree.go
+++ b/internal/data/tree.go
@@ -223,6 +223,12 @@ func (t *TreeWriter) Finalize(ctx context.Context) (restic.ID, error) {
 	return id, err
 }
 
+// Count returns the number of nodes in the tree
+func (t *TreeWriter) Count() int {
+
+	return t.builder.countNodes
+}
+
 func SaveTree(ctx context.Context, saver restic.BlobSaver, nodes TreeNodeIterator) (restic.ID, error) {
 	treeWriter := NewTreeWriter(saver)
 	for item := range nodes {