sqlstmt 0.1.0

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2007 Makani Mason, Kem Mason
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/lib/sqlstmt/error.rb

@@ -0,0 +1,5 @@
+module SqlStmt
+
+class Error < StandardError; end
+
+end

data/lib/sqlstmt/insert_select.rb

@@ -0,0 +1,29 @@
+require 'sqlstmt/query'
+
+module SqlStmt
+
+class InsertSelect < FieldValueQuery
+  def initialize
+    super
+    @into_table = nil
+  end
+
+  def into(table)
+    @into_table = table
+    self
+  end
+
+private
+  def verify_minimum_requirements
+    super
+    raise SqlStmt::Error, "unable to build sql - must first call :into" if @into_table.nil?
+  end
+
+  def build_stmt
+    into_str = @fields.join(',')
+    select_str = @values.join(',')
+    "INSERT INTO #@into_table (#{into_str}) SELECT #{select_str} FROM #{build_table_list}#{build_where_clause}"
+  end
+end
+
+end

data/lib/sqlstmt/query.rb

@@ -0,0 +1,89 @@
+require 'doh/core_ext/class/force_deep_copy'
+require 'sqlstmt/error'
+require 'sqlstmt/to_sql'
+
+module SqlStmt
+
+class Query
+  force_deep_copy :fields, :tables, :wheres
+
+  def initialize
+    @fields = []
+    @tables = []
+    @wheres = []
+    @use_wheres = true
+  end
+
+  def table(table)
+    @tables.push(table)
+    self
+  end
+
+  def where(*sql)
+    @wheres.concat(sql)
+    self
+  end
+
+  def no_where
+    @use_wheres = false
+    self
+  end
+
+  def to_s
+    verify_minimum_requirements
+    build_stmt
+  end
+
+  def group_by(clause)
+    @group_by = clause
+    self
+  end
+
+  def order_by(clause)
+    @order_by = clause
+    self
+  end
+
+  def limit(clause)
+    @limit = clause
+    self
+  end
+
+private
+  def verify_minimum_requirements
+    raise SqlStmt::Error, "unable to build sql - must first call :field" if @fields.empty?
+    raise SqlStmt::Error, "unable to build sql - must first call :table" if @tables.empty?
+    raise SqlStmt::Error, "unable to build sql - must first call :where or :no_where" if @use_wheres && @wheres.empty?
+    raise SqlStmt::Error, "unable to build sql - :where and :no_where must not be called on same builder instance" if !@use_wheres && !@wheres.empty?
+  end
+
+  def build_table_list
+    @tables.join(',')
+  end
+
+  def build_where_clause
+    if @wheres.empty? then '' else " WHERE #{@wheres.join(' AND ')}" end
+  end
+end
+
+class FieldValueQuery < Query
+  force_deep_copy :values
+
+  def initialize
+    super
+    @values = []
+  end
+
+  def field(field, value)
+    raise "trying to include field #{field} again" if @fields.include?(field)
+    @fields.push(field)
+    @values.push(if value.is_a?(String) then value else value.to_sql end)
+    self
+  end
+
+  def fieldq(field, value)
+    field(field, value.to_sql)
+  end
+end
+
+end

data/lib/sqlstmt/select.rb

