sql_beautifier 0.10.2 → 0.10.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 918b0bcf08b9f13dabc26250b303f2b35244cf996877939ac6c31ef7d263dd24
-  data.tar.gz: b8daaffc147aaf1556823c3bbc9ca95efdf5070a0747638b2dcc2af14b090e0e
+  metadata.gz: 35c33bf6ef1f1c6095643a9e6661aa16a93a29ff44534aab07fddd0befec5b43
+  data.tar.gz: aab36afe287416dc61b4633f3d4c07975736d6111a427118bf975be2742325a6
 SHA512:
-  metadata.gz: 3e9188e453a6055bbd79f7c1b4eb0b9aed091a7123d3ce83613e64a82629898022c090845fa36f6a68ff7a51cb692ad97b07dae55b3f1ec5927d196daa267ed7
-  data.tar.gz: 89a38090f7fd926e77a54d46d340a50d7b45bd461f49eda0fda865d6801f62ad8ccac2c751f4535cffd6418f1e2f9146f43fd9573e398bcb5113eb01ab022a5e
+  metadata.gz: 58f7b763cc122816a919b2caa1ba0281efb5fa8dc1bfbd4e97f27c6838728e8a8db2a6018d819d1f2314359851f1cd138f28bb0f0fb21986defa4a1e11ad724a
+  data.tar.gz: db2e7de085a689a9331e831432f5ed31f8a58304c9f20f4cca2376acf68681da9ba84741c18bf0fb9b9002c85fb4f1d87c3e8bf92fdf31a384455caba9cccb23

data/CHANGELOG.md

@@ -2,6 +2,14 @@
 
 ## [X.X.X] - YYYY-MM-DD
 
+## [0.10.3] - 2026-03-30
+
+- Add `DROP TABLE` formatting — `DROP TABLE [IF EXISTS] table_name` statements are now recognized and rendered with proper keyword casing and PascalCase table names instead of passing through as normalized text
+- Add `CREATE TABLE` (DDL) formatting — `CREATE [TEMP|TEMPORARY|UNLOGGED|LOCAL] TABLE [IF NOT EXISTS] table_name (column_defs)` statements with column definitions are now recognized and rendered with proper keyword casing and PascalCase table names
+- Fix `INSERT INTO` with a single column rendering the column list on multiple lines — single-column lists now render inline (e.g. `insert into Table (id)` instead of expanding to three lines)
+- Reject `DROP TABLE` statements with trailing text (e.g. `CASCADE`, `RESTRICT`, multiple table names) — these now return `nil` instead of silently dropping the trailing text
+- Reject `CREATE TABLE` (DDL) statements with trailing clauses after column definitions (e.g. `WITH (...)`, `TABLESPACE ...`) — these now return `nil` instead of silently dropping the trailing text
+
 ## [0.10.2] - 2026-03-30
 
 - Fix `INSERT INTO ... (columns) (SELECT ...)` not being recognized — `InsertQuery.parse_body` now unwraps parenthesized SELECT subqueries, supporting PostgreSQL's valid `INSERT INTO table (cols) (SELECT ...)` syntax

data/README.md

