tsql_parser 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a0a85b6dba179ecd613d8c18923b25c14ea4ae1b7a06f0ddba09eb212c21502b
-  data.tar.gz: dbc151c034be90b2837d133b3c87e4fe476585b0108f9afd382f01a0471d4259
+  metadata.gz: 62f81b6f4b6195e663c823dbb1fda82a8a7530d534ed73a9ba27a048fdacb42d
+  data.tar.gz: c1af50b7c5421316290044ca548bd9983b568c8f6446402ba1d6b25ef46920ea
 SHA512:
-  metadata.gz: f77f801453284b9a941b2d1f8ca22718e6783a3b733181c697f0f6ff09bfc0373afdc50b39ea7915006c16c81b0b60b2641dd0926ecd13df5800c6a65a6a9a4f
-  data.tar.gz: eaf630332538b7cf7fa6d677d400882a1f7f194c95882f1f630fbbb59e3d3b902f3bc8708eecd792e1d8f51489c79831298d4673589a2d6583ceea975bf3b038
+  metadata.gz: 24c95e57e20af4b3dff6d7b15cb38e0736bddc683f025c9bc5300e6662992045e110990343497c13952524de2868fdc5341214ab744d6305416a2f001388d555
+  data.tar.gz: 06d257666a82067a42990e8c0f9a4ac8dd1644ec399f7f8aa453cfe7d2465dc680a349c7623bcf92e6f1f0d412a6a4ea2440f818bdf1b239510896b601e891cd

data/lib/parsing/formatter.rb

@@ -17,7 +17,7 @@
 module TSqlParser::Parsing
   require_relative "iterator"
   require_relative "parser"
-  require_relative "text_formatter"
+  require_relative "formatters/text_formatter"
   require_relative "model/sql_container"
   require_relative "model/flat_sql_container"
 

data/lib/parsing/formatters/strategy/base_formatter.rb

@@ -0,0 +1,28 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/base_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::SetFormatter
+
+module TSqlParser::Parsing::Formatters
+  class BaseFormatter
+    def self.get_tab_count(line, tab = "    ")
+      tab_count = 0
+      while line.start_with? tab
+        tab_count += 1
+        line = line.sub(tab, "")
+      end
+      tab_count
+    end
+  end
+end

data/lib/parsing/formatters/strategy/insert_formatter.rb

@@ -0,0 +1,66 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/insert_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::InsertFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative "base_formatter"
+
+  class InsertFormatter < BaseFormatter
+    def self.format(text, tab = "    ")
+      formatted = []
+      lines = text.split("\n")
+      search = "INSERT INTO"
+      lines.each do |line|
+        first = line.strip.split(" ").first
+        if first != "INSERT"
+          formatted << line
+          next
+        end
+
+        tab_count = self.get_tab_count(line, tab)
+        insert = line.strip[search.size + 1..]
+        new_insert = self.format_insert(insert, tab_count, tab)
+        if new_insert.nil?
+          formatted << line
+          next
+        end
+        formatted << line.sub(insert, new_insert)
+      end
+      formatted.join("\n")
+    end
+
+    private
+
+    def self.format_insert(s, tab_count = 0, tab = "    ")
+      return s if s.nil?
+      formatted = []
+      if s.include? ") VALUES ("
+        tokens = s.split(") VALUES (")
+        table = tokens[0][..tokens[0].index("(") - 2]
+        columns = tokens[0][tokens[0].index("(") + 1..]
+        values = tokens[1][..-2]
+        formatted << "\n#{tab * (tab_count + 1)}#{table}"
+        formatted << "#{tab * (tab_count + 2)}(#{columns})"
+        formatted << "#{tab * (tab_count + 1)}VALUES"
+        if s.end_with? ");"
+          formatted << "#{tab * (tab_count + 2)}(#{values}"
+        else
+          formatted << "#{tab * (tab_count + 2)}(#{values})"
+        end
+      end
+      formatted.join("\n") unless formatted.empty?
+    end
+  end
+end

