arfy 0.2 → 0.2.1

data/lib/arfy/generator.rb

@@ -1,49 +1,14 @@
 require 'erb'
 
-module StringPimped
-  #code from: http://sequel.rubyforge.org/rdoc-plugins/classes/String.html
-  def underscore
-    gsub(/::/, '/').gsub(/([A-Z]+)([A-Z][a-z])/,'\1_\2').
-      gsub(/([a-z\d])([A-Z])/,'\1_\2').tr("-", "_").downcase
-  end
-
-  def camelize(first_letter_in_uppercase =:upper)
-    s = gsub(/\/(.?)/){|x| "::#{x[-1..-1].upcase unless x == '/'}"}.
-      gsub(/(^|_)(.)/){|x| x[-1..-1].upcase}
-    s[0...1] = s[0...1].downcase unless first_letter_in_uppercase == :upper
-    s
-  end
-
-  def classify
-    str = sub(/.*\./, '')
-    class << str; include StringPimped; end
-    str.camelize
-  end
-
-  def respond_to? method
-    @i_know ||= [:underscore, :camelize, :classify]
-    if @i_know.include? method
-      return true
-    end
-    super
-  end
-end
-
 module Arfy
   class Generator
     def generate_migration_file(migration_name, dir)
+      dir ||= Rails.application.config.paths['db/migrate']
       migration_path = File.join(dir, new_migration_file_name_for(migration_name))
       File.open(migration_path, "w") do |file|
         file << code_for_migration(migration_name)
       end
-
-      if migration_path
-        puts "migration generated on: #{file}"
-      else
-        puts "FAIL. Migration not generated."
-      end
-
-      File.exists? migration_path
+      migration_path
     end
 
     private
@@ -56,7 +21,7 @@ module Arfy
       new_string = String.new(string)
       unless new_string.respond_to? method
         class << new_string
-          include StringPimped
+          include StringNamefyExtension
         end
       end
       new_string.send(method)
@@ -92,3 +57,4 @@ module Arfy
 
   end
 end
+

data/lib/arfy/migration_builder.rb

@@ -0,0 +1,105 @@
+require 'arfy/migration_builder/template_handler'
+require 'arfy/migration_builder/builders/column'
+require 'arfy/migration_builder/builders/create_table'
+require 'arfy/migration_builder/builders/drop_table'
+require 'arfy/migration_builder/builders/rename_table'
+require 'arfy/migration_builder/builders/add_column'
+require 'arfy/migration_builder/builders/change_column'
+require 'arfy/migration_builder/builders/rename_column'
+require 'arfy/migration_builder/builders/remove_column'
+require 'arfy/migration_builder/builders/add_index'
+require 'arfy/migration_builder/builders/remove_index'
+
+module Arfy
+  module MigrationBuilder
+    class Builder
+      def initialize(name, work_dir, id_generator)
+        @id_generator = id_generator
+        @work_dir = work_dir
+        @name = augment_string(name)
+        @builder = nil
+      end
+
+      def build(builder = nil)
+        @builder = builder
+        create_file
+        tpl_handler = TemplateHandler.new(migration_template, template_vars)
+        tpl_code = tpl_handler.render
+
+        File.open(final_file_path, "w") do |f|
+          f << tpl_code
+        end
+
+        tpl_code
+      end
+
+      def create_table(table_name)
+        CreateTable.new(table_name)
+      end
+
+      def drop_table(table_name)
+        DropTable.new(table_name)
+      end
+
+      private
+      def migration_template
+%{
+class <%= klass %> < ActiveRecord::Migration
+  <% if !change_code.nil? && change_code.length > 0 %>
+  def change
+    <%= change_code %>
+  end
+  <% else %>
+  def up
+    <%= up_code %>
+  end
+
+  def down
+    <%= down_code %>
+  end
+  <% end %>
+end
+}
+      end
+
+      def template_vars
+        vars = {}
+        unless @builder.nil?
+          vars.merge! @builder.code_for_template
+        end
+        vars[:klass] = @name.classify
+        vars
+      end
+
+      def augment_string(original_string)
+        string = String.new(original_string)
+        unless string.respond_to?(:underscore)
+          class << string
+            include StringNamefyExtension
+          end
+        end
+        string
+      end
+
+      def file_name
+        @file_identifier = @id_generator.identify
+        "#{@file_identifier}_#{@name.underscore}.rb"
+      end
+
+      def file_path
+        @work_dir
+      end
+
+      def final_file_path
+        @final_file_path ||= File.join(file_path, file_name)
+      end
+
+      def create_file
+        FileUtils.mkdir_p(file_path) unless Dir.exists? file_path
+        FileUtils.touch(final_file_path)
+        final_file_path
+      end
+    end
+  end
+end
+

