require 'simple_object'

module Migrant
  # Converts the following DSL:
  #
  # class MyModel < ActiveRecord::Base
  #   structure do
  #     my_field "some string"
  #    end
  # end
  # into a schema on that model class by calling method_missing(my_field)
  # and deciding what the best schema type is for the user's requiredments
  class Schema
    attr_accessor :indexes, :columns, :validations, :methods_to_alias

    def initialize
      @proxy = SchemaProxy.new(self)
      @columns = Hash.new
      @indexes = Array.new
      @validations = Hash.new
      @methods_to_alias = Array.new
    end

    def define_structure(&block)
      # Runs method_missing on columns given in the model "structure" DSL
      @proxy.translate_fancy_dsl(&block) if block_given?
    end

    def add_associations(associations)
      associations.each do |association|
        # Rails 3.1 changes primary_key_name to foreign_key (correct behaviour), so this is essentially backwards compatibility for Rails 3.0
        field = (association.respond_to?(:foreign_key))? association.foreign_key.to_sym : association.primary_key_name.to_sym
        case association.macro
          when :belongs_to
            if association.options[:polymorphic]
              @columns[(association.name.to_s+'_type').to_sym] = DataType::Polymorphic.new(:field => field)
              @indexes << [(association.name.to_s+'_type').to_sym, field]
            end
            @columns[field] = DataType::ForeignKey.new(:field => field)
            @indexes << field
        end
      end
    end

    def requires_migration?
      !(@columns.blank? && @indexes.blank?)
    end

    def column_migrations
      @columns.collect {|field, data| [field, data.column] } # All that needs to be migrated
    end

    # This is where we decide what the best schema is based on the structure requirements
    # The output of this is essentially a formatted schema hash that is processed
    # on each model by Migrant::MigrationGenerator

    def add_field(field, data_type = nil, options = {})
      data_type = DataType::String if data_type.nil?

      # Fields that do special things go here.
      if field == :timestamps
        add_field(:updated_at, :datetime)
        add_field(:created_at, :datetime)
        return true
      end

      # Add index if explicitly asked
      @indexes << field if options.delete(:index) || data_type.class.to_s == 'Hash' && data_type.delete(:index)
      @validations[field] = options.delete(:validates) if options[:validates]
      options.merge!(:field => field)

      # Matches: description DataType::Paragraph, :index => true
      if data_type.is_a?(Class) && data_type.respond_to?(:migrant_data_type?)
        @columns[field] = data_type.new(options)
      # Matches: description :index => true, :unique => true
      else
        begin
          # Eg. "My field" -> String -> DataType::String
          @columns[field] = "DataType::#{data_type.class.to_s}".constantize.new(options.merge(:value => data_type))
        rescue NameError
          # We don't have a matching type, throw a warning and default to string
          puts "MIGRATION WARNING: No migration implementation for class #{data_type.class.to_s} on field '#{field}', defaulting to string..."
          @columns[field] = DataType::Base.new(options)
        end
      end
      puts [":#{field}", "#{@columns[field].class}", "#{options.inspect}"].collect { |s| s.ljust(25) }.join if ENV['DEBUG']
    end
  end

  class InheritedSchema < Schema
    attr_accessor :parent_schema

    def initialize(parent_schema)
      @parent_schema = parent_schema
      @columns = Array.new
      @indexes = Array.new
    end

    def requires_migration?
      false # All added to base table
    end
  end

  # Why does this class exist? Excellent question.
  # Basically, Kernel gives a whole bunch of global methods, like system, puts, etc. This is bad because
  # our DSL relies on being able to translate any arbitrary method into a method_missing call.
  # So, we call method missing in this happy bubble where these magic methods don't exist.
  # The reason we don't inherit Schema itself in this way, is that we'd lose direct access to all other classes
  # derived from Object. Normally this would just mean scope resolution operators all over the class, but the real
  # killer is that the DSL would need to reference classes in that manner as well, which would reduce sexy factor by at least 100.
  class SchemaProxy < BasicObject
    def initialize(binding)
      @binding = binding
    end

    def translate_fancy_dsl(&block)
      self.instance_eval(&block)
    end

    def method_missing(*args, &block)
      field = args.slice!(0)
      data_type = args.slice!(0) unless args.first.nil? || args.first.respond_to?(:keys)

      @binding.add_field(field, data_type, args.extract_options!)
    end
  end
end