feat: Extend the span statistics

And use it for deciding when to inline single/simple blocks

Author: lu-zero

src/formatter.rs

@@ -233,7 +233,15 @@ impl<'a> Formatter<'a> {
             self.add_new_line(query);
         }
         query.push_str(&self.equalize_whitespace(&self.format_reserved_word(token.value)));
-        if newline_after && token.alias != "CREATE" {
+
+        if newline_after
+            && token.alias != "CREATE"
+            && (self.options.max_inline_top_level.is_none()
+                || self
+                    .next_non_whitespace_token(1)
+                    .is_some_and(|t| t.kind != TokenKind::OpenParen)
+                || span_info.arguments != 0)
+        {
             self.indentation.increase_top_level(span_info);
             self.add_new_line(query);
         } else {
@@ -553,6 +561,17 @@ impl<'a> Formatter<'a> {
         }
     }
 
+    fn next_non_whitespace_token(&self, idx: usize) -> Option<&Token<'_>> {
+        let index = self.index.checked_add(idx);
+        if let Some(index) = index {
+            self.tokens[index..]
+                .iter()
+                .find(|t| t.kind != TokenKind::Whitespace)
+        } else {
+            None
+        }
+    }
+
     fn top_level_tokens_info(&self) -> SpanInfo {
         let mut block_level = self.block_level;
         let mut full_span = 0;

src/lib.rs

@@ -688,12 +688,11 @@ mod tests {
             "
             INSERT INTO t(id, a, min, max)
             SELECT input.id, input.a, input.min, input.max
-            FROM
-              (
-                SELECT id, a, min, max
-                FROM foo
-                WHERE a IN ('a', 'b')
-              ) AS input
+            FROM (
+              SELECT id, a, min, max
+              FROM foo
+              WHERE a IN ('a', 'b')
+            ) AS input
             WHERE (SELECT true FROM condition)
             ON CONFLICT ON CONSTRAINT a_id_key DO UPDATE SET
               id = EXCLUDED.id,
@@ -2283,8 +2282,7 @@ mod tests {
             cc AS (
               INSERT INTO
                 C (a, b, c, d)
-              VALUES
-                (1, 2, 3, 4)
+              VALUES (1, 2, 3, 4)
             )
             SELECT b, field
             FROM a, aa;"