From f5099356a12a363bc31df9756ea379671112e62c Mon Sep 17 00:00:00 2001
From: Tom Alexander <tom@fizz.buzz>
Date: Fri, 21 Apr 2023 21:39:26 -0400
Subject: [PATCH] Add support for parsing the planning element in sections.

---
 src/parser/document.rs | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/parser/document.rs b/src/parser/document.rs
index 28b4157..9020903 100644
--- a/src/parser/document.rs
+++ b/src/parser/document.rs
@@ -6,6 +6,7 @@ use crate::parser::object_parser::standard_set_object;
 use crate::parser::parser_context::ContextElement;
 use crate::parser::parser_context::ContextTree;
 use crate::parser::parser_context::ExitMatcherNode;
+use crate::parser::planning::planning;
 use crate::parser::property_drawer::property_drawer;
 use crate::parser::util::blank_line;
 use crate::parser::util::maybe_consume_trailing_whitespace_if_not_exiting;
@@ -168,20 +169,27 @@ fn section<'r, 's>(context: Context<'r, 's>, mut input: &'s str) -> Res<&'s str,
     let element_matcher = parser_with_context!(element)(&parser_context);
     let exit_matcher = parser_with_context!(exit_matcher_parser)(&parser_context);
     // TODO: Match whatever a planning is.
-    let (mut remaining, property_drawer_element) =
-        opt(parser_with_context!(property_drawer)(&parser_context))(input)?;
-    if property_drawer_element.is_none() {
+    let (mut remaining, (planning_element, property_drawer_element)) = tuple((
+        opt(parser_with_context!(planning)(&parser_context)),
+        opt(parser_with_context!(property_drawer)(&parser_context)),
+    ))(input)?;
+    if planning_element.is_none() && property_drawer_element.is_none() {
         let (remain, _ws) = many0(blank_line)(remaining)?;
         remaining = remain;
         input = remain;
     }
     let (remaining, (mut children, _exit_contents)) = verify(
         many_till(element_matcher, exit_matcher),
-        |(children, _exit_contents)| !children.is_empty() || property_drawer_element.is_some(),
+        |(children, _exit_contents)| {
+            !children.is_empty() || property_drawer_element.is_some() || planning_element.is_some()
+        },
     )(remaining)?;
     property_drawer_element
         .map(Element::PropertyDrawer)
         .map(|ele| children.insert(0, ele));
+    planning_element
+        .map(Element::Planning)
+        .map(|ele| children.insert(0, ele));
 
     let (remaining, _trailing_ws) =
         maybe_consume_trailing_whitespace_if_not_exiting(context, remaining)?;