rails_steroids 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 656c3847e194bf888f0b4c12482d3d7ba06200ab1af54d433bada03401d84398
-  data.tar.gz: 7bd0c50707fb1e44a256f09c5645cc1e8410bc0f7f131c20ea8f4cf70f22431b
+  metadata.gz: 9e8022f4d9c1864d8fb7886a799f32b7f8d0f7db85152a7f29db9810ce104367
+  data.tar.gz: 04571a62c83ac2d43d3294189265a9563e351fd43350dea47c413e2738d1b76e
 SHA512:
-  metadata.gz: cce4f6c5f46464d33fb3c9e8721a5d6db48595be7369bba3487882ee3c98c77fa98b57f653d6490254ad0815fed235f0b8c37534e4d4415b0d69d8bb7423bc73
-  data.tar.gz: 98e2c27907407481adf9c168116b2cc5e46c992d9c5c12d2f5892f30e41cfe38c9eba051b4208bdef7ba9c8fed0b8e19b5d82f6435335f62fb2497a36f18d19d
+  metadata.gz: 30fb811636d668a114df763e0f0eb50dffae19cef6ab6e2246a12772cfbabd8e6f3f4d8b66e948bbcd80f646d9233aca061259456b39712c16ecec78b91f36bd
+  data.tar.gz: 902918a36f478d5da29a785da787ab4eaf19896354f6c24fe2ca3143aba778b07535f6a689902fea8042085220f22b5e270a846f7f0c50cf37d9d42799e8dbb9

data/CHANGELOG.md

@@ -1,8 +1,20 @@
 ## [Unreleased]
 
+## [0.7.0] - 2024-02-08
+
+- Improvement in steroid recipe: migration
+  - Create new migration interactively for modify table with
+    * add_column, remove_column
+    * add_index, remove_index
+    * add_remove, remove_reference
+
 ## [0.6.0] - 2024-02-07
 
-- New steroid recipe: migration (Create new migration interactively)
+- New steroid recipe: migration
+  - Create new migration interactively for
+    * Create table
+    * Create join table
+    * Drop table
 
 ## [0.5.0] - 2024-02-06
 

data/lib/generators/steroid/migration/USAGE

@@ -19,3 +19,7 @@ What will this do?:
             - index and unique index
             - id column, custom join table name for join table creation
             - Addition to original migration generator is we can set default value as well
+        * Modify table with
+            - add_column, remove_column
+            - add_index, remove_index
+            - add_remove, remove_reference

data/lib/generators/steroid/migration/migration_generator.rb

@@ -15,25 +15,19 @@ module Steroid
     def add_migration
       say "Injecting steroid: Migration", :green
       cmd = ["rails generate migration"]
-      prompt = TTY::Prompt.new
-      boolean_choices = [{name: "yes", value: true}, {name: "no", value: false}]
 
       action_choices = [
         {name: 'Create table', value: 'create_table'},
         {name: 'Create join table', value: 'create_join_table'},
         {name: 'Drop table', value: 'drop_table'},
-        # {name: 'Modify table columns/index', value: 'modify_table'},
-        # {name: 'Add column', value: 'add_column'},
-        # {name: 'Remove column', value: 'remove_column'},
-        # {name: 'Add index', value: 'add_index'},
-        # {name: 'Remove index', value: 'remove_index'},
+        {name: 'Modify table columns/index', value: 'modify_table'},
       ]
-      action = prompt.select("What would you like to do?", action_choices)
+      @action = prompt.select("What would you like to do?", action_choices)
 
-      case action
+      case @action
       when 'create_table'
         table_name = prompt.ask("What is the name of table to be created?", required: true) { |q| q.modify :remove }
-        @content = content_for_create_table(table_name, collect_columns_data, need_timestamps?)
+        @content = content_for_create_table(table_name, collect_columns_data)
         migration_template "migration.rb", "#{db_migrate_path}/create_#{table_name}.rb"
       when 'create_join_table'
         table1_name = prompt.ask("What is the name of first table to be joined?", required: true) { |q| q.modify :remove }
