From 57b2f7c625ad83fbbc2d08012a84c46ffc27c3c8 Mon Sep 17 00:00:00 2001
From: Leeingnyo <dyd1928@naver.com>
Date: Tue, 14 Sep 2021 23:08:07 +0900
Subject: [PATCH] Get nested ids when title removed

---
 src/library/library.cr | 3 ++-
 src/library/title.cr   | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/library/library.cr b/src/library/library.cr
index 0a04b1a..6cf8c32 100644
--- a/src/library/library.cr
+++ b/src/library/library.cr
@@ -153,8 +153,9 @@ class Library
       title = @title_hash[title_id]
       existence = title.examine examine_context
       unless existence
+        examine_context["deleted_title_ids"].concat title.deep_titles.map &.id
+        examine_context["deleted_entry_ids"].concat title.deep_entries.map &.id
         @title_hash.delete title_id
-        examine_context["deleted_title_ids"] << title_id
       end
       existence
     end
diff --git a/src/library/title.cr b/src/library/title.cr
index fcc5f2f..db2ca30 100644
--- a/src/library/title.cr
+++ b/src/library/title.cr
@@ -102,8 +102,9 @@ class Title
       title = Library.default.get_title! title_id
       existence = title.examine context
       unless existence
+        context["deleted_title_ids"].concat title.deep_titles.map &.id
+        context["deleted_entry_ids"].concat title.deep_entries.map &.id
         Library.default.title_hash.delete title_id
-        context["deleted_title_ids"] << title_id
       end
       existence
     end