pg_enums 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6984b2a6b097469fc73dd768518c7e03d6439fb6
+  data.tar.gz: b6e6e20cb417f77fa99fe23ff7485a8497c380bb
+SHA512:
+  metadata.gz: c3dc8b830126e25c6b97133f059a8efa3eac46bfaaf486211c9b45d0605dca93e594f398d47d462c22cbfe9cbd8146ee9b681c04298da3de15f562b0c1e7bd8f
+  data.tar.gz: 869fe82f72c9c76cefc8b04f315c71e989350d8ff04d7eeb08cd9ea9967a7978640796f637d9892e62e6e4b92ea590f18ab298217e944b977fdda6518fd93d25

data/lib/pg_enums.rb

@@ -0,0 +1,112 @@
+# frozen_string_literal: true
+module PGEnums
+
+  def update_enum(table:, column:, enum_type:, old_value:, new_value:)
+    # Select default store it and drop it if defined
+    default = default(table, column, enum_type, old_value, new_value)
+    drop_default(table, column) if default
+    # Generate the new labels
+    new_enumlabels = new_enumlabels(enum_type, old_value, new_value)
+    # Add the new label to the enum
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TYPE #{enum_type} ADD VALUE IF NOT EXISTS '#{new_value}';
+    SQL
+    # update the enum and the column
+    change_enum_labels(table, column, enum_type, old_value, new_value, new_enumlabels)
+    # restaure the default value
+    restaure_default(table, column, default, enum_type) if default
+  end
+
+  def delete_from_enum(table:, column:, enum_type:, value_to_drop:, map_to:)
+    # Select default store it and drop it if defined
+    default = default(table, column, enum_type, value_to_drop, map_to)
+    drop_default(table, column) if default
+    # Generate the new labels
+    new_enumlabels = new_enumlabels(enum_type, value_to_drop)
+    # update the enum and the column
+    change_enum_labels(table, column, enum_type, value_to_drop, map_to, new_enumlabels)
+    # restaure the default value
+    restaure_default(table, column, default, enum_type) if default
+  end
+
+  def add_to_enum(enum_type:, new_value:)
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TYPE #{enum_type} ADD VALUE IF NOT EXISTS '#{new_value}';
+    SQL
+  end
+
+  def rename_enum(enum:, new_name:)
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TYPE #{enum} RENAME TO #{new_name};
+    SQL
+  end
+
+  def delete_enum(enum:, table:, column:)
+    remove_column table, column
+    ActiveRecord::Base.connection.execute <<-SQL
+      DROP TYPE #{enum};
+    SQL
+  end
+
+  def create_enum(enum_type:, table:, column:, values:)
+    ActiveRecord::Base.connection.execute <<-SQL
+      CREATE TYPE #{enum_type} AS ENUM (#{"'" + values.join("', '") + "'"});
+    SQL
+    add_column table, column, enum_type
+  end
+
+  private
+
+  def change_enum_labels(table, column, enum_type, old_value, new_value, new_enumlabels)
+    # Rename the enum
+    # Create a new enum with the new labels
+    # Update the table to set each old value to the new one
+    # Update the table to map the column to the new enum
+    # Drop the old enum
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TYPE #{enum_type} RENAME TO old_#{enum_type};
+      CREATE TYPE #{enum_type} AS ENUM (#{new_enumlabels});
+      UPDATE #{table} SET #{column} = '#{new_value}' WHERE #{table}.#{column} = '#{old_value}';
+      ALTER TABLE #{table} ALTER COLUMN #{column} TYPE #{enum_type} USING #{column}::text::#{enum_type};
+      DROP TYPE old_#{enum_type};
+    SQL
+  end
+
+  # Fetch the enum labels from the data base
+  def new_enumlabels(enum_type, old_value, new_value = nil)
+    enumlabels = ActiveRecord::Base.connection.execute <<-SQL
+      SELECT enumlabel from pg_enum
+      WHERE enumtypid=(
+        SELECT oid FROM pg_type WHERE typname='#{enum_type}'
+      )
+      ORDER BY enumsortorder;
+    SQL
+    enumlabels = enumlabels.map { |e| "'#{e["enumlabel"]}'" } - ["'#{old_value}'"]
+    enumlabels << "'#{new_value}'" if new_value
+    enumlabels.uniq.join(", ").chomp(", ")
+  end
+
+  def default(table, column, enum_type, old_value, new_value)
+    default = ActiveRecord::Base.connection.execute <<-SQL
+      SELECT column_default FROM information_schema.columns
+      WHERE (table_schema, table_name, column_name) = ('public', '#{table}', '#{column}')
+    SQL
+    return unless default.first["column_default"]
+    default = default.first["column_default"]
+    default.slice!("::#{enum_type}")
+    (default == "'#{old_value}'") ? "'#{new_value}'" : default
+  end
+
+  def drop_default(table, column)
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TABLE #{table} ALTER #{column} DROP default
+    SQL
+  end
+
+  def restaure_default(table, column, default, enum_type)
+    ActiveRecord::Base.connection.execute <<-SQL
+      ALTER TABLE #{table} ALTER #{column} SET DEFAULT #{default}::#{enum_type}
+    SQL
+  end
+
+end

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification
+name: pg_enums
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- François Sevaistre
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2017-09-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.1'
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+description: Simple usage of postgreSQL for Rails
+email: frasevaistre+github@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/pg_enums.rb
+homepage: http://rubygems.org/gems/pg_enums
+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.6.13
+signing_key: 
+specification_version: 4
+summary: PG Enums
+test_files: []