@@ -51,48 +45,81 @@ module Steroid
         end
         if prompt.select("Add composite index for #{table1_name} and #{table2_name} foreign_key?", boolean_choices)
           uniq_index_option = prompt.select("Unique combination index?", boolean_choices) ? {meta: {unique: true}} : {}
-          columns_data << {name: "[:#{table2_name.singularize}_id, :#{table2_name.singularize}_id]", type: 'index'}.merge(uniq_index_option)
+          columns_data << {name: ["#{table2_name.singularize}_id", "#{table2_name.singularize}_id"], type: 'index'}.merge(uniq_index_option)
         end
-        @content = content_for_create_join_table(table1_name, table2_name, table_name, (columns_data + collect_columns_data), need_timestamps?)
+        @content = content_for_create_join_table(table1_name, table2_name, table_name, (columns_data + collect_columns_data))
         migration_template "migration.rb", "#{db_migrate_path}/create_join_table_#{table_name}.rb"
       when 'drop_table'
         table_name = prompt.ask("What is the name of table to be dropped?", required: true) { |q| q.modify :remove }
         @content = content_for_drop_table(table_name)
         migration_template "migration.rb", "#{db_migrate_path}/drop_#{table_name}.rb"
-      # when 'modify_table'
-      #   table_name = prompt.ask("What is the name of table to add/remove columns?", required: true) { |q| q.modify :remove }
-      #   columns_to_be_added = collect_columns_data
-      #   columns_to_be_removed
-      #   migration_template "migration.rb", "#{db_migrate_path}/drop_#{table_name}.rb"
-      # when 'add_column'
-      # when 'remove_column'
-      # when 'add_index'
-      # when 'remove_index'
+      when 'modify_table'
+        table_name = prompt.ask("What is the name of table to add/remove columns/index?", required: true) { |q| q.modify :remove }
+        modify_table_choices = [
+          {name: 'Add column', value: 'add_column'}, {name: 'Remove column', value: 'remove_column'},
+          {name: 'Add index', value: 'add_index'}, {name: 'Remove index', value: 'remove_index'},
+          {name: 'Add reference', value: 'add_reference'}, {name: 'Remove reference', value: 'remove_reference'},
+          # {name: 'Add timestamps', value: 'add_timestamps'}, {name: 'Remove timestamps', value: 'remove_timestamps'},
+          {name: 'Exit', value: 'exit'}
+        ]
+        @content = []
+        file_name = []
+        while (modify_table_action = prompt.select("What would you like to do?", modify_table_choices)) != 'exit'
+          for_removal = modify_table_action.start_with?('remove_')
+          if modify_table_action.end_with?('_index') || modify_table_action.end_with?('_reference')
+            data = modify_table_action.end_with?('_index') ? ask_index_data(for_removal: for_removal) : ask_reference_data(for_removal: for_removal)
+            file_name << modify_table_action << data[:name]
+          else
+            data = ask_column_data(for_removal: for_removal)
+            file_name << (for_removal ? 'remove' : 'add') << data[:name]
+          end
+          @content << format_statement_for_modify_table(modify_table_action, table_name, data)
+        end
+        @content = content_for_modify_table(@content.join("\n    "))
+        migration_template "migration.rb", "#{db_migrate_path}/#{file_name.join('_')}_in_#{table_name}.rb"
       end
     end
 
     private
 
+    def prompt
+      TTY::Prompt.new
+    end
+
+    def boolean_choices
+      [{name: "yes", value: true}, {name: "no", value: false}]
+    end
+
     def collect_columns_data
-      prompt = TTY::Prompt.new
-      boolean_choices = [{name: "yes", value: true}, {name: "no", value: false}]
-      columns = []
+      columns_data = []
       while prompt.select("Would you like to add model attributes(columns)?", boolean_choices)
-        columns_data = { meta: {} }
-        column_name = prompt.ask("Specify name of column:", required: true) { |q| q.modify :remove }
+        columns_data << ask_column_data
+      end
+      timestamps_data = ask_timestamps_data
+      columns_data << timestamps_data if timestamps_data
+      columns_data
+    end
 