data/lib/parsing/formatters/strategy/join_formatter.rb

@@ -0,0 +1,44 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/join_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::JoinFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative "base_formatter"
+
+  class JoinFormatter < BaseFormatter
+    def self.format(text, tab = "    ")
+      text = text.gsub(/INNER\s+JOIN/, "INNER JOIN")
+                 .gsub(/LEFT\s+JOIN/, "LEFT JOIN")
+      lines = text.split("\n")
+      new_text = []
+
+      lines.each do |line|
+        first = line.strip.split(" ").first
+        if line.include? " WHERE " and first != "WHERE" and not first.start_with? "--" and not first.start_with? "/*"
+          tab_count = self.get_tab_count(line, tab)
+          where_parts = line.strip.split(" WHERE ")
+          where_text = []
+          where_text << "#{tab * tab_count}#{where_parts[0]}"
+          where_text << "#{tab * tab_count}WHERE #{where_parts[1]}"
+          new_text << where_text.join("\n")
+        else
+          new_text << line
+        end
+      end
+
+      new_text.join("\n")
+    end
+  end
+end

data/lib/parsing/formatters/strategy/select_formatter.rb

@@ -0,0 +1,52 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/select_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::SelectFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative "base_formatter"
+
+  class SelectFormatter < BaseFormatter
+    def self.format(text, tab = "    ")
+      formatted = []
+      lines = text.split("\n")
+      lines.each do |line|
+        first = line.strip.split(" ").first
+        if first != "SELECT"
+          formatted << line
+          next
+        end
+
+        tab_count = self.get_tab_count(line, tab)
+        select_sql = line.strip[first.size + 1..]
+        new_select = self.format_select(select_sql, tab_count, tab)
+        if new_select.nil?
+          formatted << line
+          next
+        end
+        formatted << line.sub(select_sql, new_select)
+      end
+      formatted.join("\n")
+    end
+
+    private
+
+    def self.format_select(s, tab_count = 0, tab = "    ")
+      return s if s.nil?
+
+      tokens = s.split(", ")
+      "\n#{tokens.map { |t| "#{tab * (tab_count + 1)}#{t}" }.join(",\n")}"
+    end
+  end
+end

data/lib/parsing/formatters/strategy/set_formatter.rb

