Collect sections in book.

1 files changed, 10 insertions, 0 deletions

diff --git a/scripts/MarkdownBook/Book.pm b/scripts/MarkdownBook/Book.pm
index f758aa3..3328cae 100644
--- a/scripts/MarkdownBook/Book.pm
+++ b/scripts/MarkdownBook/Book.pm
@@ -40,6 +40,8 @@ sub new
 	}
 	$self->{'dir'} = $dir;
 	$self->{'docs'} = [];
+	$self->{'sections'} = [];
+	$self->{'sections_by_id'} = {};
 
 	open($control_fh, '<', $dir . '/control');
 	while (<$control_fh>) {
@@ -142,6 +144,14 @@ sub create_documents
 	}
 }
 
+sub add_section
+{
+	my ($self, $section) = @_;
+
+	push(@{$self->{'sections'}}, $section);
+	$self->{'sections_by_id'}->{$section->id()} = $section;
+}
+
 sub parse
 {
 	my ($self) = @_;