postgres_with 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: ca230790590213616e6fece28510b13c3b4d17f66f33f8913132ea26f41ed493
+  data.tar.gz: 146f38e6d41e10b20d42810d114183304fc327e2db0703fac2de54e0e3da4ba2
+SHA512:
+  metadata.gz: d14d96e9e8f9667c9d703cf52aed70fdda8102924ea12006d21b2fd08ab15d1f7a200e64e3f35026886d31941961a9f954c820518036089661e0cf0a56ed3f40
+  data.tar.gz: cf84a1c1a64aba00f183badc893fbe0399f0f1d5ac832ff284c15a727e5f62f92be9f041605e73dfda71be0e9cf3e71799b2c0a8de33b63388d49f05a7e23c2f

data/.gitignore

@@ -0,0 +1,25 @@
+.env
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+bin/
+.rbenv-version*
+.ruby-version
+.ruby-gemset
+gemfiles/*.lock
+
+spec/dummy/config/database.yml

data/Gemfile

@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gemspec

data/README.md

@@ -0,0 +1,9 @@
+# PostgresWith
+
+Adds CTEs for Postgres
+
+## Usage
+
+Just `require 'postgres_with'` and use ActiveRecord as you normally would! postgres\_ext extends
+ActiveRecord's data type handling and query methods in both Arel and
+ActiveRecord.

data/Rakefile

@@ -0,0 +1,8 @@
+#!/usr/bin/env rake
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+Bundler::GemHelper.install_tasks

data/lib/postgres_with/active_record/cte_proxy.rb

@@ -0,0 +1,35 @@
+class CTEProxy
+  include ActiveRecord::Querying
+  include ActiveRecord::Sanitization::ClassMethods
+  include ActiveRecord::Reflection::ClassMethods
+
+  attr_accessor :reflections, :current_scope
+  attr_reader :connection, :arel_table
+
+  def initialize(name, model)
+    @name = name
+    @arel_table = Arel::Table.new(name)
+    @model = model
+    @connection = model.connection
+    @_reflections = {}
+  end
+
+  def name
+    @name
+  end
+
+  def table_name
+    name
+  end
+
+  delegate :column_names, :columns_hash, :model_name, :primary_key, :attribute_alias?,
+    :aggregate_reflections, :instantiate, :type_for_attribute, :relation_delegate_class, to: :@model
+
+  private
+
+  def reflections
+    @_reflections
+  end
+
+  alias _reflections reflections
+end

data/lib/postgres_with/active_record/querying.rb

@@ -0,0 +1,11 @@
+module ActiveRecord::Querying
+  delegate :with, to: :all
+
+  def from_cte(name, expression)
+    table = Arel::Table.new(name)
+
+    cte_proxy = CTEProxy.new(name, self)
+    relation = ActiveRecord::Relation.new cte_proxy, cte_proxy.arel_table
+    relation.with name => expression
+  end
+end

data/lib/postgres_with/active_record/relation/merger.rb

@@ -0,0 +1,9 @@
+module ActiveRecord
+  class Relation
+    class Merger # :nodoc:
+      def normal_values
+        NORMAL_VALUES + [:with]
+      end
+    end
+  end
+end

data/lib/postgres_with/active_record/relation/query_methods.rb

@@ -0,0 +1,98 @@
+module ActiveRecord
+  module QueryMethods
+    # WithChain objects act as placeholder for queries in which #with does not have any parameter.
+    # In this case, #with must be chained with #recursive to return a new relation.
+    class WithChain
+      def initialize(scope)
+        @scope = scope
+      end
+
+      # Returns a new relation expressing WITH RECURSIVE
+      def recursive(*args)
+        @scope.with_values += args
+        @scope.recursive_value = true
+        @scope
+      end
+    end
+
+    [:with].each do |name|
+      class_eval <<-CODE, __FILE__, __LINE__ + 1
+       def #{name}_values                   # def select_values
+         @values[:#{name}] || []            #   @values[:select] || []
+       end                                  # end
+                                            #
+       def #{name}_values=(values)          # def select_values=(values)
+         raise ImmutableRelation if @loaded #   raise ImmutableRelation if @loaded
+         @values[:#{name}] = values         #   @values[:select] = values
+       end                                  # end
+      CODE
+    end
+
+    [:recursive].each do |name|
+      class_eval <<-CODE, __FILE__, __LINE__ + 1
+        def #{name}_value=(value)            # def readonly_value=(value)
+          raise ImmutableRelation if @loaded #   raise ImmutableRelation if @loaded
+          @values[:#{name}] = value          #   @values[:readonly] = value
+        end                                  # end
+
+        def #{name}_value                    # def readonly_value
+          @values[:#{name}]                  #   @values[:readonly]
+        end                                  # end
+      CODE
+    end
+
+    def with(opts = :chain, *rest)
+      if opts == :chain
+        WithChain.new(spawn)
+      elsif opts.blank?
+        self
+      else
+        spawn.with!(opts, *rest)
+      end
+    end
+
+    def with!(opts = :chain, *rest) # :nodoc:
+      if opts == :chain
+        WithChain.new(self)
+      else
+        self.with_values += [opts] + rest
+        self
+      end
+    end
+
+    def build_arel_with_extensions
+      arel = build_arel_without_extensions
+
+      with_statements = with_values.flat_map do |with_value|
+        case with_value
+        when String
+          with_value
+        when Hash
+          with_value.map  do |name, expression|
+            case expression
+            when String
+              select = Arel::Nodes::SqlLiteral.new "(#{expression})"
+            when ActiveRecord::Relation, Arel::SelectManager
+              select = Arel::Nodes::SqlLiteral.new "(#{expression.to_sql})"
+            end
+            Arel::Nodes::As.new Arel::Nodes::SqlLiteral.new("\"#{name.to_s}\""), select
+          end
+        when Arel::Nodes::As
+          with_value
+        end
+      end
+      unless with_statements.empty?
+        if recursive_value
+          arel.with :recursive, with_statements
+        else
+          arel.with with_statements
+        end
+      end
+
+      arel
+    end
+
+    alias_method :build_arel_without_extensions, :build_arel
+    alias_method :build_arel, :build_arel_with_extensions
+  end
+end

data/lib/postgres_with/active_record/relation.rb

@@ -0,0 +1,2 @@
+require 'postgres_with/active_record/relation/merger'
+require 'postgres_with/active_record/relation/query_methods'

data/lib/postgres_with/active_record.rb

@@ -0,0 +1,4 @@
+require 'active_record'
+require 'postgres_with/active_record/relation'
+require 'postgres_with/active_record/cte_proxy'
+require 'postgres_with/active_record/querying'

data/lib/postgres_with/version.rb

@@ -0,0 +1,3 @@
+module PostgresWith
+  VERSION = '0.0.1'
+end

data/lib/postgres_with.rb

@@ -0,0 +1,2 @@
+require 'postgres_with/version'
+require 'postgres_with/active_record'

data/postgres_with.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/postgres_with/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Derrick Neier"]
+  gem.email         = ["derrick.neier@gmail.com"]
+  gem.description   = %q{Adds support for CTEs for PostgreSQL and ActiveRecord}
+  gem.summary       = %q{Extends ActiveRecord to handle PostgreSQL CTEs}
+  gem.homepage      = 'https://github.com/dneier/postgres_with'
+  gem.licenses      = ['MIT']
+
+  gem.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  gem.files         = `git ls-files`.split("\n")
+  gem.name          = "postgres_with"
+  gem.require_paths = ["lib"]
+  gem.version       = PostgresWith::VERSION
+
+  gem.add_dependency 'activerecord', '>= 4.0'
+  gem.add_dependency 'arel', '>= 4.0'
+
+  gem.add_development_dependency 'pg', '>= 0.13'
+end

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification
+name: postgres_with
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Derrick Neier
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-07-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: arel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.0'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.13'
+description: Adds support for CTEs for PostgreSQL and ActiveRecord
+email:
+- derrick.neier@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- README.md
+- Rakefile
+- lib/postgres_with.rb
+- lib/postgres_with/active_record.rb
+- lib/postgres_with/active_record/cte_proxy.rb
+- lib/postgres_with/active_record/querying.rb
+- lib/postgres_with/active_record/relation.rb
+- lib/postgres_with/active_record/relation/merger.rb
+- lib/postgres_with/active_record/relation/query_methods.rb
+- lib/postgres_with/version.rb
+- postgres_with.gemspec
+homepage: https://github.com/dneier/postgres_with
+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.7.6
+signing_key: 
+specification_version: 4
+summary: Extends ActiveRecord to handle PostgreSQL CTEs
+test_files: []