@@ -0,0 +1,130 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/set_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::SetFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative 'base_formatter'
+
+  class SetFormatter < BaseFormatter
+    def self.format(text, tab = "    ")
+      formatted = []
+      lines = text.split("\n")
+      wait = false
+      set_lines = []
+      special_set_keywords = %w[ANSI_DEFAULTS ANSI_NULL_DFLT_OFF ANSI_NULL_DFLT_ON ANSI_NULLS ANSI_PADDING ANSI_WARNINGS ARITHABORT ARITHIGNORE CONCAT_NULL_YIELDS_NULL CURSOR_CLOSE_ON_COMMIT DATEFIRST DATEFORMAT DEADLOCK_PRIORITY FIPS_FLAGGER FMTONLY FORCEPLAN IDENTITY_INSERT IMPLICIT_TRANSACTIONS LANGUAGE LOCK_TIMEOUT NOCOUNT NOEXEC NUMERIC_ROUNDABORT OFFSETS PARSEONLY QUERY_GOVERNOR_COST_LIMIT QUOTED_IDENTIFIER REMOTE_PROC_TRANSACTIONS ROWCOUNT SHOWPLAN_ALL SHOWPLAN_TEXT SHOWPLAN_XML STATISTICS TEXTSIZE TRANSACTION XACT_ABORT]
+
+      lines.each do |line|
+        tokens = line.strip.split(" ")
+        first = tokens.first
+        next_token = tokens[1] if tokens.size > 1
+
+        if %w[FROM WHERE].include? first and wait
+          wait = false
+          tab_count = self.get_tab_count(line, tab)
+          set_text = set_lines.join("\n")
+          first = set_text.strip.split(" ").first
+          set = set_text.strip[first.size + 1..]
+          new_set = self.format_set(set, tab_count, tab)
+          if new_set.nil?
+            formatted << line
+            next
+          end
+          formatted << "#{tab * tab_count}SET #{new_set}"
+          formatted << line
+          set_lines = []
+          next
+        end
+
+        if first == "SET" and not line.strip.start_with? "SET @"
+          if not next_token.nil? and not special_set_keywords.include? next_token
+            wait = true
+            set_lines << line
+          else
+            formatted << line
+            formatted << ""
+          end
+        elsif first != "SET" and line.include? " SET "
+          parts = line.strip.split(" SET ")
+          tab_count = self.get_tab_count(line, tab)
+          formatted << "#{tab * tab_count}#{parts[0]}\n"
+          parts[1..].each { |p| formatted << "#{tab * tab_count}SET #{p}" }
+        elsif wait
+          set_lines << line
+        else
+          formatted << line
+        end
+      end
+      formatted.join("\n")
+    end
+
+    private
+
+    def self.format_set(s, tab_count = 0, tab = "    ")
+      return s if s.nil?
+      parts = []
+      builder = ""
+      parenthesis = 0
+      tokens = s.split("")
+      comment = false
+      skip_count = 0
+      tokens.each_with_index do |c, i|
+        if skip_count > 0
+          skip_count -= 1
+          next
+        end
+
+        if comment
+          if c == "\n"
+            comment = false
+            parts << builder unless builder.empty?
+            builder = ""
+          else
+            builder << c
+          end
+          next
+        end
+
+        parenthesis += 1 if c == "("
+        parenthesis -= 1 if c == ")"
+        next_c = tokens[i + 1] if tokens.size > i + 1
+        if c == ","
+          if parenthesis > 0
+            builder << c
+          else
+            parts << builder unless builder.empty?
+            builder = ""
+          end
+        elsif c == "\n"
+          parts << builder unless builder.empty?
+          builder = ""
+        elsif c == "-"
+          if next_c == "-"
+            comment = true
+            skip_count = 1
+            parts << builder unless builder.empty?
+            builder = "--"
+          else
+            builder << c
+          end
+        else
+          builder << c
+        end
+      end
+      parts << builder unless builder.empty?
+      parts = parts.map { |p| p.strip }.select { |p| not p.empty? }
+      "\n#{parts.map { |p| "#{tab * (tab_count + 1)}#{p.strip}" }.join(",\n")}"
+    end
+  end
+end

data/lib/parsing/formatters/strategy/update_formatter.rb

@@ -0,0 +1,50 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/update_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::UpdateFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative "base_formatter"
+
+  class UpdateFormatter < BaseFormatter
+    def self.format(text, tab = "    ")
+      formatted = []
+      lines = text.split("\n")
+      lines.each do |line|
+        first = line.strip.split(" ").first
+        if first != "UPDATE"
+          formatted << line
+          next
+        end
+
+        tab_count = self.get_tab_count(line, tab)
+        update = line.strip[first.size + 1..]
+        new_update = self.format_update(update, tab_count, tab)
+        if new_update.nil?
+          formatted << line
+          next
+        end
+        formatted << line.sub(update, new_update)
+      end
+      formatted.join("\n")
+    end
+
+    private
+
+    def self.format_update(s, tab_count = 0, tab = "    ")
+      return s if s.nil?
+      "\n#{tab * (tab_count + 1)}#{s}"
+    end
+  end
+end

data/lib/parsing/formatters/strategy/where_formatter.rb

