require 'rubypants'
require 'digest/md5'
$KCODE = 'U'

# A collection of simple helpers for improving web
# typograhy. Based on TypographyHelper by Luke Hartman and Typogrify.
#
# @example Using all filters
# require 'typogruby'
# Typogruby.improve('my text')
#
# @example Using a single filter
# Typogruby.initial_quotes('my text')
#
# @see http://github.com/hunter/typography-helper
# @see http://code.google.com/p/typogrify
# @author Arjan van der Gaag <arjan.vandergaag@gmail.com>
module Typogruby

  # Get the current gem version number
  # @return [String]
  def version
    File.read(File.join(File.dirname(__FILE__), *%w{.. VERSION}))
  end

  # Applies smartypants to a given piece of text
  #
  # @example
  # smartypants('The "Green" man')
  # # => 'The &#8220;Green&#8221; man'
  #
  # @param [String] text input text
  # @return [String] input text with smartypants applied
  def smartypants(text)
    ::RubyPants.new(text).to_html
  end

  # converts a & surrounded by optional whitespace or a non-breaking space
  # to the HTML entity and surrounds it in a span with a styled class.
  #
  # @example
  # amp('One & two')
  # # => 'One <span class="amp">&amp;</span> two'
  # amp('One &amp; two')
  # # => 'One <span class="amp">&amp;</span> two'
  # amp('One &#38; two')
  # # => 'One <span class="amp">&amp;</span> two'
  # amp('One&nbsp;&amp;&nbsp;two')
  # # => 'One&nbsp;<span class="amp">&amp;</span>&nbsp;two'
  #
  # @example It won't mess up & that are already wrapped, in entities or URLs
  #
  # amp('One <span class="amp">&amp;</span> two')
  # # => 'One <span class="amp">&amp;</span> two'
  # amp('&ldquo;this&rdquo; & <a href="/?that&amp;test">that</a>')
  # # => '&ldquo;this&rdquo; <span class="amp">&amp;</span> <a href="/?that&amp;test">that</a>'
  #
  # @example It should ignore standalone amps that are in attributes
  # amp('<link href="xyz.html" title="One & Two">xyz</link>')
  # # => '<link href="xyz.html" title="One & Two">xyz</link>'
  #
  # @param [String] text input text
  # @return [String] input text with ampersands wrapped
  def amp(text)
    # $1 is an excluded HTML tag, $2 is the part before the caps and $3 is the amp match
    ignore_scripts(text) do |t|
      t.gsub(/<(code|pre).+?<\/\1>|(\s|&nbsp;)&(?:amp;|#38;)?(\s|&nbsp;)/) { |str|
        $1 ? str : $2 + '<span class="amp">&amp;</span>' + $3
      }.gsub(/(\w+)="(.*?)<span class="amp">&amp;<\/span>(.*?)"/, '\1="\2&amp;\3"')
    end
  end

  # replaces space(s) before the last word (or tag before the last word)
  # before an optional closing element (<tt>a</tt>, <tt>em</tt>,
  # <tt>span</tt>, strong) before a closing tag (<tt>p</tt>, <tt>h[1-6]</tt>,
  # <tt>li</tt>, <tt>dt</tt>, <tt>dd</tt>) or the end of the string.
  #
  # @example
  # > widont('A very simple test')
  # # => 'A very simple&nbsp;test'
  #
  # @example Single word items shouldn't be changed
  # widont('Test')
  # # => 'Test'
  # widont(' Test')
  # # => ' Test'
  # widont('<ul><li>Test</p></li><ul>')
  # # => '<ul><li>Test</p></li><ul>'
  # widont('<ul><li> Test</p></li><ul>')
  # # => '<ul><li> Test</p></li><ul>'
  #
  # @example Nested tags
  # widont('<p>In a couple of paragraphs</p><p>paragraph two</p>')
  # # => '<p>In a couple of&nbsp;paragraphs</p><p>paragraph&nbsp;two</p>'
  # widont('<h1><a href="#">In a link inside a heading</i> </a></h1>')
  # # => '<h1><a href="#">In a link inside a&nbsp;heading</i> </a></h1>'
  # widont('<h1><a href="#">In a link</a> followed by other text</h1>')
  # # => '<h1><a href="#">In a link</a> followed by other&nbsp;text</h1>'
  #
  # @example Empty HTMLs shouldn't error
  # widont('<h1><a href="#"></a></h1>')
  # # => '<h1><a href="#"></a></h1>'
  #
  # @example Excluded tags
  # widont('<div>Divs get no love!</div>')
  # # => '<div>Divs get no love!</div>'
  # widont('<pre>Neither do PREs</pre>')
  # # => '<pre>Neither do PREs</pre>'
  # widont('<div><p>But divs with paragraphs do!</p></div>')
  # # => '<div><p>But divs with paragraphs&nbsp;do!</p></div>'
  #
  # @see http://mucur.name/posts/widon-t-and-smartypants-helpers-for-rails
  # @see http://shauninman.com/archive/2006/08/22/widont_wordpress_plugin
  # @param [String] text input text
  # @return [String] input text with non-breaking spaces inserted
  def widont(text)
    ignore_scripts(text) do |t|
      t.gsub(%r{
        ((?:</?(?:a|em|span|strong|i|b)[^>]*>)|[^<>\s]) # must be proceeded by an approved inline opening or closing tag or a nontag/nonspace
        \s+                                             # the space to replace
        (([^<>\s]+)                                     # must be flollowed by non-tag non-space characters
        \s*                                             # optional white space!
        (</(a|em|span|strong|i|b)>\s*)*                 # optional closing inline tags with optional white space after each
        ((</(p|h[1-6]|li|dt|dd)>)|$))                   # end with a closing p, h1-6, li or the end of the string
      }x) { |match| $1 + (match.include?('&nbsp;') ? ' ' : '&nbsp;') + $2 } # Make sure to not add another nbsp before one already there
    end
  end

  # surrounds two or more consecutive captial letters, perhaps with interspersed digits and periods
  # in a span with a styled class.
  #
  # @example
  # caps("A message from KU")
  # # => 'A message from <span class="caps">KU</span>'
  #
  # @example Allows digits
  # caps("A message from 2KU2 with digits")
  # # => 'A message from <span class="caps">2KU2</span> with digits'
  #
  # @example All caps with with apostrophes in them shouldn't break. Only handles dump apostrophes though.
  # caps("JIMMY'S")
  # # => '<span class="caps">JIMMY\\'S</span>'
  # caps("<i>D.O.T.</i>HE34T<b>RFID</b>")
  # # => '<i><span class="caps">D.O.T.</span></i><span class="caps">HE34T</span><b><span class="caps">RFID</span></b>'
  #
  # @param [String] text input text
  # @return [String] input text with caps wrapped
  def caps(text)
    ignore_scripts(text) do |t|
      # $1 is an excluded HTML tag, $2 is the part before the caps and $3 is the caps match
      t.gsub(/(?i:<(:code|pre).+?<\/\1>)|(\s|&nbsp;|^|'|"|>)([A-Z\d][A-Z\d\.']{1,})(?!\w)/)  do |str|
        excluded, before, caps = $1, $2, $3
        if excluded
          str
        elsif $3 =~ /^[\d\.]+$/
          before + caps
        else
          before + '<span class="caps">' + caps + '</span>'
        end
      end
    end
  end

  # encloses initial single or double quote, or their entities
  # (optionally preceeded by a block element and perhaps an inline element)
  # with a span that can be styled.
  #
  # @example
  # initial_quotes('"With primes"')
  # # => '<span class="dquo">"</span>With primes"'
  # initial_quotes("'With single primes'")
  # # => '<span class="quo">\\'</span>With single primes\\''
  #
  # @example With primes and links
  # initial_quotes('<a href="#">"With primes and a link"</a>')
  # # => '<a href="#"><span class="dquo">"</span>With primes and a link"</a>'
  #
  # @example with Smartypants-quotes
  # initial_quotes('&#8220;With smartypanted quotes&#8221;')
  # # => '<span class="dquo">&#8220;</span>With smartypanted quotes&#8221;'
  #
  # @param [String] text input text
  # @return [String] input text with initial quotes wrapped
  def initial_quotes(text)
    # $1 is the initial part of the string, $2 is the quote or entitity, and $3 is the double quote
    ignore_scripts(text) do |t|
      t.gsub(/((?:<(?:h[1-6]|p|li|dt|dd)[^>]*>|^)\s*(?:<(?:a|em|strong|span)[^>]*>)?)('|&#8216;|&lsquo;|("|&#8220;|&ldquo;))/) {$1 + "<span class=\"#{'d' if $3}quo\">#{$2}</span>"}
    end
  end

  # Converts special characters (excluding HTML tags) to HTML entities.
  #
  # @example
  # entities("Aloë Vera") # => "Alo&euml; Vera"
  #
  # @param [String] text input text
  # @return [String] input text with all special characters converted to
  #   HTML entities.
  def entities(text)
    o = ''
    text.scan(/(?x)

        ( <\?(?:[^?]*|\?(?!>))*\?>
        | <!-- (?m:.*?) -->
        | <\/? (?i:a|abbr|acronym|address|applet|area|b|base|basefont|bdo|big|blockquote|body|br|button|caption|center|cite|code|col|colgroup|dd|del|dfn|dir|div|dl|dt|em|fieldset|font|form|frame|frameset|h1|h2|h3|h4|h5|h6|head|hr|html|i|iframe|img|input|ins|isindex|kbd|label|legend|li|link|map|menu|meta|noframes|noscript|object|ol|optgroup|option|p|param|pre|q|s|samp|script|select|small|span|strike|strong|style|sub|sup|table|tbody|td|textarea|tfoot|th|thead|title|tr|tt|u|ul|var)\b
            (?:[^>"']|"[^"]*"|'[^']*')*
          >
        | &(?:[a-zA-Z0-9]+|\#[0-9]+|\#x[0-9a-fA-F]+);
        )
        |([^<&]+|[<&])

      /x) do |tag, text|
      o << tag.to_s
      o << encode(text.to_s)
    end
    o
  end

  # main function to do all the functions from the method.
  # @param [String] text input text
  # @return [String] input text with all filters applied
  def improve(text)
    initial_quotes(caps(smartypants(widont(amp(text)))))
  end

private

  # Convert characters from the map in ./lib/characters.txt
  # Code taken from TextMate HTML bundle
  # @param [String] text input text
  # @return [String] input text with all special characters converted to
  #   HTML entities.
  def encode(text)
    @char_to_entity ||= begin
      map = {}
      File.read(File.join(File.dirname(__FILE__), 'characters.txt')).scan(/^(\d+)\s*(.+)$/) do |key, value|
        map[[key.to_i].pack('U')] = value
      end
      map
    end

    text.gsub(/[^\x00-\x7F]|["'<>&]/) do |ch|
      ent = @char_to_entity[ch]
      ent ? "&#{ent};" : sprintf("&#x%02X;", ch.unpack("U")[0])
    end
  end

  # Hackish text filter that will make sure our text filters leave inline
  # javascript alone without resorting to a full-blown HTML parser.
  #
  # The idea is simple: every text filter is applied as a block to this
  # method. This will preprocess the text and replace any inline scripts
  # with a MD5 hash of its entire contents. Then the filter is called,
  # and then the hashes are replaced back with their original content.
  def ignore_scripts(text)
    @ignored_scripts = {}
    modified_text = text.gsub(/<script[^>]*>.*?<\/script>/mi) do |script|
      hash = Digest::MD5.hexdigest(script)
      @ignored_scripts[hash] = script
      hash
    end
    yield(modified_text).gsub(/#{@ignored_scripts.keys.join('|')}/) do |h|
      @ignored_scripts.delete(h)
    end
  end

  extend self
end