-        column_type = prompt.select("Choose type of column:", %w(references boolean string text integer decimal float binary date time datetime primary_key digest token))
+    def ask_column_data(for_removal: false)
+      column_data = { meta: {} }
+      column_name = prompt.ask("Specify name of column:", required: true) { |q| q.modify :remove }
 
-        if column_type == 'references'
-          columns_data[:meta][:polymorphic] = true if prompt.select("Polymorphic association?", boolean_choices)
-        end
+      column_type_choices = %w(boolean string text integer decimal float binary date time datetime primary_key digest token)
+      column_type_choices << 'references' unless @action == 'modify_table'
+      column_type = prompt.select("Choose type of column:", column_type_choices)
+
+      if column_type == 'references'
+        column_data[:meta][:foreign_key] = true if prompt.select("Foreign_key to be #{for_removal ? 'removed' : 'added'}?", boolean_choices)
+        column_data[:meta][:polymorphic] = true if prompt.select("Polymorphic association?", boolean_choices)
+      end
+      unless for_removal
         if %w(integer string text binary).include?(column_type)
           if prompt.select("Set limit?", boolean_choices)
             limit = prompt.ask("Specify limit:", required: true) do |q|
               q.modify :remove
               q.convert(:int, "Invalid input! Please provide integer value.")
             end
-            columns_data[:meta][:limit] = limit
+            column_data[:meta][:limit] = limit
           end
         end
         if column_type == 'decimal'
@@ -101,66 +128,106 @@ module Steroid
               q.modify :remove
               q.convert(:int, "Invalid input! Please provide integer value.")
             end
-            columns_data[:meta][:precision] = precision
+            column_data[:meta][:precision] = precision
             scale = prompt.ask("Specify scale:", required: true) do |q|
               q.modify :remove
               q.convert(:int, "Invalid input! Please provide integer value.")
             end
-            columns_data[:meta][:scale] = scale
+            column_data[:meta][:scale] = scale
           end
         end
 
         if %w(references primary_key digest token).exclude?(column_type) && prompt.select("Add default value?", boolean_choices)
           acceptable_value_type = { 'text' => :string }[column_type] || column_type.to_sym
-          columns_data[:meta][:default] = prompt.ask("Specify default value:", required: true) do |q|
+          column_data[:meta][:default] = prompt.ask("Specify default value:", required: true) do |q|
             q.modify :remove
             q.convert(acceptable_value_type, "Invalid input! Please provide %{type} value.")
           end
         end
 
         if prompt.select("Add index?", boolean_choices)
-          columns_data[:meta][:index] = prompt.select("Unique index?", boolean_choices) ? :unique : true
+          column_data[:meta][:index] = prompt.select("Unique index?", boolean_choices) ? :unique : true
         end
+      end
+
+      column_data.merge!(name: column_name, type: column_type)
+      column_data
+    end
+
+    def ask_reference_data(for_removal: false)
+      column_data = { meta: {} }
+      column_data[:name] = prompt.ask("Specify name of reference:", required: true) { |q| q.modify :remove }
+      column_data[:meta][:foreign_key] = true if prompt.select("Foreign_key to be #{for_removal ? 'removed' : 'added'}?", boolean_choices)
+      column_data[:meta][:polymorphic] = true if prompt.select("Polymorphic association?", boolean_choices)
+      column_data
+    end
 
-        columns_data.merge!(name: column_name, type: column_type)
-        columns << columns_data
+    def ask_index_data(for_removal: false)
+      column_data = { meta: {} }
+      names = []
+      names << prompt.ask("Specify name of column:", required: true) { |q| q.modify :remove }
+      while prompt.select("Add more columns for Composite index?", boolean_choices)
+        names << prompt.ask("Specify name of another column:", required: true) { |q| q.modify :remove }
+      end
+      column_data[:name] = names
+      unless for_removal
+        column_data[:meta][:unique] = true if prompt.select("Unique index?", boolean_choices)
       end