data/lib/arfy/migration_builder/builders/add_column.rb

@@ -0,0 +1,17 @@
+require 'arfy/migration_builder/builders/column.rb'
+
+module Arfy
+  module MigrationBuilder
+    class AddColumn < ColumnMigration
+      include Reversible
+
+      private
+      def template_up_code
+%{
+    add_column :<%= table_name %>, :<%= column_name %>, :<%= column_type %><% if options && options.length > 0 %>, <%= options %><% end %>
+}
+      end
+    end
+  end
+end
+

data/lib/arfy/migration_builder/builders/add_index.rb

@@ -0,0 +1,41 @@
+require 'arfy/migration_builder/builders/column.rb'
+
+module Arfy
+  module MigrationBuilder
+    class AddIndex < ColumnMigration
+      include Reversible
+
+      def initialize(table_name, column_name, options = nil)
+        super(table_name, column_name)
+        unless options.nil?
+          @name = options[:name]
+          @unique = options[:unique]
+        end
+      end
+
+      private
+      def vars_for_direction(direction)
+        vars = super
+        if direction == :up
+          vars[:options] = all_options
+        end
+        vars
+      end
+
+      def all_options
+        options = ""
+        options << ":name => \"#{@name}\"" unless @name.nil?
+        unless @unique.nil?
+          options << ", " if options.length > 0
+          options << ":unique => #{@unique ? "true" : "false"}"
+        end
+        options
+      end
+
+      def template_up_code
+        %{    add_index :<%= table_name %>, :<%= column_name %><% if options.length > 0 %>, <%= options %><% end %>}
+      end
+    end
+  end
+end
+

data/lib/arfy/migration_builder/builders/change_column.rb

@@ -0,0 +1,16 @@
+require 'arfy/migration_builder/builders/column.rb'
+
+module Arfy
+  module MigrationBuilder
+    class ChangeColumn < ColumnMigration
+
+      private
+      def template_up_code
+%{
+    change_column :<%= table_name %>, :<%= column_name %>, :<%= column_type %><% if options && options.length > 0 %>, <%= options %><% end %>
+}
+      end
+    end
+  end
+end
+

data/lib/arfy/migration_builder/builders/column.rb

