fix: do not push expectation errors to EOF (#539)

Author: juleswritescode

.sqlx/query-aced4382cedbfc359bb1c1ab71cf2472fa8afc0b24bf02ee1bfdfdbf111acfc8.json

@@ -92,7 +92,7 @@ impl Lexed<'_> {
     }
 
     pub(crate) fn text_range(&self, i: usize) -> TextRange {
-        assert!(i < self.len());
+        assert!(i < self.len() - 1);
         let lo = self.start[i];
         let hi = self.start[i + 1];
         TextRange::new(lo.into(), hi.into())

crates/pgt_lexer/src/lexed.rs

@@ -553,4 +553,12 @@ values ('insert', new.id, now());",
             "select\n            email,\n\n\n        from\n            auth.users;",
         ]);
     }
+
+    #[test]
+    fn does_not_panic_on_eof_expectation() {
+        Tester::from("insert").expect_errors(vec![SplitDiagnostic::new(
+            "Expected INTO_KW".to_string(),
+            TextRange::new(0.into(), 6.into()),
+        )]);
+    }
 }

crates/pgt_statement_splitter/src/lib.rs

@@ -168,9 +168,15 @@ impl<'a> Splitter<'a> {
         if self.current() == kind {
             self.advance();
         } else {
+            let token = if self.current() == SyntaxKind::EOF {
+                self.current_pos - 1
+            } else {
+                self.current_pos
+            };
+
             self.errors.push(SplitError {
                 msg: format!("Expected {:#?}", kind),
-                token: self.current_pos,
+                token,
             });
         }
     }