declare_schema 0.10.0 → 0.12.0.pre.2

data/lib/declare_schema/schema_change/all.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module DeclareSchema
+  module SchemaChange
+    module All
+    end
+    autoload :Base, 'declare_schema/schema_change/base'
+    autoload :ColumnAdd, 'declare_schema/schema_change/column_add'
+    autoload :ColumnChange, 'declare_schema/schema_change/column_change'
+    autoload :ColumnRename, 'declare_schema/schema_change/column_rename'
+    autoload :ForeignKeyAdd, 'declare_schema/schema_change/foreign_key_add'
+    autoload :ForeignKeyRemove, 'declare_schema/schema_change/foreign_key_remove'
+    autoload :IndexAdd, 'declare_schema/schema_change/index_add'
+    autoload :IndexRemove, 'declare_schema/schema_change/index_remove'
+    autoload :PrimaryKeyChange, 'declare_schema/schema_change/primary_key_change'
+    autoload :TableAdd, 'declare_schema/schema_change/table_add'
+    autoload :TableChange, 'declare_schema/schema_change/table_change'
+    autoload :TableRemove, 'declare_schema/schema_change/table_remove'
+    autoload :TableRename, 'declare_schema/schema_change/table_rename'
+  end
+end
+

data/lib/declare_schema/schema_change/base.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class Base
+      class << self
+        def format_options(options)
+          options.map do |k, v|
+            value =
+              if v.is_a?(Hash)
+                "{ #{format_options(v).join(', ')} }"
+              else
+                v.inspect
+              end
+            if k.is_a?(Symbol)
+              "#{k}: #{value}"
+            else
+              "#{k.inspect} => #{value}"
+            end
+          end
+        end
+      end
+
+      def up
+        up_command + spacing(up_command)
+      end
+
+      def down
+        down_command + spacing(down_command)
+      end
+
+      private
+
+      def spacing(command)
+        if command["\n"]
+          "\n\n"
+        else
+          "\n"
+        end
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/column_add.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class ColumnAdd < Base
+      def initialize(table_name, column_name, column_type, **column_options)
+        @table_name = table_name or raise ArgumentError, "must provide table_name"
+        @column_name = column_name or raise ArgumentError, "must provide column_name"
+        @column_type = column_type or raise ArgumentError, "must provide column_type"
+        @column_options = column_options
+      end
+
+      def up_command
+        "add_column #{[@table_name.to_sym.inspect,
+                       @column_name.to_sym.inspect,
+                       @column_type.to_sym.inspect,
+                       *self.class.format_options(@column_options)].join(", ")}"
+      end
+
+      def down_command
+        "remove_column #{@table_name.to_sym.inspect}, #{@column_name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/column_change.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class ColumnChange < Base
+      def initialize(table_name, column_name, old_type:, old_options:, new_type:, new_options:)
+        @table_name = table_name
+        @column_name = column_name
+        @old_type = old_type
+        @old_options = old_options
+        @new_type = new_type
+        @new_options = new_options
+      end
+
+      def up_command
+        "change_column #{[@table_name.to_sym.inspect,
+                          @column_name.to_sym.inspect,
+                          @new_type.to_sym.inspect,
+                          *self.class.format_options(@new_options)].join(", ")}"
+      end
+
+      def down_command
+        "change_column #{[@table_name.to_sym.inspect,
+                          @column_name.to_sym.inspect,
+                          @old_type.to_sym.inspect,
+                          *self.class.format_options(@old_options)].join(", ")}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/column_remove.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative 'column_add'
+
+module DeclareSchema
+  module SchemaChange
+    class ColumnRemove < ColumnAdd
+      alias column_add_up_command up_command
+      alias column_add_down_command down_command
+
+      def up_command
+        column_add_down_command
+      end
+
+      def down_command
+        column_add_up_command
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/column_rename.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class ColumnRename < Base
+      def initialize(table_name, old_name, new_name)
+        @table_name = table_name
+        @old_name = old_name
+        @new_name = new_name
+      end
+
+      def up_command
+        "rename_column #{@table_name.to_sym.inspect}, #{@old_name.to_sym.inspect}, #{@new_name.to_sym.inspect}"
+      end
+
+      def down_command
+        "rename_column #{@table_name.to_sym.inspect}, #{@new_name.to_sym.inspect}, #{@old_name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/foreign_key_add.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class ForeignKeyAdd < Base
+      def initialize(table_name, parent_table_name, column_name:, name:)
+        @table_name = table_name
+        @parent_table_name = parent_table_name
+        @column_name = column_name
+        @name = name
+      end
+
+      def up_command
+        "add_foreign_key #{@table_name.to_sym.inspect}, #{@parent_table_name.to_sym.inspect}, " +
+          "column: #{@column_name.to_sym.inspect}, name: #{@name.to_sym.inspect}"
+      end
+
+      def down_command
+        "remove_foreign_key #{@table_name.to_sym.inspect}, name: #{@name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/foreign_key_remove.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative 'foreign_key_add'
+
+module DeclareSchema
+  module SchemaChange
+    class ForeignKeyRemove < ForeignKeyAdd
+      alias index_add_up_command up_command
+      alias index_add_down_command down_command
+
+      def up_command
+        index_add_down_command
+      end
+
+      def down_command
+        index_add_up_command
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/index_add.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class IndexAdd < Base
+      def initialize(table_name, column_names, name:, unique:, where: nil)
+        @table_name = table_name
+        @column_names = column_names
+        @name = name
+        @unique = unique
+        @where = where.presence
+      end
+
+      def up_command
+        options = {
+          name: @name.to_sym,
+        }
+        options[:unique] = true if @unique
+        options[:where] = @where if @where
+
+        "add_index #{[@table_name.to_sym.inspect,
+                      @column_names.map(&:to_sym).inspect,
+                      *self.class.format_options(options)].join(', ')}"
+      end
+
+      def down_command
+        "remove_index #{@table_name.to_sym.inspect}, name: #{@name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/index_remove.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative 'index_add'
+
+module DeclareSchema
+  module SchemaChange
+    class IndexRemove < IndexAdd
+      alias index_add_up_command up_command
+      alias index_add_down_command down_command
+
+      def up_command
+        index_add_down_command
+      end
+
+      def down_command
+        index_add_up_command
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/primary_key_change.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class PrimaryKeyChange < Base
+      def initialize(table_name, old_column_names, new_column_names)
+        @table_name = table_name
+        @old_column_names = old_column_names.presence
+        @new_column_names = new_column_names.presence
+      end
+
+      def up_command
+        alter_primary_key(@old_column_names, @new_column_names)
+      end
+
+      def down_command
+        alter_primary_key(@new_column_names, @old_column_names)
+      end
+
+      private
+
+      def alter_primary_key(old_col_names, new_col_names)
+        drop_command = "DROP PRIMARY KEY" if old_col_names
+        add_command = "ADD PRIMARY KEY (#{new_col_names.join(', ')})" if new_col_names
+        commands = [drop_command, add_command].compact.join(', ')
+        statement = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(@table_name)} #{commands}"
+        "execute #{statement.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/table_add.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class TableAdd < Base
+      def initialize(table_name, fields, create_table_options, sql_options: nil)
+        @table_name = table_name
+        fields.all? do |type, name, options|
+          type.is_a?(Symbol) && name.is_a?(Symbol) && options.is_a?(Hash)
+        end or raise ArgumentError, "fields must be Array(Array(Symbol, Symbol, Hash)); got #{fields.inspect}"
+        @fields = fields
+        @create_table_options = create_table_options
+        @create_table_options = @create_table_options.merge(options: sql_options) if sql_options.present?
+      end
+
+      def up_command
+        longest_field_type_length = @fields.map { |type, _name, _option| type.to_s.length }.max
+
+        <<~EOS.strip
+          create_table #{[@table_name.to_sym.inspect, *self.class.format_options(@create_table_options)].join(', ')} do |t|
+          #{@fields.map do |type, name, options|
+              padded_type = format("%-*s", longest_field_type_length, type)
+              args = [name.inspect, *self.class.format_options(options)].join(', ')
+              "  t.#{padded_type} #{args}"
+            end.join("\n")}
+          end
+        EOS
+      end
+
+      def down_command
+        "drop_table #{@table_name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/table_change.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class TableChange < Base
+      def initialize(table_name, old_options, new_options)
+        @table_name = table_name
+        @old_options = old_options
+        @new_options = new_options
+      end
+
+      def up_command
+        alter_table(@table_name, @new_options)
+      end
+
+      def down_command
+        alter_table(@table_name, @old_options)
+      end
+
+      private
+
+      TABLE_OPTIONS_TO_SQL_MAPPINGS = {
+        charset:   'CHARACTER SET',
+        collation: 'COLLATE'
+      }.freeze
+
+      def alter_table(table_name, options)
+        sql_options = options.map { |key, value| [TABLE_OPTIONS_TO_SQL_MAPPINGS[key], value] }
+        statement = "ALTER TABLE #{ActiveRecord::Base.connection.quote_table_name(table_name)} #{sql_options.join(' ')}"
+        "execute #{statement.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/table_remove.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class TableRemove < Base
+      def initialize(table_name, add_table_back)
+        @table_name = table_name
+        @add_table_back = add_table_back
+      end
+
+      def up_command
+        "drop_table #{@table_name.to_sym.inspect}"
+      end
+
+      def down_command
+        @add_table_back
+      end
+    end
+  end
+end

data/lib/declare_schema/schema_change/table_rename.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative 'base'
+
+module DeclareSchema
+  module SchemaChange
+    class TableRename < Base
+      def initialize(old_name, new_name)
+        @old_name = old_name
+        @new_name = new_name
+      end
+
+      def up_command
+        "rename_table #{@old_name.to_sym.inspect}, #{@new_name.to_sym.inspect}"
+      end
+
+      def down_command
+        "rename_table #{@new_name.to_sym.inspect}, #{@old_name.to_sym.inspect}"
+      end
+    end
+  end
+end

data/lib/declare_schema/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module DeclareSchema
-  VERSION = "0.10.0"
+  VERSION = "0.12.0.pre.2"
 end