began to add document support

Author: RPG-Alex

src/dialect.rs

@@ -0,0 +1,75 @@
+//! Module for handling dialect selection for parsing
+use sqlparser::dialect::AnsiDialect;
+use sqlparser::dialect::BigQueryDialect;
+use sqlparser::dialect::ClickHouseDialect;
+use sqlparser::dialect::DatabricksDialect;
+use sqlparser::dialect::DuckDbDialect;
+use sqlparser::dialect::GenericDialect;
+use sqlparser::dialect::HiveDialect;
+use sqlparser::dialect::MsSqlDialect;
+use sqlparser::dialect::MySqlDialect;
+use sqlparser::dialect::OracleDialect;
+use sqlparser::dialect::PostgreSqlDialect;
+use sqlparser::dialect::RedshiftSqlDialect;
+use sqlparser::dialect::SQLiteDialect;
+use sqlparser::dialect::SnowflakeDialect;
+
+/// Enum used for SQL dialect selection
+#[derive(PartialEq, Eq)]
+pub enum Dialect {
+    /// ANSI SQL dialect
+    Ansi,
+    /// Google BigQuery SQL dialect
+    BigQuery,
+    /// ClickHouse SQL dialect
+    ClickHouse,
+    /// Databricks SQL dialect
+    Databricks,
+    /// DuckDB SQL dialect
+    DuckDb,
+    /// Generic SQL dialect
+    Generic,
+    /// Apache Hive SQL dialect
+    Hive,
+    /// Microsoft SQL Server (T-SQL) dialect
+    MsSql,
+    /// MySQL SQL dialect
+    MySql,
+    /// Oracle SQL dialect
+    Oracle,
+    /// PostgreSQL SQL dialect
+    PostgreSql,
+    /// Amazon Redshift SQL dialect
+    RedshiftSql,
+    /// SQLite SQL dialect
+    SQLite,
+    /// Snowflake SQL dialect
+    Snowflake,
+}
+
+use sqlparser::dialect::Dialect as SqlDialectTrait;
+
+impl Dialect {
+    /// Execute a function with the selected `sqlparser` dialect.
+    pub fn with_sqlparser_dialect<R>(
+        self,
+        f: impl FnOnce(&dyn SqlDialectTrait) -> R,
+    ) -> R {
+        match self {
+            Dialect::Ansi => f(&AnsiDialect {}),
+            Dialect::BigQuery => f(&BigQueryDialect {}),
+            Dialect::ClickHouse => f(&ClickHouseDialect {}),
+            Dialect::Databricks => f(&DatabricksDialect {}),
+            Dialect::DuckDb => f(&DuckDbDialect {}),
+            Dialect::Generic => f(&GenericDialect {}),
+            Dialect::Hive => f(&HiveDialect {}),
+            Dialect::MsSql => f(&MsSqlDialect {}),
+            Dialect::MySql => f(&MySqlDialect {}),
+            Dialect::Oracle => f(&OracleDialect {}),
+            Dialect::PostgreSql => f(&PostgreSqlDialect {}),
+            Dialect::RedshiftSql => f(&RedshiftSqlDialect {}),
+            Dialect::SQLite => f(&SQLiteDialect {}),
+            Dialect::Snowflake => f(&SnowflakeDialect {}),
+        }
+    }
+}

src/lib.rs

@@ -18,3 +18,4 @@ pub mod files;
 pub mod source;
 pub mod sql_doc;
 pub use sql_doc::SqlDoc;
+pub mod dialect;

src/sql_doc.rs

@@ -6,6 +6,8 @@ use std::{
     vec,
 };
 
+use sqlparser::dialect::Dialect;
+
 use crate::{
     ast::ParsedSqlFile,
     comments::{Comments, LeadingCommentCapture, MultiFlatten},
@@ -286,6 +288,8 @@ pub struct SqlDocBuilder<'a> {
     multiline_flat: MultiFlatten<'a>,
     /// Tracks the chosen setting for leading comment collection
     leading_type: LeadingCommentCapture,
+    /// The dialect specified to use
+    dialect: dyn Dialect,
 }
 
 /// Enum for specifying a file doc source as a `directory` or a specific `file`