@@ -0,0 +1,72 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatters/where_formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatters::WhereFormatter
+
+module TSqlParser::Parsing::Formatters
+  require_relative "base_formatter"
+
+  class WhereFormatter < BaseFormatter
+    def self.format(text, tab = "   ")
+      formatted = []
+      text.split("\n").each do |line|
+        first = line.strip.split(" ").first
+        if first != "WHERE"
+          formatted << line
+          next
+        end
+
+        tab_count = self.get_tab_count(line, tab)
+        predicate = line.strip[first.size + 1..]
+        new_predicate = self.format_predicate(predicate, tab_count, tab)
+        if new_predicate.nil?
+          formatted << line
+          next
+        end
+        formatted << line.sub(predicate, new_predicate)
+      end
+
+      formatted.join("\n")
+    end
+
+    private
+
+    def self.format_predicate(s, tab_count = 0, tab = "    ")
+      return s if s.nil?
+      indented = []
+      formatted = []
+      builder = []
+
+      tokens = s.split(" ")
+      tokens.each do |t|
+        if %w[AND OR].include? t
+          formatted << builder.join(" ") unless builder.empty?
+          builder = [t]
+        else
+          builder << t
+        end
+      end
+      formatted << builder.join(" ")
+
+      level = tab_count
+      formatted.each_with_index do |f, i|
+        indented << "#{tab * (level + 1)}#{f}"
+        level -= f.count(")")
+        level += f.count("(")
+      end
+
+      "\n#{indented.join("\n")}"
+    end
+  end
+end

data/lib/parsing/formatters/text_formatter.rb

@@ -0,0 +1,50 @@
+#   __               .__
+# _/  |_  ___________|  |           ___________ _______  ______ ___________
+# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
+#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
+#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
+#            \/   |__|              |__|       \/           \/     \/
+#
+# A very light-weight and opinionated T-SQL parser and formatter.
+#
+# github.com/scstauf
+#
+# path:
+#   parsing/formatter.rb
+# object:
+#   TSqlParser::Parsing::Formatter
+
+module TSqlParser::Parsing
+  require_relative 'strategy/set_formatter'
+  require_relative 'strategy/join_formatter'
+  require_relative 'strategy/insert_formatter'
+  require_relative 'strategy/select_formatter'
+  require_relative 'strategy/update_formatter'
+  require_relative 'strategy/where_formatter'
+
+  class TextFormatter
+    def self.format_sets(text, tab = "    ")
+      Formatters::SetFormatter.format(text, tab)
+    end
+
+    def self.format_joins(text, tab = "    ")
+      Formatters::JoinFormatter.format(text, tab)
+    end
+
+    def self.format_updates(text, tab = "    ")
+      Formatters::UpdateFormatter.format(text, tab)
+    end
+
+    def self.format_inserts(text, tab = "    ")
+      Formatters::InsertFormatter.format(text, tab)
+    end
+
+    def self.format_selects(text, tab = "    ")
+      Formatters::SelectFormatter.format(text, tab)
+    end
+
+    def self.format_wheres(text, tab = "   ")
+      Formatters::WhereFormatter.format(text, tab)
+    end
+  end
+end

data/lib/parsing/tokenizer.rb

@@ -94,7 +94,7 @@ module TSqlParser::Parsing
           next
         end
 
-        if Parser.is_comment_start?(c, next_c)
+        if Parser.is_comment_start?(c, next_c) and not comment
           comment = true
           skip_count = 1
           specific_tokens << builder unless builder.empty?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tsql_parser
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - Scott Stauffer
@@ -17,12 +17,19 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/parsing/formatter.rb
+- lib/parsing/formatters/strategy/base_formatter.rb
+- lib/parsing/formatters/strategy/insert_formatter.rb
+- lib/parsing/formatters/strategy/join_formatter.rb
+- lib/parsing/formatters/strategy/select_formatter.rb
+- lib/parsing/formatters/strategy/set_formatter.rb
+- lib/parsing/formatters/strategy/update_formatter.rb
+- lib/parsing/formatters/strategy/where_formatter.rb
+- lib/parsing/formatters/text_formatter.rb
 - lib/parsing/iterator.rb
 - lib/parsing/keyword.rb
 - lib/parsing/model/flat_sql_container.rb
 - lib/parsing/model/sql_container.rb
 - lib/parsing/parser.rb
-- lib/parsing/text_formatter.rb
 - lib/parsing/tokenizer.rb
 - lib/tsql_parser.rb
 homepage: https://rubygems.org/gems/tsql_parser

