ronin-sql 0.1.1 → 0.2.0

data/lib/ronin/code/sql/{builder.rb → drop.rb}

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -22,37 +22,33 @@
 #
 
 require 'ronin/code/sql/statement'
-require 'ronin/code/sql/dialect'
-require 'ronin/code/sql/common_dialect'
-require 'ronin/code/sql/style'
 
 module Ronin
   module Code
     module SQL
-      class Builder < Statement
+      class Drop < Statement
 
-        def initialize(style,options={},&block)
-          @commands = []
+        def initialize(dialect,type,name=nil,options={},&block)
+          @type = type
+          @name = name
+          @if_exists = options[:if_exists]
 
-          super(style,&block)
+          super(dialect,&block)
         end
 
-        def compile
-          @style.compile_statements(@commands)
-        end
-
-        protected
-
-        def command(cmd)
-          @commands << cmd
+        def if_exists
+          @if_exists = true
           return self
         end
 
-        def method_missing(sym,*args,&block)
-          result = super(sym,*args,&block)
+        def emit
+          tokens = emit_token('DROP')
+          tokens += emit_token(@type)
+
+          tokens += emit_token('IF EXISTS') if @if_exists
+          tokens += emit_value(@name)
 
-          @commands << result if result.kind_of?(Statement)
-          return result
+          return tokens
         end
 
       end

data/lib/ronin/code/sql/drop_index.rb

@@ -0,0 +1,43 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/code/sql/drop'
+
+module Ronin
+  module Code
+    module SQL
+      class DropIndex < Drop
+
+        def initialize(dialect,index=nil,options={},&block)
+          super(dialect,'INDEX',index,options,&block)
+        end
+
+        def index(name)
+          @name = name
+          return self
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/drop_table.rb

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -21,30 +21,22 @@
 #++
 #
 
-require 'ronin/code/sql/statement'
+require 'ronin/code/sql/drop'
 
 module Ronin
   module Code
     module SQL
-      class DropTable < Statement
+      class DropTable < Drop
 
-        option :if_exists, "IF EXISTS"
-
-        def initialize(style,table=nil,&block)
-          @table = table
-          @exists = false
-
-          super(style,&block)
+        def initialize(dialect,table=nil,options={},&block)
+          super(dialect,'TABLE',table,options,&block)
         end
 
-        def compile
-          compile_expr(keyword_drop,if_exists?,@table)
+        def table(name)
+          @name = name
+          return self
         end
 
-        protected
-
-        keyword :drop, 'DROP TABLE'
-
       end
     end
   end

data/lib/ronin/code/sql/drop_view.rb

@@ -0,0 +1,43 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/code/sql/drop'
+
+module Ronin
+  module Code
+    module SQL
+      class DropView < Drop
+
+        def initialize(dialect,view=nil,options={},&block)
+          super(dialect,'VIEW',view,options,&block)
+        end
+
+        def view(name)
+          @name = name
+          return self
+        end
+
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/emittable.rb

@@ -0,0 +1,102 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+require 'ronin/code/emittable'
+
+module Ronin
+  module Code
+    module SQL
+      module Emittable
+        include Code::Emittable
+
+        protected
+
+        def emit_token(value)
+          value.to_s.split(/\s/).map { |word| Token.new(word) }
+        end
+
+        def emit_value(value)
+          if value.kind_of?(Statement)
+            tokens = []
+            
+            tokens << Token.open_paren
+            tokens += value.emit
+            tokens << Token.close_paren
+
+            return tokens
+          else
+            return super(value)
+          end
+        end
+
+        def emit_values(values)
+          tokens = []
+
+          values.each { |value| tokens += emit_value(value) }
+
+          return tokens
+        end
+
+        #
+        # Emits the comma separated list of the specified _values_.
+        #
+        def emit_list(values)
+          tokens = []
+
+          (values.length - 1).times do |index|
+            tokens << emit_value(values[index])
+            tokens << Token.new(',')
+          end
+
+          tokens << emit_value(values.last)
+          return tokens
+        end
+
+        #
+        # Emits the specified SQL _row_.
+        #
+        def emit_row(row)
+          case row.length
+          when 0
+            return []
+          when 1
+            return emit_list(row)
+          else
+            return [Token.new('(')] + emit_list(row) + [Token.new(')')]
+          end
+        end
+
+        #
+        # Emits the specified _statement_.
+        #
+        def emit_statement(statement)
+          if statement.kind_of?(Statement)
+            return statement.emit
+          else
+            return statement
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/exceptions/unknown_clause.rb

@@ -0,0 +1,31 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+module Ronin
+  module Code
+    module SQL
+      class UnknownClause < RuntimeError
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/exceptions/unknown_dialect.rb

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -24,7 +24,7 @@
 module Ronin
   module Code
     module SQL
-      class DialectNotFound < RuntimeError
+      class UnknownDialect < RuntimeError
       end
     end
   end

data/lib/ronin/code/sql/exceptions/unknown_statement.rb

@@ -0,0 +1,31 @@
+#
+#--
+# Ronin SQL - A Ronin library providing support for SQL related security
+# tasks.
+#
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#++
+#
+
+module Ronin
+  module Code
+    module SQL
+      class UnknownStatement < RuntimeError
+      end
+    end
+  end
+end

data/lib/ronin/code/sql/exceptions.rb

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -22,3 +22,5 @@
 #
 
 require 'ronin/code/sql/exceptions/unknown_dialect'
+require 'ronin/code/sql/exceptions/unknown_statement'
+require 'ronin/code/sql/exceptions/unknown_clause'

data/lib/ronin/code/sql/expr.rb

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -21,7 +21,7 @@
 #++
 #
 