@@ -0,0 +1,71 @@
+require 'sqlstmt/query'
+
+module SqlStmt
+
+class Select < Query
+  force_deep_copy :joins
+
+  def initialize
+    super
+    @joins = []
+    @group_by = nil
+    @order_by = nil
+    @limit = nil
+  end
+
+  def field(*field_exprs)
+    @fields.concat(field_exprs)
+    self
+  end
+
+  def join(table, expr)
+    @joins.push("JOIN #{table} ON #{expr}")
+    self
+  end
+
+  def join_using(table, *fields)
+    @joins.push("JOIN #{table} USING (#{fields.join(',')})")
+    self
+  end
+
+  def left_join(table, expr)
+    @joins.push("LEFT JOIN #{table} ON #{expr}")
+    self
+  end
+
+  def left_join_using(table, *fields)
+    @joins.push("LEFT JOIN #{table} USING (#{fields.join(',')})")
+    self
+  end
+
+  def group_by(clause)
+    @group_by = clause
+    self
+  end
+
+  def order_by(clause)
+    @order_by = clause
+    self
+  end
+
+  def limit(clause)
+    @limit = clause
+    self
+  end
+
+private
+  def simple_clause(keywords, value)
+    if value then " #{keywords} #{value}" else '' end
+  end
+
+  def build_stmt
+    select_str = @fields.join(',')
+    join_clause = if @joins.empty? then '' else " #{@joins.join(' ')}" end
+    group_clause = simple_clause('GROUP BY', @group_by)
+    order_clause = simple_clause('ORDER BY', @order_by)
+    limit_clause = simple_clause('LIMIT', @limit)
+    "SELECT #{select_str} FROM #{build_table_list}#{join_clause}#{build_where_clause}#{group_clause}#{order_clause}#{limit_clause}"
+  end
+end
+
+end

data/lib/sqlstmt/to_sql.rb

@@ -0,0 +1,65 @@
+require 'bigdecimal'
+require 'date'
+
+class String
+  unless method_defined?(:to_sql)
+    def to_sql
+       str = gsub('\\', '\\\\').gsub('\'', '\\\'').gsub("'", "\\'")
+      "'#{str}'"
+    end
+  end
+end
+
+class NilClass
+  def to_sql
+    'NULL'
+  end
+end
+
+class Numeric
+  def to_sql
+    to_s
+  end
+end
+
+class Date
+  def to_sql
+    "'#{strftime('%Y-%m-%d')}'"
+  end
+end
+
+class DateTime
+  def to_sql
+    "'#{strftime('%Y-%m-%d %H:%M:%S')}'"
+  end
+end
+
+class Time
+  def to_sql
+    "'#{strftime('%Y-%m-%d %H:%M:%S')}'"
+  end
+end
+
+class TrueClass
+  def to_sql
+    '1'
+  end
+end
+
+class FalseClass
+  def to_sql
+    '0'
+  end
+end
+
+class BigDecimal
+  def to_sql
+    to_s('F')
+  end
+end
+
+class Array
+  def to_sql
+    '(' + collect { |elem| elem.to_sql }.join(',') + ')'
+  end
+end

data/lib/sqlstmt/update.rb

@@ -0,0 +1,20 @@
+require 'sqlstmt/query'
+
+module SqlStmt
+
+class Update < FieldValueQuery
+private
+  def build_set_clause
+    set_exprs = []
+    @fields.each_with_index do |field, index|
+      set_exprs.push("#{field} = #{@values[index]}")
+    end
+    set_exprs.join(', ')
+  end
+
+  def build_stmt
+    "UPDATE #{build_table_list} SET #{build_set_clause}#{build_where_clause}"
+  end
+end
+
+end

data/test/insert_select.dt.rb

