From 1bca9ccecb9edb583a0a94bd3611147b66909ab9 Mon Sep 17 00:00:00 2001
From: Laurenz <laurmaedje@gmail.com>
Date: Wed, 4 May 2022 23:26:15 +0200
Subject: Improve diagnostics

---
 src/parse/mod.rs    | 12 ++++++++----
 src/parse/parser.rs |  8 +++++++-
 2 files changed, 15 insertions(+), 5 deletions(-)

(limited to 'src/parse')

diff --git a/src/parse/mod.rs b/src/parse/mod.rs
index 8bf7b1c5..b473c9af 100644
--- a/src/parse/mod.rs
+++ b/src/parse/mod.rs
@@ -604,6 +604,7 @@ fn collection(p: &mut Parser, keyed: bool) -> (CollectionKind, usize) {
                 missing_coma = Some(p.trivia_start());
             }
         } else {
+            p.eat_if(NodeKind::Comma);
             kind = Some(CollectionKind::Group);
         }
     }
@@ -638,11 +639,15 @@ fn item(p: &mut Parser, keyed: bool) -> ParseResult<NodeKind> {
                 p.eat();
                 marker.perform(p, NodeKind::Keyed, expr)?;
             }
-            _ => {
+            kind => {
                 let mut msg = EcoString::from("expected identifier");
                 if keyed {
                     msg.push_str(" or string");
                 }
+                if let Some(kind) = kind {
+                    msg.push_str(", found ");
+                    msg.push_str(kind.as_str());
+                }
                 let error = NodeKind::Error(ErrorPos::Full, msg);
                 marker.end(p, error);
                 p.eat();
@@ -661,8 +666,7 @@ fn item(p: &mut Parser, keyed: bool) -> ParseResult<NodeKind> {
 /// expressions.
 fn array(p: &mut Parser, marker: Marker) {
     marker.filter_children(p, |x| match x.kind() {
-        NodeKind::Named => Err("expected expression, found named pair"),
-        NodeKind::Keyed => Err("expected expression, found keyed pair"),
+        NodeKind::Named | NodeKind::Keyed => Err("expected expression"),
         _ => Ok(()),
     });
     marker.end(p, NodeKind::ArrayExpr);
@@ -685,7 +689,7 @@ fn dict(p: &mut Parser, marker: Marker) {
             Ok(())
         }
         NodeKind::Spread | NodeKind::Comma | NodeKind::Colon => Ok(()),
-        _ => Err("expected named or keyed pair, found expression"),
+        _ => Err("expected named or keyed pair"),
     });
     marker.end(p, NodeKind::DictExpr);
 }
diff --git a/src/parse/parser.rs b/src/parse/parser.rs
index 50bb5711..f3a3ffd1 100644
--- a/src/parse/parser.rs
+++ b/src/parse/parser.rs
@@ -4,6 +4,7 @@ use std::ops::Range;
 
 use super::{TokenMode, Tokens};
 use crate::syntax::{ErrorPos, Green, GreenData, GreenNode, NodeKind};
+use crate::util::EcoString;
 
 /// A convenient token-based parser.
 pub struct Parser<'s> {
@@ -474,7 +475,12 @@ impl Marker {
             }
 
             if let Err(msg) = f(child) {
-                let error = NodeKind::Error(ErrorPos::Full, msg.into());
+                let mut msg = EcoString::from(msg);
+                if msg.starts_with("expected") {
+                    msg.push_str(", found ");
+                    msg.push_str(child.kind().as_str());
+                }
+                let error = NodeKind::Error(ErrorPos::Full, msg);
                 let inner = mem::take(child);
                 *child = GreenNode::with_child(error, inner).into();
             }
-- 
cgit v1.2.3