baza 0.0.20 → 0.0.21

data/lib/baza/sql_queries/select.rb

@@ -0,0 +1,170 @@
+class Baza::SqlQueries::Select
+  def initialize(args)
+    @db = args.fetch(:db)
+    @selects = []
+    @froms = []
+    @joins = []
+    @wheres = []
+    @groups = []
+    @orders = []
+  end
+
+  def select(arg)
+    @selects << arg
+    self
+  end
+
+  def from(arg)
+    @froms << arg
+    self
+  end
+
+  def join(arg)
+    @joins << arg
+    self
+  end
+
+  def where(*args)
+    @wheres << args
+    self
+  end
+
+  def group(arg)
+    @groups << arg
+    self
+  end
+
+  def order(arg)
+    @orders << arg
+    self
+  end
+
+  def limit(limit)
+    @limit = limit
+    self
+  end
+
+  def offset(offset)
+    @offset = offset
+    self
+  end
+
+  def to_sql
+    "#{select_sql} #{from_sql} #{where_sql} #{group_sql} #{limit_sql}"
+  end
+
+  def to_a
+    each.to_a
+  end
+
+  def each(&blk)
+    query(&blk)
+  end
+
+  def each_row
+    query do |data|
+      yield Baza::Row.new(
+        db: @db,
+        table: first_from,
+        data: data
+      )
+    end
+  end
+
+  def to_enum
+    Enumerator.new do |yielder|
+      query do |data|
+        yielder << data
+      end
+    end
+  end
+
+  def query(&blk)
+    @db.query(to_sql, &blk)
+  end
+
+private
+
+  def select_sql
+    sql = "SELECT"
+
+    if @selects.empty?
+      sql << " *"
+    else
+      first = true
+      @selects.each do |select|
+        sql << "," unless first
+        first = false if first
+
+        if select.is_a?(Symbol)
+          select << " #{@db.sep_col}#{@db.escape_column(select)}#{@db.sep_col}"
+        else
+          select << @db.sqlval(select)
+        end
+      end
+    end
+
+    sql
+  end
+
+  def from_sql
+    sql = "FROM"
+
+    first = true
+    @froms.each do |from|
+      sql << "," unless first
+      first = false if first
+      sql << " #{@db.sep_table}#{@db.escape_table(from)}#{@db.sep_table}"
+    end
+
+    sql
+  end
+
+  def first_from
+    @first_from ||= @froms.first
+  end
+
+  def where_sql
+    return if @wheres.empty?
+
+    sql = " WHERE"
+
+    first = true
+    @wheres.each do |args|
+      where = args.shift
+
+      sql << " AND " unless first
+      first = false if first
+
+      if where.is_a?(Hash)
+        where.each do |key, value|
+          sql << "#{@db.sep_col}#{@db.escape_column(key)}#{@db.sep_col} = #{@db.sqlval(value)}"
+        end
+      elsif where.is_a?(String)
+        sql_arg = where.clone
+        args.each do |arg|
+          sql_arg.sub!("?", @db.sqlval(arg))
+        end
+
+        sql << sql_arg
+      else
+        raise "Dont know what to do with that argument: #{where}"
+      end
+    end
+
+    sql
+  end
+
+  def group_sql
+    return if @groups.empty?
+  end
+
+  def limit_sql
+    if @limit
+      sql = "LIMIT #{@db.sqlval(@limit)}"
+      sql << ", #{@db.sqlval(@offset)}" if @offset
+    end
+
+    sql
+  end
+end

data/lib/baza/sql_queries/sqlite_upsert_duplicate_key.rb

@@ -0,0 +1,99 @@
+class Baza::SqlQueries::SqliteUpsertDuplicateKey
+  def initialize(args)
+    @db = args.fetch(:db)
+    @table_name = args.fetch(:table_name)
+    @updates = StringCases.stringify_keys(args.fetch(:updates))
+    @terms = StringCases.stringify_keys(args.fetch(:terms))
+    @return_id = args[:return_id]
+  end
+
+  def execute
+    return insert_or_handle_duplicate if @terms.empty?
+
+    @db.transaction do
+      @db.query(insert_sql)
+      @db.query(update_sql)
+
+      if @return_id
+        data = @db.single(@table_name, @terms)
+        raise "Couldn't find the updated data" unless data
+        return data.fetch(primary_column).to_i
+      end
+    end
+  end
+
+private
+
+  def primary_column
+    @primary_column ||= @db.tables[@table_name.to_s].columns.find(&:primarykey?).name.to_sym
+  end
+
+  def insert_or_handle_duplicate
+    @db.insert(@table_name, @updates)
+    return @db.last_id if @return_id
+  rescue => e
+    if (match = e.message.match(/UNIQUE constraint failed: #{Regexp.escape(@table_name)}\.(.+?)(:|\Z|\))/))
+      column_name = match[1]
+    elsif (match = e.message.match(/column (.+?) is not unique/))
+      column_name = match[1]
+    else
+      raise e
+    end
+
+    conflicting_value = @updates.fetch(column_name)
+    @db.update(@table_name, @updates, column_name => conflicting_value)
+
+    if @return_id
+      data = @db.single(@table_name, column_name => conflicting_value)
+      raise "Couldn't find the updated data" unless data
+      return data.fetch(primary_column).to_i
+    end
+  end
+
+  def insert_sql
+    sql = "INSERT OR IGNORE INTO #{@db.sep_table}#{@db.escape_table(@table_name)}#{@db.sep_table} ("
+
+    combined_data = @updates.merge(@terms)
+
+    first = true
+    combined_data.each_key do |column_name|
+      sql << ", " unless first
+      first = false if first
+      sql << "#{@db.sep_col}#{@db.escape_column(column_name)}#{@db.sep_col}"
+    end
+
+    sql << ") VALUES ("
+
+    first = true
+    combined_data.each_value do |value|
+      sql << ", " unless first
+      first = false if first
+      sql << @db.sqlval(value).to_s
+    end
+
+    sql << ")"
+    sql
+  end
+
+  def update_sql
+    sql = "UPDATE OR IGNORE #{@db.sep_table}#{@db.escape_table(@table_name)}#{@db.sep_table} SET "
+
+    first = true
+    @updates.each do |key, value|
+      sql << ", " unless first
+      first = false if first
+      sql << "#{@db.sep_col}#{@db.escape_column(key)}#{@db.sep_col} = #{@db.sqlval(value)}"
+    end
+
+    sql << " WHERE "
+
+    first = true
+    @terms.each do |key, value|
+      sql << " AND " unless first
+      first = false if first
+      sql << "#{@db.sep_col}#{@db.escape_column(key)}#{@db.sep_col} = #{@db.sqlval(value)}"
+    end
+
+    sql
+  end
+end

data/lib/baza/table.rb

@@ -1,4 +1,5 @@
 class Baza::Table
+  include Comparable
   include Baza::DatabaseModelFunctionality
 
   attr_reader :db
@@ -24,17 +25,43 @@ class Baza::Table
   end
 
   def row(id)
-    data = rows({id: id}, limit: 1).fetch
-    raise Baza::Errors::RowNotFound unless data
-
-    Baza::Row.new(
-      db: db,
-      table: name,
-      data: data
-    )
+    row = rows({id: id}, limit: 1).first
+    raise Baza::Errors::RowNotFound unless row
+    row
   end
 
   def to_param
     name
   end
+
+  def insert(data, args = {})
+    @db.insert(name, data, args)
+  end
+
+  def upsert_duplicate_key(data, terms = {}, args = {})
+    @db.upsert_duplicate_key(name, data, terms, args)
+  end
+
+  def rows_count
+    sql = "SELECT COUNT(*) AS count FROM #{@db.sep_table}#{@db.escape_table(name)}#{@db.sep_table}"
+    @db.query(sql).fetch.fetch(:count).to_i
+  end
+
+  def truncate
+    @db.query("TRUNCATE #{@db.sep_table}#{@db.escape_table(name)}#{@db.sep_table}")
+    self
+  end
+
+  def <=>(other)
+    return false unless other.is_a?(Baza::Table)
+    return false unless other.db.opts.fetch(:db) == db.opts.fetch(:db)
+    other.name <=> name
+  end
+
+  def create_columns(col_arr)
+    col_arr.each do |col_data|
+      sql = "ALTER TABLE #{db.sep_col}#{name}#{db.sep_col} ADD COLUMN #{@db.columns.data_sql(col_data)};"
+      @db.query(sql)
+    end
+  end
 end

data/spec/active_record/models/user.rb

@@ -0,0 +1,3 @@
+class User < ActiveRecord::Base
+  validates_presence_of :email
+end

data/spec/drivers/active_record_mysql2_spec.rb

@@ -1,9 +1,9 @@
-require "spec_helper"
-
 unless RUBY_PLATFORM == "java"
+  require "spec_helper"
+  Baza.load_driver("active_record")
+
   describe Baza::Driver::ActiveRecord do
     let(:constant) do
-      name = described_class.name.split("::").last
       const_name = "InfoActiveRecordMysql2"
       require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
       raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
@@ -14,5 +14,7 @@ unless RUBY_PLATFORM == "java"
     it_should_behave_like "a baza tables driver"
     it_should_behave_like "a baza columns driver"
     it_should_behave_like "a baza indexes driver"
+    it_should_behave_like "a baza users driver"
+    it_should_behave_like "an active record driver"
   end
 end

data/spec/drivers/active_record_mysql_spec.rb

@@ -2,7 +2,6 @@ require "spec_helper"
 
 describe Baza::Driver::ActiveRecord do
   let(:constant) do
-    name = described_class.name.split("::").last
     const_name = "InfoActiveRecordMysql"
     require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
     raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
@@ -13,4 +12,6 @@ describe Baza::Driver::ActiveRecord do
   it_should_behave_like "a baza tables driver"
   it_should_behave_like "a baza columns driver"
   it_should_behave_like "a baza indexes driver"
+  it_should_behave_like "a baza users driver"
+  it_should_behave_like "an active record driver"
 end

data/spec/drivers/active_record_pg_spec.rb

@@ -0,0 +1,20 @@
+require "spec_helper"
+
+unless RUBY_PLATFORM == "java"
+  Baza.load_driver("active_record")
+
+  describe Baza::Driver::ActiveRecord do
+    let(:constant) do
+      const_name = "InfoActiveRecordPg"
+      require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
+      raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
+      Baza.const_get(const_name)
+    end
+
+    it_behaves_like "a baza driver"
+    it_should_behave_like "a baza tables driver"
+    it_should_behave_like "a baza columns driver"
+    it_should_behave_like "a baza indexes driver"
+    it_should_behave_like "an active record driver"
+  end
+end

data/spec/drivers/active_record_sqlite3_spec.rb

@@ -1,8 +1,8 @@
 require "spec_helper"
+require_relative "../../lib/baza/driver/active_record"
 
 describe Baza::Driver::ActiveRecord do
   let(:constant) do
-    name = described_class.name.split("::").last
     const_name = "InfoActiveRecordSqlite3"
     require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
     raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
@@ -13,4 +13,5 @@ describe Baza::Driver::ActiveRecord do
   it_should_behave_like "a baza tables driver"
   it_should_behave_like "a baza columns driver"
   it_should_behave_like "a baza indexes driver"
+  it_should_behave_like "an active record driver"
 end

data/spec/drivers/mysql2_spec.rb

@@ -2,7 +2,6 @@ require "spec_helper"
 
 describe Baza::Driver::Mysql2 do
   let(:constant) do
-    name = described_class.name.split("::").last
     const_name = "InfoMysql2"
     require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
     raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
@@ -14,4 +13,5 @@ describe Baza::Driver::Mysql2 do
   it_should_behave_like "a baza tables driver"
   it_should_behave_like "a baza columns driver"
   it_should_behave_like "a baza indexes driver"
+  it_should_behave_like "a baza users driver"
 end

data/spec/drivers/mysql_spec.rb

@@ -2,7 +2,6 @@ require "spec_helper"
 
 describe Baza.const_get(:Driver).const_get(:Mysql) do
   let(:constant) do
-    name = described_class.name.split("::").last
     const_name = "InfoMysql"
     require "#{File.dirname(__FILE__)}/../#{StringCases.camel_to_snake(const_name)}"
     raise "Constant was not defined: '#{const_name}'." unless Baza.const_defined?(const_name)
@@ -14,6 +13,7 @@ describe Baza.const_get(:Driver).const_get(:Mysql) do
   it_should_behave_like "a baza tables driver"
   it_should_behave_like "a baza columns driver"
   it_should_behave_like "a baza indexes driver"
+  it_should_behave_like "a baza users driver"
 
   it "should dump to sqlite3" do
     require "info_sqlite3"
@@ -29,25 +29,25 @@ describe Baza.const_get(:Driver).const_get(:Mysql) do
 
     begin
       db1.tables.create(:test_table, columns: [
-        {name: :id, type: :int, autoincr: true, primarykey: true},
-        {name: :name, type: :varchar, maxlength: 100}
-      ])
+                          {name: :id, type: :int, autoincr: true, primarykey: true},
+                          {name: :name, type: :varchar, maxlength: 100}
+                        ])
       test_table = db1.tables[:test_table]
 
 
       db1.copy_to(db2)
 
       table_sqlite = db2.tables[:test_table]
-      table_sqlite.columns.length.should eq test_table.columns.length
+      expect(table_sqlite.columns.length).to eq test_table.columns.length
 
       col_id_sqlite = table_sqlite.column(:id)
-      col_id_sqlite.type.should eq :int
-      col_id_sqlite.autoincr?.should eq true
-      col_id_sqlite.primarykey?.should eq true
+      expect(col_id_sqlite.type).to eq :int
+      expect(col_id_sqlite.autoincr?).to eq true
+      expect(col_id_sqlite.primarykey?).to eq true
 
       col_name_sqlite = table_sqlite.column(:name)
-      col_name_sqlite.type.should eq :varchar
-      col_name_sqlite.maxlength.to_i.should eq 100
+      expect(col_name_sqlite.type).to eq :varchar
+      expect(col_name_sqlite.maxlength.to_i).to eq 100
     ensure
       driver1.after
       driver2.after