-      columns
+      custom_name = prompt.ask("Specify custom name for index (optional):") { |q| q.modify :remove }
+      column_data[:meta][:name] = custom_name if custom_name.present?
+      column_data
     end
 
-    def need_timestamps?
-      prompt = TTY::Prompt.new
-      boolean_choices = [{name: "yes", value: true}, {name: "no", value: false}]
-      prompt.select("Add timestamps?", boolean_choices)
+    def ask_timestamps_data
+      if prompt.select("Add timestamps?", boolean_choices)
+        { type: 'timestamps' }
+      end
     end
 
-    def content_for_create_table(table_name, columns_data, need_timestamps)
-      columns_data_content = columns_data.map do |c|
-        column_row = ["t.#{c[:type]} :#{c[:name]}"]
-        c[:meta].each { |key, value| column_row << "#{key}: #{value.inspect}" }
-        column_row.join(', ')
+    def format_statement_for_modify_table(modify_table_action, table_name, column_data)
+      statement_data = ["#{modify_table_action} :#{table_name}"]
+      statement_data << if column_data[:name].is_a?(Array)
+        col_name = column_data[:name].size > 1 ? "[#{column_data[:name].map { |n| ":#{n}" }.join(', ')}]" : ":#{column_data[:name].first}"
+        "column: #{col_name}" if modify_table_action == 'remove_index'
+      else
+        ":#{column_data[:name]}"
       end
-      columns_data_content << "t.timestamps" if need_timestamps
+      statement_data << ":#{column_data[:type]}" if column_data[:type]
+      column_data[:meta].each { |key, value| statement_data << "#{key}: #{value.inspect}" }
+      statement_data = statement_data.join(', ')
+    end
+
+    def format_statement_for_create_table(columns_data)
+      columns_data.map do |c|
+        column_row = if c[:name].is_a?(Array)
+          ["t.#{c[:type]} [#{c[:name].map { |n| ":#{n}" }.join(', ')}]"]
+        elsif c[:name].nil?
+          ["t.#{c[:type]}"]
+        else
+          ["t.#{c[:type]} :#{c[:name]}"]
+        end
+        c[:meta].each { |key, value| column_row << "#{key}: #{value.inspect}" } if c[:meta]
+        column_row.join(', ')
+      end.join("\n      ")
+    end
+
+    def content_for_create_table(table_name, columns_data)
+      columns_data_content = format_statement_for_create_table(columns_data)
       <<-CONTENT
   def change
     create_table :#{table_name}#{primary_key_type} do |t|
-      #{columns_data_content.join("\n      ")}
+      #{columns_data_content}
     end
   end
       CONTENT
     end
 
-    def content_for_create_join_table(table1_name, table2_name, table_name, columns_data, need_timestamps)
-      columns_data_content = columns_data.map do |c|
-        column_row = ["t.#{c[:type]} :#{c[:name]}"]
-        c[:meta]&.each { |key, value| column_row << "#{key}: #{value.inspect}" }
-        column_row.join(', ')
-      end
-      columns_data_content << "t.timestamps" if need_timestamps
+    def content_for_create_join_table(table1_name, table2_name, table_name, columns_data)
+      columns_data_content = format_statement_for_create_table(columns_data)
       <<-CONTENT
   def change
     create_join_table :#{table1_name}, :#{table2_name}, table_name: :#{table_name} do |t|
-      #{columns_data_content.join("\n      ")}
+      #{columns_data_content}
     end
   end
       CONTENT
@@ -174,5 +241,13 @@ module Steroid
       CONTENT
     end
 
+    def content_for_modify_table(statements)
+      <<-CONTENT
+  def change
+    #{statements}
+  end
+      CONTENT
+    end
+
   end
 end

data/lib/rails_steroids/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsSteroids
-  VERSION = "0.6.0"
+  VERSION = "0.7.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rails_steroids
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Anand Bait
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-02-07 00:00:00.000000000 Z
+date: 2024-02-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails