From 5c26b0d6dca327807105fcf4a539b597665ccef6 Mon Sep 17 00:00:00 2001 From: Alex Ling Date: Thu, 11 Jun 2020 10:03:34 +0000 Subject: [PATCH] Handle nested titles in the recently added section --- src/library.cr | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/library.cr b/src/library.cr index 9eaf4f7..71a520a 100644 --- a/src/library.cr +++ b/src/library.cr @@ -191,6 +191,12 @@ class Title @title_ids.map { |tid| @library.get_title! tid } end + # Get all entries, including entries in nested titles + def deep_entries + return @entries if title_ids.empty? + @entries + titles.map { |t| t.deep_entries }.flatten + end + def parents ary = [] of Title tid = @parent_id @@ -535,7 +541,7 @@ class Library def get_recently_added_entries(username) # Get all entries added within the last three months - entries = titles.map { |t| t.entries } + entries = titles.map { |t| t.deep_entries } .flatten .select { |e| e.date_added > 3.months.ago }