returning 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile*.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,8 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in returning.gemspec
+gemspec
+
+gem 'activerecord', '~> 3.0'
+gem 'rake'
+gem 'pg'

data/Gemfile3.1

@@ -0,0 +1,8 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in returning.gemspec
+gemspec
+
+gem 'activerecord', '~> 3.1.rc6'
+gem 'rake'
+gem 'pg'

data/README.md

@@ -0,0 +1,24 @@
+Hi there,
+
+Returning
+=========
+
+It's a simple implementation of the +returning+ for ActiveRecord 3.0 and 3.1.
+
+How to use
+----------
+
+Add it to your Gemfile
+
+    gem 'returning'
+
+And then
+
+    # This sends UPDATE objects SET field = 42 WHERE id = 42 RETURNING generation
+    object.save(:returning => 'generation') # => #<0xblah Object generation=2>
+    # This sends DELETE FROM objects WHERE id = 42 RETURNING generation
+    object.destroy(:returning => 'generation') # => #<0xblah Object generation=3>
+
+It returns readonly object with fields that are specified in the returning option.
+
+Cheers

data/Rakefile

@@ -0,0 +1,7 @@
+require 'bundler/gem_tasks'
+
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/lib/returning/active_record/adapter.rb

@@ -0,0 +1,45 @@
+module Returning
+  module ActiveRecord
+    module Adapter
+      def to_sql(arel)
+        if arel.respond_to?(:ast)
+          visitor.accept(arel.ast)
+        else
+          arel
+        end
+      end
+
+      def returning?
+        !!@_returning
+      end
+
+      def update(sql, name = nil, binds = [])
+        if @_returning
+          records = @_returning[1].find_by_sql("#{to_sql(sql)} RETURNING #{@_returning[0]}", binds)
+          records.each { |r| r.readonly! }
+          records
+        else
+          super
+        end
+      end
+
+      def delete(arel, name = nil, binds = [])
+        if @_returning
+          records = @_returning[1].find_by_sql("#{to_sql(arel)} RETURNING #{@_returning[0]}", binds)
+          records.each { |r| r.readonly! }
+          records
+        else
+          super
+        end
+      end
+
+
+      def returning(columns, klass)
+        old_returning, @_returning = @_returning, [columns, klass]
+        yield
+      ensure
+        @_returning = old_returning
+      end
+    end
+  end
+end

data/lib/returning/active_record/returning.rb

@@ -0,0 +1,49 @@
+require 'active_support/concern'
+
+module Returning
+  module ActiveRecord
+    module Returning
+      extend ActiveSupport::Concern
+
+      included do
+        if method(:find_by_sql).arity == 1
+          class_eval do
+            def self.find_by_sql(sql, binds = [])
+              connection.select_all(sanitize_sql(sql), "#{name} Load").collect! { |record| instantiate(record) }
+            end
+          end
+        end
+      end
+
+      def save(options = {})
+        if r = options[:returning]
+          connection.returning(r, self.class) do
+            super()
+          end
+        else
+          super()
+        end
+      end
+
+      def create_or_update
+        if connection.returning? && !new_record?
+          fields = update
+          # if no attributes were changed return self
+          fields == 0 ? self : fields[0]
+        else
+          super
+        end
+      end
+
+      def destroy(options = {})
+        if r = options[:returning]
+          connection.returning(r, self.class) do
+            super()
+          end
+        else
+          super()
+        end
+      end
+    end
+  end
+end

data/lib/returning/version.rb

@@ -0,0 +1,3 @@
+module Returning
+  VERSION = "0.0.1"
+end

data/lib/returning.rb

@@ -0,0 +1,14 @@
+require "returning/version"
+require 'returning/active_record/returning'
+require 'returning/active_record/adapter'
+
+require 'active_record'
+require 'active_record/connection_adapters/postgresql_adapter'
+
+module Returning
+  # Your code goes here...
+end
+
+
+ActiveRecord::Base.send(:include, Returning::ActiveRecord::Returning)
+ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.send(:include, Returning::ActiveRecord::Adapter)