@@ -0,0 +1,147 @@
+require 'arfy/migration_builder/generic_migration.rb'
+
+module Arfy
+  module MigrationBuilder
+    class InvalidColumnTypeError < Exception
+    end
+
+    COLUMN_VALID_TYPES = [:primary_key, :string, :text, :integer, :float, :decimal,
+      :datetime, :timestamp, :time, :date, :binary, :boolean].to_enum
+
+    COLUMN_VALID_OPTIONS = [:limit, :default, :null, :precision, :scale].to_enum
+
+    class ColumnMigration < GenericMigration
+      def initialize(table_name, column, type = nil, options = nil)
+        super table_name
+        @column = Column.new(column, type)
+        unless options.nil?
+          options.each do |name, value|
+            @column.add_option name, value
+          end
+        end
+      end
+
+      private
+      def method_missing(method, *args)
+        if @column.respond_to? method
+          return @column.send method, *args
+        end
+        super
+      end
+
+      def vars_for_direction(direction)
+        vars = {}
+        if direction == :up
+          vars = {
+            :table_name => @table_name,
+            :column_name => @column.name,
+            :column_type => @column.type,
+            :options => @column.options_string
+          }
+        end
+        vars
+      end
+    end
+
+    class Option
+      attr_reader :name, :value
+      def initialize(name, value=nil)
+        super
+        @name = name
+        @value = value
+      end
+
+      def value_string
+        value_string = ""
+        if @value.class == String
+          value_string = "\"#{@value}\""
+        elsif @value.class == Symbol
+          value_string = ":#{@value}"
+        else
+          value_string = @value.to_s
+        end
+        value_string
+      end
+    end
+
+    class Column
+      attr_reader :name, :type
+      def initialize(name, type=nil)
+        unless type.nil?
+          raise InvalidColumnTypeError, type unless valid_types.include? type
+        end
+        @name = name
+        @type = type
+      end
+
+      def respond_to?(method, includes_private_methods=false)
+        valid_option = is_valid_option_getter?(method) || is_valid_option_setter?(method)
+        return valid_option || super
+      end
+
+      def options_string
+        return "" if @options.nil?
+        processed_options.join(", ")
+      end
+
+      def add_option(option_name, value=nil)
+        @options ||= []
+        if value
+          option = Option.new(option_name, value)
+        else
+          # seria bom conferir se é um option mesmo?
+          option = option_name
+        end
+        @options << option
+        option
+      end
+
+      private
+      def processed_options
+        options = @options.map do |option|
+          unless option.value.nil?
+            ":#{option.name} => #{option.value_string}"
+          end
+        end
+        options.compact
+      end
+
+      def valid_types
+        COLUMN_VALID_TYPES
+      end
+
+      def valid_options
+        COLUMN_VALID_OPTIONS
+      end
+
+      def is_valid_option_getter?(option_getter)
+        valid_options.include? option_getter
+      end
+
+      def is_valid_option_setter?(option_setter)
+        valid_option_setters = valid_options.map {|opt| "#{opt.to_s}=".to_sym}
+        valid_option_setters.include? option_setter
+      end
+
+      def option_for(option_name)
+        return "" if @options.nil?
+        options = @options.select do |item|
+          item.name == option_name
+        end
+        return "" if options.length <= 0
+        options[0].value
+      end
+
+      def method_missing(method, *args)
+        if is_valid_option_setter? method
+          option_name = method.to_s.gsub("=", "").to_sym
+          add_option(option_name, args[0])
+        elsif is_valid_option_getter? method
+          return option_for method
+        else
+          super
+        end
+      end
+    end
+  end
+end

data/lib/arfy/migration_builder/builders/create_table.rb

@@ -0,0 +1,30 @@
+require 'arfy/migration_builder/generic_migration.rb'
+module Arfy
+  module MigrationBuilder
+  
+    class CreateTable < GenericMigration
+      include Reversible
+
+      private
+      def vars_for_direction(direction)
+        vars = super
+        if direction == :up
+          vars[:columns] = @columns || {}
+        end
+        vars
+      end
+
+      def template_up_code
+        %{  create_table :<%= table_name %> do |t|
+    <% columns.each do |column| %>
+    t.<%= column.type %> :<%= column.name %><% if column.options_string.length > 0 %>, <%= column.options_string %><% end %>
+    <% end %>
+  end}
+      end
+
+      def template_down_code
+        %{  drop_table :<%= table_name %>}
+      end
+    end
+  end
+end

data/lib/arfy/migration_builder/builders/drop_table.rb

@@ -0,0 +1,16 @@
+require 'arfy/migration_builder/generic_migration.rb'
+
+module Arfy
+  module MigrationBuilder
+    class DropTable < GenericMigration
+
+      private
+      def template_up_code
+%{
+    drop_table :<%= table_name %>
+}
+      end
+    end
+  end
+end
+

data/lib/arfy/migration_builder/builders/remove_column.rb

@@ -0,0 +1,14 @@
+require 'arfy/migration_builder/builders/column.rb'
+
+module Arfy
+  module MigrationBuilder
+    class RemoveColumn < ColumnMigration
+      private
+      def template_up_code
+        %{    remove_column :<%= table_name %>, :<%= column_name %>}
+      end
+    end
+  end
+end
+
+