1 files changed, 73 insertions, 49 deletions

diff --git a/src/parser.rs b/src/parser.rs
index c7a8abe..297230c 100644
--- a/src/parser.rs
+++ b/src/parser.rs
@@ -40,16 +40,18 @@ struct Context<'a> {
     cur: [Option<Symbol<'a>>;2],
     icur: usize,
     ll: Lexer<'a>,
+    source: usize,
 }
 
 impl<'a> Context<'a> {
-    pub fn new( mut ll: Lexer<'a> ) -> Self {
+    pub fn new( mut ll: Lexer<'a>, source: usize ) -> Self {
         let cur = [ll.next(), ll.next()];
         //println!(" - cur: {:?}, peek: {:?}", cur[0], cur[1] );
         Self {
             cur,
             icur: 0,
-            ll
+            ll,
+            source,
         }
     }
     
@@ -75,6 +77,10 @@ impl<'a> Context<'a> {
             Ok(f( self.cur().unwrap().symbol(), self.peek().unwrap().symbol()))
         }
     }
+
+    pub fn source(&self) -> usize {
+        self.source
+    }
 }
 
 trait SymbolHelper {
@@ -195,7 +201,7 @@ impl Parser {
         }
     }
 
-    fn lex_error(&self, error: &Symbol) -> ParseResult<Token> {
+    fn lex_error<T>(&self, error: &Symbol) -> ParseResult<T> {
         if let SymbolType::Error{what} = error.symbol() {
             Err(ParseError {
                 position: error.start().clone(),
@@ -209,15 +215,15 @@ impl Parser {
         }
     }
 
-    pub fn parse(&self, src: &str ) -> ParseResult<Token> {
+    pub fn parse(&self, src: &str, source: usize ) -> ParseResult<Vec<View>> {
         let ll = Lexer::new( src );
-        let mut ctx = Context::new( ll );
+        let mut ctx = Context::new( ll, source );
 
         // Parse the root of the file, the input context
         self.p_input( &mut ctx )
     }
 
-    fn p_input(&self, ctx: &mut Context ) -> ParseResult<Token> {
+    fn p_input(&self, ctx: &mut Context ) -> ParseResult<Vec<View>> {
         let mut children = Vec::new();
         loop {
             if ctx.cur().is_none() {
@@ -229,7 +235,7 @@ impl Parser {
                     children.push( self.parse_tag_view( ctx )? );
                 }
                 SymbolType::Error{..} => {
-                    return self.lex_error( &ctx.cur().unwrap() );
+                    self.lex_error( &ctx.cur().unwrap() )?;
                 }
                 _ => {
 
@@ -237,11 +243,11 @@ impl Parser {
             }
             ctx.next();
         }
-        Ok(Token::Root(children))
+        Ok(children)
     }
 
-    fn parse_tag_view(&self, ctx: &mut Context) -> ParseResult<Token> {
-        let mut tb = self.parse_open_tag( ctx )?;
+    fn parse_tag_view(&self, ctx: &mut Context) -> ParseResult<View> {
+        let tb = self.parse_open_tag( ctx )?;
         if !tb.is_name( &SymbolType::View ) {
             return Err(ParseError{
                 position: tb.start_pos(),
@@ -250,9 +256,10 @@ impl Parser {
         }
 
         if tb.is_unary() {
-            return tb.build();
+            return tb.build_view(ctx, Vec::new());
         }
         let mut children = Vec::new();
+        let mut outputs = Vec::new();
 
         //println!("--parse-tag-view-- tag parsed, next token: {:?}", ctx.cur() );
         loop {
@@ -265,47 +272,37 @@ impl Parser {
                     ctx.next();
                 }
                 SymbolType::StartFlat => {
-                    //child.is_name(SymbolType::Output)
-                    children.push( self.parse_tag_body( ctx )? );
+                    let subtb = self.parse_tag_body( ctx )?;
+                    if subtb.is_name(&SymbolType::Output) {
+                        outputs.push(subtb.build_output(ctx)?);
+                    } else {
+                        children.push(subtb.build()?);
+                    }
                 }
                 SymbolType::StartPoint => {
                     let end = self.parse_end_tag( ctx )?;
                     //println!("End tag: {:?}, open tag: {:?}", end, tb.name() );
                     if tb.is_name(end.symbol()) {
                         // They match, time to decide what we're doing.
-                        let any_outputs = children.iter().any(
-                            |t| matches!(t, Token::Output(..))
+                        let any_outputs = outputs.len() > 0;
+                        let all_text = children.iter().all(
+                            |t| matches!(t, Token::Text(..))
                             );
-                        if children.iter().all(|t| matches!(t, Token::Text(..)) || matches!(t, Token::Output(..))) {
-                            // Everything is either text or output
-                            if !any_outputs {
-                                // Special case, if it's only text then we create an implicit output.
-                                tb.add_child(Token::Output("content".to_string(), /*Properties::new(),*/ children ));
-                            }
-                            else {
-                                // Standard case, outputs are included, text is
-                                // skipped.
-                                for token in children {
-                                    if matches!(token, Token::Output(..)) {
-                                        tb.add_child( token );
-                                    }
-                                }
-                            }
+                        if !any_outputs {
+                            // Create an implicit output called "content"
+                            outputs.push(Output{
+                                name: "content".to_string(),
+                                code: children,
+                            });
                         } else {
-                            // We have another mix, now check to see if there
-                            // are any outptus, if so this is an error.
-                            if any_outputs {
+                            if !all_text {
                                 return Err(ParseError{
                                     position: *end.start(),
                                     what: "You cannot mix non-output and output tags in a view.".to_string()
                                 });
-                            } else {
-                                // No outputs at all, we create an implict output
-                                tb.add_child(Token::Output("content".to_string(), /*Properties::new(), */children ));
                             }
                         }
-
-                        return tb.build();
+                        return tb.build_view(ctx,outputs);
                     } else {
                         // They don't match, complain.
                         return Err(ParseError{
@@ -392,10 +389,10 @@ impl Parser {
         Ok(name)
     }
 
-    fn parse_tag_body(&self, ctx: &mut Context ) -> ParseResult<Token> {
+    fn parse_tag_body<'a>(&self, ctx: &mut Context<'a> ) -> ParseResult<TagBuilder<'a>> {
         let mut tb = self.parse_open_tag( ctx )?;
         if tb.is_unary() {
-            return tb.build();
+            return Ok(tb);
         }
 
         loop {
@@ -408,13 +405,13 @@ impl Parser {
                     ctx.next();
                 }
                 SymbolType::StartFlat => {
-                    tb.add_child( self.parse_tag_body( ctx )? );
+                    tb.add_child( self.parse_tag_body( ctx )?.build()? );
                 }
                 SymbolType::StartPoint => {
                     let end = self.parse_end_tag( ctx )?;
                     if tb.is_name(end.symbol()) {
                         // They match, end.
-                        return tb.build();
+                        return Ok(tb);
                     } else {
                         // They don't match, complain.
                         return Err(ParseError{
@@ -424,7 +421,7 @@ impl Parser {
                     }
                 }
                 SymbolType::Error{..} => {
-                    self.lex_error( &ctx.cur().unwrap() )?;
+                    return self.lex_error( &ctx.cur().unwrap() );
                 }
                 _ => {
 
@@ -565,15 +562,42 @@ impl<'a> TagBuilder<'a> {
         self.tag_type
     }
 
+    pub fn build_view(mut self, ctx: &Context, outputs: Vec<Output>) -> ParseResult<View> {
+        if let Some(sym) = self.name {
+            if *sym.symbol() == SymbolType::View {
+                Ok(View {
+                    name: self.params.swap_remove(0),
+                    source: ctx.source(),
+                    outputs: outputs,
+                    //theme: Option<String>
+                    layout: self.props.get("layout").cloned(),
+                })
+            } else {
+                Err(ParseError::new(self.start_pos, &format!("Expected tag type view, found {:?}", sym.symbol())))
+           }
+        } else {
+            Err(ParseError::new(self.start_pos, "Broken Parser? No tag type found when building view."))
+        }
+    }
+
+    pub fn build_output(mut self, ctx: &Context) -> ParseResult<Output> {
+        if let Some(sym) = self.name {
+            if *sym.symbol() == SymbolType::Output {
+                Ok(Output {
+                    name: self.params.swap_remove(0),
+                    code: self.children,
+                })
+            } else {
+                Err(ParseError::new(self.start_pos, &format!("Expected tag type view, found {:?}", sym.symbol())))
+            }
+        } else {
+            Err(ParseError::new(self.start_pos, "Broken Parser? No tag type found when building view."))
+        }
+    }
+
     pub fn build(mut self) -> ParseResult<Token> {
         if let Some(sym) = self.name {
             match sym.symbol() {
-                SymbolType::View => {
-                    Ok(Token::View(self.params.swap_remove(0), self.props, self.children))
-                }
-                SymbolType::Output => {
-                    Ok(Token::Output(self.params.swap_remove(0), /*self.props,*/ self.children))
-                }
                 SymbolType::Show => {
                     Ok(Token::Show(self.params.swap_remove(0), self.props))
                 }