data/returning.gemspec

@@ -0,0 +1,23 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "returning/version"
+
+Gem::Specification.new do |s|
+  s.name        = "returning"
+  s.version     = Returning::VERSION
+  s.authors     = ["Eugene Pimenov"]
+  s.email       = ["eugene@soocial.com"]
+  s.homepage    = "http://github.com/libc/returning"
+  s.summary     = %q{Add PostgreSQL RETURNING thingie to ActiveRecord}
+  s.description = %q{Simple implementation of RETURNING for ActiveRecord}
+
+  s.rubyforge_project = "returning"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'activerecord', '>= 3.0'
+  s.add_development_dependency 'rspec', '~> 2.0'
+end

data/spec/returning_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+describe Returning do
+  before do
+    Post.create :name => "hello", :author => "josh"
+  end
+
+  describe '#save' do
+    let(:post) { Post.first }
+    context 'when called with :returning => name' do
+      subject do
+        post.name = 'hello world'
+        post.save(:returning => 'name')
+      end
+
+      its('name') { should == 'hello world' }
+      it { should be_readonly }
+    end
+
+    context 'when nothing changed' do
+      it 'returns self' do
+        post.save(:returning => 'name').should eql(post)
+      end
+    end
+  end
+
+  describe '#destroy' do
+    it 'returns the column passed to returning' do
+      post = Post.first
+      post.name = 'hello world'
+      post.destroy(:returning => 'name').name.should == 'hello world'
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'bundler/setup'
+
+Bundler.require
+
+Dir['spec/support/**/*.rb'].each { |f| require f }
+
+RSpec.configure do
+end

data/spec/support/ar.rb

@@ -0,0 +1,36 @@
+require 'active_record'
+
+config = {:adapter => 'postgresql', 'database' => 'returning_test', :host => 'localhost', :encoding => 'unicode'}
+begin
+  ActiveRecord::Base.establish_connection config
+  ActiveRecord::Base.connection
+rescue
+  begin
+    ActiveRecord::Base.establish_connection(config.merge('database' => 'postgres', 'schema_search_path' => 'public'))
+    ActiveRecord::Base.connection.create_database(config['database'], config.merge('encoding' => 'utf8'))
+    ActiveRecord::Base.establish_connection(config)
+  rescue => e
+    STDERR.puts "**** Couldn't create database returning_test"
+    STDERR.puts "#{e.class}: #{e.message}"
+    STDERR.puts e.backtrace.join("\n")
+    exit 1
+  end
+end
+
+
+class Migration < ActiveRecord::Migration
+  def self.up
+    create_table :posts, :force => true do |t|
+      t.string :name, :author
+    end
+  end
+
+  def self.down
+    drop_table :posts
+  end
+end
+
+Migration.up
+
+class Post < ActiveRecord::Base
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification 
+name: returning
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Eugene Pimenov
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-08-22 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 7
+        segments: 
+        - 3
+        - 0
+        version: "3.0"
+  name: activerecord
+  prerelease: false
+  version_requirements: *id001
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 2
+        - 0
+        version: "2.0"
+  name: rspec
+  prerelease: false
+  version_requirements: *id002
+  type: :development
+description: Simple implementation of RETURNING for ActiveRecord
+email: 
+- eugene@soocial.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Gemfile3.1
+- README.md
+- Rakefile
+- lib/returning.rb
+- lib/returning/active_record/adapter.rb
+- lib/returning/active_record/returning.rb
+- lib/returning/version.rb
+- returning.gemspec
+- spec/returning_spec.rb
+- spec/spec_helper.rb
+- spec/support/ar.rb
+homepage: http://github.com/libc/returning
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: returning
+rubygems_version: 1.8.5
+signing_key: 
+specification_version: 3
+summary: Add PostgreSQL RETURNING thingie to ActiveRecord
+test_files: 
+- spec/returning_spec.rb
+- spec/spec_helper.rb
+- spec/support/ar.rb