@@ -0,0 +1,55 @@
+require 'sqlstmt/insert_select'
+
+module SqlStmt
+
+class TC_insert_select < DohTest::TestGroup
+  def test_minimum_requirements
+    assert_raises(SqlStmt::Error) { InsertSelect.new.into('target').to_s }
+    assert_raises(SqlStmt::Error) { InsertSelect.new.into('target').no_where.to_s }
+    assert_raises(SqlStmt::Error) { InsertSelect.new.into('target').no_where.field('blah', 'blee').to_s }
+  end
+
+  def test_simple
+    assert_equal('INSERT INTO target (blah) SELECT blee FROM source', InsertSelect.new.into('target').table('source').field('blah', 'blee').no_where.to_s)
+    assert_equal('INSERT INTO target (blah) SELECT blee FROM source WHERE source_id = 1', InsertSelect.new.into('target').table('source').field('blah', 'blee').where('source_id = 1').to_s)
+  end
+
+  def test_dup
+    shared_builder = SqlStmt::InsertSelect.new.into('target')
+    first_builder = shared_builder
+    other_builder = shared_builder.dup
+
+    first_builder.where('status="bad"')
+    first_builder.field('created_at', 'NOW()')
+    first_builder.table('some_tbl s')
+
+    other_builder.no_where
+    other_builder.field('info', 'o.info')
+    other_builder.field('created_at', 'then')
+    other_builder.table('other_tbl o')
+
+    assert_equal('INSERT INTO target (created_at) SELECT NOW() FROM some_tbl s WHERE status="bad"', first_builder.to_s)
+    assert_equal('INSERT INTO target (info,created_at) SELECT o.info,then FROM other_tbl o', other_builder.to_s)
+  end
+
+  def test_complex
+    shared_builder = SqlStmt::InsertSelect.new.into('target')
+    shared_builder.table('shared_tbl')
+    shared_builder.field('created_at', 'NOW()').field('duration', 5).fieldq('is_bad', 'b')
+
+    first_builder = shared_builder
+    other_builder = shared_builder.dup
+
+    first_builder.where("status='bad'")
+
+    other_builder.table('other_tbl o')
+    other_builder.field('info', 'o.info')
+    other_builder.field('data', 'o.data')
+    other_builder.where('s.id=o.shared_id', "status='good'")
+
+    assert_equal("INSERT INTO target (created_at,duration,is_bad) SELECT NOW(),5,'b' FROM shared_tbl WHERE status='bad'", first_builder.to_s)
+    assert_equal("INSERT INTO target (created_at,duration,is_bad,info,data) SELECT NOW(),5,'b',o.info,o.data FROM shared_tbl,other_tbl o WHERE s.id=o.shared_id AND status='good'", other_builder.to_s)
+  end
+end
+
+end

data/test/select.dt.rb

@@ -0,0 +1,23 @@
+require 'sqlstmt/select'
+
+module SqlStmt
+
+class Test_Select < DohTest::TestGroup
+  def notest_minimum_requirements
+    assert_raises(SqlStmt::Error) { Select.new('target').to_s }
+    assert_raises(SqlStmt::Error) { Select.new('target').no_where.to_s }
+  end
+
+  def test_stuff
+    assert_equal('SELECT blah FROM source', Select.new.table('source').field('blah').no_where.to_s)
+    assert_equal('SELECT blah FROM source WHERE source_id = 1', Select.new.table('source').field('blah').where('source_id = 1').to_s)
+    assert_equal('SELECT blah FROM source s', Select.new.table('source s').field('blah').no_where.to_s)
+    assert_equal('SELECT blah FROM source s JOIN other o ON s.blah_id = o.blah_id', Select.new.table('source s').join('other o', 's.blah_id = o.blah_id').field('blah').no_where.to_s)
+    assert_equal('SELECT blah FROM source s JOIN other o USING (blah_id)', Select.new.table('source s').join_using('other o', 'blah_id').field('blah').no_where.to_s)
+    assert_equal('SELECT blah FROM source s LEFT JOIN other o ON s.blah_id = o.blah_id', Select.new.table('source s').left_join('other o', 's.blah_id = o.blah_id').field('blah').no_where.to_s)
+    assert_equal('SELECT blah FROM source s LEFT JOIN other o USING (blah_id)', Select.new.table('source s').left_join_using('other o', 'blah_id').field('blah').no_where.to_s)
+    assert_equal('SELECT blah,blee FROM source', Select.new.table('source').field('blah','blee').no_where.to_s)
+  end
+end
+
+end

data/test/to_sql.dt.rb

@@ -0,0 +1,20 @@
+require 'sqlstmt/to_sql'
+
+module SqlStmt
+
+class Test_to_sql < DohTest::TestGroup
+  def test_stuff
+    assert_equal("'blah'", 'blah'.to_sql)
+    assert_equal('NULL', nil.to_sql)
+    assert_equal('3', 3.to_sql)
+    assert_equal("'2008-09-24'", Date.new(2008,9,24).to_sql)
+    assert_equal("'2008-09-24 09:30:04'", DateTime.new(2008,9,24,9,30,4).to_sql)
+    assert_equal('1', true.to_sql)
+    assert_equal('0', false.to_sql)
+    assert_equal('10.0', BigDecimal.new('10').to_sql)
+    assert_equal("('a','b','c')", ['a', 'b', 'c'].to_sql)
+  end
+end
+
+end
+