@@ -411,6 +411,36 @@ where   users.account_id = accounts.id
 returning users.id;
 ```
 
+### DROP TABLE
+
+`DROP TABLE` statements are recognized and formatted with proper keyword casing and table name formatting:
+
+```ruby
+SqlBeautifier.call("DROP TABLE IF EXISTS persons")
+```
+
+Produces:
+
+```sql
+drop table if exists Persons;
+```
+
+### CREATE TABLE (DDL)
+
+`CREATE TABLE` statements with column definitions are recognized and formatted with proper keyword casing and table name formatting. Column definitions are preserved as-is:
+
+```ruby
+SqlBeautifier.call("CREATE TEMPORARY TABLE persons (id bigint)")
+```
+
+Produces:
+
+```sql
+create temporary table Persons (id bigint);
+```
+
+Modifiers (`TEMP`, `TEMPORARY`, `UNLOGGED`, `LOCAL`) and `IF NOT EXISTS` are supported.
+
 ### Set Operators (UNION, INTERSECT, EXCEPT)
 
 Compound queries joined by set operators are detected and each segment is formatted independently. The operator keyword appears on its own line with blank-line separation:

data/lib/sql_beautifier/constants.rb

@@ -42,6 +42,13 @@ module SqlBeautifier
     CLOSE_PARENTHESIS = ")"
     COMMA = ","
 
+    TABLE_MODIFIERS = %w[
+      temp
+      temporary
+      unlogged
+      local
+    ].freeze
+
     LATERAL_PREFIX_PATTERN = %r{\Alateral\s+}i
 
     WHITESPACE_REGEX = %r{\s+}

data/lib/sql_beautifier/create_table.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module SqlBeautifier
+  class CreateTable < Base
+    extend CreateTableParsing
+
+    option :modifier, default: -> {}
+    option :if_not_exists, type: Types::Bool
+    option :table_name
+    option :column_definitions
+
+    def self.parse(normalized_sql, **)
+      scanner = Scanner.new(normalized_sql)
+      return nil unless scanner.keyword_at?("create")
+
+      scanner.skip_past_keyword!("create")
+      modifier = detect_modifier(scanner)
+      scanner.skip_past_keyword!(modifier) if modifier
+
+      return nil unless scanner.keyword_at?("table")
+
+      scanner.skip_past_keyword!("table")
+
+      if_not_exists = detect_if_not_exists?(scanner)
+      skip_past_if_not_exists!(scanner) if if_not_exists
+
+      table_name = scanner.read_identifier!
+      return nil unless table_name
+
+      scanner.skip_whitespace!
+      return nil if scanner.finished?
+      return nil unless scanner.current_char == Constants::OPEN_PARENTHESIS
+
+      column_definitions = extract_column_definitions(normalized_sql, scanner)
+      return nil unless column_definitions
+
+      new(modifier: modifier, if_not_exists: if_not_exists, table_name: table_name, column_definitions: column_definitions)
+    end
+
+    def self.extract_column_definitions(normalized_sql, scanner)
+      closing = scanner.find_matching_parenthesis(scanner.position)
+      return nil unless closing
+
+      inner_text = normalized_sql[(scanner.position + 1)...closing].strip
+      return nil if inner_text.empty?
+
+      trailing_text = normalized_sql[(closing + 1)..].strip
+      return nil unless trailing_text.empty?
+
+      inner_text
+    end
+
+    private_class_method :extract_column_definitions
+
+    def render
+      parts = [Util.format_keyword("create")]
+      parts << Util.format_keyword(@modifier) if @modifier
+      parts << Util.format_keyword("table")
+      parts << "#{Util.format_keyword('if')} #{Util.format_keyword('not')} #{Util.format_keyword('exists')}" if @if_not_exists
+      parts << Util.format_table_name(@table_name)
+
+      "#{parts.join(' ')} (#{@column_definitions})\n"
+    end
+  end
+end

data/lib/sql_beautifier/create_table_as.rb

@@ -2,12 +2,7 @@
 
 module SqlBeautifier
   class CreateTableAs < Base
-    MODIFIERS = %w[
-      temp
-      temporary
-      unlogged
-      local
-    ].freeze
+    extend CreateTableParsing
 
     WITH_DATA_SUFFIX_REGEX = %r{\s+(with\s+(?:no\s+)?data)\s*\z}i
 
@@ -47,27 +42,6 @@ module SqlBeautifier
       new(modifier: modifier, if_not_exists: if_not_exists, table_name: table_name, body_sql: body_sql, suffix: suffix, depth: depth)
     end
 
-    def self.detect_modifier(scanner)
-      MODIFIERS.detect { |modifier| scanner.keyword_at?(modifier) }
-    end
-
-    def self.detect_if_not_exists?(scanner)
-      return false unless scanner.keyword_at?("if")
-
-      probe = Scanner.new(scanner.source, position: scanner.position)
-      probe.skip_past_keyword!("if")
-      return false unless probe.keyword_at?("not")
-
-      probe.skip_past_keyword!("not")
-      probe.keyword_at?("exists")
-    end
-
-    def self.skip_past_if_not_exists!(scanner)
-      scanner.skip_past_keyword!("if")
-      scanner.skip_past_keyword!("not")
-      scanner.skip_past_keyword!("exists")
-    end
-
     def self.extract_body(sql, position)
       scanner = Scanner.new(sql, position: position)
       scanner.skip_whitespace!

data/lib/sql_beautifier/create_table_parsing.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module SqlBeautifier
+  module CreateTableParsing
+    def self.extended(base)
+      base.private_class_method :detect_modifier
+      base.private_class_method :detect_if_not_exists?
+      base.private_class_method :skip_past_if_not_exists!
+    end
+
+    def detect_modifier(scanner)
+      Constants::TABLE_MODIFIERS.detect { |modifier| scanner.keyword_at?(modifier) }
+    end
+
+    def detect_if_not_exists?(scanner)
+      return false unless scanner.keyword_at?("if")
+
+      probe = Scanner.new(scanner.source, position: scanner.position)
+      probe.skip_past_keyword!("if")
+      return false unless probe.keyword_at?("not")
+
+      probe.skip_past_keyword!("not")
+      probe.keyword_at?("exists")
+    end
+
+    def skip_past_if_not_exists!(scanner)
+      scanner.skip_past_keyword!("if")
+      scanner.skip_past_keyword!("not")
+      scanner.skip_past_keyword!("exists")
+    end
+  end
+end

data/lib/sql_beautifier/drop_table.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module SqlBeautifier
+  class DropTable < Base
+    option :table_name
+    option :if_exists, type: Types::Bool
+
+    def self.parse(normalized_sql, **)
+      scanner = Scanner.new(normalized_sql)
+      return nil unless scanner.keyword_at?("drop")
+
+      scanner.skip_past_keyword!("drop")
+      return nil unless scanner.keyword_at?("table")
+
+      scanner.skip_past_keyword!("table")
+
+      if_exists = false
+
+      if scanner.keyword_at?("if")
+        return nil unless detect_if_exists?(scanner)
+
+        skip_past_if_exists!(scanner)
+        if_exists = true
+      end
+
+      table_name = scanner.read_identifier!
+      return nil unless table_name
+
+      scanner.skip_whitespace!
+      return nil unless scanner.finished?
+
+      new(table_name: table_name, if_exists: if_exists)
+    end
+
+    def self.detect_if_exists?(scanner)
+      probe = Scanner.new(scanner.source, position: scanner.position)
+      probe.skip_past_keyword!("if")
+      probe.keyword_at?("exists")
+    end
+
+    def self.skip_past_if_exists!(scanner)
+      scanner.skip_past_keyword!("if")
+      scanner.skip_past_keyword!("exists")
+    end
+
+    private_class_method :detect_if_exists?, :skip_past_if_exists!
+
+    def render
+      parts = [Util.format_keyword("drop"), Util.format_keyword("table")]
+      parts << "#{Util.format_keyword('if')} #{Util.format_keyword('exists')}" if @if_exists
+      parts << Util.format_table_name(@table_name)
+
+      "#{parts.join(' ')}\n"
+    end
+  end
+end

data/lib/sql_beautifier/formatter.rb

@@ -23,12 +23,18 @@ module SqlBeautifier
       @leading_sentinels = extract_leading_sentinels!
       return unless @normalized_value.present?
 
+      drop_table_result = DropTable.parse(@normalized_value)&.render
+      return prepend_sentinels(drop_table_result) if drop_table_result
+
       cte_result = CteQuery.parse(@normalized_value, depth: @depth)&.render
       return prepend_sentinels(cte_result) if cte_result
 
       create_table_as_result = CreateTableAs.parse(@normalized_value, depth: @depth)&.render
       return prepend_sentinels(create_table_as_result) if create_table_as_result
 
+      create_table_result = CreateTable.parse(@normalized_value)&.render
+      return prepend_sentinels(create_table_result) if create_table_result
+
       compound_result = CompoundQuery.parse(@normalized_value, depth: @depth)&.render
       return prepend_sentinels(compound_result) if compound_result
 

data/lib/sql_beautifier/insert_query.rb

@@ -199,8 +199,9 @@ module SqlBeautifier
 
     def render_column_list
       columns = Tokenizer.split_by_top_level_commas(@column_list)
-      indent = Util.whitespace(SqlBeautifier.config_for(:indent_spaces) || 4)
+      return " (#{columns.first.strip})" if columns.length == 1
 
+      indent = Util.whitespace(SqlBeautifier.config_for(:indent_spaces) || 4)
       formatted_columns = columns.map { |column| "#{indent}#{column.strip}" }.join(",\n")
 
       " (\n#{formatted_columns}\n)"

data/lib/sql_beautifier/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SqlBeautifier
-  VERSION = "0.10.2"
+  VERSION = "0.10.3"
 end

data/lib/sql_beautifier.rb

@@ -24,6 +24,9 @@ require_relative "sql_beautifier/sort_expression"
 require_relative "sql_beautifier/condition"
 require_relative "sql_beautifier/cte_definition"
 require_relative "sql_beautifier/cte_query"
+require_relative "sql_beautifier/drop_table"
+require_relative "sql_beautifier/create_table_parsing"
+require_relative "sql_beautifier/create_table"
 require_relative "sql_beautifier/create_table_as"
 require_relative "sql_beautifier/compound_query"
 require_relative "sql_beautifier/dml_rendering"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sql_beautifier
 version: !ruby/object:Gem::Version
-  version: 0.10.2
+  version: 0.10.3
 platform: ruby
 authors:
 - Kinnell Shah
@@ -82,11 +82,14 @@ files:
 - lib/sql_beautifier/condition.rb
 - lib/sql_beautifier/configuration.rb
 - lib/sql_beautifier/constants.rb
+- lib/sql_beautifier/create_table.rb
 - lib/sql_beautifier/create_table_as.rb
+- lib/sql_beautifier/create_table_parsing.rb
 - lib/sql_beautifier/cte_definition.rb
 - lib/sql_beautifier/cte_query.rb
 - lib/sql_beautifier/delete_query.rb
 - lib/sql_beautifier/dml_rendering.rb
+- lib/sql_beautifier/drop_table.rb
 - lib/sql_beautifier/expression.rb
 - lib/sql_beautifier/formatter.rb
 - lib/sql_beautifier/insert_query.rb