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,
  });
  }
  }