sqlglot_rust/
lib.rs

1//! # sqlglot-rust
2//!
3//! A SQL parser, optimizer, and transpiler library written in Rust,
4//! inspired by Python's [sqlglot](https://github.com/tobymao/sqlglot).
5//!
6//! ## Features
7//!
8//! - Parse SQL strings into a structured AST
9//! - Generate SQL from AST nodes
10//! - Transpile between SQL dialects (30 dialects including MySQL, PostgreSQL, BigQuery, Snowflake, DuckDB, Hive, Spark, Presto, Trino, T-SQL, Oracle, ClickHouse, Redshift, and more)
11//! - Optimize SQL queries
12//! - CTEs, subqueries, UNION/INTERSECT/EXCEPT
13//! - Window functions, CAST, EXTRACT, EXISTS
14//! - Pretty-print SQL output
15//! - AST traversal (walk, find, transform)
16//!
17//! ## Quick Start
18//!
19//! ```rust
20//! use sqlglot_rust::{parse, generate, transpile, Dialect};
21//!
22//! // Parse a SQL query
23//! let ast = parse("SELECT a, b FROM t WHERE a > 1", Dialect::Ansi).unwrap();
24//!
25//! // Generate SQL for a specific dialect
26//! let sql = generate(&ast, Dialect::Postgres);
27//! assert_eq!(sql, "SELECT a, b FROM t WHERE a > 1");
28//!
29//! // One-step transpile between dialects
30//! let result = transpile("SELECT a, b FROM t", Dialect::Ansi, Dialect::Postgres).unwrap();
31//! ```
32
33pub mod ast;
34pub mod dialects;
35pub mod errors;
36pub mod generator;
37pub mod optimizer;
38pub mod parser;
39pub mod schema;
40pub mod tokens;
41
42pub use ast::{Expr, QuoteStyle, Statement};
43pub use dialects::Dialect;
44pub use errors::SqlglotError;
45pub use generator::{generate, generate_pretty};
46pub use parser::parse;
47
48/// Transpile a SQL string from one dialect to another.
49///
50/// This is the primary high-level API, corresponding to Python sqlglot's
51/// `sqlglot.transpile()`.
52///
53/// # Example
54///
55/// ```rust
56/// use sqlglot_rust::{transpile, Dialect};
57///
58/// let result = transpile(
59///     "SELECT CAST(x AS INT) FROM t",
60///     Dialect::Ansi,
61///     Dialect::Postgres,
62/// ).unwrap();
63/// ```
64///
65/// # Errors
66///
67/// Returns a [`SqlglotError`] if parsing fails.
68pub fn transpile(
69    sql: &str,
70    read_dialect: Dialect,
71    write_dialect: Dialect,
72) -> errors::Result<String> {
73    let ast = parse(sql, read_dialect)?;
74    let transformed = dialects::transform(&ast, read_dialect, write_dialect);
75    Ok(generate(&transformed, write_dialect))
76}
77
78/// Transpile a SQL string, returning multiple statements if the input
79/// contains semicolons.
80///
81/// # Errors
82///
83/// Returns a [`SqlglotError`] if parsing fails.
84pub fn transpile_statements(
85    sql: &str,
86    read_dialect: Dialect,
87    write_dialect: Dialect,
88) -> errors::Result<Vec<String>> {
89    let stmts = parser::parse_statements(sql, read_dialect)?;
90    let mut results = Vec::with_capacity(stmts.len());
91    for stmt in &stmts {
92        let transformed = dialects::transform(stmt, read_dialect, write_dialect);
93        results.push(generate(&transformed, write_dialect));
94    }
95    Ok(results)
96}
sqlglot_rust/lib.rs

sqlglot_rust/
lib.rs