data/lib/parsing/text_formatter.rb

@@ -1,250 +0,0 @@
-#   __               .__
-# _/  |_  ___________|  |           ___________ _______  ______ ___________
-# \   __\/  ___/ ____/  |    ______ \____ \__  \\_  __ \/  ___// __ \_  __ \
-#  |  |  \___ < <_|  |  |__ /_____/ |  |_> > __ \|  | \/\___ \\  ___/|  | \/
-#  |__| /____  >__   |____/         |   __(____  /__|  /____  >\___  >__|
-#            \/   |__|              |__|       \/           \/     \/
-#
-# A very light-weight and opinionated T-SQL parser and formatter.
-#
-# github.com/scstauf
-#
-# path:
-#   parsing/formatter.rb
-# object:
-#   TSqlParser::Parsing::Formatter
-
-module TSqlParser::Parsing
-  class TextFormatter
-    def self.format_sets(text, tab = "    ")
-      formatted = []
-      lines = text.split("\n")
-      lines.each do |line|
-        first = line.strip.split(" ").first
-        if first == "SET" and not line.strip.start_with? "SET @"
-          tab_count = self.get_tab_count(line, tab)
-          set = line.strip[first.size + 1..]
-          new_set = self.format_set(set, tab_count, tab)
-          if new_set.nil?
-            formatted << line
-            next
-          end
-          formatted << line.sub(set, new_set)
-        elsif first != "SET" and line.include? " SET "
-          parts = line.strip.split(" SET ")
-          tab_count = self.get_tab_count(line, tab)
-          formatted << "#{tab * tab_count}#{parts[0]}\n"
-          parts[1..].each { |p| formatted << "#{tab * tab_count}SET #{p}" }
-        else
-          formatted << line
-        end
-      end
-      formatted.join("\n")
-    end
-
-    def self.format_joins(text, tab = "    ")
-      text = text.gsub(/INNER\s+JOIN/, "INNER JOIN")
-                 .gsub(/LEFT\s+JOIN/, "LEFT JOIN")
-      lines = text.split("\n")
-      new_text = []
-
-      lines.each do |line|
-        first = line.strip.split(" ").first
-        if line.include? " WHERE " and first != "WHERE" and not first.start_with? "--" and not first.start_with? "/*"
-          tab_count = self.get_tab_count(line, tab)
-          where_parts = line.strip.split(" WHERE ")
-          where_text = []
-          where_text << "#{tab * tab_count}#{where_parts[0]}"
-          where_text << "#{tab * tab_count}WHERE #{where_parts[1]}"
-          new_text << where_text.join("\n")
-        else
-          new_text << line
-        end
-      end
-
-      new_text.join("\n")
-    end
-
-    def self.format_updates(text, tab = "    ")
-      formatted = []
-      lines = text.split("\n")
-      lines.each do |line|
-        first = line.strip.split(" ").first
-        if first != "UPDATE"
-          formatted << line
-          next
-        end
-
-        tab_count = self.get_tab_count(line, tab)
-        update = line.strip[first.size + 1..]
-        new_update = self.format_update(update, tab_count, tab)
-        if new_update.nil?
-          formatted << line
-          next
-        end
-        formatted << line.sub(update, new_update)
-      end
-      formatted.join("\n")
-    end
-
-    def self.format_inserts(text, tab = "    ")
-      formatted = []
-      lines = text.split("\n")
-      search = "INSERT INTO"
-      lines.each do |line|
-        first = line.strip.split(" ").first
-        if first != "INSERT"
-          formatted << line
-          next
-        end
-
-        tab_count = self.get_tab_count(line, tab)
-        insert = line.strip[search.size + 1..]
-        new_insert = self.format_insert(insert, tab_count, tab)
-        if new_insert.nil?
-          formatted << line
-          next
-        end
-        formatted << line.sub(insert, new_insert)
-      end
-      formatted.join("\n")
-    end
-
-    def self.format_selects(text, tab = "    ")
-      formatted = []
-      lines = text.split("\n")
-      lines.each do |line|
-        first = line.strip.split(" ").first
-        if first != "SELECT"
-          formatted << line
-          next
-        end
-
-        tab_count = self.get_tab_count(line, tab)
-        select_sql = line.strip[first.size + 1..]
-        new_select = self.format_select(select_sql, tab_count, tab)
-        if new_select.nil?
-          formatted << line
-          next
-        end
-        formatted << line.sub(select_sql, new_select)
-      end
-      formatted.join("\n")
-    end
-
-    def self.format_wheres(text, tab = "   ")
-      formatted = []
-      text.split("\n").each do |line|
-        first = line.strip.split(" ").first
-        if first != "WHERE"
-          formatted << line
-          next
-        end
-
-        tab_count = self.get_tab_count(line, tab)
-        predicate = line.strip[first.size + 1..]
-        new_predicate = self.format_predicate(predicate, tab_count, tab)
-        if new_predicate.nil?
-          formatted << line
-          next
-        end
-        formatted << line.sub(predicate, new_predicate)
-      end
-
-      formatted.join("\n")
-    end
-
-    private
-
-    def self.format_set(s, tab_count = 0, tab = "    ")
-      return s if s.nil?
-      parts = []
-      builder = ""
-      parenthesis = 0
-      s.split("").each do |c|
-        parenthesis += 1 if c == "("
-        parenthesis -= 1 if c == ")"
-        if c == ","
-          if parenthesis > 0
-            builder << c
-          else
-            parts << builder
-            builder = ""
-          end
-        else
-          builder << c
-        end
-      end
-      parts << builder unless builder.empty?
-      "\n#{parts.map { |p| "#{tab * (tab_count + 1)}#{p.strip}" }.join(",\n")}"
-    end
-
-    def self.format_update(s, tab_count = 0, tab = "    ")
-      return s if s.nil?
-      "\n#{tab * (tab_count + 1)}#{s}"
-    end
-
-    def self.format_insert(s, tab_count = 0, tab = "    ")
-      return s if s.nil?
-      formatted = []
-      if s.include? ") VALUES ("
-        tokens = s.split(") VALUES (")
-        table = tokens[0][..tokens[0].index("(") - 2]
-        columns = tokens[0][tokens[0].index("(") + 1..]
-        values = tokens[1][..-2]
-        formatted << "\n#{tab * (tab_count + 1)}#{table}"
-        formatted << "#{tab * (tab_count + 2)}(#{columns})"
-        formatted << "#{tab * (tab_count + 1)}VALUES"
-        if s.end_with? ");"
-          formatted << "#{tab * (tab_count + 2)}(#{values}"
-        else
-          formatted << "#{tab * (tab_count + 2)}(#{values})"
-        end
-      end
-      formatted.join("\n") unless formatted.empty?
-    end
-
-    def self.format_select(s, tab_count = 0, tab = "    ")
-      return s if s.nil?
-
-      tokens = s.split(", ")
-      "\n#{tokens.map { |t| "#{tab * (tab_count + 1)}#{t}" }.join(",\n")}"
-    end
-
-    def self.format_predicate(s, tab_count = 0, tab = "    ")
-      return s if s.nil?
-      indented = []
-      formatted = []
-      builder = []
-
-      tokens = s.split(" ")
-      tokens.each do |t|
-        if %w[AND OR].include? t
-          formatted << builder.join(" ") unless builder.empty?
-          builder = [t]
-        else
-          builder << t
-        end
-      end
-      formatted << builder.join(" ")
-
-      level = tab_count
-      formatted.each_with_index do |f, i|
-        indented << "#{tab * (level + 1)}#{f}"
-        level -= f.count(")")
-        level += f.count("(")
-      end
-
-      "\n#{indented.join("\n")}"
-    end
-
-    def self.get_tab_count(line, tab = "    ")
-      tab_count = 0
-      while line.start_with? tab
-        tab_count += 1
-        line = line.sub(tab, "")
-      end
-      tab_count
-    end
-  end
-end