<gh_stars>10-100
# -*- coding: utf-8 -*-
"""
    readtime.utils
    ~~~~~~~~~~~~~~

    Utility and non-public methods.

    :copyright: (c) 2016 <NAME>.
    :license: BSD, see LICENSE for more details.
"""


from __future__ import division

import lxml
import math
import markdown2
import re
from pyquery import PyQuery as pq

from .result import Result
from ._compat import u

DEFAULT_WPM = 265  # Medium says they use 275 WPM but they actually use 265
WORD_DELIMITER = re.compile(r'\W+')


def read_time(content, format=None, wpm=None):
    """Returns the read time of some content.

    :param content: String of content.
    :param format: Format of the content (html, markdown, or text).
    """

    try:
        format = format.lower()
    except:
        pass

    if not wpm:
        wpm = DEFAULT_WPM

    if format == 'text':
        seconds = read_time_as_seconds(content, wpm=wpm)

    elif format == 'markdown':
        html = markdown2.markdown(content)
        el = pq(html)
        text, images = parse_html(el)
        seconds = read_time_as_seconds(text, images=images, wpm=wpm)

    elif format == 'html':
        el = pq(content)
        text, images = parse_html(el)
        seconds = read_time_as_seconds(text, images=images, wpm=wpm)

    else:
        raise Exception(u('Unsupported format: {0}').format(format))

    return Result(seconds=seconds, wpm=wpm)


def read_time_as_seconds(text, images=0, wpm=None):
    """Returns the read time as seconds of some plain text.

    :param text:   String of plain text.
    :param images: The number of inline images in the text.
    """

    if not wpm:
        wpm = DEFAULT_WPM

    try:
        num_words = len(re.split(WORD_DELIMITER, text.strip()))
    except (AttributeError, TypeError):
        num_words = 0

    seconds = int(math.ceil(num_words / wpm * 60))

    # add extra seconds for inline images
    delta = 12
    for img in range(images):
        seconds += delta
        if delta > 3:
            delta -= 1

    return seconds


def parse_html(el):
    """Converts HTML to plain text.

    Returns a tuple of (plain_text, num_images).

    :param el: A PyQuery DOM object.
    """

    text = []
    images = []
    paragraphs = ['h1', 'h2', 'h3', 'h4', 'h5']

    def add_text(tag, no_tail=False):
        if tag.tag == 'img':
            images.append(tag)
        if tag.text and not isinstance(tag, lxml.etree._Comment):
            text.append(tag.text)
        for child in tag.getchildren():
            add_text(child)
        if tag.tag in paragraphs and len(text) > 0 and not text[-1].strip().endswith('.'):
            text.append('.')
        if not no_tail and tag.tail:
            text.append(tag.tail)

    for tag in el:
        add_text(tag, no_tail=True)

    plain_text = re.sub(r'\s+', ' ', ''.join([t for t in text if t])).strip()

    return plain_text, len(images)