-require 'ronin/code/sql/keyword'
+require 'ronin/code/sql/emittable'
 require 'ronin/extensions/meta'
 
 module Ronin
@@ -29,15 +29,10 @@ module Ronin
     module SQL
       class Expr
 
-        # The style to use
-        attr_reader :style
-
-        def initialize(style)
-          @style = style
-        end
+        include Emittable
 
         def in?(*range)
-          In.new(@style,self,*range)
+          In.new(self,*range)
         end
 
         def ===(*range)
@@ -48,38 +43,12 @@ module Ronin
           in?(*range).not!
         end
 
-        def compile
-          # place holder
-        end
-
-        def to_s
-          compile
-        end
-
         protected
 
-        def keyword(value)
-          keyword_cache[value.to_sym]
-        end
-
-        def keywords(*values)
-          values.map { |value| keyword(value) }
-        end
-
-        def self.keyword(name,value=name.to_s.upcase)
-          name = name.to_s.downcase
-
-          class_def("keyword_#{name}") do
-            keyword(value)
-          end
-
-          return self
-        end
-
         def self.binary_op(op,*names)
           names.each do |name|
             class_def(name) do |expr|
-              BinaryExpr.new(@style,op,self,expr)
+              BinaryExpr.new(op,self,expr)
             end
           end
 
@@ -104,7 +73,7 @@ module Ronin
         def self.like_op(op,*names)
           names.each do |name|
             class_def(name) do |expr,escape|
-              LikeExpr.new(@style,op,self,expr,escape)
+              Like.new(op,self,expr,escape)
             end
           end
 
@@ -119,7 +88,7 @@ module Ronin
         def self.unary_op(op,*names)
           names.each do |name|
             class_def(name) do
-              UnaryExpr.new(@style,op,self)
+              UnaryExpr.new(op,self)
             end
           end
 
@@ -129,64 +98,6 @@ module Ronin
         unary_op 'NOT', :not!
         unary_op 'EXISTS', :exists?
 
-        def compile_space
-          @style.compile_space
-        end
-
-        def preappend_space(str)
-          @style.preappend_space(str)
-        end
-
-        def append_space(str)
-          @style.append_space(str)
-        end
-
-        def space(*str)
-          @style.space(*str)
-        end
-
-        def compile_newline
-          @style.compile_newline
-        end
-
-        def quote_string(data)
-          @style.quote_string(data)
-        end
-
-        def compile_keyword(name)
-          @style.compile_keyword(name)
-        end
-
-        def compile_list(*expr)
-          @style.compile_list(*expr)
-        end
-
-        def compile_datalist(*expr)
-          @style.compile_list(*expr)
-        end
-
-        def compile_row(*expr)
-          @style.compile_row(*expr)
-        end
-
-        def compile_data(data)
-          @style.compile_data(data)
-        end
-
-        def compile_expr(*expr)
-          @style.compile_expr(*expr)
-        end
-
-        def compile_statements(*statements)
-          @style.compile_statements(*statements)
-        end
-
-        private
-
-        def keyword_cache
-          @keyword_cache ||= Hash.new { |hash,key| hash[key] = Keyword.new(@style,key) }
-        end
-
       end
     end
   end

data/lib/ronin/code/sql/field.rb

@@ -3,7 +3,7 @@
 # Ronin SQL - A Ronin library providing support for SQL related security
 # tasks.
 #
-# Copyright (c) 2007-2008 Hal Brodigan (postmodern.mod3 at gmail.com)
+# Copyright (c) 2007-2009 Hal Brodigan (postmodern.mod3 at gmail.com)
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -22,26 +22,41 @@
 #
 
 require 'ronin/code/sql/expr'
+require 'ronin/code/sql/as'
 require 'ronin/code/sql/between'
+require 'ronin/code/sql/asc'
+require 'ronin/code/sql/desc'
 
 module Ronin
   module Code
     module SQL
       class Field < Expr
 
-        def initialize(style,name,prefix=nil)
-          super(style)
-
+        def initialize(symbols,name,prefix=nil)
+          @symbols = symbols
           @prefix = prefix
           @name = name
         end
 
-        def *
-          field_cache[:"*"]
+        def field(name)
+          sym = @symbols.symbol("#{path}.#{name}")
+          sym.value ||= Field.new(@symbols,name,self)
+
+          return sym
         end
 
+        def all
+          field('*')
+        end
+
+        alias * all
+
         def id
-          field_cache[:id]
+          field('id')
+        end
+        
+        def as(name)
+          As.new(self,name)
         end
 
         def between(start,stop)
@@ -52,32 +67,34 @@ module Ronin
           between(range.begin,range.end)
         end
 
-        def compile
-          if @prefix
-            return "#{@prefix}.#{@name}"
-          else
-            return @name.to_s
-          end
+        def asc
+          Asc.new(self)
         end
 
-        def to_sym
-          compile.to_sym
+        def desc
+          Desc.new(self)
+        end
+
+        def emit
+          [path.to_sym]
         end
 
         protected
 
-        def method_missing(sym,*args)
-          if (args.length==0 && @prefix.nil?)
-            return field_cache[sym]
+        def path
+          if @prefix
+            return "#{@prefix}.#{@name}"
+          else
+            return "#{@name}"
           end
-
-          raise(NoMethodError,sym.id2name)
         end
 
-        private
+        def method_missing(name,*arguments,&block)
+          if (arguments.empty? && @prefix.nil? && block.nil?)
+            return field(name)
+          end
 
-        def field_cache
-          @field_cache ||= Hash.new { |hash,key| hash[key] = Field.new(@style,key,self) }
+          raise(NoMethodError,sym.id2name)
         end
 
       end