# -*- encoding : utf-8 -*-

require 'map'

module Current
  def Current.version
    '1.5.0'
  end

  def Current.data
    Thread.current[:current] ||= Map.new
  end

  def Current.clear
    data.clear
    self
  end

  def Current.reset
    attribute_names.each{|name| undefine_attribute_method(name)}
    attribute_names.clear
    generators.clear
    clear
  end

  def Current.generators
    @generators ||= Map.new
  end

  def Current.attribute(name, *args, &block)
    options = Map.options_for!(args)

    name = name.to_s

    attribute_names.push(name)
    attribute_names.uniq!

    if options.has_key?(:default)
      default = options[:default]
      if default.respond_to?(:call)
        block ||= default
      else
        data[name] = default
      end
    end

    if !args.empty?
      value = args.shift
      data[name] = value
    end

    if block
      generators[name] = block
    end

    define_attribute_method(name)

    self
  end

  def Current.define_attribute_method(name)
    name = name.to_s
    unless respond_to?(name)
      singleton_class.module_eval do
        define_method(name) do
          value = data[name]
          if value.nil? and generator = generators[name]
            value = generator.call
            data[name] = value
          end
          value
        end

        define_method(name + '=') do |value|
          data[name] = value
        end

        define_method(name + '?') do |*args|
          send(name)
        end
      end
    end
  end

  def Current.undefine_attribute_method(name)
    if respond_to?(name)
      singleton_class.module_eval do
        remove_method(name)
        remove_method(name + '=')
        remove_method(name + '?')
      end
    end
  end

  def Current.singleton_class
    @singleton_class ||= class << self; self; end
  end

  def Current.attribute?(name)
    attribute_names.include?(name.to_s)
  end

  def Current.attribute_names
    @attribute_names ||= []
  end

  def Current.attributes
    attribute_names.inject(Map.new){|map, name| map.update(name => send(name))}
  end

  def Current.method_missing(method, *args, &block)
    case method.to_s
      when /^(.*)[=]$/
        name = $1
        value = args.shift
        attribute(name, value)
        value

      when /^(.*)[?]$/
        nil

      else
        if block
          name = method.to_s
          attribute(name, &block)
          block
        else
          nil
        end
    end
  end

  def method_missing(method, *args, &block)
    case method.to_s
      when /^current_(.*)$/
        msg = $1
        value = Current.send(msg, *args, &block)

      else
        super
    end
  end
end

def Current(*args, &block)
  Current.attribute(*args, &block)
end

if defined?(Rails)

##
#
  module Current
    attribute(:controller)
    attribute(:user)
  end

##
#
  module Current
    def Current.install_before_filter!
      if defined?(::ActionController::Base)
        ::ActionController::Base.module_eval do
          prepend_before_filter do |controller|
            Current.clear
            Current.controller = controller
          end

          extend Current
          include Current
          helper{ include Current }
        end
      end
    end
  end

##
#
  if defined?(Rails::Engine)
    class Engine < Rails::Engine
      config.before_initialize do
        ActiveSupport.on_load(:action_controller) do
          Current.install_before_filter!
        end
      end
    end
  else
    Current.install_before_filter!
  end

end

::Rails_current = ::Current

BEGIN {
  Object.send(:remove_const, :Current) if defined?(::Current)
  Object.send(:remove_const, :Rails_current) if defined?(::Rails_current)
}