lax-sql 0.1.2

Lax SQL formatter that never reinterprets your code. Dialect agnostic by construction. Usable as a library or a dprint plugin.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

use std::path::Path;

use anyhow::Result;
use dprint_core::configuration::resolve_new_line_kind;
use dprint_core::formatting::PrintOptions;

use crate::configuration::Configuration;
use crate::generation;

pub fn format_text(_path: &Path, text: &str, config: &Configuration) -> Result<Option<String>> {
  let result = format_text_inner(text, config)?;
  if result == text { Ok(None) } else { Ok(Some(result)) }
}

fn format_text_inner(text: &str, config: &Configuration) -> Result<String> {
  let text = text.strip_prefix('\u{FEFF}').unwrap_or(text);
  let tokens = generation::tokenize(text);
  if has_ignore_file_comment(&tokens, &config.ignore_file_comment_text) {
    return Ok(text.to_string());
  }
  let statements = generation::parse(&tokens, text);
  if statements.is_empty() {
    return Ok(String::new());
  }
  Ok(dprint_core::formatting::format(
    || generation::generate(&statements, text, config),
    PrintOptions {
      indent_width: config.indent_width,
      max_width: config.line_width,
      use_tabs: config.use_tabs,
      new_line_text: resolve_new_line_kind(text, config.new_line_kind),
    },
  ))
}

/// True when a comment in the first comment cluster of the file contains the
/// ignore file directive. The cluster starts on the first line and ends at
/// the first blank line or non comment construct.
fn has_ignore_file_comment(tokens: &[generation::Token], directive: &str) -> bool {
  for (index, token) in tokens.iter().enumerate() {
    match token.kind {
      generation::TokenKind::Whitespace { newlines } => {
        if newlines >= 2 || (index == 0 && newlines >= 1) {
          return false;
        }
      }
      generation::TokenKind::LineComment | generation::TokenKind::BlockComment => {
        if token.text.contains(directive) {
          return true;
        }
      }
      _ => return false,
    }
  }
  false
}