data/test/update.dt.rb

@@ -0,0 +1,55 @@
+require 'sqlstmt/update'
+
+module SqlStmt
+
+class TC_update < DohTest::TestGroup
+  def test_minimum_requirements
+    assert_raises(SqlStmt::Error) { Update.new.table('target').to_s }
+    assert_raises(SqlStmt::Error) { Update.new.table('target').no_where.to_s }
+  end
+
+  def test_simple
+    assert_equal('UPDATE target SET blah = blee', Update.new.table('target').field('blah', 'blee').no_where.to_s)
+    assert_equal('UPDATE target SET blah = blee WHERE target_id = 1', Update.new.table('target').field('blah', 'blee').where('target_id = 1').to_s)
+  end
+
+  def test_dup
+    shared_builder = SqlStmt::Update.new.table('target')
+    first_builder = shared_builder
+    other_builder = shared_builder.dup
+
+    first_builder.where('status="bad"')
+    first_builder.field('created_at', 'NOW()')
+    first_builder.table('some_tbl s')
+
+    other_builder.no_where
+    other_builder.field('info', 'o.info')
+    other_builder.field('created_at', 'then')
+    other_builder.table('other_tbl o')
+
+    assert_equal('UPDATE target,some_tbl s SET created_at = NOW() WHERE status="bad"', first_builder.to_s)
+    assert_equal('UPDATE target,other_tbl o SET info = o.info, created_at = then', other_builder.to_s)
+  end
+
+  def test_complex
+    shared_builder = SqlStmt::Update.new.table('target')
+    shared_builder.table('shared_tbl')
+    shared_builder.field('created_at', 'NOW()').field('duration', 5).field('is_bad', 1)
+
+    first_builder = shared_builder
+    other_builder = shared_builder.dup
+
+    first_builder.where('status="bad"')
+
+    other_builder.table('other_tbl o')
+    other_builder.field('info', 'o.info')
+    other_builder.field('data', 'o.data')
+    other_builder.where('s.id=o.shared_id')
+    other_builder.where('status="good"')
+
+    assert_equal('UPDATE target,shared_tbl SET created_at = NOW(), duration = 5, is_bad = 1 WHERE status="bad"', first_builder.to_s)
+    assert_equal('UPDATE target,shared_tbl,other_tbl o SET created_at = NOW(), duration = 5, is_bad = 1, info = o.info, data = o.data WHERE s.id=o.shared_id AND status="good"', other_builder.to_s)
+  end
+end
+
+end

metadata

@@ -0,0 +1,75 @@
+--- !ruby/object:Gem::Specification
+name: sqlstmt
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Makani Mason
+- Kem Mason
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-27 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: dohutil
+  requirement: &70289917880740 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 0.1.5
+  type: :runtime
+  prerelease: false
+  version_requirements: *70289917880740
+description: mysql centric (for now) object helpers for building SQL statements
+email:
+- devinfo@atpsoft.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- MIT-LICENSE
+files:
+- lib/sqlstmt/error.rb
+- lib/sqlstmt/insert_select.rb
+- lib/sqlstmt/query.rb
+- lib/sqlstmt/select.rb
+- lib/sqlstmt/to_sql.rb
+- lib/sqlstmt/update.rb
+- test/insert_select.dt.rb
+- test/select.dt.rb
+- test/to_sql.dt.rb
+- test/update.dt.rb
+- MIT-LICENSE
+homepage: https://github.com/atpsoft/sqlstmt
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: helper for building SQL statements
+test_files:
+- test/insert_select.dt.rb
+- test/select.dt.rb
+- test/to_sql.dt.rb
+- test/update.dt.rb
+has_rdoc: