nobrainer 0.8.0

data/lib/no_brainer/query_runner/run_options.rb

@@ -0,0 +1,34 @@
+class NoBrainer::QueryRunner::RunOptions < NoBrainer::QueryRunner::Middleware
+  # XXX NoBrainer::Database#drop() uses Thread.current[:nobrainer_options]
+  # We should fix that hack.
+
+  def self.with_database(db_name, &block)
+    with_options(:db => db_name, &block)
+  end
+
+  def self.with_options(options={}, &block)
+    old_options = Thread.current[:nobrainer_options]
+    Thread.current[:nobrainer_options] = (old_options || {}).merge(options.symbolize_keys)
+    block.call if block
+  ensure
+    Thread.current[:nobrainer_options] = old_options
+  end
+
+  def call(env)
+    env[:options].symbolize_keys!
+    if Thread.current[:nobrainer_options]
+      env[:options] = env[:options].reverse_merge(Thread.current[:nobrainer_options])
+    end
+
+    if NoBrainer::Config.durability.to_s != 'hard'
+      env[:options] = env[:options].reverse_merge(:durability => NoBrainer::Config.durability)
+    end
+
+    if env[:options][:db] && !env[:options][:db].is_a?(RethinkDB::RQL)
+      env[:db_name] = env[:options][:db].to_s
+      env[:options][:db] = RethinkDB::RQL.new.db(env[:db_name])
+    end
+
+    @runner.call(env)
+  end
+end

data/lib/no_brainer/query_runner/table_on_demand.rb

@@ -0,0 +1,44 @@
+class NoBrainer::QueryRunner::TableOnDemand < NoBrainer::QueryRunner::Middleware
+  def call(env)
+    @runner.call(env)
+  rescue RuntimeError => e
+    if NoBrainer::Config.auto_create_tables &&
+       e.message =~ /^Table `(.+)` does not exist\.$/
+      auto_create_table(env, $1)
+      retry
+    end
+    raise
+  end
+
+  private
+
+  def auto_create_table(env, table_name)
+    if env[:auto_create_table] == table_name
+      raise "Auto table creation is not working with #{table_name}"
+    end
+    env[:auto_create_table] = table_name
+
+    # FIXME This stinks.
+    database_names = find_db_names(env[:query])
+    case database_names.size
+    when 0 then NoBrainer.table_create(table_name)
+    when 1 then NoBrainer.with_database(database_names.first) { NoBrainer.table_create(table_name) }
+    else raise "Ambiguous database name for creation on demand: #{database_names}"
+    end
+  rescue RuntimeError => e
+    # We might have raced with another table create
+    raise unless e.message =~ /Table `#{table_name}` already exists/
+  end
+
+  def find_db_names(terms)
+    terms = terms.body.args if terms.is_a?(RethinkDB::RQL)
+    terms.map do |term|
+      next unless term.is_a?(Term)
+      if term.type == Term::TermType::DB
+        term.args.first.datum.r_str
+      else
+        find_db_names(term.args)
+      end
+    end.flatten.uniq
+  end
+end

data/lib/no_brainer/query_runner/write_error.rb

@@ -0,0 +1,28 @@
+class NoBrainer::QueryRunner::WriteError < NoBrainer::QueryRunner::Middleware
+  def call(env)
+    write_query = NoBrainer::Util.is_write_query?(env[:query])
+    @runner.call(env).tap do |result|
+      # TODO Fix rethinkdb driver: Their classes Term, Query, Response are
+      # not scoped to the RethinkDB module! (that would prevent a user from
+      # creating a Response model for example).
+
+      if write_query && (result['errors'].to_i != 0 || result['skipped'].to_i != 0)
+        raise_write_error(env, result['first_error'])
+      end
+    end
+  rescue RethinkDB::RqlRuntimeError => e
+    raise unless write_query
+
+    error_msg = e.message.split("\nBacktrace").first
+    error_msg = "Non existent document" if e.message =~ /Expected type OBJECT but found NULL/
+    raise_write_error(env, error_msg)
+  end
+
+  private
+
+  def raise_write_error(env, error_msg)
+    error_msg ||= "Unknown error"
+    error_msg += "\nQuery was: #{env[:query].inspect[0..1000]}"
+    raise NoBrainer::Error::DocumentNotSaved, error_msg
+  end
+end

data/lib/no_brainer/railtie.rb

@@ -0,0 +1,36 @@
+require "nobrainer"
+require "rails"
+
+class NoBrainer::Railtie < Rails::Railtie
+  config.action_dispatch.rescue_responses.merge!(
+    "NoBrainer::Errors::DocumentNotFound" => :not_found,
+    "NoBrainer::Errors::DocumentInvalid"  => :unprocessable_entity,
+    "NoBrainer::Errors::DocumentNotSaved" => :unprocessable_entity,
+  )
+
+  rake_tasks do
+    load "no_brainer/railtie/database.rake"
+  end
+
+  config.after_initialize do
+    NoBrainer::Config.configure unless NoBrainer::Config.configured?
+
+    if defined?(ActiveRecord) && NoBrainer::Config.warn_on_active_record
+      STDERR.puts "[NoBrainer] WARNING: ActiveRecord is loaded which is probably not what you want."
+      STDERR.puts "[NoBrainer] Follow the instructions on http://todo/ to learn how to remove ActiveRecord."
+      STDERR.puts "[NoBrainer] Configure NoBrainer with 'config.warn_on_active_record = false' to disable with warning."
+    end
+
+    if defined?(Mongoid)
+      STDERR.puts "[NoBrainer] WARNING: Mongoid is loaded, and we conflict on the symbol decorations"
+      STDERR.puts "[NoBrainer] They are used in queries like Model.where(:tags.in => ['fun', 'stuff'])"
+      STDERR.puts "[NoBrainer] This is a problem!"
+    end
+
+    ActionDispatch::Reloader.to_cleanup do
+      NoBrainer::Loader.cleanup
+    end
+  end
+
+  #config.eager_load_namespaces << NoBrainer
+end

data/lib/no_brainer/railtie/database.rake

@@ -0,0 +1,34 @@
+namespace :db do
+  desc 'Drop the database'
+  task :drop => :environment do
+    NoBrainer.drop!
+  end
+
+  desc 'Load seed data from db/seeds.rb'
+  task :seed => :environment do
+    Rails.application.load_seed
+  end
+
+  desc 'Create and drop indexes on the database'
+  task :update_indexes => :environment do
+    NoBrainer.update_indexes(:verbose => true)
+  end
+
+  task :update_indexes_quiet => :environment do
+    NoBrainer.update_indexes
+  end
+
+  desc 'Equivalent to db:update_indexes + db:seed'
+  task :setup => [ :update_indexes_quiet, :seed ]
+
+  desc 'Equivalent to db:drop + db:setup'
+  task :reset => [ :drop, :setup ]
+
+  task :create => :environment do
+    # noop
+  end
+
+  task :migrate => :environment do
+    # noop
+  end
+end

data/lib/no_brainer/util.rb

@@ -0,0 +1,23 @@
+module NoBrainer::Util
+  def self.is_write_query?(rql_query)
+    rql_type(rql_query) == :write
+  end
+
+  def self.rql_type(rql_query)
+    case rql_query.body.type
+    when Term::TermType::UPDATE,  Term::TermType::DELETE,
+         Term::TermType::REPLACE, Term::TermType::INSERT
+      :write
+    when Term::TermType::DB_CREATE,    Term::TermType::DB_DROP,
+         Term::TermType::DB_LIST,      Term::TermType::TABLE_CREATE,
+         Term::TermType::TABLE_DROP,   Term::TermType::TABLE_LIST,
+         Term::TermType::SYNC,         Term::TermType::INDEX_CREATE,
+         Term::TermType::INDEX_DROP,   Term::TermType::INDEX_LIST,
+         Term::TermType::INDEX_STATUS, Term::TermType::INDEX_WAIT
+      :management
+    else
+      # XXX Not sure if that's correct, but we'll be happy for logging colors.
+      :read
+    end
+  end
+end

data/lib/no_brainer/version.rb

@@ -0,0 +1,3 @@
+module NoBrainer
+  VERSION = '0.8.0'
+end

data/lib/nobrainer.rb

@@ -0,0 +1,59 @@
+require 'active_support/core_ext'
+
+if Gem::Version.new(RUBY_VERSION.dup) < Gem::Version.new('1.9')
+  raise 'Please use Ruby 1.9 or later'
+end
+
+module NoBrainer
+  require 'no_brainer/autoload'
+  extend NoBrainer::Autoload
+
+  autoload :Config, :Document, :Connection, :Database, :Error, :QueryRunner,
+           :Criteria, :DecoratedSymbol, :IndexManager, :Loader, :Logging, :Util, :Fork
+
+  class << self
+    # Note: we always access the connection explicitly, so that in the future,
+    # we can refactor to return a connection depending on the context.
+    # Note that a connection is tied to a database in NoBrainer.
+    def connection
+      @connection ||= begin
+        url = NoBrainer::Config.rethinkdb_url
+        raise "Please specify a database connection to RethinkDB" unless url
+        Connection.new(url).tap { |c| c.connect }
+      end
+    end
+
+    def disconnect
+      @connection.try(:disconnect)
+      @connection = nil
+    end
+
+    def disconnect_if_url_changed
+      if @connection && @connection.uri != NoBrainer::Config.rethinkdb_url
+        disconnect
+      end
+    end
+
+    # No not use modules to extend, it's nice to see the NoBrainer module API here.
+    delegate :db_create, :db_drop, :db_list, :database, :to => :connection
+    delegate :table_create, :table_drop, :table_list,
+             :drop!, :purge!, :to => :database
+    delegate :run, :to => 'NoBrainer::QueryRunner'
+    delegate :update_indexes, :to => 'NoBrainer::IndexManager'
+    delegate :with_options, :with_database, :to => 'NoBrainer::QueryRunner::RunOptions'
+    delegate :configure, :logger, :to => 'NoBrainer::Config'
+
+    def jruby?
+      RUBY_PLATFORM == 'java'
+    end
+  end
+
+  DecoratedSymbol.hook
+  Fork.hook unless jruby?
+end
+
+ActiveSupport.on_load(:i18n) do
+  I18n.load_path << File.dirname(__FILE__) + '/no_brainer/locale/en.yml'
+end
+
+require 'no_brainer/railtie' if defined?(Rails)

metadata

@@ -0,0 +1,152 @@
+--- !ruby/object:Gem::Specification
+name: nobrainer
+version: !ruby/object:Gem::Version
+  version: 0.8.0
+platform: ruby
+authors:
+- Nicolas Viennot
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-12-31 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rethinkdb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.11.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.11.0.1
+- !ruby/object:Gem::Dependency
+  name: activemodel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: '5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 3.2.0
+    - - <
+      - !ruby/object:Gem::Version
+        version: '5'
+- !ruby/object:Gem::Dependency
+  name: middleware
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+description: ORM for RethinkDB
+email:
+- nicolas@viennot.biz
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/no_brainer/document/polymorphic.rb
+- lib/no_brainer/document/id.rb
+- lib/no_brainer/document/injection_layer.rb
+- lib/no_brainer/document/store_in.rb
+- lib/no_brainer/document/timestamps.rb
+- lib/no_brainer/document/association.rb
+- lib/no_brainer/document/association/core.rb
+- lib/no_brainer/document/association/has_many.rb
+- lib/no_brainer/document/association/belongs_to.rb
+- lib/no_brainer/document/core.rb
+- lib/no_brainer/document/criteria.rb
+- lib/no_brainer/document/dirty.rb
+- lib/no_brainer/document/dynamic_attributes.rb
+- lib/no_brainer/document/persistance.rb
+- lib/no_brainer/document/serialization.rb
+- lib/no_brainer/document/validation.rb
+- lib/no_brainer/document/attributes.rb
+- lib/no_brainer/document/index.rb
+- lib/no_brainer/query_runner/connection.rb
+- lib/no_brainer/query_runner/driver.rb
+- lib/no_brainer/query_runner/database_on_demand.rb
+- lib/no_brainer/query_runner/logger.rb
+- lib/no_brainer/query_runner/run_options.rb
+- lib/no_brainer/query_runner/table_on_demand.rb
+- lib/no_brainer/query_runner/write_error.rb
+- lib/no_brainer/railtie/database.rake
+- lib/no_brainer/criteria/chainable/raw.rb
+- lib/no_brainer/criteria/chainable/core.rb
+- lib/no_brainer/criteria/chainable/limit.rb
+- lib/no_brainer/criteria/chainable/order_by.rb
+- lib/no_brainer/criteria/chainable/scope.rb
+- lib/no_brainer/criteria/chainable/where.rb
+- lib/no_brainer/criteria/termination/inc.rb
+- lib/no_brainer/criteria/termination/count.rb
+- lib/no_brainer/criteria/termination/first.rb
+- lib/no_brainer/criteria/termination/enumerable.rb
+- lib/no_brainer/criteria/termination/update.rb
+- lib/no_brainer/criteria/termination/cache.rb
+- lib/no_brainer/criteria/termination/delete.rb
+- lib/no_brainer/criteria/termination/eager_loading.rb
+- lib/no_brainer/autoload.rb
+- lib/no_brainer/railtie.rb
+- lib/no_brainer/criteria.rb
+- lib/no_brainer/database.rb
+- lib/no_brainer/decorated_symbol.rb
+- lib/no_brainer/document.rb
+- lib/no_brainer/error.rb
+- lib/no_brainer/index_manager.rb
+- lib/no_brainer/loader.rb
+- lib/no_brainer/locale/en.yml
+- lib/no_brainer/query_runner.rb
+- lib/no_brainer/util.rb
+- lib/no_brainer/config.rb
+- lib/no_brainer/connection.rb
+- lib/no_brainer/fork.rb
+- lib/no_brainer/version.rb
+- lib/nobrainer.rb
+- README.md
+- LICENSE.md
+homepage: http://github.com/nviennot/nobrainer
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.8
+signing_key: 
+specification_version: 4
+summary: ORM for RethinkDB
+test_files: []
+has_rdoc: false