Fix restore of groups/categories with docs

Ref: #2801
TypeStrong · Dec 12, 2024 · 288856c · 288856c
1 parent a7bff48
commit 288856c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,10 @@ title: Changelog
 
 ## Unreleased
 
+### Bug Fixes
+
+-   Fix restoration of groups/categories including documents, #2801.
+
 ## v0.27.4 (2024-12-09)
 
 ### Features

diff --git a/src/lib/models/ReflectionCategory.ts b/src/lib/models/ReflectionCategory.ts
@@ -76,7 +76,7 @@ export class ReflectionCategory {
                     const child = project.getReflectionById(
                         de.oldIdToNewId[childId] ?? -1,
                     );
-                    if (child?.isDeclaration()) {
+                    if (child?.isDeclaration() || child?.isDocument()) {
                         this.children.push(child);
                     }
                 }

diff --git a/src/lib/models/ReflectionGroup.ts b/src/lib/models/ReflectionGroup.ts
@@ -96,7 +96,7 @@ export class ReflectionGroup {
                     const child = project.getReflectionById(
                         de.oldIdToNewId[childId] ?? -1,
                     );
-                    if (child?.isDeclaration()) {
+                    if (child?.isDeclaration() || child?.isDocument()) {
                         this.children.push(child);
                     }
                 }