ar_schema_functions 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 6cea6bd061b0022cd749cbaed3924dc53e6c10957ec03f0a58e59dd5ab352414
+  data.tar.gz: '09e0c07f709115d6b8e75f092ce0d281b3a488e970837d4a693a6671f5c042f3'
+SHA512:
+  metadata.gz: 12ac604674a76566f4dde3286ab352df2626df7b1f27597d340c648049087cd87a2e255a30e6adff73656f8892a4fa5fc59f89d9f7ef77556e6234502d9822c8
+  data.tar.gz: c812aaceadfe1b99e1c7901735b72727ec446589be3a74d0f88533060c6eef5c08ab86e7361694940dbd00d806968a755a9a7faa51e338a42cbb76b1e9c501cf

data/Gemfile

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in ar_schema_functions.gemspec
+gemspec
+
+gem 'rubocop', '~> 1.7'

data/ar_schema_functions.gemspec

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require_relative 'lib/ar_schema_functions/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = 'ar_schema_functions'
+  spec.version       = ArSchemaFunctions::VERSION
+  spec.authors       = ['Dave Allie']
+  spec.email         = ['dave@visibuild.com.au']
+
+  spec.summary       = 'Populate schema.rb with stored SQL functions'
+  spec.homepage      = 'https://github.com/visibuild/ar_schema_functions'
+  spec.license       = 'MIT'
+  spec.required_ruby_version = Gem::Requirement.new('>= 2.5.0')
+
+  spec.metadata['homepage_uri'] = spec.homepage
+  spec.metadata['source_code_uri'] = 'https://github.com/visibuild/ar_schema_functions'
+  spec.metadata['changelog_uri'] = 'https://github.com/visibuild/ar_schema_functions/blob/master/CHANGELOG.md'
+
+  spec.files = Dir['lib/**/*'] + %w[Gemfile ar_schema_functions.gemspec]
+  spec.require_paths = ['lib']
+
+  spec.add_dependency 'activerecord', '>= 4.0.0', '< 7.0.0'
+end

data/lib/ar_schema_functions/extensions.rb

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+module ArSchemaFunctions
+  module Extensions
+    module SchemaDumper
+      def dump(stream)
+        header(stream)
+        extensions(stream)
+        functions(stream)
+        tables(stream)
+        trailer(stream)
+        stream
+      end
+
+      # Defined by SQL engine specific adapters (if supported)
+      def functions(stream); end
+    end
+
+    module PostgreSQLSchemaDumper
+      def functions(stream)
+        functions = @connection.functions
+
+        return unless functions.any?
+
+        functions.each do |function|
+          indented_function = function.lines.map { |line| "    #{line}" }.join
+          stream.puts "  connection.execute(<<~SQL)\n#{indented_function}  SQL"
+          stream.puts
+        end
+      end
+    end
+
+    module AbstractAdapter
+      # Defined by SQL engine specific adapters (if supported)
+      def functions
+        []
+      end
+    end
+
+    module PostgreSQLAdapter
+      def functions
+        # From https://dataedo.com/kb/query/postgresql/list-user-defined-functions
+        exec_query(<<~SQL.squish, 'SCHEMA').cast_values
+          select case when l.lanname = 'internal' then p.prosrc
+                      else pg_get_functiondef(p.oid)
+                      end as definition
+          from pg_proc p
+          left join pg_namespace n on p.pronamespace = n.oid
+          left join pg_language l on p.prolang = l.oid
+          where n.nspname not in ('pg_catalog', 'information_schema')#{' '}
+            and l.lanname != 'c'
+          order by definition;
+        SQL
+      end
+    end
+  end
+end

data/lib/ar_schema_functions/railtie.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative 'extensions'
+
+module ArSchemaFunctions
+  class Railtie < Rails::Railtie
+    initializer 'ar_schema_functions.active_record' do
+      ActiveSupport.on_load :active_record do
+        ActiveRecord::SchemaDumper.prepend(ArSchemaFunctions::Extensions::SchemaDumper)
+        ActiveRecord::ConnectionAdapters::AbstractAdapter.prepend(ArSchemaFunctions::Extensions::AbstractAdapter)
+
+        if defined?(ActiveRecord::ConnectionAdapters::PostgreSQLAdapter)
+          ActiveRecord::ConnectionAdapters::PostgreSQLAdapter.prepend(ArSchemaFunctions::Extensions::PostgreSQLAdapter)
+          ActiveRecord::ConnectionAdapters::PostgreSQL::SchemaDumper
+            .prepend(ArSchemaFunctions::Extensions::PostgreSQLSchemaDumper)
+        end
+      end
+    end
+  end
+end

data/lib/ar_schema_functions/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module ArSchemaFunctions
+  VERSION = '0.1.0'
+end

data/lib/ar_schema_functions.rb

@@ -0,0 +1,4 @@
+# frozen_string_literal: true
+
+require_relative 'ar_schema_functions/version'
+require_relative 'ar_schema_functions/railtie' if defined?(Rails::Railtie) && defined?(ActiveRecord)

metadata

@@ -0,0 +1,72 @@
+--- !ruby/object:Gem::Specification
+name: ar_schema_functions
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Dave Allie
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2021-09-30 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.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 7.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 4.0.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: 7.0.0
+description:
+email:
+- dave@visibuild.com.au
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- ar_schema_functions.gemspec
+- lib/ar_schema_functions.rb
+- lib/ar_schema_functions/extensions.rb
+- lib/ar_schema_functions/railtie.rb
+- lib/ar_schema_functions/version.rb
+homepage: https://github.com/visibuild/ar_schema_functions
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/visibuild/ar_schema_functions
+  source_code_uri: https://github.com/visibuild/ar_schema_functions
+  changelog_uri: https://github.com/visibuild/ar_schema_functions/blob/master/CHANGELOG.md
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.15
+signing_key:
+specification_version: 4
+summary: Populate schema.rb with stored SQL functions
+test_files: []