require 'markaby'

module Atchoum
  # Custom Markaby markup
  module Markup
    def tag!(*args, &block)
      h = args[-1]
      [:href, :action, :src].each do |arg|
        case h[arg]
        when :index
          h[arg] = @rendered_page == :index ? "./" : "../"
        else
          h[arg] = @rendered_page == h[arg] ? "../#{h[arg]}/" : "#{h[arg]}/"
        end if h[arg].is_a? Symbol
      end if h.is_a? Hash
      super
    end
  end
end